From 61dd0f2bf19a4f044a51cf3777c00d776dfe866c Mon Sep 17 00:00:00 2001
From: Glade Miller <glade.miller@insidesales.com>
Date: Thu, 2 May 2019 12:23:26 -0600
Subject: [PATCH] Add tests for route_data and app_data for TestRequest

---
 src/test.rs | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/src/test.rs b/src/test.rs
index 7d3321180..4e9bb6471 100644
--- a/src/test.rs
+++ b/src/test.rs
@@ -677,4 +677,25 @@ mod tests {
         let res = block_fn(|| app.call(req)).unwrap();
         assert!(res.status().is_success());
     }
+
+    #[test]
+    fn test_route_data() {
+        let req = TestRequest::get().route_data(Person { id: "12345".to_string(), name: "User name".to_string()}).uri("/people").to_http_request();
+        let extensions = req.extensions();
+        let person = extensions.get::<Person>();
+        assert!(person.is_some());
+        let person = person.expect("Person wasn't returned");
+        assert_eq!(person.id, "12345");
+        assert_eq!(person.name, "User name");
+    }
+
+   #[test]
+    fn test_app_data() {
+        let req = TestRequest::get().app_data(Person { id: "12345".to_string(), name: "User name".to_string()}).uri("/people").to_http_request();
+        let person = req.app_data::<Person>();
+        assert!(person.is_some());
+        let person = person.expect("Person wasn't returned");
+        assert_eq!(person.id, "12345");
+        assert_eq!(person.name, "User name");
+    }
 }