Add tests with state

commonsensesoftware · Dec 24, 2023 · fcbbdd0 · fcbbdd0
1 parent 7c1edfe
commit fcbbdd0
Show file tree

Hide file tree

Showing 2 changed files with 93 additions and 1 deletion.
diff --git a/src/inject.rs b/src/inject.rs
@@ -147,6 +147,7 @@ mod tests {
     use super::*;
     use crate::{RouterServiceProviderExtensions, TestClient};
     use axum::{
+        extract::State,
         routing::{get, post},
         Router,
     };
@@ -398,4 +399,50 @@ mod tests {
         // assert
         assert_eq!(&text, "3");
     }
+
+    #[tokio::test]
+    async fn inject_with_state_into_handler() {
+        // arrange
+        trait Service: Send + Sync {
+            fn do_work(&self) -> String;
+        }
+
+        #[injectable(Service)]
+        struct ServiceImpl;
+
+        impl Service for ServiceImpl {
+            fn do_work(&self) -> String {
+                "Test".into()
+            }
+        }
+
+        #[derive(Clone)]
+        struct AppState;
+
+        async fn handler(
+            Inject(service): Inject<dyn Service>,
+            State(_state): State<AppState>,
+        ) -> String {
+            service.do_work()
+        }
+
+        let provider = ServiceCollection::new()
+            .add(ServiceImpl::scoped())
+            .build_provider()
+            .unwrap();
+
+        let app = Router::new()
+            .route("/test", get(handler))
+            .with_state(AppState)
+            .with_provider(provider);
+
+        let client = TestClient::new(app);
+
+        // act
+        let response = client.get("/test").send().await;
+        let text = response.text().await;
+
+        // assert
+        assert_eq!(&text, "Test");
+    }
 }
diff --git a/src/inject_keyed.rs b/src/inject_keyed.rs
@@ -155,7 +155,7 @@ mod tests {
     use crate::{RouterServiceProviderExtensions, TestClient};
     use axum::{
         routing::{get, post},
-        Router,
+        Router, extract::State,
     };
     use di::{injectable, Injectable, ServiceCollection};
     use http::StatusCode;
@@ -301,4 +301,49 @@ mod tests {
         // assert
         assert_eq!(&text, "2");
     }
+
+    #[tokio::test]
+    async fn inject_with_key_and_state_into_handler() {
+        // arrange
+        trait Service: Send + Sync {
+            fn do_work(&self) -> String;
+        }
+
+        #[injectable(Service)]
+        struct ServiceImpl;
+
+        impl Service for ServiceImpl {
+            fn do_work(&self) -> String {
+                "Test".into()
+            }
+        }
+
+        #[derive(Clone)]
+        struct AppState;
+
+        async fn handler(
+            InjectWithKey(service): InjectWithKey<key::Basic, dyn Service>,
+            State(_state): State<AppState>) -> String {
+            service.do_work()
+        }
+
+        let provider = ServiceCollection::new()
+            .add(ServiceImpl::scoped().with_key::<key::Basic>())
+            .build_provider()
+            .unwrap();
+
+        let app = Router::new()
+            .route("/test", get(handler))
+            .with_state(AppState)
+            .with_provider(provider);
+
+        let client = TestClient::new(app);
+
+        // act
+        let response = client.get("/test").send().await;
+        let text = response.text().await;
+
+        // assert
+        assert_eq!(&text, "Test");
+    }
 }