fix: fix testcases to take push node environment from .env rather tha… (

#1196) * fix: fix testcases to take push node environment from .env rather than defining at each file * Dynamically loading the env from .env, add process-env.d.ts and remove redundant declarations of ENV
push-protocol · Apr 5, 2024 · 7a4bd8d · 7a4bd8d
1 parent 985c422
commit 7a4bd8d
Show file tree

Hide file tree

Showing 38 changed files with 210 additions and 190 deletions.
diff --git a/packages/restapi/package.json b/packages/restapi/package.json
@@ -28,7 +28,7 @@
     "viem": "^1.20.3"
   },
   "scripts": {
-    "test": "TS_NODE_PROJECT='./tsconfig.mocha.json' NODE_OPTIONS='--loader ts-node/esm' mocha -r ts-node/register 'tests/**/*.test.ts' --timeout 1200000 --require tests/root.ts --serial"
+    "test": "TS_NODE_PROJECT='./tsconfig.mocha.json' NODE_OPTIONS='--loader ts-node/esm' DOTENV_CONFIG_PATH='./tests/.env' mocha -r ts-node/register -r dotenv/config 'tests/**/*.test.ts' --timeout 1200000 --require tests/root.ts --serial"
   },
   "devDependencies": {
     "@types/chai": "^4.3.4",
@@ -43,4 +43,4 @@
     "ts-node": "^10.9.1",
     "typescript": "^5.0.2"
   }
-}
+}
diff --git a/packages/restapi/tests/lib/alias/arbitrum.test.ts b/packages/restapi/tests/lib/alias/arbitrum.test.ts
@@ -1,6 +1,3 @@
-import * as path from 'path';
-import * as dotenv from 'dotenv';
-dotenv.config({ path: path.resolve(__dirname, '../../.env') });
 import { ethers } from 'ethers';
 import { PushAPI } from '../../../src';
 import { ENV } from '../../../src/lib/constants';
