fix: Code optimization on tests

Flagsmith · Oct 20, 2023 · b1fc193 · b1fc193
1 parent fd396a4
commit b1fc193
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/tests/sdk/flagsmith.test.ts b/tests/sdk/flagsmith.test.ts
@@ -337,17 +337,14 @@ test('test_flagsmith_uses_offline_handler_if_set_and_no_api_response', async ()
     fetch.mockReturnValue(Promise.resolve(errorResponse));
 
     // When
-    await flagsmith.getEnvironmentFlags();
-    await flagsmith.getIdentityFlags('identity', {});
+    const environmentFlags:Flags = await flagsmith.getEnvironmentFlags();
+    const identityFlags:Flags = await flagsmith.getIdentityFlags('identity', {});
 
     // Then
     expect(mock_offline_handler.getEnvironment).toHaveBeenCalledTimes(1);
     expect(flagsmith.getEnvironmentFlags).toHaveBeenCalled();
     expect(flagsmith.getIdentityFlags).toHaveBeenCalled();
-
-    const environmentFlags:Flags = await flagsmith.getEnvironmentFlags();
-    const identityFlags:Flags = await flagsmith.getIdentityFlags('identity', {});
-
+
     expect(environmentFlags.isFeatureEnabled('some_feature')).toBe(true);
     expect(environmentFlags.getFeatureValue('some_feature')).toBe('offline-value');