🐛 fix mocks true on white list mode

FBerthelot · Jul 27, 2019 · 2ea5dc1 · 2ea5dc1
1 parent c744b2b
commit 2ea5dc1
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 1 deletion.
diff --git a/packages/component-test-utils-react/src/__tests__/shallow-html.spec.js b/packages/component-test-utils-react/src/__tests__/shallow-html.spec.js
@@ -145,6 +145,24 @@ describe('shallow - html', () => {
     );
   });
 
+  it('should render component that with HOC and with mock', () => {
+    const Child = ({className}) => {
+      return <button type="button" className={className}>0 Like</button>;
+    };
+
+    const withStyle = Component => () => {
+      return <Component className="class-for-button"/>;
+    };
+
+    const Component = withStyle(Child);
+
+    const cmp = shallow(<Component/>, {mocks: {Child: true}});
+
+    expect(cmp.html()).toBe(
+      '<Child class="class-for-button"><button type="button" class="class-for-button">0 Like</button></Child>'
+    );
+  });
+
   it('should render component that use useState hooks', () => {
     const Component = () => {
       const [nbLikes] = React.useState(0);

diff --git a/packages/component-test-utils-react/src/render/render.js b/packages/component-test-utils-react/src/render/render.js
@@ -27,9 +27,12 @@ const render = (reactEl, config, ShallowRender) => {
   }
 
   if (!isAlreadyMocked && shouldBeRender(reactEl, config)) {
+    const mock = config.mocks && config.mocks[reactEl.type.displayName || reactEl.type.name];
+    const el = mock === true ? reactEl.type : mock || reactEl.type;
+
     const shallowRender = new ShallowRender(
       React.createElement(
-        (config.mocks && config.mocks[reactEl.type.displayName || reactEl.type.name]) || reactEl.type,
+        el,
         reactEl.props,
         reactEl.props && reactEl.props.children
       ),