Tests for @provide decorator

acdlite · acdlite · commit 612681b9d3ce · 2015-06-12T11:36:46.000-07:00
diff --git a/src/components/createProvideDecorator.js b/src/components/createProvideDecorator.js
@@ -1,14 +1,16 @@
 import getDisplayName from '../utils/getDisplayName';
 
 export default function createProvideDecorator(React, Provider) {
+  const { Component } = React;
+
   return function provide(redux) {
-    return DecoratedComponent => class ProviderDecorator {
+    return DecoratedComponent => class ProviderDecorator extends Component {
       static displayName = `Provider(${getDisplayName(DecoratedComponent)})`;
 
       render() {
         return (
           <Provider redux={redux}>
-            {props => <DecoratedComponent {...this.props} {...props} />}
+            {() => <DecoratedComponent {...this.props} />}
           </Provider>
         );
       }
diff --git a/test/components/provide.spec.js b/test/components/provide.spec.js
@@ -0,0 +1,52 @@
+import expect from 'expect';
+import jsdom from 'mocha-jsdom';
+import React, { PropTypes, Component } from 'react/addons';
+import { createRedux } from '../../src';
+import { provide, Provider } from '../../src/react';
+
+const { TestUtils } = React.addons;
+
+describe('React', () => {
+  describe('provide', () => {
+    jsdom();
+
+    class Child extends Component {
+      static contextTypes = {
+        redux: PropTypes.object.isRequired
+      }
+
+      render() {
+        return <div />;
+      }
+    }
+
+    it('wraps component with Provider', () => {
+      const redux = createRedux({ test: () => 'test' });
+
+      @provide(redux)
+      class Container extends Component {
+        render() {
+          return <Child {...this.props} />;
+        }
+      }
+
+      const container = TestUtils.renderIntoDocument(<Container pass="through" />);
+      const child = TestUtils.findRenderedComponentWithType(container, Child);
+      expect(child.props.pass).toEqual('through');
+      expect(() => TestUtils.findRenderedComponentWithType(container, Provider))
+        .toNotThrow();
+      expect(child.context.redux).toBe(redux);
+    });
+
+    it('sets displayName correctly', () => {
+      @provide(createRedux({ test: () => 'test' }))
+      class Container extends Component {
+        render() {
+          return <div />;
+        }
+      }
+
+      expect(Container.displayName).toBe('Provider(Container)');
+    });
+  });
+});
