Use createHelper

mui · Dec 15, 2016 · 6da2cf8 · 6da2cf8
1 parent e4592fa
commit 6da2cf8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/src/internal/SwitchLabel.js → src/internal/withSwitchLabel.js b/src/internal/SwitchLabel.js → src/internal/withSwitchLabel.js
@@ -3,6 +3,7 @@
 
 import React, { Component, PropTypes } from 'react';
 import { createStyleSheet } from 'jss-theme-reactor';
+import createHelper from 'recompose/createHelper';
 import classNames from 'classnames';
 
 export const styleSheet = createStyleSheet('SwitchLabel', (theme) => {
@@ -23,12 +24,8 @@ export const styleSheet = createStyleSheet('SwitchLabel', (theme) => {
   };
 });
 
-export function withSwitchLabel(SwitchComponent) {
-  const switchComponentName = SwitchComponent.displayName || SwitchComponent.name;
-
+function withSwitchLabel(SwitchComponent) {
   return class SwitchLabel extends Component {
-    static displayName = `withSwitchLabel(${switchComponentName})`;
-
     static propTypes = {
       /**
        * If `true`, the control will be disabled.
@@ -93,3 +90,5 @@ export function withSwitchLabel(SwitchComponent) {
     }
   };
 }
+
+export default createHelper(withSwitchLabel, 'withSwitchLabel', true, true);
diff --git a/src/internal/SwitchLabel.spec.js → src/internal/withSwitchLabel.spec.js b/src/internal/SwitchLabel.spec.js → src/internal/withSwitchLabel.spec.js
@@ -5,7 +5,7 @@ import React from 'react';
 import { assert } from 'chai';
 import { spy } from 'sinon';
 import { createShallowWithContext } from 'test/utils';
-import { withSwitchLabel, styleSheet } from './SwitchLabel';
+import withSwitchLabel, { styleSheet } from './withSwitchLabel';
 
 describe('<SwitchLabel />', () => {
   let shallow;