@react-native-harness/runtime 1.0.0-alpha.19 → 1.0.0-alpha.20

package/src/mocker/registry.ts

@@ -1,32 +1,28 @@
 import { ModuleFactory, ModuleId, Require } from './types.js';
 
+const modulesCache = new Map<number, unknown>();
 const mockRegistry = new Map<number, ModuleFactory>();
-const mockCache = new Map<number, unknown>();
 
 const originalRequire = global.__r;
 
 export const mock = (moduleId: string, factory: ModuleFactory): void => {
-  mockCache.delete(moduleId as unknown as ModuleId);
+  modulesCache.delete(moduleId as unknown as ModuleId);
   mockRegistry.set(moduleId as unknown as ModuleId, factory);
 };
 
-export const clearMocks = (): void => {
-  mockRegistry.clear();
-  mockCache.clear();
+const isModuleMocked = (moduleId: number): boolean => {
+  return mockRegistry.has(moduleId);
 };
 
 const getMockImplementation = (moduleId: number): unknown | null => {
-  if (mockCache.has(moduleId)) {
-    return mockCache.get(moduleId);
-  }
-
   const factory = mockRegistry.get(moduleId);
+  
   if (!factory) {
     return null;
   }
 
   const implementation = factory();
-  mockCache.set(moduleId, implementation);
+  modulesCache.set(moduleId, implementation);
   return implementation;
 };
 
@@ -37,25 +33,32 @@ export const requireActual = <T = any>(moduleId: string): T =>
 
 export const unmock = (moduleId: string) => {
   mockRegistry.delete(moduleId as unknown as ModuleId);
-  mockCache.delete(moduleId as unknown as ModuleId);
+  modulesCache.delete(moduleId as unknown as ModuleId);
 };
 
 export const resetModules = (): void => {
-  mockCache.clear();
-
-  // Reset Metro's module cache
-  global.__resetAllModules();
+  modulesCache.clear();
+  mockRegistry.clear();
 };
 
 const mockRequire = (moduleId: string) => {
   // babel plugin will transform 'moduleId' to a number
-  const mockedModule = getMockImplementation(moduleId as unknown as ModuleId);
+  const moduleIdNumber = moduleId as unknown as ModuleId;
+  const cachedModule = modulesCache.get(moduleIdNumber);
+
+  if (cachedModule) {
+    return cachedModule;
+  }
 
-  if (mockedModule) {
+  if (isModuleMocked(moduleIdNumber)) {
+    const mockedModule = getMockImplementation(moduleIdNumber);
+    modulesCache.set(moduleIdNumber, mockedModule);
     return mockedModule;
   }
 
-  return originalRequire(moduleId as unknown as ModuleId);
+  const originalModule = originalRequire(moduleIdNumber);
+  modulesCache.set(moduleIdNumber, originalModule);
+  return originalModule;
 };
 
 Object.setPrototypeOf(mockRequire, Object.getPrototypeOf(originalRequire));