wehub-resource-sync
d25d482dc2
CI / Migrate Dev DB (push) Has been skipped
CI / Detect Version (push) Has been cancelled
CI / Migrate DB (push) Has been cancelled
CI / Build Dev ECR (./docker/app.Dockerfile, ECR_APP) (push) Has been cancelled
CI / Build Dev ECR (./docker/db.Dockerfile, ECR_MIGRATIONS) (push) Has been cancelled
CI / Build Dev ECR (./docker/pii.Dockerfile, ECR_PII) (push) Has been cancelled
CI / Build Dev ECR (./docker/realtime.Dockerfile, ECR_REALTIME) (push) Has been cancelled
CI / Deploy Trigger.dev (Dev) (push) Has been cancelled
CI / Build AMD64 (./docker/app.Dockerfile, ECR_APP, ghcr.io/simstudioai/simstudio) (push) Has been cancelled
CI / Build AMD64 (./docker/db.Dockerfile, ECR_MIGRATIONS, ghcr.io/simstudioai/migrations) (push) Has been cancelled
CI / Build AMD64 (./docker/pii.Dockerfile, ECR_PII, ghcr.io/simstudioai/pii) (push) Has been cancelled
CI / Build AMD64 (./docker/realtime.Dockerfile, ECR_REALTIME, ghcr.io/simstudioai/realtime) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/app.Dockerfile, ghcr.io/simstudioai/simstudio) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/db.Dockerfile, ghcr.io/simstudioai/migrations) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/pii.Dockerfile, ghcr.io/simstudioai/pii) (push) Has been cancelled
CI / Build ARM64 (GHCR Only) (./docker/realtime.Dockerfile, ghcr.io/simstudioai/realtime) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/migrations) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/pii) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/realtime) (push) Has been cancelled
CI / Create GHCR Manifests (ghcr.io/simstudioai/simstudio) (push) Has been cancelled
CI / Check Docs Changes (push) Has been cancelled
CI / Process Docs (push) Has been cancelled
CI / Create GitHub Release (push) Has been cancelled
CI / Test and Build (push) Has been cancelled
Publish CLI Package / publish-npm (push) Has been cancelled
Publish Python SDK / publish-pypi (push) Has been cancelled
Publish TypeScript SDK / publish-npm (push) Has been cancelled
170 lines
6.3 KiB
TypeScript
170 lines
6.3 KiB
TypeScript
/**
|
||
* @vitest-environment node
|
||
*/
|
||
import { beforeEach, describe, expect, it, vi } from 'vitest'
|
||
|
||
const {
|
||
billingFlag,
|
||
mockClaim,
|
||
mockSelectRows,
|
||
dbUpdateSpy,
|
||
sendEmailSpy,
|
||
getEmailPreferencesMock,
|
||
renderMock,
|
||
subjectMock,
|
||
isOrgAdminRoleMock,
|
||
} = vi.hoisted(() => ({
|
||
billingFlag: { enabled: true },
|
||
mockClaim: vi.fn<[], unknown[]>(() => [{ id: 'u1' }]),
|
||
mockSelectRows: vi.fn<[], unknown[]>(() => []),
|
||
dbUpdateSpy: vi.fn(),
|
||
sendEmailSpy: vi.fn(() => Promise.resolve({ success: true })),
|
||
getEmailPreferencesMock: vi.fn(() => Promise.resolve(null as unknown)),
|
||
renderMock: vi.fn(() => Promise.resolve('<html></html>')),
|
||
subjectMock: vi.fn(() => 'Subject'),
|
||
isOrgAdminRoleMock: vi.fn(() => true),
|
||
}))
|
||
|
||
vi.mock('@sim/db', () => {
|
||
const updateBuilder: Record<string, unknown> = {
|
||
set: () => updateBuilder,
|
||
where: () => updateBuilder,
|
||
returning: () => Promise.resolve(mockClaim()),
|
||
then: (f: (v: unknown) => unknown, r?: (e: unknown) => unknown) =>
|
||
Promise.resolve(undefined).then(f, r),
|
||
}
|
||
const selectBuilder: Record<string, unknown> = {
|
||
from: () => selectBuilder,
|
||
where: () => selectBuilder,
|
||
innerJoin: () => selectBuilder,
|
||
leftJoin: () => selectBuilder,
|
||
limit: () => Promise.resolve(mockSelectRows()),
|
||
then: (f: (v: unknown) => unknown, r?: (e: unknown) => unknown) =>
|
||
Promise.resolve(mockSelectRows()).then(f, r),
|
||
}
|
||
dbUpdateSpy.mockImplementation(() => updateBuilder)
|
||
return { db: { update: dbUpdateSpy, select: () => selectBuilder } }
|
||
})
|
||
|
||
vi.mock('@/lib/core/config/env-flags', () => ({
|
||
get isBillingEnabled() {
|
||
return billingFlag.enabled
|
||
},
|
||
}))
|
||
vi.mock('@/lib/core/utils/urls', () => ({ getBaseUrl: () => 'https://app.sim.ai' }))
|
||
vi.mock('@/lib/messaging/email/mailer', () => ({ sendEmail: sendEmailSpy }))
|
||
vi.mock('@/lib/messaging/email/unsubscribe', () => ({
|
||
getEmailPreferences: getEmailPreferencesMock,
|
||
}))
|
||
vi.mock('@/components/emails/render', () => ({
|
||
renderLimitThresholdEmail: renderMock,
|
||
getLimitEmailSubject: subjectMock,
|
||
}))
|
||
vi.mock('@sim/platform-authz/workspace', () => ({ isOrgAdminRole: isOrgAdminRoleMock }))
|
||
|
||
import { maybeSendLimitThresholdEmail } from '@/lib/billing/core/limit-notifications'
|
||
|
||
const baseUserParams = {
|
||
category: 'storage' as const,
|
||
scope: 'user' as const,
|
||
workspaceId: 'ws-1',
|
||
usageLabel: '4.5 GB',
|
||
limitLabel: '5 GB',
|
||
userId: 'u1',
|
||
userEmail: 'u1@example.com',
|
||
userName: 'Ada',
|
||
}
|
||
|
||
describe('maybeSendLimitThresholdEmail', () => {
|
||
beforeEach(() => {
|
||
vi.clearAllMocks()
|
||
billingFlag.enabled = true
|
||
mockClaim.mockReturnValue([{ id: 'u1' }])
|
||
mockSelectRows.mockReturnValue([])
|
||
getEmailPreferencesMock.mockResolvedValue(null)
|
||
})
|
||
|
||
it('sends a warning email when crossing 80% and the claim wins', async () => {
|
||
await maybeSendLimitThresholdEmail({ ...baseUserParams, currentUsage: 4.5, limit: 5 })
|
||
expect(sendEmailSpy).toHaveBeenCalledTimes(1)
|
||
expect(renderMock).toHaveBeenCalledWith(expect.objectContaining({ kind: 'warning' }))
|
||
expect(subjectMock).toHaveBeenCalledWith('storage', 'warning')
|
||
})
|
||
|
||
it('sends a reached email at/over 100%', async () => {
|
||
await maybeSendLimitThresholdEmail({ ...baseUserParams, currentUsage: 5, limit: 5 })
|
||
expect(renderMock).toHaveBeenCalledWith(expect.objectContaining({ kind: 'reached' }))
|
||
expect(subjectMock).toHaveBeenCalledWith('storage', 'reached')
|
||
})
|
||
|
||
it('never sends in rearmOnly mode, even when usage is above a threshold', async () => {
|
||
await maybeSendLimitThresholdEmail({
|
||
...baseUserParams,
|
||
currentUsage: 4.5,
|
||
limit: 5,
|
||
rearmOnly: true,
|
||
})
|
||
expect(mockClaim).not.toHaveBeenCalled()
|
||
expect(sendEmailSpy).not.toHaveBeenCalled()
|
||
})
|
||
|
||
it('does not send when the atomic claim is lost (already notified)', async () => {
|
||
mockClaim.mockReturnValue([])
|
||
await maybeSendLimitThresholdEmail({ ...baseUserParams, currentUsage: 4.5, limit: 5 })
|
||
expect(sendEmailSpy).not.toHaveBeenCalled()
|
||
})
|
||
|
||
it('claims without re-arming on a crossing (re-arm and claim are mutually exclusive)', async () => {
|
||
await maybeSendLimitThresholdEmail({ ...baseUserParams, currentUsage: 4.5, limit: 5 })
|
||
expect(dbUpdateSpy).toHaveBeenCalledTimes(1)
|
||
expect(mockClaim).toHaveBeenCalledTimes(1)
|
||
expect(sendEmailSpy).toHaveBeenCalledTimes(1)
|
||
})
|
||
|
||
it('does not send in the dead band (70%–80%)', async () => {
|
||
await maybeSendLimitThresholdEmail({ ...baseUserParams, currentUsage: 3.75, limit: 5 })
|
||
expect(mockClaim).not.toHaveBeenCalled()
|
||
expect(sendEmailSpy).not.toHaveBeenCalled()
|
||
})
|
||
|
||
it('re-arms below the band without claiming or sending', async () => {
|
||
await maybeSendLimitThresholdEmail({ ...baseUserParams, currentUsage: 1, limit: 5 })
|
||
expect(mockClaim).not.toHaveBeenCalled()
|
||
expect(sendEmailSpy).not.toHaveBeenCalled()
|
||
})
|
||
|
||
it('does not send OR burn the claim when the per-user toggle is off', async () => {
|
||
mockSelectRows.mockReturnValue([{ enabled: false }])
|
||
await maybeSendLimitThresholdEmail({ ...baseUserParams, currentUsage: 4.5, limit: 5 })
|
||
expect(sendEmailSpy).not.toHaveBeenCalled()
|
||
expect(mockClaim).not.toHaveBeenCalled()
|
||
})
|
||
|
||
it('does not send OR burn the claim when the recipient unsubscribed', async () => {
|
||
getEmailPreferencesMock.mockResolvedValue({ unsubscribeNotifications: true })
|
||
await maybeSendLimitThresholdEmail({ ...baseUserParams, currentUsage: 4.5, limit: 5 })
|
||
expect(sendEmailSpy).not.toHaveBeenCalled()
|
||
expect(mockClaim).not.toHaveBeenCalled()
|
||
})
|
||
|
||
it('skips entirely when billing is disabled', async () => {
|
||
billingFlag.enabled = false
|
||
await maybeSendLimitThresholdEmail({ ...baseUserParams, currentUsage: 5, limit: 5 })
|
||
expect(mockClaim).not.toHaveBeenCalled()
|
||
expect(sendEmailSpy).not.toHaveBeenCalled()
|
||
})
|
||
|
||
it('re-arms but does not send when usage is fully cleared (zero usage)', async () => {
|
||
await maybeSendLimitThresholdEmail({ ...baseUserParams, currentUsage: 0, limit: 5 })
|
||
expect(dbUpdateSpy).toHaveBeenCalledTimes(1)
|
||
expect(mockClaim).not.toHaveBeenCalled()
|
||
expect(sendEmailSpy).not.toHaveBeenCalled()
|
||
})
|
||
|
||
it('skips when the limit is non-positive', async () => {
|
||
await maybeSendLimitThresholdEmail({ ...baseUserParams, currentUsage: 4, limit: 0 })
|
||
expect(dbUpdateSpy).not.toHaveBeenCalled()
|
||
expect(sendEmailSpy).not.toHaveBeenCalled()
|
||
})
|
||
})
|