refactor(test): share env var helpers

src/test-utils/env.ts

@@ -16,3 +16,38 @@ export function captureEnv(keys: string[]) {
     },
   };
 }
+
+export function withEnv<T>(env: Record<string, string | undefined>, fn: () => T): T {
+  const snapshot = captureEnv(Object.keys(env));
+  try {
+    for (const [key, value] of Object.entries(env)) {
+      if (value === undefined) {
+        delete process.env[key];
+      } else {
+        process.env[key] = value;
+      }
+    }
+    return fn();
+  } finally {
+    snapshot.restore();
+  }
+}
+
+export async function withEnvAsync<T>(
+  env: Record<string, string | undefined>,
+  fn: () => Promise<T>,
+): Promise<T> {
+  const snapshot = captureEnv(Object.keys(env));
+  try {
+    for (const [key, value] of Object.entries(env)) {
+      if (value === undefined) {
+        delete process.env[key];
+      } else {
+        process.env[key] = value;
+      }
+    }
+    return await fn();
+  } finally {
+    snapshot.restore();
+  }
+}