resetSdkMock.ts

import { resetInstrumentationHandlers } from '@sentry/core';
import { vi } from 'vitest';

import type { Replay as ReplayIntegration } from '../../src/integration';
import type { ReplayContainer } from '../../src/replay';
import type { RecordMock } from './../index';
import { BASE_TIMESTAMP } from './../index';
import type { DomHandler } from './../types';
import type { MockSdkParams } from './mockSdk';
import { mockSdk } from './mockSdk';

export async function resetSdkMock({ replayOptions, sentryOptions, autoStart }: MockSdkParams): Promise<{
  domHandler: DomHandler;
  mockRecord: RecordMock;
  replay: ReplayContainer;
  integration: ReplayIntegration;
}> {
  let domHandler: DomHandler;

  vi.setSystemTime(new Date(BASE_TIMESTAMP));
  vi.clearAllMocks();
  vi.resetModules();

  // Clear all handlers that have been registered
  resetInstrumentationHandlers();

  const SentryBrowserUtils = await import('@sentry-internal/browser-utils');
  vi.spyOn(SentryBrowserUtils, 'addClickKeypressInstrumentationHandler').mockImplementation(handler => {
    domHandler = handler;
  });
  const { mockRrweb } = await import('./mockRrweb');
  const { record: mockRecord } = mockRrweb();

  const { replay, integration } = await mockSdk({
    replayOptions,
    sentryOptions,
    autoStart,
  });

  // XXX: This is needed to ensure `domHandler` is set
  await vi.runAllTimersAsync();
  vi.setSystemTime(new Date(BASE_TIMESTAMP));

  return {
    // @ts-expect-error use before assign
    domHandler,
    mockRecord,
    replay,
    integration,
  };
}