attempt fix ampproject#3

dreamofabear · Dec 8, 2017 · 2006c8a · 2006c8a
1 parent 01bce3e
commit 2006c8a
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/test/functional/test-resources.js b/test/functional/test-resources.js
@@ -1473,10 +1473,7 @@ describe('Resources discoverWork', () => {
     resources.discoverWork_();
 
     expect(resource1.build).to.be.calledOnce;
-    return Promise.resolve().then(() => {
-      // Pass should be scheduled after successful resource build.
-      expect(resources.schedulePass).to.be.calledWithExactly();
-    });
+    return expect(resources.schedulePass).to.eventually.be.calledWithExactly();
   });
 
   it('should build resource when not built and before doc ready', () => {
@@ -1491,10 +1488,7 @@ describe('Resources discoverWork', () => {
     resources.discoverWork_();
 
     expect(resource1.build).to.be.calledOnce;
-    return Promise.resolve().then(() => {
-      // Pass should be scheduled after successful resource build.
-      expect(resources.schedulePass).to.be.calledWithExactly();
-    });
+    return expect(resources.schedulePass).to.eventually.be.calledWithExactly();
   });
 
   it('should NOT build non-prerenderable resources in prerender', () => {