emeraldjava
diff --git a/‎src/test/groovy/graphql/parser/IDLParserTest.groovy‎
Lines changed: 3 additions & 0 deletions b/‎src/test/groovy/graphql/parser/IDLParserTest.groovy‎
Lines changed: 3 additions & 0 deletions
diff --git a/‎src/test/groovy/graphql/schema/idl/SchemaGeneratorTest.groovy‎
Lines changed: 23 additions & 0 deletions b/‎src/test/groovy/graphql/schema/idl/SchemaGeneratorTest.groovy‎
Lines changed: 23 additions & 0 deletions
@@ -282,6 +282,7 @@ three: [Number] @three
 schema @d1 @d2 {
     query: OpType1
     mutation: OpType2
+    subscription: OpType3
 }
 """
 
@@ -293,6 +294,8 @@ schema @d1 @d2 {
                 .add(new OperationTypeDefinition("query", new TypeName("OpType1")))
         schema.getOperationTypeDefinitions()
                 .add(new OperationTypeDefinition("mutation", new TypeName("OpType2")))
+        schema.getOperationTypeDefinitions()
+                .add(new OperationTypeDefinition("subscription", new TypeName("OpType3")))
 
         when:
         def document = new Parser().parseDocument(input)
 
@@ -654,4 +654,27 @@ class SchemaGeneratorTest extends Specification {
         def err = thrown(NotAnOutputTypeError.class)
         err.message == "expected OutputType, but found CharacterInput type"
     }
+
+    def "schema with subscription"() {
+        given:
+        def spec = """
+            schema {
+                query: Query
+                subscription: Subscription
+            }
+            type Query {
+                foo: String
+            }
+            
+            type Subscription {
+                foo: String 
+            }
+            """
+        when:
+        def wiring = RuntimeWiring.newRuntimeWiring()
+                .build()
+        def schema = generateSchema(spec, wiring)
+        then:
+        schema.getSubscriptionType().name == "Subscription"
+    }
 }