Added test in Spring GraphQL example app to test @RequestHeader (#2008)

Netflix · Sep 16, 2024 · 8069d80 · 8069d80
1 parent 2942c51
commit 8069d80
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/...-example-java/src/test/java/com/netflix/graphql/dgs/example/datafetcher/GreetingTest.java b/...-example-java/src/test/java/com/netflix/graphql/dgs/example/datafetcher/GreetingTest.java
@@ -174,4 +174,22 @@ void documentContextWithHeaders() {
         DocumentContext documentContext = queryExecutor.executeAndGetDocumentContext("query Hello($name: String){ hello(name: $name) }", Map.of("name", "DGS"));
         assertThat(documentContext.<String>read("data.hello")).isEqualTo("hello, DGS!");
     }
+
+    @Test
+    void withHeaders() {
+        MockHttpServletRequest servletRequest = new MockHttpServletRequest();
+        servletRequest.addHeader("demo-header", "demo-header-value");
+
+        String message = queryExecutor.executeAndExtractJsonPath("{demoHeader}", "data.demoHeader", new ServletWebRequest(servletRequest));
+        assertThat(message).isEqualTo("demo-header-value");
+    }
+
+    @Test
+    void withHeadersAndNoRequest() {
+        HttpHeaders headers = new HttpHeaders();
+        headers.add("demo-header", "demo-header-value");
+
+        String message = queryExecutor.executeAndExtractJsonPath("{demoHeader}", "data.demoHeader", headers);
+        assertThat(message).isEqualTo("demo-header-value");
+    }
 }