feat: support custom env in config

vitest-dev · May 9, 2022 · 4701e0b · 4701e0b
1 parent cd4bed5
commit 4701e0b
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 0 deletions.
diff --git a/packages/vitest/src/node/cli-api.ts b/packages/vitest/src/node/cli-api.ts
@@ -32,6 +32,9 @@ export async function startVitest(cliFilters: string[], options: CliOptions, vit
 
   process.env.VITEST_MODE = ctx.config.watch ? 'WATCH' : 'RUN'
 
+  if (ctx.config.env)
+    Object.assign(process.env, ctx.config.env)
+
   if (ctx.config.coverage.enabled) {
     if (!await ensurePackageInstalled('c8')) {
       process.exitCode = 1

diff --git a/packages/vitest/src/types/config.ts b/packages/vitest/src/types/config.ts
@@ -314,6 +314,11 @@ export interface InlineConfig {
    */
   logHeapUsage?: boolean
 
+  /**
+   * Custom environment variables assigned to `process.env` before running tests.
+   */
+  env?: Record<string, string>
+
   /**
    * Options for @sinon/fake-timers
    */

diff --git a/test/core/test/env.test.ts b/test/core/test/env.test.ts
@@ -37,3 +37,8 @@ test('can see env in "define"', () => {
 test('has worker env', () => {
   expect(process.env.VITEST_WORKER_ID).toBeDefined()
 })
+
+test('custom env', () => {
+  expect(process.env.CUSTOM_ENV).toBe('foo')
+  expect(import.meta.env.CUSTOM_ENV).toBe('foo')
+})
diff --git a/test/core/vitest.config.ts b/test/core/vitest.config.ts
@@ -45,6 +45,9 @@ export default defineConfig({
     coverage: {
       reporter: ['text', 'html'],
     },
+    env: {
+      CUSTOM_ENV: 'foo',
+    },
     resolveSnapshotPath: (path, extension) => {
       if (path.includes('moved-snapshot'))
         return path + extension