fix: add schema builder scalar test

Author: igdianov

graphql-jpa-query-schema/src/main/java/com/introproventures/graphql/jpa/query/schema/GraphQLSchemaBuilder.java

@@ -15,6 +15,7 @@
  */
 package com.introproventures.graphql.jpa.query.schema;
 
+import graphql.schema.GraphQLScalarType;
 import graphql.schema.GraphQLSchema;
 
 /**
@@ -57,7 +58,17 @@ public interface GraphQLSchemaBuilder {
      * @return this builder instance
      */
     GraphQLSchemaBuilder namingStrategy(NamingStrategy instance);
-    
+
+
+    /**
+     * Register Java type scalar
+     *
+     * @param javaType class
+     * @param scalarType GraphQL scalar instance
+     * @return this builder instance
+     */
+    GraphQLSchemaBuilder scalar(Class<?> javaType, GraphQLScalarType scalarType);
+
     /**
      * Builds {code #GraphQLSchema} instance
      * 

graphql-jpa-query-schema/src/main/java/com/introproventures/graphql/jpa/query/schema/impl/GraphQLJpaSchemaBuilder.java

@@ -24,6 +24,7 @@
 import java.util.Collection;
 import java.util.Collections;
 import java.util.HashMap;
+import java.util.LinkedHashMap;
 import java.util.List;
 import java.util.Map;
 import java.util.Optional;
@@ -133,6 +134,7 @@ public class GraphQLJpaSchemaBuilder implements GraphQLSchemaBuilder {
     private boolean enableResultStream = false;
 
     private RestrictedKeysProvider restrictedKeysProvider = (entityDescriptor) -> Optional.of(Collections.emptyList());
+    private Map<Class<?>, GraphQLScalarType> scalars = new LinkedHashMap<>();
 
     private final Relay relay = new Relay();
 
@@ -151,6 +153,8 @@ public GraphQLJpaSchemaBuilder(EntityManager entityManager) {
      */
     @Override
     public GraphQLSchema build() {
+        scalars.forEach((javaType, scalarType) -> JavaScalars.register(javaType, scalarType));
+
         GraphQLSchema.Builder schema = GraphQLSchema.newSchema()
                                                     .query(getQueryType());
 
@@ -166,7 +170,7 @@ public GraphQLSchema build() {
     }
 
     public GraphQLJpaSchemaBuilder scalar(Class<?> javaType, GraphQLScalarType scalarType) {
-        JavaScalars.register(javaType, scalarType);
+        scalars.put(javaType, scalarType);
 
         return this;
     }

graphql-jpa-query-schema/src/test/java/com/introproventures/graphql/jpa/query/schema/StarwarsSchemaBuildTest.java

@@ -16,10 +16,15 @@
 
 package com.introproventures.graphql.jpa.query.schema;
 
-import static org.assertj.core.api.Assertions.assertThat;
-
+import java.util.Optional;
 import javax.persistence.EntityManager;
-
+import com.introproventures.graphql.jpa.query.AbstractSpringBootTestSupport;
+import com.introproventures.graphql.jpa.query.schema.impl.GraphQLJpaSchemaBuilder;
+import graphql.Scalars;
+import graphql.schema.GraphQLInputObjectType;
+import graphql.schema.GraphQLObjectType;
+import graphql.schema.GraphQLScalarType;
+import graphql.schema.GraphQLSchema;
 import org.junit.Before;
 import org.junit.Test;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -28,13 +33,7 @@
 import org.springframework.boot.test.context.SpringBootTest;
 import org.springframework.context.annotation.Bean;
 
-import com.introproventures.graphql.jpa.query.AbstractSpringBootTestSupport;
-import com.introproventures.graphql.jpa.query.schema.impl.GraphQLJpaSchemaBuilder;
-
-import graphql.Scalars;
-import graphql.schema.GraphQLInputObjectType;
-import graphql.schema.GraphQLObjectType;
-import graphql.schema.GraphQLSchema;
+import static org.assertj.core.api.Assertions.assertThat;
 
 @SpringBootTest
 public class StarwarsSchemaBuildTest extends AbstractSpringBootTestSupport {
@@ -188,5 +187,25 @@ public void testBuildSchema(){
         //then
         assertThat(schema).isNotNull();
     }
+
+    @Test
+    public void scalar() {
+        // given
+        GraphQLScalarType scalarType = GraphQLScalarType.newScalar()
+                                                        .name("TestObject")
+                                                        .coercing(new JavaScalars.GraphQLObjectCoercing())
+                                                        .build();
+
+        // when
+        builder.scalar(Object.class, scalarType)
+               .build();
+
+        // then
+        Optional<GraphQLScalarType> result = JavaScalars.of(scalarType.getName());
+
+        assertThat(result).isNotEmpty()
+                          .get()
+                          .isEqualTo(scalarType);
+    }
 
 }