vitest.unit.config.ts

import { cpus } from "node:os";
import tsconfigPaths from "vite-tsconfig-paths";
import { configDefaults, defineConfig } from "vitest/config";

// Allow auth module (src/services/auth, tokens.ts) to load when any unit test imports it.
// NODE_ENV: only set when unset (?? "test"); does not override CI's NODE_ENV=production.
// API_AUTH_JWT_SECRET: when unset, this fallback prevents getSecret() from throwing TalawaRestError
// ("API_AUTH_JWT_SECRET must be set in production") so password.test.ts and other files can static-import auth.
process.env.NODE_ENV = process.env.NODE_ENV ?? "test";
if (!process.env.API_AUTH_JWT_SECRET) {
	process.env.API_AUTH_JWT_SECRET = "unit-test-secret";
}

const isCI = !!process.env.CI;
const cpuCount = cpus().length;

const MAX_CI_THREADS = 12; // Reduced to leave headroom
const MAX_LOCAL_THREADS = 16;

// Allow override via env var for CI rollout testing
const envCiThreads = process.env.VITEST_CI_THREADS
	? Number.parseInt(process.env.VITEST_CI_THREADS, 10)
	: undefined;

const computedCiThreads = Math.min(
	MAX_CI_THREADS,
	Math.max(4, Math.floor(cpuCount * 0.85)), // Increased utilization
);

const ciThreads =
	envCiThreads && !Number.isNaN(envCiThreads)
		? envCiThreads
		: computedCiThreads;

const localThreads = Math.min(MAX_LOCAL_THREADS, Math.max(4, cpuCount));

// Runtime observability for CI rollout
if (isCI) {
	// const shardId = process.env.SHARD_INDEX || "unknown";
	// console.log(
	// 	`[Vitest Config] Shard ${shardId}: cpuCount=${cpuCount}, computedCiThreads=${computedCiThreads}, ciThreads=${ciThreads}`,
	// );
	if (ciThreads < cpuCount) {
		// console.warn(
		// 	`[Vitest Config] Thread count reduced from ${cpuCount} to ${ciThreads} to prevent over-subscription`,
		// );
	}
	if (envCiThreads) {
		// console.log(
		// 	`[Vitest Config] Using VITEST_CI_THREADS override: ${ciThreads}`,
		// );
	}
}

// Skip global setup for pure unit tests that don't need server/db

export default defineConfig({
	plugins: [tsconfigPaths()],
	test: {
		// Define unit test patterns
		include: [
			"test/unit/**/*.{test,spec}.ts",
			"test/utilities/**/*.{test,spec}.ts",
			"test/services/**/*.{test,spec}.ts",
			"test/routes/**/*.{test,spec}.ts",
			"test/observability/**/*.{test,spec}.ts",
			"test/workers/**/*.{test,spec}.ts",
			"test/config/**/*.{test,spec}.ts",
			"test/fastifyPlugins/**/*.{test,spec}.ts",
			"test/plugin/**/*.{test,spec}.ts",
			"test/scripts/**/*.{test,spec}.ts",
			"test/setup/**/*.{test,spec}.ts",
			"test/unit_tests/**/*.{test,spec}.ts",
			"test/helpers/**/*.{test,spec}.ts",
			"test/*.{test,spec}.ts",
			"src/**/*.{test,spec}.ts",
		],
		exclude: [
			...configDefaults.exclude,
			"dist/**",
			"coverage/**",
			"docs/**",
			"**/*.d.ts",
			"data/**",
			"docker/**",
			"drizzle_migrations/**",
			"envFiles/**",
			"scripts/**",
			"**/scripts/**",
			"test/drizzle/**",
			"test/graphql/**",
			"test/install/**",
		],
		coverage: {
			provider: "v8",
			reporter: ["text", "lcov", "html", "json"],
			// Restrict unit coverage to unit-owned source paths
			include: [
				"src/config/**",
				"src/fastifyPlugins/**",
				"src/observability/**",
				"src/plugin/**",
				"src/plugins/**",
				"src/routes/**",
				"src/services/**",
				"src/setup/**",
				"src/types/**",
				"src/utilities/**",
				"src/workers/**",
				"src/*.ts",
			],
			// Only include files that are actually touched by tests
			all: false,
			exclude: [
				...(configDefaults.coverage?.exclude ?? []),
				"dist/**",
				"coverage/**",
				"docs/**",
				"**/*.d.ts",
				"data/**",
				"docker/**",
				"drizzle_migrations/**",
				"envFiles/**",
				"**/scripts/**",
				"vitest.*.config.ts",
			],
		},

		// https://vitest.dev/config/#globalsetup
		globalSetup: [],

		// https://vitest.dev/config/#passwithnotests
		passWithNoTests: true,

		hookTimeout: 30000, // 30 seconds for hooks
		testTimeout: 60000, // 60 seconds per test
		pool: "threads", // for faster test execution and to avoid postgres max-limit error
		isolate: true,
		maxWorkers: isCI ? ciThreads : localThreads,
		// Set maxConcurrency lower than maxWorkers to prevent single heavy test files
		// from consuming all workers and blocking other test files
		maxConcurrency: isCI
			? Math.max(1, Math.floor(ciThreads / 2))
			: Math.max(1, Math.floor(localThreads / 2)),
		fileParallelism: true,
		sequence: {
			shuffle: false,
			concurrent: false,
		},
	},
});