rocket/tests/setup.ts at develop · imdeepmind/rocket · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
import {vi} from 'vitest';

import {
  pgConnectMock,
  pgEndMock,
  pgQueryMock,
  sqliteCloseMock,
  sqliteExecMock,
  sqlitePrepareMock,
} from '@tests/helpers/db-mocks';

vi.mock('pg', () => {
  return {
    // prettier-ignore
    // eslint-disable-next-line prefer-arrow-callback
    Pool: vi.fn().mockImplementation(function () {
      return {
        query: pgQueryMock,
        end: pgEndMock,
        connect: pgConnectMock,
      };
    }),
  };
});

vi.mock('better-sqlite3', () => {
  return {
    // prettier-ignore
    // eslint-disable-next-line prefer-arrow-callback
    default: vi.fn().mockImplementation(function () {
      return {
        prepare: sqlitePrepareMock,
        close: sqliteCloseMock,
        exec: sqliteExecMock,
      };
    }),
  };
});