Function createDbMock

packages/billing/src/__tests__/org-billing.test.ts:52–85

(options?: {
  grants?: typeof mockGrants | any[]
  insert?: () => { values: () => { onConflictDoNothing: () => { returning: () => Promise<unknown[]> } } }
  update?: () => { set: () => { where: () => Promise<unknown> } }
})

}

const createDbMock = (options?: {
  grants?: typeof mockGrants | any[]
  insert?: () => { values: () => { onConflictDoNothing: () => { returning: () => Promise<unknown[]> } } }
  update?: () => { set: () => { where: () => Promise<unknown> } }
}) => {
  const { grants = mockGrants, insert, update } = options ?? {}

  return {
    select: () => ({
      from: () => ({
        where: () => ({
          orderBy: () => grants,
        }),
      }),
    }),
    insert:
      insert ??
      (() => ({
        values: () => ({
          onConflictDoNothing: () => ({
            returning: () => Promise.resolve([{ id: 'test-id' }]),
          }),
        }),
      })),
    update:
      update ??
      (() => ({
        set: () => ({
          where: () => Promise.resolve(),
        }),
      })),
    execute: () => Promise.resolve([]),
  }
}

describe('Organization Billing', () => {
  beforeEach(async () => {