From 6687b2a6e1c7e9ece2ab9bf47e0048561ca8b578 Mon Sep 17 00:00:00 2001
From: Erick Tryzelaar <erick.tryzelaar@gmail.com>
Date: Fri, 14 Nov 2014 22:50:47 -0800
Subject: [PATCH] std: add tests for the Vec<u8> Writer impl

---
 src/libstd/io/mem.rs | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/src/libstd/io/mem.rs b/src/libstd/io/mem.rs
index f27951f263da2..e084b585394e8 100644
--- a/src/libstd/io/mem.rs
+++ b/src/libstd/io/mem.rs
@@ -362,6 +362,16 @@ mod test {
     use self::test::Bencher;
     use str::StrPrelude;
 
+    #[test]
+    fn test_vec_writer() {
+        let mut writer = Vec::new();
+        writer.write(&[0]).unwrap();
+        writer.write(&[1, 2, 3]).unwrap();
+        writer.write(&[4, 5, 6, 7]).unwrap();
+        let b: &[_] = &[0, 1, 2, 3, 4, 5, 6, 7];
+        assert_eq!(writer.as_slice(), b);
+    }
+
     #[test]
     fn test_mem_writer() {
         let mut writer = MemWriter::new();
@@ -385,6 +395,8 @@ mod test {
             assert_eq!(writer.tell(), Ok(8));
             writer.write(&[]).unwrap();
             assert_eq!(writer.tell(), Ok(8));
+
+            assert!(writer.write(&[1]).is_err());
         }
         let b: &[_] = &[0, 1, 2, 3, 4, 5, 6, 7];
         assert_eq!(buf.as_slice(), b);