Function createMockDb

packages/testing/src/mocks/database.mock.ts:260–309

()

 * Creates a mock database connection.
 */
export function createMockDb() {
  // A `where(...)` result that is both awaitable (resolves to `[]`) and exposes
  // `.limit`/`.orderBy`, so `select().from()[.leftJoin()].where()[.limit()]`
  // works whether or not a terminal is chained.
  const whereResult = () => {
    const thenable: any = Promise.resolve([])
    thenable.limit = vi.fn(() => Promise.resolve([]))
    thenable.orderBy = vi.fn(() => Promise.resolve([]))
    return thenable
  }
  const fromBuilder = () => ({
    where: vi.fn(whereResult),
    leftJoin: vi.fn(() => ({ where: vi.fn(whereResult) })),
    innerJoin: vi.fn(() => ({ where: vi.fn(whereResult) })),
  })

  return {
    select: vi.fn(() => ({
      from: vi.fn(fromBuilder),
    })),
    selectDistinct: vi.fn(() => ({
      from: vi.fn(fromBuilder),
    })),
    insert: vi.fn(() => ({
      values: vi.fn(() => ({
        returning: vi.fn(() => Promise.resolve([])),
        onConflictDoUpdate: vi.fn(() => ({
          returning: vi.fn(() => Promise.resolve([])),
        })),
        onConflictDoNothing: vi.fn(() => ({
          returning: vi.fn(() => Promise.resolve([])),
        })),
      })),
    })),
    update: vi.fn(() => ({
      set: vi.fn(() => ({
        where: vi.fn(() => ({
          returning: vi.fn(() => Promise.resolve([])),
        })),
      })),
    })),
    delete: vi.fn(() => ({
      where: vi.fn(() => ({
        returning: vi.fn(() => Promise.resolve([])),
      })),
    })),
    transaction: vi.fn(async (callback) => callback(createMockDb())),
    query: vi.fn(() => Promise.resolve([])),
  }
}

/**
 * Mock module for @sim/db.