Something went wrong. Try again.
[READ-ONLY] Mirror of https://github.com/openstatusHQ/openstatus. 🫖 Status page with uptime monitoring & API monitoring as code 🫖 openstatus.dev
bun drizzle-orm monitoring monitoring-as-code nextjs observability on-call open-source shadcn-ui status-page statuspage synthetic-monitoring tinybird turso uptime uptime-checker uptime-monitor
Something went wrong. Try again.
TypeScript
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291import { db, eq } from "@openstatus/db";import { invitation, workspace } from "@openstatus/db/src/schema";import { getLimits } from "@openstatus/db/src/schema/plan/utils";import { createTestWorkspace } from "@openstatus/db/src/test/factories";import { expect } from "@std/expect";import { afterEach, beforeEach, describe, test } from "@std/testing/bdd";import { assertSpyCalls, type Stub, stub } from "@std/testing/mock";import type Stripe from "stripe";
import { stripe } from "./shared";import { TRIAL_DAYS, maybeStartSignupTrial } from "./trial";import { PLANS } from "./utils";
const STARTER_PRICE = PLANS.find((p) => p.plan === "starter")?.price.monthly .priceIds.test;const trialEnd = Math.floor(Date.now() / 1000) + TRIAL_DAYS * 86_400;
async function freeWorkspace() { return createTestWorkspace({ plan: "free", stripeId: null });}
function customer(id: string, metadata: Record<string, string> = {}) { return { id, metadata } as unknown as Stripe.Customer;}
describe("maybeStartSignupTrial", () => { let customerId: string; let customers: Stripe.Customer[]; let price: Partial<Stripe.Price>; let stubs: Stub[]; let list: Stub; let createCustomer: Stub; let createSubscription: Stub; // Runs inside `customers.create`, to interleave work with the Stripe call. let onCreateCustomer: (() => Promise<void>) | undefined;
beforeEach(() => { customerId = `cus_${crypto.randomUUID()}`; customers = []; onCreateCustomer = undefined; price = { currency: "usd", currency_options: {} }; list = stub( stripe.customers, "list", () => Promise.resolve({ data: customers } as Stripe.Response< Stripe.ApiList<Stripe.Customer> >) as Stripe.ApiListPromise<Stripe.Customer>, ); createCustomer = stub(stripe.customers, "create", async () => { await onCreateCustomer?.(); return { id: customerId } as Stripe.Response<Stripe.Customer>; }); createSubscription = stub(stripe.subscriptions, "create", () => Promise.resolve({ id: "sub_trial", status: "trialing", trial_end: trialEnd, items: { data: [ { price: { id: STARTER_PRICE }, quantity: 1, current_period_end: trialEnd, }, ], }, } as Stripe.Response<Stripe.Subscription>), ); stubs = [ list, createCustomer, createSubscription, stub(stripe.prices, "retrieve", () => Promise.resolve(price as Stripe.Response<Stripe.Price>), ), ]; });
afterEach(() => { for (const s of stubs) s.restore(); });
test("starts a Starter trial on the owner's workspace", async () => { const { workspace: ws, user } = await freeWorkspace(); price = { currency: "usd", currency_options: { eur: {} as Stripe.Price.CurrencyOptions, }, };
const result = await maybeStartSignupTrial({ userId: user.id, email: user.email ?? "", provider: "github", currency: "EUR", });
const trialEndsAt = new Date(trialEnd * 1000); expect(result).toEqual({ started: true, trialEndsAt });
assertSpyCalls(createCustomer, 1); expect(createCustomer.calls[0]?.args).toEqual([ { email: user.email, metadata: { workspaceId: String(ws.id), trialed: "true" }, }, { idempotencyKey: `trial-customer:ws_${ws.id}` }, ]);
assertSpyCalls(createSubscription, 1); const [params, options] = createSubscription.calls[0]?.args ?? []; expect(params).toMatchObject({ customer: customerId, currency: "eur", items: [{ price: STARTER_PRICE }], trial_period_days: TRIAL_DAYS, trial_settings: { end_behavior: { missing_payment_method: "cancel" } }, }); expect(options).toEqual({ idempotencyKey: `trial-sub:ws_${ws.id}` });
const after = await db .select() .from(workspace) .where(eq(workspace.id, ws.id)) .get(); expect(after?.plan).toBe("starter"); expect(after?.stripeId).toBe(customerId); expect(after?.subscriptionId).toBe("sub_trial"); expect(after?.trialEndsAt).toEqual(trialEndsAt); expect(after?.endsAt).toEqual(trialEndsAt); expect(JSON.parse(after?.limits ?? "{}")).toEqual(getLimits("starter")); });
test("falls back to the price currency when the requested one is missing", async () => { const { user } = await freeWorkspace(); price = { currency: "eur", currency_options: {} };
await maybeStartSignupTrial({ userId: user.id, email: user.email ?? "", currency: "USD", });
expect(createSubscription.calls[0]?.args[0]).toMatchObject({ currency: "eur", }); });
test("skips SSO sign-ins", async () => { const { user } = await freeWorkspace();
const result = await maybeStartSignupTrial({ userId: user.id, email: user.email ?? "", provider: "workos", currency: "USD", });
expect(result).toEqual({ started: false, reason: "sso" }); assertSpyCalls(createCustomer, 0); });
test("skips magic-link sign-ins", async () => { const { user } = await freeWorkspace();
const result = await maybeStartSignupTrial({ userId: user.id, email: user.email ?? "", provider: "resend", currency: "USD", });
expect(result).toEqual({ started: false, reason: "email" }); assertSpyCalls(createCustomer, 0); });
test("skips users with a pending invitation", async () => { const { workspace: ws, user } = await freeWorkspace(); await db.insert(invitation).values({ email: (user.email ?? "").toUpperCase(), workspaceId: ws.id, token: crypto.randomUUID(), expiresAt: new Date(Date.now() + 86_400_000), });
const result = await maybeStartSignupTrial({ userId: user.id, email: user.email ?? "", currency: "USD", });
expect(result).toEqual({ started: false, reason: "invited" }); assertSpyCalls(createCustomer, 0); });
test("skips disposable email domains", async () => { const { user } = await freeWorkspace();
const result = await maybeStartSignupTrial({ userId: user.id, email: "someone@mailinator.com", currency: "USD", });
expect(result).toEqual({ started: false, reason: "disposable" }); assertSpyCalls(list, 0); });
test("skips emails that already had a trial", async () => { const { user } = await freeWorkspace(); customers = [ customer("cus_paid"), customer("cus_old", { trialed: "true" }), ];
const result = await maybeStartSignupTrial({ userId: user.id, email: "O'Brien@example.com", currency: "USD", });
expect(result).toEqual({ started: false, reason: "already_trialed" }); expect(list.calls[0]?.args[0]).toEqual({ email: "o'brien@example.com", limit: 100, }); assertSpyCalls(createCustomer, 0); });
test("ignores customers for the email that never trialed", async () => { const { user } = await freeWorkspace(); customers = [customer("cus_paid")];
const result = await maybeStartSignupTrial({ userId: user.id, email: user.email ?? "", currency: "USD", });
expect(result).toMatchObject({ started: true }); assertSpyCalls(createCustomer, 1); });
test("drops its customer when another request linked one first", async () => { const { workspace: ws, user } = await freeWorkspace(); const linked = `cus_${crypto.randomUUID()}`; // Simulate the race: a checkout links its customer while ours is created. onCreateCustomer = async () => { await db .update(workspace) .set({ stripeId: linked }) .where(eq(workspace.id, ws.id)); }; const delCustomer = stub(stripe.customers, "del", () => Promise.resolve({} as Stripe.Response<Stripe.DeletedCustomer>), ); stubs.push(delCustomer);
const result = await maybeStartSignupTrial({ userId: user.id, email: user.email ?? "", currency: "USD", });
expect(result).toEqual({ started: false, reason: "not_eligible" }); expect(delCustomer.calls[0]?.args[0]).toBe(customerId); assertSpyCalls(createSubscription, 0); const after = await db .select({ stripeId: workspace.stripeId }) .from(workspace) .where(eq(workspace.id, ws.id)) .get(); expect(after?.stripeId).toBe(linked); });
test("leaves an already-billed workspace alone", async () => { const { user } = await createTestWorkspace({ plan: "team" });
const result = await maybeStartSignupTrial({ userId: user.id, email: user.email ?? "", currency: "USD", });
expect(result).toEqual({ started: false, reason: "not_eligible" }); assertSpyCalls(createCustomer, 0); });});