Merge pull request #14 from kpp/inc_coverage

Increase code coverage

Author: Roman Proskuryakov

src/stream.rs

@@ -926,6 +926,14 @@ mod tests {
         assert_eq!(vec![1, 2, 3], executor::block_on(collect::<_, Vec<_>>(stream)));
     }
 
+    #[test]
+    fn test_take_more_than_size() {
+        let stream = iter(1..=3);
+        let stream = take(stream, 10);
+
+        assert_eq!(vec![1, 2, 3], executor::block_on(collect::<_, Vec<_>>(stream)));
+    }
+
     #[test]
     fn test_repeat() {
         let stream = repeat(9);
@@ -962,6 +970,14 @@ mod tests {
         assert_eq!(vec![6, 7, 8, 9, 10], executor::block_on(collect::<_, Vec<_>>(stream)));
     }
 
+    #[test]
+    fn test_skip_more_than_size() {
+        let stream = iter(1..=10);
+        let stream = skip(stream, 15);
+
+        assert!(executor::block_on(collect::<_, Vec<_>>(stream)).is_empty());
+    }
+
     #[test]
     fn test_zip() {
         let stream1 = iter(1..=3);
@@ -988,6 +1004,14 @@ mod tests {
         assert_eq!(vec![1, 2, 3, 4, 5], executor::block_on(collect::<_, Vec<_>>(stream)));
     }
 
+    #[test]
+    fn test_take_while_more_than_size() {
+        let stream = iter(1..=3);
+        let stream = take_while(stream, |x| ready(*x <= 5));
+
+        assert_eq!(vec![1, 2, 3], executor::block_on(collect::<_, Vec<_>>(stream)));
+    }
+
     #[test]
     fn test_skip_while() {
         let stream = iter(1..=10);
@@ -996,6 +1020,14 @@ mod tests {
         assert_eq!(vec![6, 7, 8, 9, 10], executor::block_on(collect::<_, Vec<_>>(stream)));
     }
 
+    #[test]
+    fn test_skip_while_more_than_size() {
+        let stream = iter(1..=3);
+        let stream = skip_while(stream, |x| ready(*x <= 5));
+
+        assert!(executor::block_on(collect::<_, Vec<_>>(stream)).is_empty());
+    }
+
     #[test]
     fn test_fold() {
         let stream = iter(0..6);