refactor: use vi.stubEnv instead of process.env

apps/www/components/page/copy-button.test.tsx

@@ -1,5 +1,5 @@
-import { cleanup, render, screen } from "@testing-library/react";
-import { afterEach, describe, expect, it, vi } from "vitest";
+import { render, screen } from "@testing-library/react";
+import { describe, expect, it, vi } from "vitest";
 import { CopyButton } from "./copy-button";
 
 // Mock the useToast hook

apps/www/components/page/sail-button.test.tsx

@@ -1,6 +1,6 @@
-import { cleanup, render, screen } from "@testing-library/react";
+import { render, screen } from "@testing-library/react";
 import userEvent from "@testing-library/user-event";
-import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
+import { describe, expect, it, vi } from "vitest";
 import { SailButton } from "./sail-button";
 
 // Mock Next.js router

apps/www/components/page/star-us-button.test.tsx

@@ -1,5 +1,5 @@
-import { cleanup, render, screen } from "@testing-library/react";
-import { afterEach, describe, expect, it, vi } from "vitest";
+import { render, screen } from "@testing-library/react";
+import { describe, expect, it, vi } from "vitest";
 import { StarUsButton } from "./star-us-button";
 
 // Mock fetch to avoid real API calls

apps/www/components/page/video-demo.test.tsx

@@ -1,5 +1,5 @@
-import { cleanup, fireEvent, render, screen } from "@testing-library/react";
-import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
+import { fireEvent, render, screen } from "@testing-library/react";
+import { beforeEach, describe, expect, it, vi } from "vitest";
 import { VideoDemo } from "./video-demo";
 
 // Mock window.open

apps/www/components/ui/heading.test.tsx

@@ -1,6 +1,6 @@
-import { cleanup, render, screen } from "@testing-library/react";
+import { render, screen } from "@testing-library/react";
 import userEvent from "@testing-library/user-event";
-import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
+import { describe, expect, it, vi } from "vitest";
 import { Heading } from "./heading";
 
 // Mock the useIsMobile hook

apps/www/hooks/use-is-mobile.test.ts

@@ -1,5 +1,5 @@
 import { renderHook } from "@testing-library/react";
-import { beforeEach, describe, expect, it, vi } from "vitest";
+import { describe, expect, it, vi } from "vitest";
 import { useIsMobile } from "./use-is-mobile";
 
 // Mock window.matchMedia

apps/www/lib/utils/github.test.ts

