testing

Author: andimarek

graphql-java-spring-webmvc/src/test/java/graphql/spring/web/servlet/components/GraphQLControllerTest.java

@@ -1,8 +1,13 @@
 package graphql.spring.web.servlet.components;
 
+import graphql.ExecutionInput;
+import graphql.ExecutionResultImpl;
+import graphql.GraphQL;
 import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
+import org.mockito.ArgumentCaptor;
+import org.mockito.Mockito;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.http.MediaType;
 import org.springframework.test.context.ContextConfiguration;
@@ -13,7 +18,12 @@
 import org.springframework.test.web.servlet.setup.MockMvcBuilders;
 import org.springframework.web.context.WebApplicationContext;
 
+import java.util.LinkedHashMap;
+import java.util.Map;
+import java.util.concurrent.CompletableFuture;
+
 import static org.hamcrest.CoreMatchers.is;
+import static org.junit.Assert.assertThat;
 import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.asyncDispatch;
 import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
 import static org.springframework.test.web.servlet.result.MockMvcResultHandlers.print;
@@ -32,29 +42,50 @@ public class GraphQLControllerTest {
     @Autowired
     private WebApplicationContext wac;
 
+    @Autowired
+    GraphQL graphql;
+
     @Before
     public void setup() throws Exception {
         this.mockMvc = MockMvcBuilders.webAppContextSetup(this.wac).build();
     }
 
     @Test
     public void testGetRequest() throws Exception {
-        String query = "{foo}";
-        String variablesJson = "{\"key\":\"value\"}";
+        String variablesJson = "{\"variable\":\"variableValue\"}";
+        String query = "query myQuery {foo}";
+        String operationName = "myQuery";
+
+        ExecutionResultImpl executionResult = ExecutionResultImpl.newExecutionResult()
+                .data("bar")
+                .build();
+        CompletableFuture cf = CompletableFuture.completedFuture(executionResult);
+        ArgumentCaptor<ExecutionInput> captor = ArgumentCaptor.forClass(ExecutionInput.class);
+        Mockito.when(graphql.executeAsync(captor.capture())).thenReturn(cf);
+
 
         MvcResult mvcResult = this.mockMvc.perform(get("/graphql")
                 .param("query", query)
-                .param("variables", variablesJson))
+                .param("variables", variablesJson)
+                .param("operationName", operationName))
                 .andExpect(status().isOk())
                 .andExpect(request().asyncStarted())
                 .andReturn();
 
         this.mockMvc.perform(asyncDispatch(mvcResult))
                 .andDo(print()).andExpect(status().isOk())
                 .andExpect(content().contentType(MediaType.APPLICATION_JSON_UTF8))
-                .andExpect(jsonPath("data", is("foo")))
+                .andExpect(jsonPath("data", is("bar")))
                 .andReturn();
 
+        assertThat(captor.getAllValues().size(), is(1));
+
+        Map<String, Object> variables = new LinkedHashMap<>();
+        variables.put("variable", "variableValue");
+        assertThat(captor.getValue().getQuery(), is(query));
+        assertThat(captor.getValue().getVariables(), is(variables));
+        assertThat(captor.getValue().getOperationName(), is(operationName));
+
     }
 
 }

graphql-java-spring-webmvc/src/test/java/graphql/spring/web/servlet/components/TestAppConfig.java

@@ -1,18 +1,13 @@
 package graphql.spring.web.servlet.components;
 
 import com.fasterxml.jackson.databind.ObjectMapper;
-import graphql.ExecutionInput;
-import graphql.ExecutionResultImpl;
 import graphql.GraphQL;
-import org.mockito.ArgumentMatchers;
 import org.mockito.Mockito;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.ComponentScan;
 import org.springframework.context.annotation.Configuration;
 import org.springframework.web.servlet.config.annotation.EnableWebMvc;
 
-import java.util.concurrent.CompletableFuture;
-
 @Configuration
 @EnableWebMvc
 @ComponentScan(basePackageClasses = GraphQLController.class)
@@ -22,11 +17,6 @@ public class TestAppConfig {
     @Bean
     public GraphQL graphQL() {
         GraphQL graphql = Mockito.mock(GraphQL.class);
-        ExecutionResultImpl executionResult = ExecutionResultImpl.newExecutionResult()
-                .data("foo")
-                .build();
-        CompletableFuture cf = CompletableFuture.completedFuture(executionResult);
-        Mockito.when(graphql.executeAsync(ArgumentMatchers.<ExecutionInput>any())).thenReturn(cf);
         return graphql;
     }