Feat(vitest-pool-workers): Support specifying an environment in tests

.changeset/rich-jokes-chew.md

@@ -0,0 +1,24 @@
+---
+"@cloudflare/vitest-pool-workers": minor
+---
+
+Feat: Support specifying an environment for your worker when running tests. This allows your tests to pick up bindings & variables that are scoped to specific environments.
+
+For example:
+
+```ts
+import { defineWorkersConfig } from "@cloudflare/vitest-pool-workers/config";
+
+export default defineWorkersConfig({
+	test: {
+		poolOptions: {
+			workers: {
+				wrangler: {
+					configPath: "./wrangler.toml",
+					environment: "production",
+				},
+			},
+		},
+	},
+});
+```

fixtures/vitest-pool-workers-examples/d1/vitest.config.ts

@@ -17,6 +17,7 @@ export default defineWorkersProject(async () => {
 					singleWorker: true,
 					wrangler: {
 						configPath: "./wrangler.toml",
+						environment: "production",
 					},
 					miniflare: {
 						// Add a test-only binding for migrations, so we can apply them in a

fixtures/vitest-pool-workers-examples/d1/wrangler.toml

@@ -3,7 +3,7 @@ main = "src/index.ts"
 compatibility_date = "2024-01-01"
 compatibility_flags = ["nodejs_compat"]
 
-[[ d1_databases ]]
+[[env.production.d1_databases]]
 binding = "DATABASE"
 database_name = "database"
 database_id = "00000000-0000-0000-0000-000000000000"

packages/vitest-pool-workers/src/pool/config.ts

@@ -49,7 +49,9 @@ const WorkersPoolOptionsSchema = z.object({
 		})
 		.passthrough()
 		.optional(),
-	wrangler: z.object({ configPath: z.ostring() }).optional(),
+	wrangler: z
+		.object({ configPath: z.ostring(), environment: z.ostring() })
+		.optional(),
 });
 export type SourcelessWorkerOptions = Omit<
 	WorkerOptions,
@@ -181,7 +183,10 @@ async function parseCustomPoolOptions(
 		// Lazily import `wrangler` if and when we need it
 		const wrangler = await import("wrangler");
 		const { workerOptions, define, main } =
-			wrangler.unstable_getMiniflareWorkerOptions(configPath);
+			wrangler.unstable_getMiniflareWorkerOptions(
+				configPath,
+				options.wrangler.environment
+			);
 
 		// If `main` wasn't explicitly configured, fall back to Wrangler config's
 		options.main ??= main;

packages/wrangler/src/api/integrations/platform/index.ts

@@ -216,12 +216,18 @@ export type SourcelessWorkerOptions = Omit<
 	"script" | "scriptPath" | "modules" | "modulesRoot"
 > & { modulesRules?: ModuleRule[] };
 
-export function unstable_getMiniflareWorkerOptions(configPath: string): {
+export function unstable_getMiniflareWorkerOptions(
+	configPath: string,
+	environment?: string
+): {
 	workerOptions: SourcelessWorkerOptions;
 	define: Record<string, string>;
 	main?: string;
 } {
-	const config = readConfig(configPath, { experimentalJsonConfig: true });
+	const config = readConfig(configPath, {
+		experimentalJsonConfig: true,
+		env: environment,
+	});
 
 	const modulesRules: ModuleRule[] = config.rules
 		.concat(DEFAULT_MODULE_RULES)