こんにちは、CX事業本部 Delivery部の若槻です。
最近テスティングフレームワークの Vitest の検証を引き続き行っているのですが、テスト実行がとても軽快で気に入っています。
今回は Vitest でテスト用の環境変数を設定する方法を確認してみました。
試してみる
環境
$ npm ls vitest
cdk_sample_app@0.1.0 /Users/wakatsuki.ryuta/projects/cm-rwakatsuki/cdk_sample_app
└── vitest@0.34.4
準備
テスト対象のモジュールで環境変数ENV_KEY_1
を取得するようなコードを書いてみます。
src/hoge.ts
const ENV_KEY_1 = process.env.ENV_KEY_1;
export const hogeFunc = () => {
console.log('ENV_KEY_1_ON_HOGE', ENV_KEY_1);
};
テストファイルでも環境変数ENV_KEY_1
を取得するようにし、また前述のモジュールをインポートするようにしています。
src/hoge.test.ts
import { hogeFunc } from './hoge';
import { test, afterAll } from 'vitest';
const ENV_KEY_1 = process.env.ENV_KEY_1;
afterAll((): void => {
hogeFunc();
});
test('should pass', (): void => {
console.log('ENV_KEY_1', ENV_KEY_1);
});
npm script に vitest run
を追加します。
package.json
{
"scripts": {
"test": "vitest run"
}
}
この段階でテストを実行すると、環境変数ENV_KEY_1
は未設定なのでもちろん取得できていません。
$ npm run test
> cdk_sample_app@0.1.0 test
> vitest run
RUN v0.34.3 /Users/wakatsuki.ryuta/projects/cm-rwakatsuki/cdk_sample_app
stdout | src/hoge.test.ts > should pass
ENV_KEY_1 undefined
stdout | unknown test
ENV_KEY_1_ON_HOGE undefined
✓ src/hoge.test.ts (1)
✓ should pass
Test Files 1 passed (1)
Tests 1 passed (1)
Start at 20:49:48
Duration 261ms (transform 55ms, setup 0ms, collect 47ms, tests 1ms, environment 0ms, prepare 61ms)
環境変数を設定してみる
Vitest Configファイルで test.env
に KEY:VALUE
形式で環境変数を設定します。
vite.config.ts
import { defineConfig } from 'vitest/config';
export default defineConfig({
test: {
env: {
ENV_KEY_1: 'ENV_VALUE_1',
},
},
});
テストを実行すると、環境変数が設定されモジュールおよびテストコードで取得できていることが確認できます。
$ npm run test
> cdk_sample_app@0.1.0 test
> vitest run
RUN v0.34.3 /Users/wakatsuki.ryuta/projects/cm-rwakatsuki/cdk_sample_app
stdout | src/hoge.test.ts > should pass
ENV_KEY_1 ENV_VALUE_1
stdout | unknown test
ENV_KEY_1_ON_HOGE ENV_VALUE_1
✓ src/hoge.test.ts (1)
✓ should pass
Test Files 1 passed (1)
Tests 1 passed (1)
Start at 20:56:51
Duration 230ms (transform 22ms, setup 0ms, collect 14ms, tests 1ms, environment 0ms, prepare 55ms)
ステージごとに異なる環境変数を設定する
次に、開発環境や本番環境などステージごとに異なる環境変数を使用したい場合を想定した設定をしてみます。
まず各ステージごとの Vitest Config を作成します。それぞれで異なる環境変数を設定しています。
vite.config.development.ts
import { defineConfig } from 'vitest/config';
export default defineConfig({
test: {
env: {
ENV_KEY_1: 'ENV_VALUE_1_DEV',
},
},
});
vite.config.production.ts
import { defineConfig } from 'vitest/config';
export default defineConfig({
test: {
env: {
ENV_KEY_1: 'ENV_VALUE_1_PRD',
},
},
});
そして、vitest コマンドの --config
オプションで Vitest Config を指定して実行するようにします。
package.json
{
"scripts": {
"test:dev": "vitest run --config vite.config.development.ts",
"test:prd": "vitest run --config vite.config.production.ts"
}
}
テストを実行すると、ステージごとに異なる環境変数が設定されていることが確認できます。
$ npm run test:dev
> cdk_sample_app@0.1.0 test:dev
> vitest run --config vite.config.development.ts
RUN v0.34.3 /Users/wakatsuki.ryuta/projects/cm-rwakatsuki/cdk_sample_app
stdout | src/hoge.test.ts > should pass
ENV_KEY_1 ENV_VALUE_1_DEV
stdout | unknown test
ENV_KEY_1_ON_HOGE ENV_VALUE_1_DEV
✓ src/hoge.test.ts (1)
✓ should pass
Test Files 1 passed (1)
Tests 1 passed (1)
Start at 20:58:42
Duration 228ms (transform 24ms, setup 0ms, collect 13ms, tests 1ms, environment 0ms, prepare 54ms)
$ npm run test:prd
> cdk_sample_app@0.1.0 test:prd
> vitest run --config vite.config.production.ts
RUN v0.34.3 /Users/wakatsuki.ryuta/projects/cm-rwakatsuki/cdk_sample_app
stdout | src/hoge.test.ts > should pass
ENV_KEY_1 ENV_VALUE_1_PRD
stdout | unknown test
ENV_KEY_1_ON_HOGE ENV_VALUE_1_PRD
✓ src/hoge.test.ts (1)
✓ should pass
Test Files 1 passed (1)
Tests 1 passed (1)
Start at 20:58:47
Duration 252ms (transform 24ms, setup 0ms, collect 13ms, tests 1ms, environment 0ms, prepare 55ms)
各ステージで共通のコンフィグや環境変数を使用したい場合
前述の方法ではステージごとに Vitest Config ファイルを分けましたが、一部コンフィグや環境変数は共通化したい場合もあるかと思います。
その場合は、次のような共有用の JSON ファイルを作成します。
vite.shared.ts
import path from 'path';
import { UserConfig } from 'vitest/config';
export const configShared: UserConfig = {
test: {
alias: {
'@': path.resolve(__dirname, '.'),
},
env: {
COMMON_ENV: 'COMMON_VALUE',
},
},
};
そして各ステージの Vitest Config ファイルで共有用の JSON ファイルをインポートして使用します。
vite.config.development.ts
import { defineConfig } from 'vitest/config';
import { configShared } from './vite.shared';
export default defineConfig({
test: {
...configShared.test,
env: {
...configShared.test!.env,
ENV_KEY_1: 'ENV_VALUE_1_DEV',
},
},
});
vite.config.production.ts
import { defineConfig } from 'vitest/config';
import { configShared } from './vite.shared';
export default defineConfig({
test: {
...configShared.test,
env: {
...configShared.test!.env,
ENV_KEY_1: 'ENV_VALUE_1_PRD',
},
},
});
テストファイルを修正します。
src/hoge.test.ts
import { test } from 'vitest';
const ENV_KEY_1 = process.env.ENV_KEY_1;
const COMMON_ENV = process.env.COMMON_ENV;
test('should pass', (): void => {
console.log('ENV_KEY_1', ENV_KEY_1);
console.log('COMMON_ENV', COMMON_ENV);
});
テストを実行すると、ステージごとの環境変数と共有用の環境変数が設定されていることが確認できます。
$ npm run test:dev
> cdk_sample_app@0.1.0 test:dev
> vitest run --config vite.config.development.ts
RUN v0.34.3 /Users/wakatsuki.ryuta/projects/cm-rwakatsuki/cdk_sample_app
stdout | src/hoge.test.ts > should pass
ENV_KEY_1 ENV_VALUE_1_DEV
COMMON_ENV COMMON_VALUE
✓ src/hoge.test.ts (1)
✓ should pass
Test Files 1 passed (1)
Tests 1 passed (1)
Start at 21:14:15
Duration 236ms (transform 19ms, setup 0ms, collect 10ms, tests 2ms, environment 0ms, prepare 50ms)
おわりに
Vitest でテスト用の環境変数を設定する方法を確認してみました。
公式ドキュメントに詳しい記載が無かったので、方法を探すのに若干苦労しました。
これから Vitest を触ってみよう方の参考になれば幸いです。
以上