Use list types in schema generation

kensho-technologies · May 20, 2019 · 4974f79 · 4974f79
1 parent 2c30ca3
commit 4974f79
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 11 deletions.
diff --git a/graphql_compiler/schema_generation/graphql_schema.py b/graphql_compiler/schema_generation/graphql_schema.py
@@ -368,7 +368,7 @@ def get_graphql_schema_from_schema_graph(schema_graph, class_to_field_type_overr
     # Create the root query GraphQL type. Consists of all non-union classes, i.e.
     # all non-abstract classes (as GraphQL types) and all abstract classes (as GraphQL interfaces).
     RootSchemaQuery = GraphQLObjectType('RootSchemaQuery', OrderedDict([
-        (name, GraphQLField(value))
+        (name, GraphQLField(GraphQLList(value)))
         for name, value in sorted(six.iteritems(graphql_types), key=lambda x: x[0])
         if not isinstance(value, GraphQLUnionType)
     ]))

diff --git a/graphql_compiler/tests/test_helpers.py b/graphql_compiler/tests/test_helpers.py
@@ -157,16 +157,16 @@
     }
 
     type RootSchemaQuery {
-        Animal: Animal
-        BirthEvent: BirthEvent
-        Entity: Entity
-        Event: Event
-        FeedingEvent: FeedingEvent
-        Food: Food
-        FoodOrSpecies: FoodOrSpecies
-        Location: Location
-        Species: Species
-        UniquelyIdentifiable: UniquelyIdentifiable
+        Animal: [Animal]
+        BirthEvent: [BirthEvent]
+        Entity: [Entity]
+        Event: [Event]
+        FeedingEvent: [FeedingEvent]
+        Food: [Food]
+        FoodOrSpecies: [FoodOrSpecies]
+        Location: [Location]
+        Species: [Species]
+        UniquelyIdentifiable: [UniquelyIdentifiable]
     }
 
     type Species implements Entity, UniquelyIdentifiable {