add ID type

andimarek · andimarek · commit 3e968edf0c43 · 2015-08-15T11:55:27.000+02:00
diff --git a/src/main/java/graphql/Scalars.java b/src/main/java/graphql/Scalars.java
@@ -106,16 +106,21 @@ public Object coerceLiteral(Object input) {
     });
 
 
-//    public static GraphQLScalarType GraphQLID = new GraphQLScalarType("ID", "Built-in ID", new Coercing() {
-//        @Override
-//        public Object coerce(Object input) {
-//            return input;
-//        }
-//
-//        @Override
-//        public Object coerceLiteral(Object input) {
-//            return input;
-//        }
-//    });
+    public static GraphQLScalarType GraphQLID = new GraphQLScalarType("ID", "Built-in ID", new Coercing() {
+        @Override
+        public Object coerce(Object input) {
+            if(input instanceof String){
+                return input;
+            }
+
+            throw new GraphQLException();
+        }
+
+        @Override
+        public Object coerceLiteral(Object input) {
+            if (!(input instanceof StringValue)) return null;
+            return ((StringValue) input).getValue();
+        }
+    });
 
 }
diff --git a/src/test/groovy/graphql/ScalarsTest.groovy b/src/test/groovy/graphql/ScalarsTest.groovy
@@ -28,6 +28,24 @@ class ScalarsTest extends Specification {
         "test"        | "test"
     }
 
+    def "ID coerce literal"() {
+        expect:
+        Scalars.GraphQLID.getCoercing().coerceLiteral(literal) == result
+
+        where:
+        literal                 | result
+        new StringValue("5457486ABSBHS4w646") | "5457486ABSBHS4w646"
+    }
+
+    def "ID coerce object"() {
+        expect:
+        Scalars.GraphQLID.getCoercing().coerce(value) == result
+
+        where:
+        value         | result
+        "5457486ABSBHS4w646"        | "5457486ABSBHS4w646"
+    }
+
     def "Int coerce literal"() {
         expect:
         Scalars.GraphQLInt.getCoercing().coerceLiteral(literal) == result