@@ -10,6 +7,12 @@ describe('ARBITRUM ALIAS functionality', () => {
   let userBob: PushAPI;
   let account: string;
   let account2: string;
+
+  // accessing env dynamically using process.env
+  type EnvStrings = keyof typeof ENV;
+  const envMode = process.env.ENV as EnvStrings;
+  const _env = ENV[envMode];
+
   before(async () => {
     const provider = new ethers.providers.JsonRpcProvider(
       'https://sepolia-rollup.arbitrum.io/rpc'
@@ -20,12 +23,12 @@ describe('ARBITRUM ALIAS functionality', () => {
     );
     account = signer.address;
     userAlice = await PushAPI.initialize(signer, {
-      env: ENV.DEV,
+      env: _env,
     });
 
     const signer2 = new ethers.Wallet(ethers.Wallet.createRandom().privateKey);
     account2 = signer2.address;
-    userBob = await PushAPI.initialize(signer2, { env: ENV.DEV });
+    userBob = await PushAPI.initialize(signer2, { env: _env });
   });
 
   it.skip('Should be able to create channel', async () => {

diff --git a/packages/restapi/tests/lib/alias/berachain.test.ts b/packages/restapi/tests/lib/alias/berachain.test.ts
@@ -1,6 +1,3 @@
-import * as path from 'path';
-import * as dotenv from 'dotenv';
-dotenv.config({ path: path.resolve(__dirname, '../../.env') });
 import { ethers } from 'ethers';
 import { PushAPI } from '../../../src';
 import { ENV } from '../../../src/lib/constants';
@@ -10,6 +7,12 @@ describe('BERACHAIN ALIAS functionality', () => {
   let userBob: PushAPI;
   let account: string;
   let account2: string;
+
+  // accessing env dynamically using process.env
+  type EnvStrings = keyof typeof ENV;
+  const envMode = process.env.ENV as EnvStrings;
+  const _env = ENV[envMode];
+
   before(async () => {
     const provider = new ethers.providers.JsonRpcProvider(
       'https://artio.rpc.berachain.com' // berachain artio Provider
@@ -20,12 +23,12 @@ describe('BERACHAIN ALIAS functionality', () => {
     );
     account = signer.address;
     userAlice = await PushAPI.initialize(signer, {
-      env: ENV.DEV,
+      env: _env,
     });
 
     const signer2 = new ethers.Wallet(ethers.Wallet.createRandom().privateKey);
     account2 = signer2.address;
-    userBob = await PushAPI.initialize(signer2, { env: ENV.DEV });
+    userBob = await PushAPI.initialize(signer2, { env: _env });
   });
 
   it.skip('Should be able to create channel', async () => {

diff --git a/packages/restapi/tests/lib/alias/optimism.test.ts b/packages/restapi/tests/lib/alias/optimism.test.ts
@@ -1,6 +1,3 @@
-import * as path from 'path';
-import * as dotenv from 'dotenv';
-dotenv.config({ path: path.resolve(__dirname, '../../.env') });
 import { ethers } from 'ethers';
 import { PushAPI } from '../../../src';
 import { ENV } from '../../../src/lib/constants';
@@ -10,6 +7,12 @@ describe('OPTIMISM ALIAS functionality', () => {
   let userBob: PushAPI;
   let account: string;
   let account2: string;
+
+  // accessing env dynamically using process.env
+  type EnvStrings = keyof typeof ENV;
+  const envMode = process.env.ENV as EnvStrings;
+  const _env = ENV[envMode];
+
   before(async () => {
     const provider = new ethers.providers.JsonRpcProvider(
       'https://sepolia.optimism.io'
@@ -20,12 +23,12 @@ describe('OPTIMISM ALIAS functionality', () => {
     );
     account = signer.address;
     userAlice = await PushAPI.initialize(signer, {
-      env: ENV.DEV,
+      env: _env,
     });
 
     const signer2 = new ethers.Wallet(ethers.Wallet.createRandom().privateKey);
     account2 = signer2.address;
-    userBob = await PushAPI.initialize(signer2, { env: ENV.DEV });
+    userBob = await PushAPI.initialize(signer2, { env: _env });
   });
 
   it.skip('Should be able to create channel', async () => {

diff --git a/packages/restapi/tests/lib/benchmark/chatMessage.test.ts b/packages/restapi/tests/lib/benchmark/chatMessage.test.ts
@@ -2,7 +2,10 @@ import { ethers } from 'ethers';
 import Constants from '../../../src/lib/constants';
 import { PushAPI } from '../../../src/lib/pushapi/PushAPI';
 
-const _env = Constants.ENV.PROD;
+// accessing env dynamically using process.env
+type EnvStrings = keyof typeof Constants.ENV;
+const envMode = process.env.ENV as EnvStrings;
+const _env = Constants.ENV[envMode];
 
 /**
  * THIS TEST GROUP IS FOR BENCHMARKING CHAT PERFORMANCE

diff --git a/packages/restapi/tests/lib/benchmark/privateGroup.test.ts b/packages/restapi/tests/lib/benchmark/privateGroup.test.ts
@@ -3,7 +3,10 @@ import Constants from '../../../src/lib/constants';
 import { PushAPI } from '../../../src/lib/pushapi/PushAPI';
 import { users } from './data/did';
 
-const _env = Constants.ENV.LOCAL;
+// accessing env dynamically using process.env
+type EnvStrings = keyof typeof Constants.ENV;
+const envMode = process.env.ENV as EnvStrings;
+const _env = Constants.ENV[envMode];
 
 /**
  * THIS TEST GROUP IS FOR BENCHMARKING PUBLIC GROUP

diff --git a/packages/restapi/tests/lib/benchmark/publicGroup.test.ts b/packages/restapi/tests/lib/benchmark/publicGroup.test.ts
@@ -3,7 +3,10 @@ import Constants from '../../../src/lib/constants';
 import { PushAPI } from '../../../src/lib/pushapi/PushAPI';
 import { users } from './data/did';
 
-const _env = Constants.ENV.LOCAL;
+// accessing env dynamically using process.env
+type EnvStrings = keyof typeof Constants.ENV;
+const envMode = process.env.ENV as EnvStrings;
+const _env = Constants.ENV[envMode];
 
 /**
  * THIS TEST GROUP IS FOR BENCHMARKING PUBLIC GROUP

diff --git a/packages/restapi/tests/lib/channel/getChannelNotifications.test.ts b/packages/restapi/tests/lib/channel/getChannelNotifications.test.ts
@@ -1,24 +1,19 @@
-import * as path from 'path';
-import * as dotenv from 'dotenv';
-dotenv.config({ path: path.resolve(__dirname, '../../.env') });
 import * as PUSH_CHANNELS from '../../../src/lib/channels';
 import { expect } from 'chai';
 import { ethers } from 'ethers';
+import { ENV } from '../../../src/lib/constants';
 
 describe('PUSH_CHANNELS.sendNotification functionality', () => {
   let signer1: any;
   let account1: string;
   let signer2: any;
   let account2: string;
-  enum ENV {
-    PROD = 'prod',
-    STAGING = 'staging',
-    DEV = 'dev',
-    /**
-     * **This is for local development only**
-     */
-    LOCAL = 'local',
-  }
+
+  // accessing env dynamically using process.env
+  type EnvStrings = keyof typeof ENV;
+  const envMode = process.env.ENV as EnvStrings;
+  const _env = ENV[envMode];
+
   beforeEach(async () => {
     signer1 = new ethers.Wallet(
       '0xb9d00f786e1d024cfed08f696a775217ff75501f4aacef5ec0795fc4a2eb9df1'
@@ -33,22 +28,22 @@ describe('PUSH_CHANNELS.sendNotification functionality', () => {
     it('Should fetch notifications of a channel', async () => {
       const res = await PUSH_CHANNELS.getChannelNotifications({
         channel: '0xD8634C39BBFd4033c0d3289C4515275102423681',
-        env: ENV.DEV,
+        env: _env,
       });
       console.log(res);
     });
     it('Should should fetch notifications of a channel based on filter', async () => {
       const res = await PUSH_CHANNELS.getChannelNotifications({
         channel: '0xD8634C39BBFd4033c0d3289C4515275102423681',
-        env: ENV.DEV,
+        env: _env,
         filter: 1,
       });
       console.log(res);
     });
     it('Should should fetch notifications of a channel based on filter and in standard format', async () => {
       const res = await PUSH_CHANNELS.getChannelNotifications({
         channel: '0xD8634C39BBFd4033c0d3289C4515275102423681',
-        env: ENV.DEV,
+        env: _env,
         filter: 1,
         raw: false,
       });
@@ -57,7 +52,7 @@ describe('PUSH_CHANNELS.sendNotification functionality', () => {
     it('Should should fetch notifications of a channel based on filter and in standard format', async () => {
         const res = await PUSH_CHANNELS.getChannelNotifications({
           channel: '0xD8634C39BBFd4033c0d3289C4515275102423681',
-          env: ENV.DEV,
+          env: _env,
           filter: 1,
           raw: false,
           page: 1,

diff --git a/packages/restapi/tests/lib/channel/sendNotification.test.ts b/packages/restapi/tests/lib/channel/sendNotification.test.ts
@@ -1,9 +1,7 @@
-import * as path from 'path';
-import * as dotenv from 'dotenv';
-dotenv.config({ path: path.resolve(__dirname, '../../.env') });
 import * as PUSH_PAYLOAD from '../../../src/lib/payloads';
 import { expect } from 'chai';
 import { ethers } from 'ethers';
+import { ENV } from '../../../src/lib/constants';
 
 describe('PUSH_PAYLOAD.sendNotification functionality', () => {
   let signer1: any;
@@ -36,17 +34,13 @@ describe('PUSH_PAYLOAD.sendNotification functionality', () => {
       SUBSET = 4,
     }
 
-    enum ENV {
-      PROD = 'prod',
-      STAGING = 'staging',
-      DEV = 'dev',
-      /**
-       * **This is for local development only**
-       */
-      LOCAL = 'local',
-    }
+    // accessing env dynamically using process.env
+    type EnvStrings = keyof typeof ENV;
+    const envMode = process.env.ENV as EnvStrings;
+    const _env = ENV[envMode];
+
     const res = await PUSH_PAYLOAD.sendNotification({
-      env: ENV.DEV,
+      env: _env,
       signer: signer1,
       type: NOTIFICATION_TYPE.BROADCAST,
       identityType: IDENTITY_TYPE.DIRECT_PAYLOAD,

diff --git a/packages/restapi/tests/lib/channel/subscribeV2.test.ts b/packages/restapi/tests/lib/channel/subscribeV2.test.ts
@@ -1,6 +1,3 @@
-import * as path from 'path';
-import * as dotenv from 'dotenv';
-dotenv.config({ path: path.resolve(__dirname, '../../.env') });
 import * as PUSH_CHANNEL from '../../../src/lib/channels/';
 import { expect } from 'chai';
 import { ethers } from 'ethers';
@@ -12,6 +9,11 @@ describe('PUSH_CHANNEL.subscribeV2 functionality', () => {
   let signer2: any;
   let account2: string;
 
+  // accessing env dynamically using process.env
+  type EnvStrings = keyof typeof CONSTANTS.ENV;
+  const envMode = process.env.ENV as EnvStrings;
+  const _env = CONSTANTS.ENV[envMode];
+
   beforeEach(async () => {
     const WALLET1 = ethers.Wallet.createRandom();
     signer1 = new ethers.Wallet(WALLET1.privateKey);
@@ -27,7 +29,7 @@ describe('PUSH_CHANNEL.subscribeV2 functionality', () => {
       signer: signer1,
       channelAddress: 'eip155:11155111:0xD8634C39BBFd4033c0d3289C4515275102423681',
       userAddress: `eip155:11155111:${account1}`,
-      env: CONSTANTS.ENV.STAGING,
+      env: _env,
     });
     expect(res.status).to.be.equal(204);
   });
@@ -37,7 +39,7 @@ describe('PUSH_CHANNEL.subscribeV2 functionality', () => {
       signer: signer1,
       channelAddress: 'eip155:11155111:0xD8634C39BBFd4033c0d3289C4515275102423681',
       userAddress: `eip155:11155111:${account1}`,
-      env: CONSTANTS.ENV.STAGING,
+      env: _env,
     });
     // console.log(res)
     expect(res.status).to.be.equal(204);

diff --git a/packages/restapi/tests/lib/chat/chat.group.test.ts b/packages/restapi/tests/lib/chat/chat.group.test.ts
@@ -11,7 +11,11 @@ import {
 } from 'unique-names-generator';
 import { PushAPI } from '../../../src/lib/pushapi/PushAPI'; // Ensure correct import path
 
-const _env = Constants.ENV.DEV;
+// accessing env dynamically using process.env
+type EnvStrings = keyof typeof Constants.ENV;
+const envMode = process.env.ENV as EnvStrings;
+const _env = Constants.ENV[envMode];
+
 let groupName: string;
 let groupDescription: string;
 const groupImage =

diff --git a/packages/restapi/tests/lib/chat/chat.test.ts b/packages/restapi/tests/lib/chat/chat.test.ts
@@ -1,7 +1,3 @@
-import * as path from 'path';
-import * as dotenv from 'dotenv';
-dotenv.config({ path: path.resolve(__dirname, '../../.env') });
-
 import { PushAPI } from '../../../src/lib/pushapi/PushAPI'; // Ensure correct import path
 import { expect } from 'chai';
 import { ethers } from 'ethers';
@@ -16,7 +12,11 @@ describe('PushAPI.chat functionality', () => {
   let signer2: any;
   let account2: string;
   const MESSAGE = 'Hey There!!!';
-  const env = CONSTANTS.ENV.DEV;
+
+  // accessing env dynamically using process.env
+  type EnvStrings = keyof typeof CONSTANTS.ENV;
+  const envMode = process.env.ENV as EnvStrings;
+  const env = CONSTANTS.ENV[envMode];
 
   beforeEach(async () => {
     const WALLET1 = ethers.Wallet.createRandom();

diff --git a/packages/restapi/tests/lib/chat/initialize.test.ts b/packages/restapi/tests/lib/chat/initialize.test.ts
@@ -1,6 +1,3 @@
-import * as path from 'path';
-import * as dotenv from 'dotenv';
-dotenv.config({ path: path.resolve(__dirname, '../../.env') });
 import { ProgressHookType } from '../../../src/lib/types';
 import { PushAPI } from '../../../src/lib/pushapi/PushAPI'; // Ensure correct import path
 import { expect } from 'chai';

diff --git a/packages/restapi/tests/lib/chat/nftChat.test.ts b/packages/restapi/tests/lib/chat/nftChat.test.ts
@@ -1,5 +1,3 @@
-import * as path from 'path';
-import * as dotenv from 'dotenv';
 import { PushAPI } from '../../../src/lib/pushapi/PushAPI';
 import { ethers } from 'ethers';
 import CONSTANTS from '../../../src/lib/constantsV2';
@@ -11,9 +9,11 @@ import {
 } from 'unique-names-generator';
 import { PushStream } from '../../../src/lib/pushstream/PushStream';
 
-dotenv.config({ path: path.resolve(__dirname, '../../.env') });
+// accessing env dynamically using process.env
+type EnvStrings = keyof typeof CONSTANTS.ENV;
+const envMode = process.env.ENV as EnvStrings;
+const env = CONSTANTS.ENV[envMode];
 
-const env = CONSTANTS.ENV.DEV;
 const showAPIResponse = false;
 
 describe('PushAPI.chat functionality For NFT Profile', () => {

diff --git a/packages/restapi/tests/lib/chat/privateGroup.test.ts b/packages/restapi/tests/lib/chat/privateGroup.test.ts
@@ -13,7 +13,11 @@ import { PushAPI } from '../../../src/lib/pushapi/PushAPI'; // Ensure correct im
 import { GroupDTO, GroupInfoDTO, MessageWithCID } from '../../../src/lib/types';
 import { cache } from '../../../src/lib/helpers/cache';
 
-const _env = Constants.ENV.DEV;
+// accessing env dynamically using process.env
+type EnvStrings = keyof typeof Constants.ENV;
+const envMode = process.env.ENV as EnvStrings;
+const _env = Constants.ENV[envMode];
+
 let groupName: string;
 let groupDescription: string;
 const groupImage =

diff --git a/packages/restapi/tests/lib/chat/profile.test.ts b/packages/restapi/tests/lib/chat/profile.test.ts
@@ -1,7 +1,3 @@
-import * as path from 'path';
-import * as dotenv from 'dotenv';
-dotenv.config({ path: path.resolve(__dirname, '../../.env') });
-
 import { PushAPI } from '../../../src/lib/pushapi/PushAPI'; // Ensure correct import path
 import { expect } from 'chai';
 import { ethers } from 'ethers';

diff --git a/packages/restapi/tests/lib/chatLowLevel/chat.test.ts b/packages/restapi/tests/lib/chatLowLevel/chat.test.ts
@@ -1,13 +1,14 @@
-import * as path from 'path';
-import * as dotenv from 'dotenv';
-dotenv.config({ path: path.resolve(__dirname, '../../.env') });
 import { chat } from '../../../src/lib/chat/chat';
 import { ethers } from 'ethers';
 import { expect } from 'chai';
 import Constants from '../../../src/lib/constants';
 
 const WALLET_PRIVATE_KEY = process.env['WALLET_PRIVATE_KEY'];
-const _env = Constants.ENV.DEV;
+
+// accessing env dynamically using process.env
+type EnvStrings = keyof typeof Constants.ENV;
+const envMode = process.env.ENV as EnvStrings;
+const _env = Constants.ENV[envMode];
 
 describe('Get chat', () => {
   it('Should return {} when not chat between users', async () => {