diff --git a/packages/enzyme-test-suite/test/ShallowWrapper-spec.jsx b/packages/enzyme-test-suite/test/ShallowWrapper-spec.jsx
index 8f9d30646..354a6fce9 100644
--- a/packages/enzyme-test-suite/test/ShallowWrapper-spec.jsx
+++ b/packages/enzyme-test-suite/test/ShallowWrapper-spec.jsx
@@ -667,23 +667,23 @@ describe('shallow', () => {
});
});
- describe('shallow() on Provider and Consumer through .contextType', () => {
-
- const { Provider } = React.createContext('howdy!');
+ describeIf(is('>= 16.6'), 'shallow() on Provider and Consumer through .contextType', () => {
+ const { Provider } = createContext('howdy!');
class OuterComponent extends React.Component {
render() {
+ const { value } = this.props;
return (
-
+
);
}
}
class WrappingComponent extends React.Component {
render() {
- const { children } = this.props;
+ const { children, value } = this.props;
return (
- { children }
+ { children }
);
}
}
@@ -696,16 +696,46 @@ describe('shallow', () => {
InnerComponent.contextType = Provider;
- it('works on a Provider', () => {
- const wrapper = shallow();
- const provides = wrapper.find(Provider).dive();
- const provider = provides.find(InnerComponent).shallow();
- expect(provider.text()).to.equal('foo');
+ describe('rendering the Provider directly', () => {
+ it('renders initial context value', () => {
+ const wrapper = shallow();
+ const provides = wrapper.find(Provider).shallow();
+ const provider = provides.find(InnerComponent).dive();
+
+ expect(provider.text()).to.equal('foo');
+ });
+
+ it('renders updated context value', () => {
+ const wrapper = shallow();
+ wrapper.setProps({ value: 'bar' });
+ const provides = wrapper.find(Provider).shallow();
+ const provider = provides.find(InnerComponent).dive();
+
+ expect(provider.text()).to.equal('bar');
+ });
});
- it('works with wrappingComponent', () => {
- const wrapper = shallow(, { wrappingComponent: WrappingComponent });
- expect(wrapper.text()).to.equal('foo');
+ describe('rendering the Provider through wrappingComponent', () => {
+ it('renders initial context value', () => {
+ const wrapper = shallow(, {
+ wrappingComponent: WrappingComponent,
+ wrappingComponentProps: { value: 'foo' },
+ });
+
+ expect(wrapper.text()).to.equal('foo');
+ });
+
+ it('renders updated context value', () => {
+ const wrapper = shallow(, {
+ wrappingComponent: WrappingComponent,
+ wrappingComponentProps: { value: 'foo' },
+ });
+ const wrappingComponent = wrapper.getWrappingComponent();
+ wrappingComponent.setProps({ value: 'bar' });
+ wrappingComponent.rerender();
+
+ expect(wrapper.text()).to.equal('bar');
+ });
});
});
});