@@ -1,16 +1,9 @@
-import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
+import { beforeEach, describe, expect, it, vi } from "vitest";
 import { breakDownGithubUrl, getLinkTitleAndHref } from "./github";
 
 describe("github utilities", () => {
-	const originalEnv = process.env;
-
 	beforeEach(() => {
 		vi.resetModules();
-		process.env = { ...originalEnv };
-	});
-
-	afterEach(() => {
-		process.env = originalEnv;
 	});
 
 	describe("breakDownGithubUrl", () => {
@@ -35,7 +28,7 @@ describe("github utilities", () => {
 		});
 
 		it("should use environment variable when no URL provided", () => {
-			process.env.NEXT_PUBLIC_GITHUB_URL = "https://github.com/example/repo";
+			vi.stubEnv("NEXT_PUBLIC_GITHUB_URL", "https://github.com/example/repo");
 
 			const result = breakDownGithubUrl();
 
@@ -47,8 +40,11 @@ describe("github utilities", () => {
 		});
 
 		it("should use custom branch from environment variable", () => {
-			process.env.NEXT_PUBLIC_GITHUB_URL = "https://github.com/yamcodes/arkenv";
-			process.env.NEXT_PUBLIC_GITHUB_BRANCH = "develop";
+			vi.stubEnv(
+				"NEXT_PUBLIC_GITHUB_URL",
+				"https://github.com/yamcodes/arkenv",
+			);
+			vi.stubEnv("NEXT_PUBLIC_GITHUB_BRANCH", "develop");
 
 			const result = breakDownGithubUrl();
 
@@ -60,7 +56,7 @@ describe("github utilities", () => {
 		});
 
 		it("should throw error when no URL is configured", () => {
-			delete process.env.NEXT_PUBLIC_GITHUB_URL;
+			vi.unstubAllEnvs();
 
 			expect(() => breakDownGithubUrl()).toThrow(
 				"NEXT_PUBLIC_GITHUB_URL is not configured",
@@ -121,7 +117,7 @@ describe("github utilities", () => {
 		});
 
 		it("should use environment variable when no URL provided", () => {
-			process.env.NEXT_PUBLIC_GITHUB_URL = "https://github.com/example/repo";
+			vi.stubEnv("NEXT_PUBLIC_GITHUB_URL", "https://github.com/example/repo");
 
 			const result = getLinkTitleAndHref("test.md");
 
@@ -132,8 +128,11 @@ describe("github utilities", () => {
 		});
 
 		it("should use custom branch from environment variable", () => {
-			process.env.NEXT_PUBLIC_GITHUB_URL = "https://github.com/yamcodes/arkenv";
-			process.env.NEXT_PUBLIC_GITHUB_BRANCH = "develop";
+			vi.stubEnv(
+				"NEXT_PUBLIC_GITHUB_URL",
+				"https://github.com/yamcodes/arkenv",
+			);
+			vi.stubEnv("NEXT_PUBLIC_GITHUB_BRANCH", "develop");
 
 			const result = getLinkTitleAndHref("package.json");
 
@@ -144,7 +143,7 @@ describe("github utilities", () => {
 		});
 
 		it("should throw error when no URL is configured", () => {
-			delete process.env.NEXT_PUBLIC_GITHUB_URL;
+			vi.unstubAllEnvs();
 
 			expect(() => getLinkTitleAndHref("test.md")).toThrow(
 				"NEXT_PUBLIC_GITHUB_URL is not configured",

apps/www/vitest.config.ts

@@ -14,5 +14,6 @@ export default defineProject({
 		environment: "jsdom",
 		setupFiles: ["./tests/setup.ts"],
 		restoreMocks: true,
+		unstubEnvs: true,
 	},
 });

packages/arkenv/src/index.test.ts

@@ -1,27 +1,11 @@
-import {
-	afterEach,
-	beforeEach,
-	describe,
-	expect,
-	expectTypeOf,
-	it,
-	vi,
-} from "vitest";
+import { afterEach, describe, expect, expectTypeOf, it, vi } from "vitest";
 import arkenv, { createEnv } from "./index";
 
-// Capture snapshot of process.env at module level
-const originalEnv = { ...process.env };
-
 describe("index.ts exports", () => {
-	beforeEach(() => {
-		// Replace with a clean environment for each test
-		process.env = {};
-	});
-
 	afterEach(() => {
-		// Restore mocks and reset process.env to captured snapshot
+		// Restore mocks and unstub all environment variables
 		vi.restoreAllMocks();
-		process.env = { ...originalEnv };
+		vi.unstubAllEnvs();
 	});
 
 	it("should export createEnv as default export", () => {
@@ -40,7 +24,7 @@ describe("index.ts exports", () => {
 
 	it("should work with default import", () => {
 		// Set test environment variable
-		process.env.TEST_DEFAULT_IMPORT = "test-value";
+		vi.stubEnv("TEST_DEFAULT_IMPORT", "test-value");
 
 		const env = arkenv({
 			TEST_DEFAULT_IMPORT: "string",
@@ -52,7 +36,7 @@ describe("index.ts exports", () => {
 
 	it("should work with named import", () => {
 		// Set test environment variable
-		process.env.TEST_NAMED_IMPORT = "test-value";
+		vi.stubEnv("TEST_NAMED_IMPORT", "test-value");
 
 		const env = createEnv({
 			TEST_NAMED_IMPORT: "string",
@@ -80,7 +64,7 @@ describe("index.ts exports", () => {
 
 	it("should have same behavior for both default and named imports", () => {
 		// Set test environment variable
-		process.env.COMPARISON_TEST = "same-value";
+		vi.stubEnv("COMPARISON_TEST", "same-value");
 
 		const envFromDefault = arkenv({
 			COMPARISON_TEST: "string",

packages/arkenv/vitest.config.ts

@@ -0,0 +1,7 @@
+import { defineProject } from "vitest/config";
+
+export default defineProject({
+	test: {
+		name: "arkenv",
+	},
+});

packages/vite-plugin/package.json

@@ -20,6 +20,7 @@
 		"tsup": "^8.5.0",
 		"typescript": "^5.9.3",
 		"vite": "^7.1.9",
+		"vite-tsconfig-paths": "^5.1.4",
 		"vitest": "^3.2.4"
 	},
 	"peerDependencies": {

packages/vite-plugin/src/index.test.ts

@@ -12,9 +12,6 @@ vi.mock("arkenv", () => ({
 
 import arkenvPlugin from "./index.js";
 
-// Capture snapshot of process.env at module level
-const ORIGINAL_ENV = { ...process.env };
-
 const fixturesDir = join(__dirname, "__fixtures__");
 
 // Get the mocked functions
@@ -28,14 +25,14 @@ for (const name of readdirSync(fixturesDir)) {
 
 	describe(`Fixture: ${name}`, () => {
 		beforeEach(() => {
-			// Clean environment start and mock cleanup
-			process.env = { ...ORIGINAL_ENV };
+			// Clear environment variables and mock cleanup
+			vi.unstubAllEnvs();
 			mockCreateEnv.mockClear();
 		});
 
 		afterEach(() => {
 			// Complete cleanup: restore environment and reset mocks
-			process.env = { ...ORIGINAL_ENV };
+			vi.unstubAllEnvs();
 			mockCreateEnv.mockReset();
 		});
 
@@ -44,7 +41,9 @@ for (const name of readdirSync(fixturesDir)) {
 
 			// Set up environment variables from the fixture
 			if (config.envVars) {
-				Object.assign(process.env, config.envVars);
+				for (const [key, value] of Object.entries(config.envVars)) {
+					vi.stubEnv(key, value);
+				}
 			}
 
 			await expect(
@@ -77,12 +76,12 @@ for (const name of readdirSync(fixturesDir)) {
 // Unit tests for plugin functionality
 describe("Plugin Unit Tests", () => {
 	beforeEach(() => {
-		process.env = { ...ORIGINAL_ENV };
+		vi.unstubAllEnvs();
 		mockCreateEnv.mockClear();
 	});
 
 	afterEach(() => {
-		process.env = { ...ORIGINAL_ENV };
+		vi.unstubAllEnvs();
 		mockCreateEnv.mockReset();
 	});
 

packages/vite-plugin/vitest.config.ts

@@ -1,15 +1,9 @@
 import path from "node:path";
-import { defineConfig } from "vitest/config";
+import { defineProject } from "vitest/config";
 
-export default defineConfig({
+export default defineProject({
 	test: {
 		name: "@arkenv/vite-plugin",
-		environment: "node",
-		coverage: {
-			provider: "v8",
-			reporter: ["text", "json", "html"],
-			exclude: ["node_modules/", "dist/", "**/*.test.ts", "**/*.config.ts"],
-		},
 	},
 	resolve: {
 		alias: {

vitest.config.ts

@@ -28,5 +28,6 @@ export default defineConfig({
 				"**/public/**",
 			],
 		},
+		unstubEnvs: true,
 	},
 });