From 90dbef1d03339d88e7dfa1a3ec2ffc096d11e5aa Mon Sep 17 00:00:00 2001
From: Jonathan Reem <jonathan.reem@gmail.com>
Date: Sat, 20 Sep 2014 03:45:28 -0700
Subject: [PATCH] Add tests for double-reads for Header to ensure safety.

---
 src/header/mod.rs | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/src/header/mod.rs b/src/header/mod.rs
index 915c1312..59594a34 100644
--- a/src/header/mod.rs
+++ b/src/header/mod.rs
@@ -390,5 +390,20 @@ mod tests {
         let ContentLength(_) = headers.get::<ContentLength>().unwrap();
         assert!(headers.get::<CrazyLength>().is_none());
     }
+
+    #[test]
+    fn test_multiple_reads() {
+        let headers = Headers::from_raw(&mut mem("Content-Length: 10\r\n\r\n")).unwrap();
+        let ContentLength(one) = headers.get::<ContentLength>().unwrap();
+        let ContentLength(two) = headers.get::<ContentLength>().unwrap();
+        assert_eq!(one, two);
+    }
+
+    #[test]
+    fn test_different_reads() {
+        let headers = Headers::from_raw(&mut mem("Content-Length: 10\r\nContent-Type: text/plain\r\n\r\n")).unwrap();
+        let ContentLength(_) = headers.get::<ContentLength>().unwrap();
+        let ContentType(_) = headers.get::<ContentType>().unwrap();
+    }
 }