wix-storybook-utils/src/AutoExample/testkit.js

Utilities for automated component documentation within Storybook

import React from 'react';
import { mount } from 'enzyme';

import { Option, Code } from './components';

export default class AutoExampleTestkit {
  constructor(AutoExample) {
    this.AutoExample = AutoExample;
  }

  component;

  when = {
    created: props => (this.component = mount(<this.AutoExample {...props} />)),
  };

  get = {
    options: () => this.component.find(Option),
    codeBlock: () => this.component.find(Code),
    exists: selector => this.component.exists(selector),
  };
}