Function setupMocks

web/src/lib/__tests__/ban-conditions.test.ts:21–77

()

let mockDisputesList!: ReturnType<typeof mock>

const setupMocks = async () => {
  mockSelect = mock(() => ({
    from: mock(() => ({
      where: mock(() => ({
        limit: mock(() => Promise.resolve([])),
      })),
    })),
  }))

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

  mockDisputesList = mock(() =>
    Promise.resolve({
      data: [],
    }),
  )

  await mockModule('@codebuff/internal/db', () => ({
    default: {
      select: mockSelect,
      update: mockUpdate,
    },
  }))

  await mockModule('@codebuff/internal/db/schema', () => ({
    user: {
      id: 'id',
      banned: 'banned',
      email: 'email',
      name: 'name',
      stripe_customer_id: 'stripe_customer_id',
    },
  }))

  await mockModule('@codebuff/internal/util/stripe', () => ({
    stripeServer: {
      disputes: {
        list: mockDisputesList,
      },
    },
  }))

  await mockModule('drizzle-orm', () => ({
    eq: mock((a: any, b: any) => ({ column: a, value: b })),
  }))

  const banConditionsModule = await import('../ban-conditions')
  DISPUTE_THRESHOLD = banConditionsModule.DISPUTE_THRESHOLD
  DISPUTE_WINDOW_DAYS = banConditionsModule.DISPUTE_WINDOW_DAYS
  banUser = banConditionsModule.banUser
  evaluateBanConditions = banConditionsModule.evaluateBanConditions
  getUserByStripeCustomerId = banConditionsModule.getUserByStripeCustomerId
}