diff --git a/src/FlagProvider.test.tsx b/src/FlagProvider.test.tsx index 4061eed..8f158c8 100644 --- a/src/FlagProvider.test.tsx +++ b/src/FlagProvider.test.tsx @@ -110,7 +110,7 @@ test('A consumer that subscribes AFTER client init shows values from provider an 'consuming value isEnabled true' ); expect(screen.getByText(/consuming value updateContext/)).toHaveTextContent( - 'consuming value updateContext [object Promise]' + 'consuming value updateContext undefined' ); expect(screen.getByText(/consuming value getVariant/)).toHaveTextContent( 'consuming value getVariant A' @@ -148,7 +148,7 @@ test('A consumer should be able to get a variant when the client is passed into 'consuming value isEnabled true' ); expect(screen.getByText(/consuming value updateContext/)).toHaveTextContent( - 'consuming value updateContext [object Promise]' + 'consuming value updateContext undefined' ); expect(screen.getByText(/consuming value getVariant/)).toHaveTextContent( 'consuming value getVariant A' diff --git a/src/FlagProvider.tsx b/src/FlagProvider.tsx index 3e5a742..828ea89 100644 --- a/src/FlagProvider.tsx +++ b/src/FlagProvider.tsx @@ -107,7 +107,7 @@ const FlagProvider: FC> = ({ () => ({ on: ((event, callback, ctx) => client.current.on(event, callback, ctx)) as IFlagContextValue['on'], off: ((event, callback) => client.current.off(event, callback)) as IFlagContextValue['off'], - updateContext: async (context) => await client.current.updateContext(context), + updateContext: client.current.updateContext, isEnabled: (toggleName) => client.current.isEnabled(toggleName), getVariant: (toggleName) => client.current.getVariant(toggleName), client: client.current,