fbredius/storybook

/* eslint-disable no-unused-expressions */
/* eslint-disable jest/valid-expect */
type StorybookApps = 'official-storybook';

type Addons = 'Actions' | 'Knobs';

const getUrl = (route: string) => {
  const host = Cypress.env('location') || 'http://localhost:8001';

  return `${host}/${route}`;
};

export const visit = (route = '') => {
  cy.clearLocalStorage().visit(getUrl(route));
  return cy.get(`#storybook-preview-iframe`).then({ timeout: 15000 }, (iframe) => {
    return cy.wrap(iframe, { timeout: 10000 }).should(() => {
      const content: Document | null = (iframe[0] as HTMLIFrameElement).contentDocument;
      const element: HTMLElement | null = content !== null ? content.documentElement : null;

      expect(element).not.null;

      if (element !== null) {
        expect(element.querySelector('#root > *')).not.null;
      }
    });
  });
};

export const clickAddon = (addonName: Addons) => {
  return cy.get(`[role=tablist] button[role=tab]`).contains(addonName).click();
};

export const getStorybookPreview = () => {
  return cy.get(`#storybook-preview-iframe`).then({ timeout: 10000 }, (iframe) => {
    const content: Document | null = (iframe[0] as HTMLIFrameElement).contentDocument;
    const element: HTMLElement | null = content !== null ? content.documentElement : null;

    console.log({ element, content, iframe });

    return cy
      .wrap(iframe)
      .should(() => {
        expect(element).not.null;

        if (element !== null) {
          expect(element.querySelector('#root > *')).not.null;
        }
      })
      .then(() => {
        return cy.wrap(element).get('#root');
      });
  });
};