Custom scalar input

examples/spring/src/main/kotlin/com/expediagroup/graphql/examples/query/ScalarQuery.kt

@@ -32,6 +32,9 @@ class ScalarQuery : Query {
     @GraphQLDescription("generates random UUID")
     fun generateRandomUUID() = UUID.randomUUID()
 
+    @GraphQLDescription("Prints a string with a custom scalar as input")
+    fun printUuids(uuids: List<UUID>) = "You sent $uuids"
+
     fun findPersonById(@GraphQLID id: String) = Person(id, "Nelson")
 }
 

graphql-kotlin-schema-generator/src/test/kotlin/com/expediagroup/graphql/hooks/SchemaGeneratorHooksTest.kt

@@ -23,10 +23,9 @@ import com.expediagroup.graphql.exceptions.EmptyInterfaceTypeException
 import com.expediagroup.graphql.exceptions.EmptyObjectTypeException
 import com.expediagroup.graphql.generator.extensions.getSimpleName
 import com.expediagroup.graphql.getTestSchemaConfigWithHooks
+import com.expediagroup.graphql.test.utils.graphqlUUIDType
 import com.expediagroup.graphql.testSchemaConfig
 import com.expediagroup.graphql.toSchema
-import graphql.language.StringValue
-import graphql.schema.Coercing
 import graphql.schema.GraphQLFieldDefinition
 import graphql.schema.GraphQLInterfaceType
 import graphql.schema.GraphQLObjectType
@@ -328,21 +327,4 @@ class SchemaGeneratorHooksTest {
     }
 
     class EmptyImplementation(override val id: String) : EmptyInterface
-
-    private val graphqlUUIDType = GraphQLScalarType.newScalar()
-        .name("UUID")
-        .description("A type representing a formatted java.util.UUID")
-        .coercing(UUIDCoercing)
-        .build()
-
-    private object UUIDCoercing : Coercing<UUID, String> {
-        override fun parseValue(input: Any?): UUID = UUID.fromString(serialize(input))
-
-        override fun parseLiteral(input: Any?): UUID? {
-            val uuidString = (input as? StringValue)?.value
-            return UUID.fromString(uuidString)
-        }
-
-        override fun serialize(dataFetcherResult: Any?): String = dataFetcherResult.toString()
-    }
 }

graphql-kotlin-schema-generator/src/test/kotlin/com/expediagroup/graphql/test/integration/CustomScalarExecutionTest.kt

@@ -0,0 +1,88 @@
+/*
+ * Copyright 2020 Expedia, Inc
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.expediagroup.graphql.test.integration
+
+import com.expediagroup.graphql.TopLevelObject
+import com.expediagroup.graphql.getTestSchemaConfigWithHooks
+import com.expediagroup.graphql.hooks.SchemaGeneratorHooks
+import com.expediagroup.graphql.test.utils.graphqlUUIDType
+import com.expediagroup.graphql.toSchema
+import graphql.GraphQL
+import graphql.schema.GraphQLType
+import org.junit.jupiter.api.Test
+import java.util.UUID
+import kotlin.reflect.KClass
+import kotlin.reflect.KType
+import kotlin.test.assertFails
+import kotlin.test.assertNotNull
+
+class CustomScalarExecutionTest {
+
+    private val schema = toSchema(
+        queries = listOf(TopLevelObject(QueryObject())),
+        config = getTestSchemaConfigWithHooks(CustomScalarHooks())
+    )
+    private val graphQL: GraphQL = GraphQL.newGraphQL(schema).build()
+
+    @Test
+    fun `a custom scalar can be used as output`() {
+        val result = graphQL.execute("{ uuidOutput }")
+        val data: Map<String, String>? = result.getData()
+
+        assertNotNull(data?.get("uuidOutput"))
+    }
+
+    @Test
+    fun `a custom scalar can be used as input`() {
+        val result = graphQL.execute("{ uuidInput(uuid: \"22dba784-f7c1-471f-8e8c-84e3a59bb5b5\") }")
+        val data: Map<String, String>? = result.getData()
+
+        assertNotNull(data?.get("uuidInput"))
+    }
+
+    @Test
+    fun `an invalid custom scalar as input will throw an error`() {
+        assertFails {
+            graphQL.execute("{ uuidInput(uuid: \"foo\") }")
+        }
+    }
+
+    @Test
+    fun `a custom scalar can be used as input in a list`() {
+        val result = graphQL.execute("{ uuidListInput(uuids: [\"435f5808-936b-40ab-89b6-ec4e8cd1ba36\"]) }")
+        val data: Map<String, String>? = result.getData()
+
+        assertNotNull(data?.get("uuidListInput"))
+    }
+
+    class QueryObject {
+        fun uuidOutput(): UUID = UUID.randomUUID()
+
+        fun uuidInput(uuid: UUID) = "You sent $uuid"
+
+        fun uuidListInput(uuids: List<UUID>) = "You sent ${uuids.size} items"
+    }
+
+    class CustomScalarHooks : SchemaGeneratorHooks {
+        override fun willGenerateGraphQLType(type: KType): GraphQLType? {
+            return when (type.classifier as? KClass<*>) {
+                UUID::class -> graphqlUUIDType
+                else -> null
+            }
+        }
+    }
+}

graphql-kotlin-schema-generator/src/test/kotlin/com/expediagroup/graphql/test/utils/graphqlUuidType.kt

@@ -0,0 +1,44 @@
+/*
+ * Copyright 2020 Expedia, Inc
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.expediagroup.graphql.test.utils
+
+import graphql.language.StringValue
+import graphql.schema.Coercing
+import graphql.schema.GraphQLScalarType
+import java.util.UUID
+
+/**
+ * Custom scalar to parse a string as a [UUID] for both input and output.
+ */
+internal val graphqlUUIDType = GraphQLScalarType.newScalar()
+    .name("UUID")
+    .description("A type representing a formatted java.util.UUID")
+    .coercing(UUIDCoercing)
+    .build()
+
+private object UUIDCoercing : Coercing<UUID, String> {
+    override fun parseValue(input: Any?): UUID = UUID.fromString(serialize(input))
+
+    override fun parseLiteral(input: Any?): UUID? {
+        val uuidString: String? = (input as? StringValue)?.value
+        return if (uuidString != null) {
+            UUID.fromString(uuidString)
+        } else null
+    }
+
+    override fun serialize(dataFetcherResult: Any?): String = dataFetcherResult.toString()
+}