Function createMockDbOperations

common/src/testing/mocks/database.ts:143–218

(
  options: CreateMockDbOptions = {},
)

 * ```
 */
export function createMockDbOperations(
  options: CreateMockDbOptions = {},
): MockDbOperations {
  const { defaultSelectData = [], defaultInsertId = 'mock-id' } = options

  const createMockSelectWhereResult = <T>(
    data: T[] = defaultSelectData as T[],
  ): MockSelectWhereResult<T> => {
    const result: MockSelectWhereResult<T> = {
      then: mock((resolve) => {
        resolve(data)
        return Promise.resolve(data)
      }),
      leftJoin: mock(() => result),
      innerJoin: mock(() => result),
      orderBy: mock(() => result),
      limit: mock(() => result),
      offset: mock(() => result),
    }
    return result
  }

  const createMockSelectFromResult = <T>(
    data: T[] = defaultSelectData as T[],
  ): MockSelectFromResult<T> => {
    const whereResult = createMockSelectWhereResult(data)
    const result: MockSelectFromResult<T> = {
      where: mock(() => whereResult),
      leftJoin: mock(() => result),
      innerJoin: mock(() => result),
      orderBy: mock(() => result),
      limit: mock(() => result),
      offset: mock(() => result),
      then: mock((resolve) => {
        resolve(data)
        return Promise.resolve(data)
      }),
    }
    return result
  }

  const createMockInsertResult = <T>(): MockInsertResult<T> => {
    const result: MockInsertResult<T> = {
      values: mock(() => Promise.resolve({ id: defaultInsertId })),
      returning: mock(() => Promise.resolve([])),
      onConflictDoNothing: mock(() => result),
      onConflictDoUpdate: mock(() => result),
    }
    return result
  }

  const createMockUpdateSetResult = (): MockUpdateSetResult => ({
    where: mock(() => Promise.resolve()),
    returning: mock(() => Promise.resolve([])),
  })

  const createMockUpdateResult = <T>(): MockUpdateResult<T> => ({
    set: mock(() => createMockUpdateSetResult()),