11jest . mock ( './initLDClient' , ( ) => jest . fn ( ) ) ;
22jest . mock ( './context' , ( ) => ( { Provider : 'Provider' } ) ) ;
33
4- import * as React from 'react' ;
4+ import React , { Component } from 'react' ;
55import { create } from 'react-test-renderer' ;
6- import { shallow } from 'enzyme' ;
76import { LDClient , LDFlagChangeset , LDFlagSet , LDOptions , LDUser } from 'launchdarkly-js-client-sdk' ;
87import initLDClient from './initLDClient' ;
98import { LDReactOptions , EnhancedComponent , defaultReactOptions , ProviderConfig } from './types' ;
@@ -40,7 +39,7 @@ describe('LDProvider', () => {
4039 < App />
4140 </ LDProvider >
4241 ) ;
43- const component = create ( LaunchDarklyApp ) ;
42+ const component = create ( LaunchDarklyApp ) . toTree ( ) ?. instance as Component ;
4443 expect ( component ) . toMatchSnapshot ( ) ;
4544 } ) ;
4645
@@ -129,8 +128,8 @@ describe('LDProvider', () => {
129128 < App />
130129 </ LDProvider >
131130 ) ;
132- const component = shallow ( LaunchDarklyApp , { disableLifecycleMethods : true } ) ;
133- const initialState = component . state ( ) as HocState ;
131+ const component = create ( LaunchDarklyApp ) . toTree ( ) ?. instance as Component ;
132+ const initialState = component . state as HocState ;
134133
135134 expect ( initialState . flags ) . toEqual ( { } ) ;
136135 } ) ;
@@ -154,8 +153,8 @@ describe('LDProvider', () => {
154153 < App />
155154 </ LDProvider >
156155 ) ;
157- const component = shallow ( LaunchDarklyApp , { disableLifecycleMethods : true } ) ;
158- const initialState = component . state ( ) as HocState ;
156+ const component = create ( LaunchDarklyApp ) . toTree ( ) ?. instance as Component ;
157+ const initialState = component . state as HocState ;
159158
160159 expect ( mockInitLDClient ) . not . toHaveBeenCalled ( ) ;
161160 expect ( initialState . flags ) . toEqual ( { testFlag : true , anotherTestFlag : false } ) ;
@@ -178,8 +177,8 @@ describe('LDProvider', () => {
178177 < App />
179178 </ LDProvider >
180179 ) ;
181- const component = shallow ( LaunchDarklyApp , { disableLifecycleMethods : true } ) ;
182- const initialState = component . state ( ) as HocState ;
180+ const component = create ( LaunchDarklyApp ) . toTree ( ) ?. instance as Component ;
181+ const initialState = component . state as HocState ;
183182
184183 expect ( mockInitLDClient ) . not . toHaveBeenCalled ( ) ;
185184 expect ( initialState . flags ) . toEqual ( { 'test-flag' : true , 'another-test-flag' : false } ) ;
@@ -200,8 +199,8 @@ describe('LDProvider', () => {
200199 < App />
201200 </ LDProvider >
202201 ) ;
203- const component = shallow ( LaunchDarklyApp , { disableLifecycleMethods : true } ) ;
204- const initialState = component . state ( ) as HocState ;
202+ const component = create ( LaunchDarklyApp ) . toTree ( ) ?. instance as Component ;
203+ const initialState = component . state as HocState ;
205204
206205 expect ( mockInitLDClient ) . not . toHaveBeenCalled ( ) ;
207206 expect ( initialState . flags ) . toEqual ( { testFlag : true , anotherTestFlag : false } ) ;
@@ -221,8 +220,8 @@ describe('LDProvider', () => {
221220 < App />
222221 </ LDProvider >
223222 ) ;
224- const component = shallow ( LaunchDarklyApp , { disableLifecycleMethods : true } ) ;
225- const initialState = component . state ( ) as HocState ;
223+ const component = create ( LaunchDarklyApp ) . toTree ( ) ?. instance as Component ;
224+ const initialState = component . state as HocState ;
226225
227226 expect ( mockInitLDClient ) . not . toHaveBeenCalled ( ) ;
228227 expect ( initialState . flags ) . toEqual ( { testFlag : true , anotherTestFlag : false } ) ;
@@ -239,8 +238,8 @@ describe('LDProvider', () => {
239238 < App />
240239 </ LDProvider >
241240 ) ;
242- const component = shallow ( LaunchDarklyApp , { disableLifecycleMethods : true } ) ;
243- const initialState = component . state ( ) as HocState ;
241+ const component = create ( LaunchDarklyApp ) . toTree ( ) ?. instance as Component ;
242+ const initialState = component . state as HocState ;
244243
245244 expect ( mockInitLDClient ) . not . toHaveBeenCalled ( ) ;
246245 expect ( initialState . flags ) . toEqual ( { } ) ;
0 commit comments