Update test

Author: EmilianoSanchez

src/__tests__/withSplitTreatments.test.tsx

@@ -12,38 +12,42 @@ import { sdkBrowser } from './testUtils/sdkConfigs';
 import { withSplitFactory } from '../withSplitFactory';
 import { withSplitClient } from '../withSplitClient';
 import { withSplitTreatments } from '../withSplitTreatments';
-import { ISplitTreatmentsChildProps } from '../types';
 import { getControlTreatmentsWithConfig } from '../constants';
 
 describe('withSplitTreatments', () => {
 
   it(`passes Split props and outer props to the child.
-      In this test, the value of "props.treatments" is obteined by the function "getControlTreatmentsWithConfig",
-      and not "client.getTreatmentsWithConfig" since the client is not ready.`, (done) => {
+      In this test, the value of "props.treatments" is obtained by the function "getControlTreatmentsWithConfig",
+      and not "client.getTreatmentsWithConfig" since the client is not ready.`, () => {
     const featureFlagNames = ['split1', 'split2'];
+
     const Component = withSplitFactory(sdkBrowser)<{ outerProp1: string, outerProp2: number }>(
       ({ outerProp1, outerProp2, factory }) => {
         const SubComponent = withSplitClient('user1')<{ outerProp1: string, outerProp2: number }>(
           withSplitTreatments(featureFlagNames)(
-            (props: ISplitTreatmentsChildProps & { outerProp1: string, outerProp2: number }) => {
+            (props) => {
               const clientMock = factory!.client('user1');
-              expect(props.outerProp1).toBe('outerProp1');
-              expect(props.outerProp2).toBe(2);
               expect((clientMock.getTreatmentsWithConfig as jest.Mock).mock.calls.length).toBe(0);
-              expect(props.treatments).toEqual(getControlTreatmentsWithConfig(featureFlagNames));
-              expect(props.isReady).toBe(false);
-              expect(props.isReadyFromCache).toBe(false);
-              expect(props.hasTimedout).toBe(false);
-              expect(props.isTimedout).toBe(false);
-              expect(props.isDestroyed).toBe(false);
-              expect(props.lastUpdate).toBe(0);
-              done();
+
+              expect(props).toStrictEqual({
+                factory: factory, client: clientMock,
+                outerProp1: 'outerProp1', outerProp2: 2,
+                treatments: getControlTreatmentsWithConfig(featureFlagNames),
+                isReady: false,
+                isReadyFromCache: false,
+                hasTimedout: false,
+                isTimedout: false,
+                isDestroyed: false,
+                lastUpdate: 0
+              });
+
               return null;
             }
           )
         );
         return <SubComponent outerProp1={outerProp1} outerProp2={outerProp2} />;
       });
+
     render(<Component outerProp1='outerProp1' outerProp2={2} />);
   });