From: Ralf Jung <post@ralfj.de>
Date: Sat, 8 Jul 2017 18:02:01 +0000 (-0700)
Subject: Merge pull request #26 from TorstenScheck/fix-typos-plus-clarifications
X-Git-Url: https://git.ralfj.de/rust-101.git/commitdiff_plain/7c40ab364e37362f07302da9074f2cabacffe8c7?hp=2dac37ce49abc1a890aa79fb15bf837b4cf1ca64

Merge pull request #26 from TorstenScheck/fix-typos-plus-clarifications

Fix typos plus clarifications
---

diff --git a/solutions/src/list.rs b/solutions/src/list.rs
index 3206982..1c794fc 100644
--- a/solutions/src/list.rs
+++ b/solutions/src/list.rs
@@ -53,6 +53,8 @@ impl<T> LinkedList<T> {
             if new_last.is_null() {
                 // The list is now empty.
                 self.first = new_last;
+            } else {
+                unsafe { (*new_last).next = ptr::null_mut() };
             }
             let last = unsafe { raw_into_box(last) } ;
             Some(last.data)
@@ -86,6 +88,8 @@ impl<T> LinkedList<T> {
             if new_first.is_null() {
                 // The list is now empty.
                 self.last = new_first;
+            } else {
+                unsafe { (*new_first).prev = ptr::null_mut() };
             }
             let first = unsafe { raw_into_box(first) } ;
             Some(first.data)
@@ -200,4 +204,20 @@ mod tests {
         }
         assert_eq!(count.count.get(), 20);
     }
+
+    #[test]
+    fn test_iter_mut() {
+        let mut l = LinkedList::<i32>::new();
+        for i in 0..5 {
+            l.push_back(i);
+        }
+
+        assert_eq!(l.pop_front(), Some(0));
+        assert_eq!(l.pop_back(), Some(4));
+
+        for (n, i) in l.iter_mut().enumerate() {
+            *i-=1;
+            assert_eq!(n as i32, *i);
+        }
+    }
 }