From 8069d80bee4fc2be770f93bb64a040dae44d0bbe Mon Sep 17 00:00:00 2001 From: Paul Bakker Date: Sun, 15 Sep 2024 20:11:18 -0700 Subject: [PATCH] Added test in Spring GraphQL example app to test @RequestHeader (#2008) --- .../dgs/example/datafetcher/GreetingTest.java | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/graphql-dgs-spring-graphql-example-java/src/test/java/com/netflix/graphql/dgs/example/datafetcher/GreetingTest.java b/graphql-dgs-spring-graphql-example-java/src/test/java/com/netflix/graphql/dgs/example/datafetcher/GreetingTest.java index a3e6dfbea..bf3ad18e0 100644 --- a/graphql-dgs-spring-graphql-example-java/src/test/java/com/netflix/graphql/dgs/example/datafetcher/GreetingTest.java +++ b/graphql-dgs-spring-graphql-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.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"); + } }