Merge pull request #1211 from LimeChain/near-workspaces-integration

Use near-workspaces for the tests during the CI process
near · Jan 22, 2024 · ab4f8d3 · ab4f8d3
2 parents ae9600b + df0cd87
commit ab4f8d3
Show file tree

Hide file tree

Showing 10 changed files with 865 additions and 346 deletions.
diff --git a/packages/accounts/package.json b/packages/accounts/package.json
@@ -37,7 +37,8 @@
     "jest": "26.0.1",
     "near-hello": "0.5.1",
     "ts-jest": "26.5.6",
-    "typescript": "4.9.4"
+    "typescript": "4.9.4",
+    "near-workspaces": "3.4.0"
   },
   "files": [
     "lib"

diff --git a/packages/accounts/test/config.js b/packages/accounts/test/config.js
@@ -1,4 +1,7 @@
-module.exports = function getConfig(env) {
+const { Worker } = require('near-workspaces');
+const fs = require('fs');
+let worker;
+module.exports = async function getConfig(env) {
     switch (env) {
         case 'production':
         case 'mainnet':
@@ -32,12 +35,17 @@ module.exports = function getConfig(env) {
                 walletUrl: 'http://localhost:4000/wallet',
             };
         case 'test':
-        case 'ci':
+        case 'ci': {
+            if (!worker) worker = await Worker.init();
+            const keyFile = fs.readFileSync(`${worker.rootAccount.manager.config.homeDir}/validator_key.json`);
+            const keyPair = JSON.parse(keyFile.toString());
             return {
-                networkId: 'shared-test',
-                nodeUrl: 'https://rpc.ci-testnet.near.org',
-                masterAccount: 'test.near',
+                networkId: worker.config.network,
+                nodeUrl: worker.manager.config.rpcAddr,
+                masterAccount: worker.rootAccount._accountId,
+                secretKey: keyPair.secret_key || keyPair.private_key
             };
+        }
         default:
             throw Error(`Unconfigured environment '${env}'. Can be configured in src/config.js.`);
     }

diff --git a/packages/accounts/test/providers.test.js b/packages/accounts/test/providers.test.js
diff --git a/packages/accounts/test/test-utils.js b/packages/accounts/test/test-utils.js
@@ -51,14 +51,15 @@ async function loadGuestBookContractCode() {
 }
 async function setUpTestConnection() {
     const keyStore = new InMemoryKeyStore();
-    const config = Object.assign(require('./config')(process.env.NODE_ENV || 'test'), {
+    const config = Object.assign(await require('./config')(process.env.NODE_ENV || 'test'), {
         networkId,
         keyStore
     });
 
     if (config.masterAccount) {
         // full accessKey on ci-testnet, dedicated rpc for tests.
-        await keyStore.setKey(networkId, config.masterAccount, KeyPair.fromString('ed25519:2wyRcSwSuHtRVmkMCGjPwnzZmQLeXLzLLyED1NDMt4BjnKgQL6tF85yBx6Jr26D2dUNeC716RBoTxntVHsegogYw'));
+        const secretKey = config.secretKey || 'ed25519:2wyRcSwSuHtRVmkMCGjPwnzZmQLeXLzLLyED1NDMt4BjnKgQL6tF85yBx6Jr26D2dUNeC716RBoTxntVHsegogYw';
+        await keyStore.setKey(networkId, config.masterAccount, KeyPair.fromString(secretKey));
     }
 
     const connection = Connection.fromConfig({

diff --git a/packages/near-api-js/package.json b/packages/near-api-js/package.json
@@ -49,7 +49,8 @@
         "rimraf": "3.0.2",
         "semver": "7.1.1",
         "ts-jest": "26.5.6",
-        "uglifyify": "5.0.1"
+        "uglifyify": "5.0.1",
+        "near-workspaces": "3.4.0"
     },
     "keywords": [],
     "license": "(MIT AND Apache-2.0)",

diff --git a/packages/near-api-js/test/config.js b/packages/near-api-js/test/config.js
@@ -1,4 +1,7 @@
-module.exports = function getConfig(env) {
+const { Worker } = require('near-workspaces');
+const fs = require('fs');
+let worker;
+module.exports = async function getConfig(env) {
     switch (env) {
     case 'production':
     case 'mainnet':
@@ -33,10 +36,14 @@ module.exports = function getConfig(env) {
         };
     case 'test':
     case 'ci':
+        if (!worker) worker = await Worker.init();
+        const keyFile = fs.readFileSync(`${worker.rootAccount.manager.config.homeDir}/validator_key.json`);
+        const keyPair = JSON.parse(keyFile.toString());
         return {
-            networkId: 'shared-test',
-            nodeUrl: 'https://rpc.ci-testnet.near.org',
-            masterAccount: 'test.near',
+            networkId: worker.config.network,
+            nodeUrl: worker.manager.config.rpcAddr,
+            masterAccount: worker.rootAccount._accountId,
+            secretKey: keyPair.secret_key || keyPair.private_key
         };
     default:
         throw Error(`Unconfigured environment '${env}'. Can be configured in src/config.js.`);

diff --git a/packages/near-api-js/test/test-utils.js b/packages/near-api-js/test/test-utils.js
@@ -18,14 +18,15 @@ const RANDOM_ACCOUNT_LENGTH = 40;
 
 async function setUpTestConnection() {
     const keyStore = new nearApi.keyStores.InMemoryKeyStore();
-    const config = Object.assign(require('./config')(process.env.NODE_ENV || 'test'), {
-        networkId: networkId,
+    const config = Object.assign(await require('./config')(process.env.NODE_ENV || 'test'), {
+        networkId,
         keyStore
     });
 
     if (config.masterAccount) {
         // full accessKey on ci-testnet, dedicated rpc for tests.
-        await keyStore.setKey(networkId, config.masterAccount, nearApi.utils.KeyPair.fromString('ed25519:2wyRcSwSuHtRVmkMCGjPwnzZmQLeXLzLLyED1NDMt4BjnKgQL6tF85yBx6Jr26D2dUNeC716RBoTxntVHsegogYw'));
+        const secretKey = config.secretKey || 'ed25519:2wyRcSwSuHtRVmkMCGjPwnzZmQLeXLzLLyED1NDMt4BjnKgQL6tF85yBx6Jr26D2dUNeC716RBoTxntVHsegogYw';
+        await keyStore.setKey(networkId, config.masterAccount, KeyPair.fromString(secretKey));
     }
     return nearApi.connect(config);
 }

diff --git a/packages/providers/package.json b/packages/providers/package.json
@@ -26,6 +26,7 @@
   "devDependencies": {
     "@types/node": "18.11.18",
     "jest": "26.0.1",
+    "near-workspaces": "3.4.0",
     "ts-jest": "26.5.6",
     "typescript": "4.9.4"
   },