fix test

andimarek · andimarek · commit 69ae51da0470 · 2017-05-24T10:01:40.000+02:00
add MapEnumValuesProvider for simple map based mappings
diff --git a/src/main/java/graphql/schema/idl/MapEnumValuesProvider.java b/src/main/java/graphql/schema/idl/MapEnumValuesProvider.java
@@ -0,0 +1,23 @@
+package graphql.schema.idl;
+
+import graphql.Assert;
+import graphql.PublicApi;
+
+import java.util.Map;
+
+@PublicApi
+public class MapEnumValuesProvider implements EnumValuesProvider {
+
+
+    private final Map<String, Object> values;
+
+    public MapEnumValuesProvider(Map<String, Object> values) {
+        Assert.assertNotNull(values, "values can't be null");
+        this.values = values;
+    }
+
+    @Override
+    public Object getValue(String name) {
+        return values.get(name);
+    }
+}
diff --git a/src/test/groovy/graphql/schema/DataFetcherSelectionTest.groovy b/src/test/groovy/graphql/schema/DataFetcherSelectionTest.groovy
@@ -5,6 +5,7 @@ import graphql.StarWarsData
 import graphql.execution.FieldCollector
 import graphql.language.AstPrinter
 import graphql.language.Field
+import graphql.schema.idl.MapEnumValuesProvider
 import graphql.schema.idl.RuntimeWiring
 import graphql.schema.idl.SchemaGenerator
 import graphql.schema.idl.SchemaParser
@@ -49,7 +50,7 @@ class DataFetcherSelectionTest extends Specification {
             //
             def selectionSet = environment.getSelectionSet().get()
 
-            if (! selectionSet.isEmpty()) {
+            if (!selectionSet.isEmpty()) {
                 String subSelection = captureSubSelection(selectionSet)
                 captureList.add(subSelection)
             }
@@ -73,6 +74,9 @@ class DataFetcherSelectionTest extends Specification {
         return new SelectionCapturingDataFetcher(delegate, captureList)
     }
 
+    def episodeValuesProvider = new MapEnumValuesProvider([NEWHOPE: 4, EMPIRE: 5, JEDI: 6])
+    def episodeWiring = newTypeWiring("Episode").enumValues(episodeValuesProvider).build()
+
     RuntimeWiring wiring = RuntimeWiring.newRuntimeWiring()
             .type(newTypeWiring("QueryType")
             .dataFetchers(
@@ -92,6 +96,7 @@ class DataFetcherSelectionTest extends Specification {
             .type(newTypeWiring("Character")
             .typeResolver(StarWarsData.getCharacterTypeResolver())
     )
+            .type(episodeWiring)
             .build()
 
     def executableStarWarsSchema = load("starWarsSchema.graphqls", wiring)
