cfw-autumn/server/tests/unit/billing/create-schedule/handle-create-schedule-errors.spec.ts

import { describe, expect, test } from "bun:test";
import type { CreateScheduleBillingContext } from "@autumn/shared";
import { ms } from "@autumn/shared";
import chalk from "chalk";
import type Stripe from "stripe";
import { handleCreateScheduleErrors } from "@/internal/billing/v2/actions/createSchedule/errors/handleCreateScheduleErrors";

const buildContext = ({
	immediateStartsAt,
	currentEpochMs,
	existingSchedule,
}: {
	immediateStartsAt: number;
	currentEpochMs: number;
	existingSchedule?: Stripe.SubscriptionSchedule;
}) =>
	({
		currentEpochMs,
		immediatePhase: {
			starts_at: immediateStartsAt,
			plans: [{ plan_id: "plan" }],
		},
		stripeSubscriptionSchedule: existingSchedule,
	}) as unknown as CreateScheduleBillingContext;

describe(chalk.yellowBright("handleCreateScheduleErrors"), () => {
	test("allows an immediate phase within the tolerance window", () => {
		const now = Date.now();

		expect(() =>
			handleCreateScheduleErrors({
				billingContext: buildContext({
					immediateStartsAt: now,
					currentEpochMs: now,
				}),
			}),
		).not.toThrow();
	});

	test("rejects creation when the immediate phase is far in the past", () => {
		const now = Date.now();

		expect(() =>
			handleCreateScheduleErrors({
				billingContext: buildContext({
					immediateStartsAt: now - ms.hours(1),
					currentEpochMs: now,
				}),
			}),
		).toThrow("The first phase must start immediately");
	});

	test("rejects creation when the immediate phase is far in the future", () => {
		const now = Date.now();

		expect(() =>
			handleCreateScheduleErrors({
				billingContext: buildContext({
					immediateStartsAt: now + ms.hours(1),
					currentEpochMs: now,
				}),
			}),
		).toThrow("The first phase must start immediately");
	});

	test("skips the immediate-start guard on updates (existing schedule)", () => {
		// Regression: when editing an existing schedule, the frontend preserves
		// the persisted starts_at for phase 0. Downstream Stripe execution anchors
		// the first phase to the schedule's current_phase.start_date anyway, so
		// the tolerance check should not reject a historical starts_at here.
		const now = Date.now();

		expect(() =>
			handleCreateScheduleErrors({
				billingContext: buildContext({
					immediateStartsAt: now - ms.days(30),
					currentEpochMs: now,
					existingSchedule: {
						id: "sub_sched_existing",
					} as unknown as Stripe.SubscriptionSchedule,
				}),
			}),
		).not.toThrow();
	});
});