Add another test for wrappingComponent

enzymejs · Mar 14, 2021 · 46a2382 · 46a2382
1 parent 6707c28
commit 46a2382
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/packages/enzyme-test-suite/test/ShallowWrapper-spec.jsx b/packages/enzyme-test-suite/test/ShallowWrapper-spec.jsx
@@ -679,6 +679,15 @@ describe('shallow', () => {
           }
         }
 
+        class WrappingComponent extends React.Component {
+          render() {
+            const { children } = this.props;
+            return (
+              <Provider value="foo">{ children }</Provider>
+            );
+          }
+        }
+
         class InnerComponent extends React.Component {
           render() {
             return this.context;
@@ -693,6 +702,11 @@ describe('shallow', () => {
           const provider = provides.find(InnerComponent).shallow();
           expect(provider.text()).to.equal('foo');
         });
+
+        it('works with wrappingComponent', () => {
+          const wrapper = shallow(<InnerComponent />, { wrappingComponent: WrappingComponent });
+          expect(wrapper.text()).to.equal('foo');
+        });
       });
     });