immich/server/test/repositories/database.repository.mock.ts

import { IDatabaseRepository } from 'src/interfaces/database.interface';
import { Version } from 'src/utils/version';

export const newDatabaseRepositoryMock = (): jest.Mocked<IDatabaseRepository> => {
  return {
    getExtensionVersion: jest.fn(),
    getAvailableExtensionVersion: jest.fn(),
    getPreferredVectorExtension: jest.fn(),
    getPostgresVersion: jest.fn().mockResolvedValue(new Version(14, 0, 0)),
    createExtension: jest.fn().mockImplementation(() => Promise.resolve()),
    updateExtension: jest.fn(),
    updateVectorExtension: jest.fn(),
    reindex: jest.fn(),
    shouldReindex: jest.fn(),
    runMigrations: jest.fn(),
    withLock: jest.fn().mockImplementation((_, function_: <R>() => Promise<R>) => function_()),
    tryLock: jest.fn(),
    isBusy: jest.fn(),
    wait: jest.fn(),
  };
};