update graphql controller

Author: tvd12

ezyhttp-server-graphql/src/main/java/com/tvd12/ezyhttp/server/graphql/constants/GraphQLConstants.java

@@ -4,6 +4,12 @@ public final class GraphQLConstants {
 
     public static final String ALL_FIELDS = "*";
 
+    public static final String ARGUMENT_NAME_GRAPHQL_QUERY_NAME =
+        "graphQLQueryName";
+
+    public static final String ARGUMENT_NAME_GRAPHQL_QUERY_GROUP_NAME =
+        "graphQLQueryGroupName";
+
     public static final String DEFAULT_QL_GROUP_NAME = "default";
 
     public static final String PREFIX_QUERY = "query";

ezyhttp-server-graphql/src/main/java/com/tvd12/ezyhttp/server/graphql/controller/GraphQLController.java

@@ -28,6 +28,8 @@
 import java.util.Map;
 
 import static com.tvd12.ezyfox.io.EzyStrings.isNotBlank;
+import static com.tvd12.ezyhttp.server.graphql.constants.GraphQLConstants.ARGUMENT_NAME_GRAPHQL_QUERY_GROUP_NAME;
+import static com.tvd12.ezyhttp.server.graphql.constants.GraphQLConstants.ARGUMENT_NAME_GRAPHQL_QUERY_NAME;
 import static java.util.Collections.singletonMap;
 
 @Api
@@ -141,6 +143,18 @@ private Object fetch(
             String queryGroup = dataFetcherManager.getGroupNameByQueryName(
                 queryName
             );
+            arguments.setArgument(
+                ARGUMENT_NAME_GRAPHQL_QUERY_NAME,
+                queryName
+            );
+            arguments.setArgument(
+                ARGUMENT_NAME_GRAPHQL_QUERY_GROUP_NAME,
+                queryGroup
+            );
+            arguments.setArgument(
+                GraphQLDataFetcherManager.class,
+                dataFetcherManager
+            );
             for (GraphQLInterceptor interceptor : interceptors) {
                 boolean ok = interceptor.preHandle(
                     arguments,

ezyhttp-server-graphql/src/test/java/com/tvd12/ezyhttp/server/graphql/test/controller/GraphQLControllerTest.java

@@ -100,6 +100,11 @@ public void test() throws Exception {
             any(GraphQLDataFetcher.class)
         );
         verifyNoMoreInteractions(interceptor);
+
+        verify(arguments, times(6)).setArgument(
+            any(Object.class),
+            any(Object.class)
+        );
         verifyNoMoreInteractions(arguments);
     }
 
@@ -171,6 +176,11 @@ public void getAllFieldsTest() {
             any(GraphQLDataFetcher.class)
         );
         verifyNoMoreInteractions(interceptor);
+
+        verify(arguments, times(3)).setArgument(
+            any(Object.class),
+            any(Object.class)
+        );
         verifyNoMoreInteractions(arguments);
     }
 
@@ -246,6 +256,11 @@ public void getAllFriendFields() {
             any(GraphQLDataFetcher.class)
         );
         verifyNoMoreInteractions(interceptor);
+
+        verify(arguments, times(6)).setArgument(
+            any(Object.class),
+            any(Object.class)
+        );
         verifyNoMoreInteractions(arguments);
     }
 
@@ -331,6 +346,11 @@ public void testInterceptorFalse() {
             any(GraphQLDataFetcher.class)
         );
         verifyNoMoreInteractions(interceptor);
+
+        verify(arguments, times(3)).setArgument(
+            any(Object.class),
+            any(Object.class)
+        );
         verifyNoMoreInteractions(arguments);
     }
 
@@ -407,6 +427,11 @@ public void testQueryWithVariables() {
             any(GraphQLDataFetcher.class)
         );
         verifyNoMoreInteractions(interceptor);
+
+        verify(arguments, times(6)).setArgument(
+            any(Object.class),
+            any(Object.class)
+        );
         verifyNoMoreInteractions(arguments);
     }
 
@@ -476,6 +501,11 @@ public void testQueryWithNullVariableType() {
             any(GraphQLDataFetcher.class)
         );
         verifyNoMoreInteractions(interceptor);
+
+        verify(arguments, times(6)).setArgument(
+            any(Object.class),
+            any(Object.class)
+        );
         verifyNoMoreInteractions(arguments);
     }
 
@@ -536,6 +566,11 @@ public void testInvalidScheme() {
             any(GraphQLDataFetcher.class)
         );
         verifyNoMoreInteractions(interceptor);
+
+        verify(arguments, times(3)).setArgument(
+            any(Object.class),
+            any(Object.class)
+        );
         verifyNoMoreInteractions(arguments);
     }