From c1a8e3bd0204ed9dde5ef1f84f69b6731de9cf5b Mon Sep 17 00:00:00 2001
From: Marius Seritan <github@seritan.net>
Date: Mon, 25 May 2015 08:26:42 -0700
Subject: [PATCH] test(connection): increasing test coverage of connection
 module

---
 src/header/common/connection.rs | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/src/header/common/connection.rs b/src/header/common/connection.rs
index b7fe0294..cfe5059d 100644
--- a/src/header/common/connection.rs
+++ b/src/header/common/connection.rs
@@ -88,3 +88,23 @@ impl Connection {
 bench_header!(close, Connection, { vec![b"close".to_vec()] });
 bench_header!(keep_alive, Connection, { vec![b"keep-alive".to_vec()] });
 bench_header!(header, Connection, { vec![b"authorization".to_vec()] });
+
+#[cfg(test)]
+mod tests {
+    use super::{Connection,ConnectionHeader};
+    use header::Header;
+    use unicase::UniCase;
+
+    fn parse_option(header: Vec<u8>) -> Connection {
+        let val = vec![header];
+        let connection: Connection = Header::parse_header(&val[..]).unwrap();
+        connection
+    }
+
+    #[test]
+    fn test_parse() {
+        assert_eq!(Connection::close(),parse_option(b"close".to_vec()));
+        assert_eq!(Connection::keep_alive(),parse_option(b"keep-alive".to_vec()));
+        assert_eq!(Connection(vec![ConnectionHeader(UniCase("upgrade".to_owned()))]),parse_option(b"upgrade".to_vec()));
+    }
+}