subscribe: improve coverage (#2355)

Author: IvanGoncharov

src/subscription/__tests__/subscribe-test.js

@@ -139,6 +139,7 @@ async function createSubscription(
 async function expectPromiseToThrow(promise, message) {
   try {
     await promise();
+    /* istanbul ignore next */
     expect.fail('promise should have thrown but did not');
   } catch (error) {
     expect(error && error.message).to.equal(message);
@@ -280,6 +281,7 @@ describe('Subscription Initialization Phase', () => {
         },
         nonImportantEmail: {
           type: EmailEventType,
+          /* istanbul ignore next (shouldn't be called) */
           subscribe() {
             didResolveNonImportantEmail = true;
             return eventEmitterAsyncIterator(new EventEmitter(), 'event');
@@ -516,27 +518,9 @@ describe('Subscription Initialization Phase', () => {
       }
     `);
 
-    const pubsub = new EventEmitter();
-    const rootValue = {
-      inbox: {
-        emails: [
-          {
-            from: '[email protected]',
-            subject: 'Hello',
-            message: 'Hello World',
-            unread: false,
-          },
-        ],
-      },
-      importantEmail() {
-        return eventEmitterAsyncIterator(pubsub, 'importantEmail');
-      },
-    };
-
     const result = await subscribe({
       schema: emailSchema,
       document: ast,
-      rootValue,
       variableValues: { priority: 'meow' },
     });
 
@@ -806,6 +790,68 @@ describe('Subscription Publish Phase', () => {
     });
   });
 
+  it('should not trigger when subscription is thrown', async () => {
+    const pubsub = new EventEmitter();
+    const { sendImportantEmail, subscription } = await createSubscription(
+      pubsub,
+    );
+    let payload = subscription.next();
+
+    // A new email arrives!
+    expect(
+      sendImportantEmail({
+        from: '[email protected]',
+        subject: 'Alright',
+        message: 'Tests are good',
+        unread: true,
+      }),
+    ).to.equal(true);
+
+    expect(await payload).to.deep.equal({
+      done: false,
+      value: {
+        data: {
+          importantEmail: {
+            email: {
+              from: '[email protected]',
+              subject: 'Alright',
+            },
+            inbox: {
+              unread: 1,
+              total: 2,
+            },
+          },
+        },
+      },
+    });
+
+    payload = subscription.next();
+
+    // Throw error
+    let caughtError;
+    try {
+      await subscription.throw('ouch');
+    } catch (e) {
+      caughtError = e;
+    }
+    expect(caughtError).to.equal('ouch');
+
+    // A new email arrives!
+    expect(
+      sendImportantEmail({
+        from: '[email protected]',
+        subject: 'Alright 2',
+        message: 'Tests are good 2',
+        unread: true,
+      }),
+    ).to.equal(false);
+
+    expect(await payload).to.deep.equal({
+      done: true,
+      value: undefined,
+    });
+  });
+
   it('event order is correct for multiple publishes', async () => {
     const pubsub = new EventEmitter();
     const { sendImportantEmail, subscription } = await createSubscription(