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' ;
@@ -80,7 +79,7 @@ describe('LDProvider', () => {
8079 const user1 : LDUser = { key : 'yus' , name : 'yus ng' } ;
8180 const user2 : LDUser = { key : 'launch' , name : 'darkly' } ;
8281 const options : LDOptions = { bootstrap : { } } ;
83- const ldClient : Promise < LDClient > = new Promise ( async resolve => {
82+ const ldClient : Promise < LDClient > = new Promise ( async ( resolve ) => {
8483 resolve ( ( await initLDClient ( clientSideID , user1 , defaultReactOptions , options , undefined ) ) . ldClient ) ;
8584
8685 return ;
@@ -101,7 +100,7 @@ describe('LDProvider', () => {
101100 test ( 'ld client is created if passed in promise resolves as undefined' , async ( ) => {
102101 const user : LDUser = { key : 'yus' , name : 'yus ng' } ;
103102 const options : LDOptions = { bootstrap : { } } ;
104- const ldClient : Promise < undefined > = new Promise ( async resolve => {
103+ const ldClient : Promise < undefined > = new Promise ( async ( resolve ) => {
105104 resolve ( undefined ) ;
106105
107106 return ;
@@ -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