Merge pull request #2279 from storybooks/fix-background-propType

Fix propTypes in addon-background
storybookjs · Nov 9, 2017 · 16cb94d · 16cb94d
2 parents a8dc0fd + 3baba88
commit 16cb94d
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/addons/background/src/BackgroundPanel.js b/addons/background/src/BackgroundPanel.js
@@ -1,7 +1,6 @@
 import React, { Component } from 'react';
 import PropTypes from 'prop-types';
 import addons from '@storybook/addons';
-import EventEmitter from 'events';
 
 import Swatch from './Swatch';
 
@@ -118,7 +117,11 @@ BackgroundPanel.propTypes = {
     getQueryParam: PropTypes.func,
     setQueryParams: PropTypes.func,
   }).isRequired,
-  channel: PropTypes.instanceOf(EventEmitter),
+  channel: PropTypes.shape({
+    emit: PropTypes.func,
+    on: PropTypes.func,
+    removeListener: PropTypes.func,
+  }),
 };
 BackgroundPanel.defaultProps = {
   channel: undefined,

diff --git a/addons/background/src/index.js b/addons/background/src/index.js
@@ -1,6 +1,5 @@
 import React from 'react';
 import PropTypes from 'prop-types';
-import EventEmitter from 'events';
 
 import addons from '@storybook/addons';
 
@@ -63,7 +62,11 @@ export class BackgroundDecorator extends React.Component {
 }
 BackgroundDecorator.propTypes = {
   backgrounds: PropTypes.arrayOf(PropTypes.object),
-  channel: PropTypes.instanceOf(EventEmitter),
+  channel: PropTypes.shape({
+    emit: PropTypes.func,
+    on: PropTypes.func,
+    removeListener: PropTypes.func,
+  }),
   story: PropTypes.func.isRequired,
 };
 BackgroundDecorator.defaultProps = {