refactor default className propType

.npmignore

@@ -0,0 +1 @@
+src/

package.json

@@ -5,7 +5,7 @@
   "main": "lib/pacomo.js",
   "scripts": {
     "compile": "babel --stage 0 -d lib/ src/",
-    "prepublish": "npm run compile",
+    "prepublish": "npm run test",
     "test": "npm run compile && mocha --compilers js:babel/register"
   },
   "repository": {
@@ -27,14 +27,17 @@
     "url": "https://github.com/jamesknelson/react-pacomo/issues"
   },
   "homepage": "https://github.com/jamesknelson/react-pacomo#readme",
+  "peerDependencies": {
+    "react": "^0.14.0-beta3"
+  },
   "dependencies": {
     "classnames": "^2.1.3",
-    "invariant": "^2.1.0",
-    "react": "^0.14.0-beta3"
+    "invariant": "^2.1.0"
   },
   "devDependencies": {
     "babel": "^5.8.23",
     "mocha": "^2.3.0",
+    "react": "^0.14.0-beta3",
     "react-addons-test-utils": "^0.14.0-beta3"
   }
 }

src/pacomo.js

@@ -12,7 +12,16 @@ export function createDecorator(prefix) {
       "pacomo must be applied to a class with no `pacomo` property"
     )
 
-    const decoratedComponent = class extends component {
+    return class extends component {
+      static displayName = componentName
+
+      // Add `className` propType, if none exists
+      static propTypes =
+        Object.assign(
+          {className: PropTypes.string},
+          component.propTypes
+        )
+
       pacomo(...args) {
         return (
           classNames(...args)
@@ -31,18 +40,5 @@ export function createDecorator(prefix) {
         )
       }
     }
-
-    // Add `className` propType, if none exists
-    if (!decoratedComponent.propTypes) {
-      decoratedComponent.propTypes = {}
-    }
-    if (!decoratedComponent.propTypes.className) {
-      decoratedComponent.propTypes = Object.assign(
-        {className: PropTypes.string},
-        decoratedComponent.propTypes
-      )
-    }
-
-    return decoratedComponent
   }
 }

test.js

@@ -111,8 +111,8 @@ describe('#pacomo', () => {
     const pacomo = createDecorator('prefix')(BareComponent).prototype.pacomo
 
     assert.equal(
-      pacomo('test'),
-      'prefix-BareComponent-test',
+      pacomo('test1', 'test2'),
+      'prefix-BareComponent-test1 prefix-BareComponent-test2',
       "`pacomo` produces the correct string when a string is passed in"
     )
   })