test: add insta

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

Author: gusinacio

Cargo.lock

@@ -67,6 +67,7 @@ tower-test = "0.4.0"
 tower-service = "0.3.3"
 tokio-test = "0.4.4"
 wiremock.workspace = true
+insta = "1.41.1"
 
 [build-dependencies]
 build-info-build = { version = "0.0.39", default-features = false }

crates/service/Cargo.toml

@@ -305,10 +305,10 @@ impl ServiceRouter {
                 let free_query = Bearer::new(free_auth_token);
                 let result = free_query.or(tap_auth);
                 let auth_layer = AsyncRequireAuthorizationLayer::new(result);
-                handler = handler.layer(auth_layer);
+                handler = handler.route_layer(auth_layer);
             } else {
                 let auth_layer = AsyncRequireAuthorizationLayer::new(tap_auth);
-                handler = handler.layer(auth_layer);
+                handler = handler.route_layer(auth_layer);
             }
 
             let deployment_to_allocation = deployment_to_allocation(allocations);
@@ -346,7 +346,7 @@ impl ServiceRouter {
                 // create attestation
                 .layer(from_fn(attestation_middleware));
 
-            handler.layer(service_builder)
+            handler.route_layer(service_builder)
         };
 
         // setup cors

crates/service/src/service/router.rs

@@ -4,15 +4,15 @@
 use std::time::{Duration, SystemTime};
 
 use alloy::primitives::Address;
-use axum::http::Request;
+use axum::{body::to_bytes, http::Request};
 use axum_extra::headers::Header;
 use indexer_config::{BlockchainConfig, GraphNodeConfig, IndexerConfig, NonZeroGRT};
 use indexer_monitor::EscrowAccounts;
 use indexer_service_rs::{
     service::{ServiceRouter, TapReceipt},
     QueryBody,
 };
-use reqwest::{StatusCode, Url};
+use reqwest::{Method, StatusCode, Url};
 use sqlx::PgPool;
 use test_assets::{create_signed_receipt, INDEXER_ALLOCATIONS, TAP_EIP712_DOMAIN};
 use tokio::sync::watch;
@@ -113,6 +113,7 @@ async fn full_integration_test(database: PgPool) {
     };
 
     let request = Request::builder()
+        .method(Method::POST)
         .uri(format!("/subgraphs/id/{deployment}"))
         .header(TapReceipt::name(), serde_json::to_string(&receipt).unwrap())
         .body(serde_json::to_string(&query).unwrap())
@@ -121,6 +122,12 @@ async fn full_integration_test(database: PgPool) {
     // with deployment
     let res = app.oneshot(request).await.unwrap();
     assert_eq!(res.status(), StatusCode::OK);
+
+    let graphql_response = res.into_body();
+    let bytes = to_bytes(graphql_response, usize::MAX).await.unwrap();
+    let res = String::from_utf8(bytes.into()).unwrap();
+
+    insta::assert_snapshot!(res);
 }
 
 fn leak<T>(thing: T) -> &'static T {

crates/service/tests/router_test.rs

@@ -0,0 +1,6 @@
+---
+source: crates/service/tests/router_test.rs
+expression: res
+snapshot_kind: text
+---
+{"graphQLResponse":"\n                {\n                    \"data\": {\n                        \"graphNetwork\": {\n                            \"currentEpoch\": 960\n                        }\n                    }\n                }\n                ","attestation":null}