storybook/addons/a11y/src/components/WrapStory.js

import { Component } from 'react';
import { findDOMNode } from 'react-dom';
import PropTypes from 'prop-types';
import axe from 'axe-core';

class WrapStory extends Component {
  static propTypes = {
    context: PropTypes.shape({}),
    storyFn: PropTypes.func,
    channel: PropTypes.shape({}),
  };
  static defaultProps = {
    context: {},
    storyFn: () => {},
    channel: {},
  };

  constructor(props) {
    super(props);
    this.runA11yCheck = this.runA11yCheck.bind(this);
  }

  componentDidMount() {
    const { channel } = this.props;
    channel.on('addon:a11y:rerun', this.runA11yCheck);
    this.runA11yCheck();
  }

  componentWillUnmount() {
    const { channel } = this.props;
    channel.removeListener('addon:a11y:rerun', this.runA11yCheck);
  }

  /* eslint-disable react/no-find-dom-node */
  runA11yCheck() {
    const { channel } = this.props;
    const wrapper = findDOMNode(this);

    if (wrapper !== null) {
      axe.a11yCheck(wrapper, {}, results => {
        channel.emit('addon:a11y:check', results);
      });
    }
  }

  render() {
    const { storyFn, context } = this.props;

    return storyFn(context);
  }
}

export default WrapStory;