Fix Reassure tests

Author: fabioh8010

tests/perf-test/OnyxUtils.perf-test.ts

@@ -3,7 +3,6 @@ import {randBoolean} from '@ngneat/falso';
 import createRandomReportAction, {getRandomReportActions} from '../utils/collections/reportActions';
 import type {Selector} from '../../lib';
 import Onyx from '../../lib';
-import StorageMock from '../../lib/storage';
 import OnyxUtils, {clearOnyxUtilsInternals} from '../../lib/OnyxUtils';
 import type GenericCollection from '../utils/GenericCollection';
 import type {OnyxUpdate} from '../../lib/Onyx';
@@ -112,7 +111,7 @@ describe('OnyxUtils', () => {
         test('10k calls with heavy objects', async () => {
             await measureAsyncFunction(() => Promise.all(mockedReportActionsKeys.map((key) => OnyxUtils.get(key))), {
                 beforeEach: async () => {
-                    await StorageMock.multiSet(Object.entries(mockedReportActionsMap).map(([k, v]) => [k, v]));
+                    await Onyx.multiSet(mockedReportActionsMap);
                 },
                 afterEach: clearOnyxAfterEachMeasure,
             });
@@ -123,7 +122,7 @@ describe('OnyxUtils', () => {
         test('one call with 50k heavy objects', async () => {
             await measureAsyncFunction(() => OnyxUtils.getAllKeys(), {
                 beforeEach: async () => {
-                    await StorageMock.multiSet(Object.entries(mockedReportActionsMap).map(([k, v]) => [k, v]));
+                    await Onyx.multiSet(mockedReportActionsMap);
                 },
                 afterEach: clearOnyxAfterEachMeasure,
             });
@@ -463,7 +462,7 @@ describe('OnyxUtils', () => {
 
             await measureAsyncFunction(() => OnyxUtils.initializeWithDefaultKeyStates(), {
                 beforeEach: async () => {
-                    await StorageMock.multiSet(Object.entries(changedReportActions).map(([k, v]) => [k, v]));
+                    await Onyx.multiSet(changedReportActions);
                     OnyxUtils.initStoreValues(ONYXKEYS, mockedReportActionsMap);
                 },
                 afterEach: async () => {
@@ -481,15 +480,6 @@ describe('OnyxUtils', () => {
     });
 
     describe('multiGet', () => {
-        test('one call getting 10k heavy objects from storage', async () => {
-            await measureAsyncFunction(() => OnyxUtils.multiGet(mockedReportActionsKeys), {
-                beforeEach: async () => {
-                    await StorageMock.multiSet(Object.entries(mockedReportActionsMap).map(([k, v]) => [k, v]));
-                },
-                afterEach: clearOnyxAfterEachMeasure,
-            });
-        });
-
         test('one call getting 10k heavy objects from cache', async () => {
             await measureAsyncFunction(() => OnyxUtils.multiGet(mockedReportActionsKeys), {
                 beforeEach: async () => {
@@ -540,7 +530,7 @@ describe('OnyxUtils', () => {
                 },
                 {
                     beforeEach: async () => {
-                        await StorageMock.multiSet(Object.entries(mockedReportActionsMap).map(([k, v]) => [k, v]));
+                        await Onyx.multiSet(mockedReportActionsMap);
                     },
                     afterEach: async () => {
                         OnyxUtils.unsubscribeFromKey(subscriptionID);
@@ -567,7 +557,7 @@ describe('OnyxUtils', () => {
                 },
                 {
                     beforeEach: async () => {
-                        await StorageMock.multiSet(Object.entries(mockedReportActionsMap).map(([k, v]) => [k, v]));
+                        await Onyx.multiSet(mockedReportActionsMap);
                     },
                     afterEach: async () => {
                         OnyxUtils.unsubscribeFromKey(subscriptionID);