feat: set up test and run tests easier

2025-10-20 10:32:26 +01:00
parent 54aafa5734
commit dc90d1dab4
21 changed files with 1165 additions and 216 deletions
--- a/scripts/setupTestUtils/createTestOrg.ts
+++ b/scripts/setupTestUtils/createTestOrg.ts
@@ -0,0 +1,128 @@
+import { type OrgConfig, member, organizations, user } from "@autumn/shared";
+import chalk from "chalk";
+import { eq } from "drizzle-orm";
+import type { DrizzleCli } from "@server/db/initDrizzle.js";
+import { createKey } from "@server/internal/dev/api-keys/apiKeyUtils.js";
+
+const TEST_ORG_CONFIG = {
+	id: "org_2sWv2S8LJ9iaTjLI6UtNsfL88Kt",
+	slug: "unit-test-org",
+	name: "Unit Test Org",
+	createdAt: new Date(1738583937426).toISOString(),
+	created_at: 1738583937426,
+};
+
+/**
+ * Creates a test organization in the database and generates an API key
+ */
+export async function createTestOrg({
+	db,
+}: {
+	db: DrizzleCli;
+}): Promise<string | null> {
+	console.log(
+		chalk.magentaBright(
+			"\n================ Creating Test Organization ================\n",
+		),
+	);
+
+	// Check if org already exists
+	const existingOrg = await db.query.organizations.findFirst({
+		where: eq(organizations.id, TEST_ORG_CONFIG.id),
+	});
+
+	if (existingOrg) {
+		console.log(
+			chalk.yellowBright(
+				`Test organization '${TEST_ORG_CONFIG.slug}' already exists. Creating new API key.`,
+			),
+		);
+
+		// Always create a new API key for existing org
+		const apiKey = await createKey({
+			db,
+			env: "sandbox" as any,
+			name: "Unit Test Key",
+			orgId: TEST_ORG_CONFIG.id,
+			prefix: "am_sk_test",
+			meta: {
+				createdBy: "setup-test-script",
+				createdAt: new Date().toISOString(),
+			},
+			userId: undefined,
+		});
+
+		console.log(chalk.greenBright("✅ Created API key for existing organization"));
+		return apiKey;
+	}
+
+	// Create the test organization
+	const org = {
+		id: TEST_ORG_CONFIG.id,
+		slug: TEST_ORG_CONFIG.slug,
+		name: TEST_ORG_CONFIG.name,
+		createdAt: new Date(TEST_ORG_CONFIG.created_at),
+		created_at: TEST_ORG_CONFIG.created_at,
+		stripe_connected: false,
+		default_currency: "usd",
+		config: {} as OrgConfig,
+		onboarded: true,
+	};
+
+	await db.insert(organizations).values(org);
+
+	console.log(
+		chalk.greenBright(
+			`✅ Created test organization: ${TEST_ORG_CONFIG.slug} (${TEST_ORG_CONFIG.id})`,
+		),
+	);
+
+	// Get first 5 users from database and create memberships
+	const users = await db.select().from(user).limit(5);
+
+	if (users.length > 0) {
+		const { generateId } = await import("@server/utils/genUtils.js");
+
+		const memberships = users.map((u) => ({
+			id: generateId("mem"),
+			organizationId: TEST_ORG_CONFIG.id,
+			userId: u.id,
+			role: "owner",
+			createdAt: new Date(),
+		}));
+
+		await db.insert(member).values(memberships);
+
+		console.log(
+			chalk.greenBright(
+				`✅ Created ${memberships.length} membership(s) for test organization`,
+			),
+		);
+	} else {
+		console.log(
+			chalk.yellowBright(
+				"⚠ No users found in database. Skipping membership creation.",
+			),
+		);
+	}
+
+	// Create API key for the new org
+	const apiKey = await createKey({
+		db,
+		env: "sandbox" as any,
+		name: "Unit Test Key",
+		orgId: TEST_ORG_CONFIG.id,
+		prefix: "am_sk_test",
+		meta: {
+			createdBy: "setup-test-script",
+			createdAt: new Date().toISOString(),
+		},
+		userId: undefined,
+	});
+
+	console.log(chalk.greenBright("✅ Created API key for test organization"));
+
+	return apiKey;
+}
+
+export { TEST_ORG_CONFIG };