fix: plat-5545 fixing unit test by reading new env var

OpenZeppelin · Jan 10, 2025 · 28787f7 · 28787f7
1 parent 7aeb571
commit 28787f7
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/packages/kvstore/src/index.test.ts b/packages/kvstore/src/index.test.ts
@@ -1,3 +1,4 @@
+import { beforeEach } from 'node:test';
 import { KeyValueStoreClient, KeyValueStoreCreateParams } from '.';
 import { KeyValueStoreActionClient } from './action';
 import { KeyValueStoreLocalClient } from './local';
@@ -10,13 +11,23 @@ class TestClient extends KeyValueStoreClient {
 }
 
 describe('KeyValueStoreClient', () => {
+  const envs = process.env;
+  beforeEach(() => {
+    process.env = { ...envs };
+  });
+
+  afterEach(() => {
+    process.env = envs;
+  });
+
   describe('create', () => {
     test('creates a local client', async () => {
       const client = new TestClient({ path: '/tmp/foo' });
       expect(client.getImplementation()).toBeInstanceOf(KeyValueStoreLocalClient);
     });
 
     test('creates an autotask client', async () => {
+      process.env.DEFENDER_ENV = 'action';
       const credentials = JSON.stringify({
         AccessKeyId: 'keyId',
         SecretAccessKey: 'accessKey',