From 49240ce1d5a81a594aa12895c1ef4091c0fd8e9b Mon Sep 17 00:00:00 2001
From: YUAN LYU <lyuyuan92@gmail.com>
Date: Fri, 2 Apr 2021 15:29:21 -0400
Subject: [PATCH] Add unit tests to raise coverage score

Signed-off-by: YUAN LYU <lyuyuan92@gmail.com>
---
 coverage_config.json  |  2 +-
 src/common/headers.rs | 10 ++++++++++
 src/common/mod.rs     |  8 ++++++++
 src/response.rs       | 11 +++++++++++
 4 files changed, 30 insertions(+), 1 deletion(-)

diff --git a/coverage_config.json b/coverage_config.json
index 750586a..ade818e 100644
--- a/coverage_config.json
+++ b/coverage_config.json
@@ -1 +1 @@
-{"coverage_score": 92.5, "exclude_path": "", "crate_features": ""}
+{"coverage_score": 93.1, "exclude_path": "", "crate_features": ""}
diff --git a/src/common/headers.rs b/src/common/headers.rs
index 5397542..89b99f1 100644
--- a/src/common/headers.rs
+++ b/src/common/headers.rs
@@ -432,6 +432,7 @@ mod tests {
         assert_eq!(headers.content_length(), 0);
         assert_eq!(headers.chunked(), false);
         assert_eq!(headers.expect(), false);
+        assert_eq!(headers.accept(), MediaType::PlainText);
     }
 
     #[test]
@@ -726,4 +727,13 @@ mod tests {
         let header = Header::try_from(b"Accept").unwrap();
         assert_eq!(header.raw(), b"Accept");
     }
+
+    #[test]
+    fn test_set_accept() {
+        let mut headers = Headers::default();
+        assert_eq!(headers.accept(), MediaType::PlainText);
+
+        headers.set_accept(MediaType::ApplicationJson);
+        assert_eq!(headers.accept(), MediaType::ApplicationJson);
+    }
 }
diff --git a/src/common/mod.rs b/src/common/mod.rs
index 9cd1c97..e406995 100644
--- a/src/common/mod.rs
+++ b/src/common/mod.rs
@@ -522,6 +522,14 @@ mod tests {
         );
     }
 
+    #[test]
+    fn test_display_route_error() {
+        assert_eq!(
+            format!("{}", RouteError::HandlerExist("test".to_string())),
+            "handler for test already exists"
+        );
+    }
+
     #[test]
     fn test_method_to_str() {
         let val = Method::Get;
diff --git a/src/response.rs b/src/response.rs
index 93e4d4f..5312d9d 100644
--- a/src/response.rs
+++ b/src/response.rs
@@ -383,4 +383,15 @@ mod tests {
         response.allow_method(Method::Put);
         assert_eq!(response.allow(), vec![Method::Get, Method::Put]);
     }
+
+    #[test]
+    fn test_equal() {
+        let response = Response::new(Version::Http10, StatusCode::MethodNotAllowed);
+        let another_response = Response::new(Version::Http10, StatusCode::MethodNotAllowed);
+        assert_eq!(response, another_response);
+
+        let response = Response::new(Version::Http10, StatusCode::OK);
+        let another_response = Response::new(Version::Http10, StatusCode::BadRequest);
+        assert_ne!(response, another_response);
+    }
 }