@stacks/rendezvous 0.7.0 → 0.7.1

package/dist/citizen.js

@@ -12,7 +12,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getTestContractSource = exports.buildRendezvousData = exports.getContractSource = exports.groupContractsByEpochFromSimnetPlan = exports.issueFirstClassCitizenship = void 0;
+exports.getSbtcBalancesFromSimnet = exports.getTestContractSource = exports.buildRendezvousData = exports.getContractSource = exports.groupContractsByEpochFromSimnetPlan = exports.issueFirstClassCitizenship = void 0;
 exports.scheduleRendezvous = scheduleRendezvous;
 const fs_1 = require("fs");
 const path_1 = require("path");
@@ -48,18 +48,7 @@ const issueFirstClassCitizenship = (manifestDir, manifestPath, remoteDataSetting
         const balanceHex = simnet.runSnippet(`(stx-get-balance '${address})`);
         return [address, (0, transactions_1.cvToValue)((0, transactions_1.hexToCV)(balanceHex))];
     }));
-    const sbtcBalancesMap = new Map(simnetAddresses.map((address) => {
-        try {
-            const { result: getBalanceResult } = simnet.callReadOnlyFn("SM3VDXK3WZZSA84XXFKAFAF15NNZX32CTSG82JFQ4.sbtc-token", "get-balance", [transactions_1.Cl.principal(address)], address);
-            // If the previous read-only call works, the user is working with
-            // sBTC. This means we can proceed with restoring sBTC balances.
-            const sbtcBalance = (0, transactions_1.cvToJSON)(getBalanceResult).value.value;
-            return [address, sbtcBalance];
-        }
-        catch (e) {
-            return [address, 0];
-        }
-    }));
+    const sbtcBalancesMap = (0, exports.getSbtcBalancesFromSimnet)(simnet);
     yield simnet.initEmptySession(remoteDataSettings);
     simnetAddresses.forEach((address) => {
         simnet.mintSTX(address, stxBalancesMap.get(address));
@@ -260,6 +249,31 @@ function scheduleRendezvous(targetContractSource, tests) {
     (ok (map-set context function-name {called: called})))`;
     return `${targetContractSource}\n\n${context}\n\n${tests}`;
 }
+/**
+ * Maps the simnet accounts to their sBTC balances. The function tries to call
+ * the `get-balance` function of the `sbtc-token` contract for each address. If
+ * the call fails, it returns a balance of 0 for that address. The call fails
+ * if the user is not working with sBTC.
+ * @param simnet The simnet instance.
+ * @returns A map of addresses to their sBTC balances.
+ */
+const getSbtcBalancesFromSimnet = (simnet) => new Map([...simnet.getAccounts().values()].map((address) => {
+    try {
+        const { result: getBalanceResult } = simnet.callReadOnlyFn("SM3VDXK3WZZSA84XXFKAFAF15NNZX32CTSG82JFQ4.sbtc-token", "get-balance", [transactions_1.Cl.principal(address)], address);
+        // If the previous read-only call works, the user is working with
+        // sBTC. This means we can proceed with restoring sBTC balances.
+        const sbtcBalanceJSON = (0, transactions_1.cvToJSON)(getBalanceResult);
+        // The `get-balance` function returns a response containing the uint
+        // balance of the address. In the JSON representation, the balance is
+        // represented as a string. We need to parse it to an integer.
+        const sbtcBalance = parseInt(sbtcBalanceJSON.value.value, 10);
+        return [address, sbtcBalance];
+    }
+    catch (e) {
+        return [address, 0];
+    }
+}));
+exports.getSbtcBalancesFromSimnet = getSbtcBalancesFromSimnet;
 /**
  * Utility function that restores the test wallets' initial sBTC balances in
  * the re-initialized first-class citizenship simnet.

package/dist/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@stacks/rendezvous",
-    "version": "0.7.0",
+    "version": "0.7.1",
     "description": "Meet your contract's vulnerabilities head-on.",
     "main": "app.js",
     "bin": {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@stacks/rendezvous",
-  "version": "0.7.0",
+  "version": "0.7.1",
   "description": "Meet your contract's vulnerabilities head-on.",
   "main": "app.js",
   "bin": {