WIP

Author: mydea

packages/browser-utils/src/getNativeImplementation.ts

@@ -66,6 +66,18 @@ export function clearCachedImplementation(name: keyof CacheableImplementations):
   delete cachedImplementations[name];
 }
 
+/**
+ * Sets a cached implementation.
+ * This should NOT be used, and is only here
+ * @hidden
+ */
+export function setCachedImplementation<T extends keyof CacheableImplementations>(
+  name: T,
+  impl: CacheableImplementations[T],
+): void {
+  cachedImplementations[name] = impl;
+}
+
 /**
  * A special usecase for incorrectly wrapped Fetch APIs in conjunction with ad-blockers.
  * Whenever someone wraps the Fetch API and returns the wrong promise chain,

packages/browser-utils/src/index.ts

@@ -18,7 +18,13 @@ export { addClickKeypressInstrumentationHandler } from './instrument/dom';
 
 export { addHistoryInstrumentationHandler } from './instrument/history';
 
-export { fetch, setTimeout, clearCachedImplementation, getNativeImplementation } from './getNativeImplementation';
+export {
+  fetch,
+  setTimeout,
+  clearCachedImplementation,
+  getNativeImplementation,
+  setCachedImplementation,
+} from './getNativeImplementation';
 
 export {
   addXhrInstrumentationHandler,

packages/replay-internal/test/unit/coreHandlers/handleClick.test.ts

@@ -3,8 +3,9 @@ import type { Breadcrumb } from '@sentry/types';
 import { BASE_TIMESTAMP } from '../..';
 import { ClickDetector, ignoreElement } from '../../../src/coreHandlers/handleClick';
 import type { ReplayContainer } from '../../../src/types';
+import { useFakeTimers } from '../../utils/use-fake-timers';
 
-vi.useFakeTimers();
+useFakeTimers();
 
 describe('Unit | coreHandlers | handleClick', () => {
   describe('ClickDetector', () => {

packages/replay-internal/test/unit/coreHandlers/handleNetworkBreadcrumbs.test.ts

@@ -13,8 +13,9 @@ import { beforeAddNetworkBreadcrumb } from '../../../src/coreHandlers/handleNetw
 import type { EventBufferArray } from '../../../src/eventBuffer/EventBufferArray';
 import type { ReplayContainer, ReplayNetworkOptions } from '../../../src/types';
 import { setupReplayContainer } from '../../utils/setupReplayContainer';
+import { useFakeTimers } from '../../utils/use-fake-timers';
 
-vi.useFakeTimers();
+useFakeTimers();
 
 async function waitForReplayEventBuffer() {
   // Need one Promise.resolve() per await in the util functions

packages/replay-internal/test/unit/coreHandlers/util/addBreadcrumbEvent.test.ts

@@ -2,8 +2,9 @@ import { BASE_TIMESTAMP } from '../../..';
 import { addBreadcrumbEvent } from '../../../../src/coreHandlers/util/addBreadcrumbEvent';
 import type { EventBufferArray } from '../../../../src/eventBuffer/EventBufferArray';
 import { setupReplayContainer } from '../../../utils/setupReplayContainer';
+import { useFakeTimers } from '../../../utils/use-fake-timers';
 
-vi.useFakeTimers();
+useFakeTimers();
 
 describe('Unit | coreHandlers | util | addBreadcrumbEvent', function () {
   beforeEach(function () {
@@ -23,12 +24,6 @@ describe('Unit | coreHandlers | util | addBreadcrumbEvent', function () {
     const replay = setupReplayContainer();
     addBreadcrumbEvent(replay, breadcrumb);
 
-    await undefined;
-    await undefined;
-    await undefined;
-    await undefined;
-    await undefined;
-
     expect((replay.eventBuffer as EventBufferArray).events).toEqual([
       {
         type: 5,

packages/replay-internal/test/unit/coreHandlers/util/networkUtils.test.ts

@@ -8,8 +8,9 @@ import {
   getFullUrl,
   parseContentLengthHeader,
 } from '../../../../src/coreHandlers/util/networkUtils';
+import { useFakeTimers } from '../../../utils/use-fake-timers';
 
-vi.useFakeTimers();
+useFakeTimers();
 
 describe('Unit | coreHandlers | util | networkUtils', () => {
   describe('parseContentLengthHeader()', () => {

packages/replay-internal/test/unit/util/createPerformanceEntry.test.ts

@@ -1,6 +1,8 @@
 import { vi } from 'vitest';
 
-vi.useFakeTimers();
+import { useFakeTimers } from '../../utils/use-fake-timers';
+
+useFakeTimers();
 vi.setSystemTime(new Date('2023-01-01'));
 
 vi.mock('@sentry/utils', async () => ({

packages/replay-internal/test/unit/util/debounce.test.ts

@@ -1,7 +1,8 @@
 import { debounce } from '../../../src/util/debounce';
+import { useFakeTimers } from '../../utils/use-fake-timers';
 
 describe('Unit | util | debounce', () => {
-  vi.useFakeTimers();
+  useFakeTimers();
   it('delay the execution of the passed callback function by the passed minDelay', () => {
     const callback = vi.fn();
     const debouncedCallback = debounce(callback, 100);

packages/replay-internal/test/unit/util/throttle.test.ts

@@ -1,7 +1,8 @@
 import { BASE_TIMESTAMP } from '../..';
 import { SKIPPED, THROTTLED, throttle } from '../../../src/util/throttle';
+import { useFakeTimers } from '../../utils/use-fake-timers';
 
-vi.useFakeTimers();
+useFakeTimers();
 
 describe('Unit | util | throttle', () => {
   it('executes when not hitting the limit', () => {