add tests for customizer

Author: andimarek

build.gradle

@@ -43,6 +43,7 @@ subprojects {
         springVersion = "5.1.2.RELEASE"
         springBootVersion = "2.1.0.RELEASE"
         jacksonVersion = "2.9.6"
+        assertJVersion = "3.11.1"
     }
 
     repositories {

graphql-java-spring-webflux/build.gradle

@@ -6,6 +6,7 @@ dependencies {
     compile "com.fasterxml.jackson.core:jackson-databind:$jacksonVersion"
     compile "com.graphql-java:graphql-java:$graphqlJavaVersion"
 
+    testCompile("org.assertj:assertj-core:$assertJVersion")
     testCompile group: 'junit', name: 'junit', version: '4.12'
     testCompile "org.springframework:spring-test:$springVersion"
     testCompile group: 'javax.servlet', name: 'javax.servlet-api', version: '4.0.1'

graphql-java-spring-webflux/src/main/java/graphql/spring/web/reactive/GraphQLInvocation.java

@@ -8,6 +8,6 @@
 @PublicApi
 public interface GraphQLInvocation {
 
-    Mono<ExecutionResult> invoke(GraphQLInvocationData invocationData, ServerWebExchange webRequest);
+    Mono<ExecutionResult> invoke(GraphQLInvocationData invocationData, ServerWebExchange serverWebExchange);
 
 }

graphql-java-spring-webflux/src/main/java/graphql/spring/web/reactive/components/DefaultGraphQLInvocation.java

@@ -17,7 +17,7 @@
 public class DefaultGraphQLInvocation implements GraphQLInvocation {
 
     @Autowired
-    private GraphQL graphQL;
+    GraphQL graphQL;
 
     @Autowired
     ExecutionInputCustomizer executionInputCustomizer;

graphql-java-spring-webflux/src/test/java/graphql/spring/web/reactive/components/DefaultGraphQLInvocationTest.java

@@ -0,0 +1,62 @@
+package graphql.spring.web.reactive.components;
+
+import graphql.ExecutionInput;
+import graphql.ExecutionResult;
+import graphql.GraphQL;
+import graphql.spring.web.reactive.ExecutionInputCustomizer;
+import graphql.spring.web.reactive.GraphQLInvocationData;
+import org.junit.Test;
+import org.mockito.ArgumentCaptor;
+import org.springframework.web.server.ServerWebExchange;
+import reactor.core.publisher.Mono;
+
+import java.util.LinkedHashMap;
+import java.util.Map;
+
+import static java.util.concurrent.CompletableFuture.completedFuture;
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+public class DefaultGraphQLInvocationTest {
+
+
+    @Test
+    public void testCustomizerIsCalled() {
+
+        String query = "query myQuery {foo}";
+        String operationName = "myQuery";
+        Map<String, Object> variables = new LinkedHashMap<>();
+
+        DefaultGraphQLInvocation defaultGraphQLInvocation = new DefaultGraphQLInvocation();
+        ExecutionInputCustomizer executionInputCustomizer = mock(ExecutionInputCustomizer.class);
+        defaultGraphQLInvocation.executionInputCustomizer = executionInputCustomizer;
+        GraphQL graphQL = mock(GraphQL.class);
+        defaultGraphQLInvocation.graphQL = graphQL;
+        ExecutionResult executionResult = mock(ExecutionResult.class);
+        when(graphQL.executeAsync(any(ExecutionInput.class))).thenReturn(completedFuture(executionResult));
+
+        GraphQLInvocationData graphQLInvocationData = new GraphQLInvocationData(query, operationName, variables);
+        ServerWebExchange serverWebExchange = mock(ServerWebExchange.class);
+
+        ArgumentCaptor<ExecutionInput> captor1 = ArgumentCaptor.forClass(ExecutionInput.class);
+        ArgumentCaptor<ServerWebExchange> captor2 = ArgumentCaptor.forClass(ServerWebExchange.class);
+        ExecutionInput executionInputResult = mock(ExecutionInput.class);
+        when(executionInputCustomizer.customizeExecutionInput(captor1.capture(), captor2.capture())).thenReturn(Mono.just(executionInputResult));
+
+        Mono<ExecutionResult> invoke = defaultGraphQLInvocation.invoke(graphQLInvocationData, serverWebExchange);
+
+        assertThat(captor1.getValue().getQuery()).isEqualTo(query);
+        assertThat(captor1.getValue().getOperationName()).isEqualTo(operationName);
+        assertThat(captor1.getValue().getVariables()).isSameAs(variables);
+
+        invoke.block();
+
+        verify(graphQL).executeAsync(executionInputResult);
+
+    }
+
+
+}

graphql-java-spring-webmvc/build.gradle

@@ -6,6 +6,7 @@ dependencies {
     compile "com.fasterxml.jackson.core:jackson-databind:$jacksonVersion"
     compile "com.graphql-java:graphql-java:$graphqlJavaVersion"
 
+    testCompile("org.assertj:assertj-core:$assertJVersion")
     testCompile group: 'junit', name: 'junit', version: '4.12'
     testCompile "org.springframework:spring-test:$springVersion"
     testCompile group: 'javax.servlet', name: 'javax.servlet-api', version: '4.0.1'

graphql-java-spring-webmvc/src/main/java/graphql/spring/web/servlet/components/DefaultGraphQLInvocation.java

@@ -18,7 +18,7 @@
 public class DefaultGraphQLInvocation implements GraphQLInvocation {
 
     @Autowired
-    private GraphQL graphQL;
+    GraphQL graphQL;
 
     @Autowired
     ExecutionInputCustomizer executionInputCustomizer;

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

@@ -0,0 +1,59 @@
+package graphql.spring.web.servlet.components;
+
+import graphql.ExecutionInput;
+import graphql.ExecutionResult;
+import graphql.GraphQL;
+import graphql.spring.web.servlet.ExecutionInputCustomizer;
+import graphql.spring.web.servlet.GraphQLInvocationData;
+import org.junit.Test;
+import org.mockito.ArgumentCaptor;
+import org.springframework.web.context.request.WebRequest;
+
+import java.util.LinkedHashMap;
+import java.util.Map;
+import java.util.concurrent.CompletableFuture;
+
+import static java.util.concurrent.CompletableFuture.completedFuture;
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+public class DefaultGraphQLInvocationTest {
+
+
+    @Test
+    public void testCustomizerIsCalled() {
+
+        String query = "query myQuery {foo}";
+        String operationName = "myQuery";
+        Map<String, Object> variables = new LinkedHashMap<>();
+
+        DefaultGraphQLInvocation defaultGraphQLInvocation = new DefaultGraphQLInvocation();
+        ExecutionInputCustomizer executionInputCustomizer = mock(ExecutionInputCustomizer.class);
+        defaultGraphQLInvocation.executionInputCustomizer = executionInputCustomizer;
+        GraphQL graphQL = mock(GraphQL.class);
+        defaultGraphQLInvocation.graphQL = graphQL;
+        ExecutionResult executionResult = mock(ExecutionResult.class);
+        when(graphQL.executeAsync(any(ExecutionInput.class))).thenReturn(completedFuture(executionResult));
+
+        GraphQLInvocationData graphQLInvocationData = new GraphQLInvocationData(query, operationName, variables);
+        WebRequest webRequest = mock(WebRequest.class);
+
+        ArgumentCaptor<ExecutionInput> captor1 = ArgumentCaptor.forClass(ExecutionInput.class);
+        ArgumentCaptor<WebRequest> captor2 = ArgumentCaptor.forClass(WebRequest.class);
+        ExecutionInput executionInputResult = mock(ExecutionInput.class);
+        when(executionInputCustomizer.customizeExecutionInput(captor1.capture(), captor2.capture())).thenReturn(completedFuture(executionInputResult));
+
+        CompletableFuture<ExecutionResult> invoke = defaultGraphQLInvocation.invoke(graphQLInvocationData, webRequest);
+
+        assertThat(captor1.getValue().getQuery()).isEqualTo(query);
+        assertThat(captor1.getValue().getOperationName()).isEqualTo(operationName);
+        assertThat(captor1.getValue().getVariables()).isSameAs(variables);
+
+        verify(graphQL).executeAsync(executionInputResult);
+
+    }
+
+}