Stabilize some tests

GoogleCloudPlatform · Jan 7, 2017 · 14945e0 · 14945e0
1 parent 028a5d7
commit 14945e0
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/pubsub/system-test/subscriptions.test.js b/pubsub/system-test/subscriptions.test.js
@@ -126,10 +126,12 @@ test.serial(`should pull ordered messages`, async (t) => {
   results = await pubsub.topic(topicNameTwo).publish({ data: expected, attributes: { counterId: '1' } }, { raw: true });
   results = await pubsub.topic(topicNameTwo).publish({ data: expected, attributes: { counterId: '2' } }, { raw: true });
   publishedMessageIds.push(results[0][0]);
-  await subscriptions.pullOrderedMessages(subscriptionNameThree);
-  t.is(console.log.callCount, 3);
-  t.deepEqual(console.log.secondCall.args, [`* %d %j %j`, publishedMessageIds[2], expected, { counterId: '2' }]);
-  t.deepEqual(console.log.thirdCall.args, [`* %d %j %j`, publishedMessageIds[0], expected, { counterId: '3' }]);
+  await tryTest(async () => {
+    await subscriptions.pullOrderedMessages(subscriptionNameThree);
+    t.is(console.log.callCount, 3);
+    t.deepEqual(console.log.secondCall.args, [`* %d %j %j`, publishedMessageIds[2], expected, { counterId: '2' }]);
+    t.deepEqual(console.log.thirdCall.args, [`* %d %j %j`, publishedMessageIds[0], expected, { counterId: '3' }]);
+  });
 });
 
 test.serial(`should set the IAM policy for a subscription`, async (t) => {