Extract mockWindowDimensions from TestRunner (#6343)

## Summary Extract `mockWindowDimensions` and `unmockWindowDimensions` from `TestRunner` ## Test plan
software-mansion · Jul 31, 2024 · 0616843 · 0616843
1 parent a85657a
commit 0616843
Show file tree

Hide file tree

Showing 3 changed files with 58 additions and 48 deletions.
diff --git a/apps/common-app/src/examples/RuntimeTests/ReJest/RuntimeTestsApi.ts b/apps/common-app/src/examples/RuntimeTests/ReJest/RuntimeTestsApi.ts
@@ -8,6 +8,7 @@ import { DescribeDecorator, TestDecorator } from './types';
 export { Presets } from './Presets';
 
 const testRunner = new TestRunner();
+const windowDimensionsMocker = testRunner.getWindowDimensionsMocker();
 const animationRecorder = testRunner.getAnimationUpdatesRecorder();
 const valueRegistry = testRunner.getValueRegistry();
 
@@ -179,11 +180,11 @@ export async function unmockAnimationTimer() {
 }
 
 export async function mockWindowDimensions() {
-  await testRunner.mockWindowDimensions();
+  await windowDimensionsMocker.mockWindowDimensions();
 }
 
 export async function unmockWindowDimensions() {
-  await testRunner.unmockWindowDimensions();
+  await windowDimensionsMocker.unmockWindowDimensions();
 }
 
 export async function recordAnimationUpdates() {

diff --git a/apps/common-app/src/examples/RuntimeTests/ReJest/TestRunner/TestRunner.ts b/apps/common-app/src/examples/RuntimeTests/ReJest/TestRunner/TestRunner.ts
@@ -4,18 +4,13 @@ import type { BuildFunction, TestCase, TestConfiguration, TestSuite, TestValue,
 import { DescribeDecorator, TestDecorator } from '../types';
 import { TestComponent } from '../TestComponent';
 import { applyMarkdown, formatTestName } from '../utils/stringFormatUtils';
-import type {
-  LayoutAnimationStartFunction,
-  LayoutAnimationType,
-  SharedTransitionAnimationsValues,
-  LayoutAnimation,
-} from 'react-native-reanimated';
 import { Matchers } from '../matchers/Matchers';
 import { assertMockedAnimationTimestamp, assertTestCase, assertTestSuite } from './Asserts';
 import { makeMutable, runOnJS } from 'react-native-reanimated';
 import { RenderLock, SyncUIRunner } from '../utils/SyncUIRunner';
 import { ValueRegistry } from './ValueRegistry';
 import { TestSummaryLogger } from './TestSummaryLogger';
+import { WindowDimensionsMocker } from './WindowDimensionsMocker';
 import { AnimationUpdatesRecorder } from './AnimationUpdatesRecorder';
 export { Presets } from '../Presets';
 
@@ -42,9 +37,14 @@ export class TestRunner {
   private _syncUIRunner: SyncUIRunner = new SyncUIRunner();
   private _renderLock: RenderLock = new RenderLock();
   private _testSummary: TestSummaryLogger = new TestSummaryLogger();
+  private _windowDimensionsMocker: WindowDimensionsMocker = new WindowDimensionsMocker();
   private _animationRecorder = new AnimationUpdatesRecorder();
   private _valueRegistry = new ValueRegistry();
 
+  public getWindowDimensionsMocker() {
+    return this._windowDimensionsMocker;
+  }
+
   public getAnimationUpdatesRecorder() {
     return this._animationRecorder;
   }
@@ -324,46 +324,6 @@ export class TestRunner {
     this._currentTestSuite.afterEach = job;
   }
 
-  public async unmockWindowDimensions() {
-    await this._syncUIRunner.runOnUIBlocking(() => {
-      'worklet';
-      if (global.originalLayoutAnimationsManager) {
-        global.LayoutAnimationsManager = global.originalLayoutAnimationsManager;
-      }
-    });
-  }
-
-  public async mockWindowDimensions() {
-    await this._syncUIRunner.runOnUIBlocking(() => {
-      'worklet';
-      const originalLayoutAnimationsManager = global.LayoutAnimationsManager;
-
-      const startLayoutAnimation: LayoutAnimationStartFunction = (
-        tag: number,
-        type: LayoutAnimationType,
-        _yogaValues: Partial<SharedTransitionAnimationsValues>,
-        config: (arg: Partial<SharedTransitionAnimationsValues>) => LayoutAnimation,
-      ) => {
-        originalLayoutAnimationsManager.start(
-          tag,
-          type,
-          {
-            ..._yogaValues,
-            windowHeight: 852,
-            windowWidth: 393,
-          },
-          config,
-        );
-      };
-
-      global.originalLayoutAnimationsManager = originalLayoutAnimationsManager;
-      global.LayoutAnimationsManager = {
-        start: startLayoutAnimation,
-        stop: originalLayoutAnimationsManager.stop,
-      };
-    });
-  }
-
   public wait(delay: number) {
     return new Promise(resolve => {
       setTimeout(resolve, delay);

diff --git a/apps/common-app/src/examples/RuntimeTests/ReJest/TestRunner/WindowDimensionsMocker.ts b/apps/common-app/src/examples/RuntimeTests/ReJest/TestRunner/WindowDimensionsMocker.ts
@@ -0,0 +1,49 @@
+import { SyncUIRunner } from '../utils/SyncUIRunner';
+import type {
+  LayoutAnimationStartFunction,
+  LayoutAnimationType,
+  SharedTransitionAnimationsValues,
+  LayoutAnimation,
+} from 'react-native-reanimated';
+
+export class WindowDimensionsMocker {
+  public async unmockWindowDimensions() {
+    await new SyncUIRunner().runOnUIBlocking(() => {
+      'worklet';
+      if (global.originalLayoutAnimationsManager) {
+        global.LayoutAnimationsManager = global.originalLayoutAnimationsManager;
+      }
+    });
+  }
+
+  public async mockWindowDimensions() {
+    await new SyncUIRunner().runOnUIBlocking(() => {
+      'worklet';
+      const originalLayoutAnimationsManager = global.LayoutAnimationsManager;
+
+      const startLayoutAnimation: LayoutAnimationStartFunction = (
+        tag: number,
+        type: LayoutAnimationType,
+        _yogaValues: Partial<SharedTransitionAnimationsValues>,
+        config: (arg: Partial<SharedTransitionAnimationsValues>) => LayoutAnimation,
+      ) => {
+        originalLayoutAnimationsManager.start(
+          tag,
+          type,
+          {
+            ..._yogaValues,
+            windowHeight: 852,
+            windowWidth: 393,
+          },
+          config,
+        );
+      };
+
+      global.originalLayoutAnimationsManager = originalLayoutAnimationsManager;
+      global.LayoutAnimationsManager = {
+        start: startLayoutAnimation,
+        stop: originalLayoutAnimationsManager.stop,
+      };
+    });
+  }
+}