unit test graphql

Author: tvd12

ezyhttp-server-graphql/src/main/java/com/tvd12/ezyhttp/server/graphql/data/GraphQLError.java

@@ -1,16 +1,86 @@
 package com.tvd12.ezyhttp.server.graphql.data;
 
-import lombok.Builder;
+import com.tvd12.ezyfox.builder.EzyBuilder;
 import lombok.Getter;
 
-import java.util.List;
-import java.util.Map;
+import java.util.*;
 
 @Getter
-@Builder
 public class GraphQLError {
-    private String message;
-    private List<Map<String, Object>> locations;
-    private List<String> path;
-    private Map<String, Object> extensions;
+
+    private final String message;
+    private final List<Map<String, Object>> locations;
+    private final List<String> path;
+    private final Map<String, Object> extensions;
+
+    protected GraphQLError(Builder builder) {
+        this.message = builder.message;
+        this.locations = builder.locations;
+        this.path = builder.path;
+        this.extensions = builder.extensions;
+    }
+
+    public static Builder builder() {
+        return new Builder();
+    }
+
+    public static class Builder implements EzyBuilder<GraphQLError> {
+        protected String message;
+        protected List<Map<String, Object>> locations;
+        protected List<String> path;
+        protected Map<String, Object> extensions;
+
+        public Builder message(String message) {
+            this.message = message;
+            return this;
+        }
+
+        public Builder location(Map<String, Object> location) {
+            if (locations == null) {
+                locations = new ArrayList<>();
+            }
+            locations.add(location);
+            return this;
+        }
+
+        public Builder locations(
+            List<Map<String, Object>> locations
+        ) {
+            if (this.locations == null) {
+                this.locations = new ArrayList<>();
+            }
+            this.locations.addAll(locations);
+            return this;
+        }
+
+        public Builder path(String... paths) {
+            return path(Arrays.asList(paths));
+        }
+
+        public Builder path(List<String> path) {
+            this.path = path;
+            return this;
+        }
+
+        public Builder extension(String key, Object value) {
+            if (extensions == null) {
+                extensions = new HashMap<>();
+            }
+            extensions.put(key, value);
+            return this;
+        }
+
+        public Builder extensions(Map<String, Object> extensions) {
+            if (this.extensions == null) {
+                this.extensions = new HashMap<>();
+            }
+            this.extensions.putAll(extensions);
+            return this;
+        }
+
+        @Override
+        public GraphQLError build() {
+            return new GraphQLError(this);
+        }
+    }
 }

ezyhttp-server-graphql/src/main/java/com/tvd12/ezyhttp/server/graphql/exception/GraphQLFetcherException.java

@@ -1,8 +1,10 @@
 package com.tvd12.ezyhttp.server.graphql.exception;
 
+import com.tvd12.ezyfox.builder.EzyBuilder;
 import com.tvd12.ezyhttp.server.graphql.data.GraphQLError;
 import lombok.Getter;
 
