Simplify jest.setup.js

jest.setup.js

@@ -1,9 +1,4 @@
 module.exports = async function () {
   // NOTE (pradeep): Can this be synced with the definitions in webpack configs?
   globalThis.__DEV__ = true;
-
-  // NOTE (pradeep): `requestAnimationFrame` and `setInterval` is already defined by jsdom.
-  // However, I'm redefining it to run the callbacks synchronously to make testing easier.
-  globalThis.requestAnimationFrame = (callback) => callback();
-  globalThis.setInterval = (callback, _interval) => callback();
 };

test/FixedDataTableRoot-test.js

@@ -13,8 +13,17 @@ import {
 
 import { Table, Column } from '../src/index';
 import Scrollbar from '../src/plugins/Scrollbar';
+import * as requestAnimationFramePolyfill from '../src/vendor_upstream/core/requestAnimationFramePolyfill';
 
 describe('FixedDataTableRoot', function () {
+  beforeEach(() => {
+    // NOTE (pradeep): Call the callback passed to requestAnimationFrame immediately.
+    // This simplifies the tests by making them behave synchronous.
+    jest
+      .spyOn(requestAnimationFramePolyfill, 'default')
+      .mockImplementation((callback) => callback());
+  });
+
   afterEach(() => {
     // restore all spys created with spyOn
     jest.restoreAllMocks();

test/ReactTouchHandler-test.js

@@ -9,7 +9,12 @@ describe('ReactTouchHandler', function () {
   let mockRequestAnimationFramePolyfill;
 
   beforeEach(function () {
-    // See https://stackoverflow.com/a/74269343
+    // NOTE (pradeep): Call the callback passed to `setInterval` immediately.
+    // This simplifies the tests by making them behave synchronous.
+    jest
+      .spyOn(globalThis, 'setInterval')
+      .mockImplementation((callback) => callback());
+
     mockRequestAnimationFramePolyfill = jest
       .spyOn(requestAnimationFrame, 'default')
       .mockImplementation(() => undefined);

test/plugins/ReorderCell-test.js

@@ -123,12 +123,6 @@ class ReorderCellTest extends React.Component {
 }
 
 describe('ReorderCell', () => {
-  beforeEach(() => {
-    jest
-      .spyOn(requestAnimationFramePolyfill, 'default')
-      .mockImplementation((callback) => setTimeout(callback, 16));
-  });
-
   afterEach(() => {
     jest.restoreAllMocks();
   });