add test for custom headers

Signed-off-by: Sahil Yeole <[email protected]>

examples/grpc-reflection.graphql

@@ -2,7 +2,7 @@
 schema
   @server(port: 8000)
   @upstream(baseURL: "http://localhost:50051", httpCache: 42, batch: {delay: 10})
-  @link(src: "http://localhost:50051", type: Grpc, headers: [{key: "authorization", value: "secret"}]) {
+  @link(src: "http://localhost:50051", type: Grpc, headers: [{key: "authorization", value: "Bearer 123"}]) {
   query: Query
 }
 

src/core/proto_reader/fetch.rs

@@ -382,4 +382,42 @@ mod grpc_fetch {
 
         Ok(())
     }
+
+    #[tokio::test]
+    async fn test_custom_headers_resp_list_all() -> Result<()> {
+        let server = start_mock_server();
+
+        let http_reflection_service_not_found = server.mock(|when, then| {
+            when.method(httpmock::Method::POST)
+                .path("/grpc.reflection.v1alpha.ServerReflection/ServerReflectionInfo")
+                .header("authorization", "Bearer 123");
+            then.status(200).body(get_fake_resp());
+        });
+
+        let runtime = crate::core::runtime::test::init(None);
+
+        let grpc_reflection = GrpcReflection::new(
+            format!("http://localhost:{}", server.port()),
+            Some(vec![KeyValue {
+                key: "authorization".to_string(),
+                value: "Bearer 123".to_string(),
+            }]),
+            runtime,
+        );
+
+        let resp = grpc_reflection.list_all_files().await?;
+
+        assert_eq!(
+            [
+                "news.NewsService".to_string(),
+                "grpc.reflection.v1alpha.ServerReflection".to_string()
+            ]
+            .to_vec(),
+            resp
+        );
+
+        http_reflection_service_not_found.assert();
+
+        Ok(())
+    }
 }

tailcall-upstream-grpc/src/main.rs

@@ -216,16 +216,11 @@ fn init_tracer() -> Result<(), Error> {
     Ok(())
 }
 
+/// Intercepts the request and checks if the token is valid.
 fn intercept(req: Request<()>) -> Result<Request<()>, Status> {
-    println!("Intercepting request: {:?}", req);
-    if let Some(token) = req.metadata().get("authorization") {
-        if token != "secret" {
-            Err(Status::permission_denied("Unauthorized"))
-        } else {
-            Ok(req)
-        }
-    } else {
-        Err(Status::permission_denied("Unauthorized"))
+    match req.metadata().get("authorization") {
+        Some(token) if token == "Bearer 123" => Ok(req),
+        _ => Err(Status::permission_denied("Unauthorized")),
     }
 }