Fix visState integration tests, Part 2 (#9815)

* Fix part 2 * Skip on Safari
ampproject · Jun 9, 2017 · 73bb468 · 73bb468
1 parent 79cd9eb
commit 73bb468
Showing 1 changed file with 12 additions and 8 deletions.
diff --git a/test/integration/test-visibility-states.js b/test/integration/test-visibility-states.js
@@ -19,6 +19,7 @@ import {documentStateFor} from '../../src/service/document-state';
 import {resourcesForDoc} from '../../src/services';
 import {VisibilityState} from '../../src/visibility-state';
 import {getVendorJsPropertyName} from '../../src/style';
+import {whenUpgradedToCustomElement} from '../../src/dom';
 import {createCustomEvent} from '../../src/event-helper';
 
 describe.configure().retryOnSaucelabs().run('Viewer Visibility State', () => {
@@ -105,6 +106,8 @@ describe.configure().retryOnSaucelabs().run('Viewer Visibility State', () => {
         img.setAttribute('layout', 'fixed');
         win.document.body.appendChild(img);
 
+        return whenUpgradedToCustomElement(img);
+      }).then(img => {
         layoutCallback = sandbox.stub(img.implementation_, 'layoutCallback');
         unlayoutCallback = sandbox.stub(img.implementation_,
             'unlayoutCallback');
@@ -136,14 +139,15 @@ describe.configure().retryOnSaucelabs().run('Viewer Visibility State', () => {
           setupSpys();
         });
 
-        it('does layout when going to PRERENDER', () => {
-          return waitForNextPass().then(() => {
-            expect(layoutCallback).to.have.been.called;
-            expect(unlayoutCallback).not.to.have.been.called;
-            expect(pauseCallback).not.to.have.been.called;
-            expect(resumeCallback).not.to.have.been.called;
-          });
-        });
+        it.configure().skipSafari().run('does layout when going to PRERENDER',
+            () => {
+              return waitForNextPass().then(() => {
+                expect(layoutCallback).to.have.been.called;
+                expect(unlayoutCallback).not.to.have.been.called;
+                expect(pauseCallback).not.to.have.been.called;
+                expect(resumeCallback).not.to.have.been.called;
+              });
+            });
 
         it('calls layout when going to VISIBLE', () => {
           viewer.receiveMessage('visibilitychange',