Test with clearInternals

Author: fabioh8010

lib/OnyxUtils.ts

@@ -52,13 +52,13 @@ const mergeQueue: Record<OnyxKey, Array<OnyxValue<OnyxKey>>> = {};
 const mergeQueuePromise: Record<OnyxKey, Promise<void>> = {};
 
 // Holds a mapping of all the React components that want their state subscribed to a store key
-const callbackToStateMapping: Record<string, Mapping<OnyxKey>> = {};
+let callbackToStateMapping: Record<string, Mapping<OnyxKey>> = {};
 
 // Keeps a copy of the values of the onyx collection keys as a map for faster lookups
 let onyxCollectionKeySet = new Set<OnyxKey>();
 
 // Holds a mapping of the connected key to the subscriptionID for faster lookups
-const onyxKeyToSubscriptionIDs = new Map();
+let onyxKeyToSubscriptionIDs = new Map();
 
 // Optional user-provided key value states set when Onyx initializes or clears
 let defaultKeyStates: Record<OnyxKey, OnyxValue<OnyxKey>> = {};
@@ -67,7 +67,7 @@ let batchUpdatesPromise: Promise<void> | null = null;
 let batchUpdatesQueue: Array<() => void> = [];
 
 // Used for comparison with a new update to avoid invoking the Onyx.connect callback with the same data.
-const lastConnectionCallbackData = new Map<number, OnyxValue<OnyxKey>>();
+let lastConnectionCallbackData = new Map<number, OnyxValue<OnyxKey>>();
 
 let snapshotKey: OnyxKey | null = null;
 
@@ -1431,6 +1431,12 @@ function updateSnapshots(data: OnyxUpdate[], mergeFn: typeof Onyx.merge): Array<
     return promises;
 }
 
+function clearInternals(): void {
+    callbackToStateMapping = {};
+    onyxKeyToSubscriptionIDs = new Map();
+    lastConnectionCallbackData = new Map();
+}
+
 const OnyxUtils = {
     METHOD,
     getMergeQueue,
@@ -1480,6 +1486,7 @@ const OnyxUtils = {
     addKeyToRecentlyAccessedIfNeeded,
     reduceCollectionWithSelector,
     updateSnapshots,
+    clearInternals,
 };
 
 GlobalSettings.addGlobalSettingsChangeListener(({enablePerformanceMetrics}) => {

tests/perf-test/Onyx.perf-test.ts

@@ -25,6 +25,7 @@ const mockedReportActionsMap = getRandomReportActions(collectionKey);
 
 const clearOnyxAfterEachMeasure = async () => {
     await Onyx.clear();
+    OnyxUtils.clearInternals();
 };
 
 describe('Onyx', () => {
@@ -39,6 +40,7 @@ describe('Onyx', () => {
 
     afterEach(async () => {
         await Onyx.clear();
+        OnyxUtils.clearInternals();
     });
 
     describe('set', () => {

tests/perf-test/OnyxConnectionManager.perf-test.ts

@@ -4,6 +4,7 @@ import type {Connection} from '../../lib/OnyxConnectionManager';
 import connectionManager from '../../lib/OnyxConnectionManager';
 import createDeferredTask from '../../lib/createDeferredTask';
 import {getRandomReportActions} from '../utils/collections/reportActions';
+import OnyxUtils from '../../lib/OnyxUtils';
 
 const ONYXKEYS = {
     TEST_KEY: 'test',
@@ -38,6 +39,7 @@ const resetConectionManagerAfterEachMeasure = () => {
 
 const clearOnyxAfterEachMeasure = async () => {
     await Onyx.clear();
+    OnyxUtils.clearInternals();
 };
 
 describe('OnyxConnectionManager', () => {

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

@@ -45,6 +45,7 @@ const mockedReportActionsKeys = Object.keys(mockedReportActionsMap);
 
 const clearOnyxAfterEachMeasure = async () => {
     await Onyx.clear();
+    OnyxUtils.clearInternals();
 };
 
 describe('OnyxUtils', () => {
@@ -60,6 +61,7 @@ describe('OnyxUtils', () => {
 
     afterEach(async () => {
         await Onyx.clear();
+        OnyxUtils.clearInternals();
     });
 
     describe('getMergeQueue', () => {

tests/perf-test/useOnyx.perf-test.tsx

@@ -5,6 +5,7 @@ import {measureRenders} from 'reassure';
 import type {FetchStatus, OnyxEntry, OnyxKey, OnyxValue, ResultMetadata, UseOnyxOptions} from '../../lib';
 import Onyx, {useOnyx} from '../../lib';
 import StorageMock from '../../lib/storage';
+import OnyxUtils from '../../lib/OnyxUtils';
 
 const ONYXKEYS = {
     TEST_KEY: 'test',
@@ -49,6 +50,7 @@ function UseOnyxWrapper({onyxKey, onyxOptions}: UseOnyxWrapperProps) {
 
 const clearOnyxAfterEachMeasure = async () => {
     await Onyx.clear();
+    OnyxUtils.clearInternals();
 };
 
 describe('useOnyx', () => {
@@ -61,6 +63,7 @@ describe('useOnyx', () => {
 
     afterEach(async () => {
         await Onyx.clear();
+        OnyxUtils.clearInternals();
     });
 
     describe('misc', () => {