test: update middleware tests

Signed-off-by: Gustavo Inacio <[email protected]>

Author: gusinacio

crates/monitor/src/deployment_to_allocation.rs

@@ -20,3 +20,25 @@ pub fn deployment_to_allocation(
             .collect()
     })
 }
+
+#[cfg(test)]
+mod tests {
+    use tokio::sync::watch;
+
+    use super::deployment_to_allocation;
+
+    #[tokio::test]
+    async fn test_deployment_to_allocation() {
+        let allocations = test_assets::INDEXER_ALLOCATIONS.clone();
+        let allocations_watcher = watch::channel(allocations.clone()).1;
+        let deployment = deployment_to_allocation(allocations_watcher);
+
+        let deployments = deployment.borrow();
+        // one of the allocation id point to the same subgraph
+        assert_eq!(deployments.len(), 3);
+        // check if all allocations point to the subgraph id
+        for (key, val) in deployments.iter() {
+            assert_eq!(allocations.get(val).unwrap().subgraph_deployment.id, *key);
+        }
+    }
+}

crates/service/src/middleware/inject_deployment.rs

@@ -37,12 +37,15 @@ mod tests {
     async fn test_deployment_middleware() {
         let middleware = from_fn(deployment_middleware);
 
-        async fn handle(extensions: Extensions) -> Body {
-            extensions
+        let deployment = *ESCROW_SUBGRAPH_DEPLOYMENT;
+
+        let handle = move |extensions: Extensions| async move {
+            let received_deployment = extensions
                 .get::<DeploymentId>()
                 .expect("Should contain a deployment_id");
+            assert_eq!(*received_deployment, deployment);
             Body::empty()
-        }
+        };
 
         let app = Router::new()
             .route("/:deployment_id", get(handle))
@@ -51,7 +54,7 @@ mod tests {
         let res = app
             .oneshot(
                 Request::builder()
-                    .uri(format!("/{}", *ESCROW_SUBGRAPH_DEPLOYMENT))
+                    .uri(format!("/{}", deployment))
                     .body(Body::empty())
                     .unwrap(),
             )

crates/service/src/middleware/inject_receipt.rs

@@ -40,22 +40,25 @@ mod tests {
     async fn test_receipt_middleware() {
         let middleware = from_fn(receipt_middleware);
 
-        async fn handle(extensions: Extensions) -> Body {
-            extensions
+        let receipt = create_signed_receipt(Address::ZERO, 1, 1, 1).await;
+        let receipt_json = serde_json::to_string(&receipt).unwrap();
+
+        let handle = move |extensions: Extensions| async move {
+            let received_receipt = extensions
                 .get::<SignedReceipt>()
                 .expect("Should decode tap receipt");
+            assert_eq!(received_receipt.message, receipt.message);
+            assert_eq!(received_receipt.signature, receipt.signature);
             Body::empty()
-        }
+        };
 
         let app = Router::new().route("/", get(handle)).layer(middleware);
 
-        let receipt = create_signed_receipt(Address::ZERO, 1, 1, 1).await;
-
         let res = app
             .oneshot(
                 Request::builder()
                     .uri("/")
-                    .header(TapReceipt::name(), serde_json::to_string(&receipt).unwrap())
+                    .header(TapReceipt::name(), receipt_json)
                     .body(Body::empty())
                     .unwrap(),
             )