+import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
@@ -13,19 +15,10 @@ public class GraphQLFetcherException extends RuntimeException {
     private final Map<String, Object> data;
     private final List<GraphQLError> errors;
 
-    public GraphQLFetcherException(
-        List<GraphQLError> errors
-    ) {
-        this(null, errors);
-    }
-
-    public GraphQLFetcherException(
-        Map<String, Object> data,
-        List<GraphQLError> errors
-    ) {
-        super("data: " + data + ", errors: " + errors);
-        this.data = data;
-        this.errors = errors;
+    protected GraphQLFetcherException(Builder builder) {
+        super("data: " + builder.data + ", errors: " + builder.errors);
+        this.data = builder.data;
+        this.errors = builder.errors;
     }
 
     public Map<String, Object> toDataMap() {
@@ -36,4 +29,65 @@ public Map<String, Object> toDataMap() {
         answer.put("errors", errors);
         return answer;
     }
+
+    public static Builder builder() {
+        return new Builder();
+    }
+
+    public static class Builder implements EzyBuilder<GraphQLFetcherException> {
+        protected Map<String, Object> data;
+        protected List<GraphQLError> errors;
+
+        public Builder data(Map<String, Object> data) {
+            if (this.data == null) {
+                this.data = new HashMap<>();
+            }
+            this.data.putAll(data);
+            return this;
+        }
+
+        @SuppressWarnings("unchecked")
+        public Builder errorDataFieldAndPaths(
+            String errorDataField,
+            String... paths
+        ) {
+            if (data == null) {
+                data = new HashMap<>();
+            }
+            Map<String, Object> lastObject = data;
+            for (String path : paths) {
+                Object object = lastObject.get(path);
+                if (object instanceof Map) {
+                    lastObject = (Map<String, Object>) object;
+                } else {
+                    Map<String, Object> map = new HashMap<>();
+                    lastObject.put(path, map);
+                    lastObject = map;
+                }
+            }
+            lastObject.put(errorDataField, null);
+            return this;
+        }
+
+        public Builder error(GraphQLError error) {
+            if (errors == null) {
+                errors = new ArrayList<>();
+            }
+            errors.add(error);
+            return this;
+        }
+
+        public Builder errors(List<GraphQLError> errors) {
+            if (this.errors == null) {
+                this.errors = new ArrayList<>();
+            }
+            this.errors.addAll(errors);
+            return this;
+        }
+
+        @Override
+        public GraphQLFetcherException build() {
+            return new GraphQLFetcherException(this);
+        }
+    }
 }

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

@@ -0,0 +1,85 @@
+package com.tvd12.ezyhttp.server.graphql.test.controller;
+
+import com.tvd12.ezyfox.util.EzyMapBuilder;
+import com.tvd12.ezyhttp.server.graphql.controller.GraphQLExceptionHandler;
+import com.tvd12.ezyhttp.server.graphql.data.GraphQLError;
+import com.tvd12.ezyhttp.server.graphql.exception.GraphQLFetcherException;
+import com.tvd12.ezyhttp.server.graphql.exception.GraphQLInvalidSchemeException;
+import com.tvd12.ezyhttp.server.graphql.exception.GraphQLObjectMapperException;
+import com.tvd12.test.assertion.Asserts;
+import org.testng.annotations.Test;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+public class GraphQLExceptionHandlerTest {
+
+    @Test
+    public void handleGraphQLFetcherExceptionTest() {
+        // given
+        Map<String, Object> data = new HashMap<>();
+        List<GraphQLError> errors = new ArrayList<>();
+        GraphQLFetcherException exception = GraphQLFetcherException.builder()
+            .data(data)
+            .errors(errors)
+            .build();
+
+        GraphQLExceptionHandler instance = new GraphQLExceptionHandler();
+
+        // when
+        Object actual = instance.handle(exception);
+
+        // then
+        Asserts.assertEquals(
+            actual,
+            EzyMapBuilder.mapBuilder()
+                .put("data", data)
+                .put("errors", errors)
+                .toMap(),
+            false
+        );
+    }
+
+    @Test
+    public void handleGraphQLInvalidSchemeExceptionTest() {
+        // given
+        List<GraphQLError> errors = new ArrayList<>();
+        GraphQLInvalidSchemeException exception = new GraphQLInvalidSchemeException(
+            errors
+        );
+
+        GraphQLExceptionHandler instance = new GraphQLExceptionHandler();
+
+        // when
+        Object actual = instance.handle(exception);
+
+        // then
+        Asserts.assertEquals(
+            actual,
+            errors
+        );
+    }
+
+    @Test
+    public void handleGraphQLObjectMapperExceptionTest() {
+        // given
+        List<GraphQLError> errors = new ArrayList<>();
+        GraphQLObjectMapperException exception = new GraphQLObjectMapperException(
+            errors,
+            new Exception("test")
+        );
+
+        GraphQLExceptionHandler instance = new GraphQLExceptionHandler();
+
+        // when
+        Object actual = instance.handle(exception);
+
+        // then
+        Asserts.assertEquals(
+            actual,
+            errors
+        );
+    }
+}