Function setupDbSpies

common/src/testing/mocks/database.ts:269–303

(
  db: { insert: unknown; update: unknown },
  options: CreateMockDbOptions = {},
)

 * Accepts any object with insert and update methods.
 */
export function setupDbSpies(
  db: { insert: unknown; update: unknown },
  options: CreateMockDbOptions = {},
): DbSpies {
  const { defaultInsertId = 'test-run-id' } = options

  const mockInsertResult = {
    values: mock(() => Promise.resolve({ id: defaultInsertId })),
  }

  const mockUpdateResult = {
    set: mock(() => ({
      where: mock(() => Promise.resolve()),
    })),
  }

  // Cast db to a spyable type - the actual db module has complex types that
  // don't play well with spyOn's inference, but the spy still works at runtime
  const spyableDb = db as { insert: () => unknown; update: () => unknown }
  const insertSpy = spyOn(spyableDb, 'insert').mockReturnValue(mockInsertResult)
  const updateSpy = spyOn(spyableDb, 'update').mockReturnValue(mockUpdateResult)

  return {
    insert: insertSpy,
    update: updateSpy,
    restore: () => {
      insertSpy.mockRestore()
      updateSpy.mockRestore()
    },
    clear: () => {
      insertSpy.mockClear()
      updateSpy.mockClear()
    },
  }
}

/**
 * Creates a mock for a database query builder chain that returns specific data.