test: fix wrapper test

histoire-dev · Sep 30, 2023 · 967a6fb · 967a6fb
1 parent f8e6326
commit 967a6fb
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 15 deletions.
diff --git a/examples/vue3/cypress/e2e/controls.cy.js b/examples/vue3/cypress/e2e/controls.cy.js
@@ -1,7 +1,9 @@
 /// <reference types="cypress" />
 
 describe('Controls', () => {
-  const getIframeBody = () => cy.get('iframe[data-test-id="preview-iframe"]').iframe()
+  const getIframeBody = () => cy.get('iframe[data-test-id="preview-iframe"]')
+    .its('0.contentDocument.body').should('not.be.empty')
+    .then(cy.wrap)
 
   beforeEach(() => {
     cy.visit('/')

diff --git a/examples/vue3/cypress/e2e/wrapper.cy.js b/examples/vue3/cypress/e2e/wrapper.cy.js
@@ -1,7 +1,9 @@
 /// <reference types="cypress" />
 
 describe('Wrapper', () => {
-  const getIframeBody = () => cy.get('iframe[data-test-id="preview-iframe"]').iframe()
+  const getIframeBody = () => cy.get('iframe[data-test-id="preview-iframe"]')
+    .its('0.contentDocument.body').should('not.be.empty')
+    .then(cy.wrap)
 
   it('should display the wrapper', () => {
     cy.visit('/story/src-components-meow-story-vue?variantId=src-components-meow-story-vue-0')

diff --git a/examples/vue3/cypress/support/commands.js b/examples/vue3/cypress/support/commands.js
@@ -23,16 +23,3 @@
 //
 // -- This will overwrite an existing command --
 // Cypress.Commands.overwrite('visit', (originalFn, url, options) => { ... })
-
-Cypress.Commands.add('iframe', { prevSubject: 'element' }, $iframe => {
-  return new Cypress.Promise(resolve => {
-    const doc = $iframe.prop('contentDocument')
-    if (doc && doc.readyState === 'complete' && doc.body.innerText) {
-      resolve($iframe.contents().find('body'))
-    } else {
-      $iframe.on('load', () => {
-        resolve($iframe.contents().find('body'))
-      })
-    }
-  })
-})