firestore-vitest-mock/mocks/reactNativeFirebaseFirestore.ts

Vitest helper for mocking Google Cloud Firestore

import { vi } from "vitest";
import { type StubOptions, type StubOverrides } from "./firebase.model";
import {
  type CollectionReference,
  type DocumentReference,
  FakeFirestore,
  type FieldPath,
  type FieldValue,
  type Query,
  type Timestamp,
  type Transaction,
} from "./firestore";

import defaultOptions from "./helpers/defaultMockOptions";

declare class Firestore extends FakeFirestore {
  constructor();
}

interface GCloudFirestoreMock {
  Firestore: typeof Firestore;
  Query: typeof Query;
  CollectionReference: typeof CollectionReference;
  DocumentReference: typeof DocumentReference;
  FieldValue: typeof FieldValue;
  FieldPath: typeof FieldPath;
  Timestamp: typeof Timestamp;
  Transaction: typeof Transaction;
}

export const firestoreStub = async (
  overrides?: StubOverrides,
  options: StubOptions = defaultOptions
): Promise<GCloudFirestoreMock> => {
  const {
    FakeFirestore,
    Query,
    CollectionReference,
    DocumentReference,
    FieldValue,
    Timestamp,
    Transaction,
    FieldPath,
  } = await import("firestore-vitest-mock");

  class Firestore extends FakeFirestore {
    constructor() {
      super(overrides?.database, options);
    }
  }
  return {
    Query,
    CollectionReference,
    DocumentReference,
    FieldValue,
    FieldPath,
    Timestamp,
    Transaction,
    /** @type {Firestore.constructor} */
    Firestore,
  };
};

export const mockReactNativeFirestore = (
  overrides?: StubOverrides,
  options: StubOptions = defaultOptions
): void => {
  mockModuleIfFound("@react-native-firebase/firestore", overrides, options);
};

function mockModuleIfFound(
  moduleName: string,
  overrides?: StubOverrides,
  options?: StubOptions
): void {
  try {
    require.resolve(moduleName);
    vi.doMock(moduleName, async () => await firestoreStub(overrides, options));
  } catch (e) {
    // eslint-disable-next-line no-console
    console.info(`Module ${moduleName} not found, mocking skipped.`);
  }
}