@fuel-ts/account 0.0.0-rc-2333-20240521114009 → 0.0.0-rc-1356-20240521154745

package/dist/test-utils.global.js

@@ -32933,6 +32933,13 @@ This unreleased fuel-core build may include features and updates not yet support
   };
   var DateTime = _DateTime;
   __publicField3(DateTime, "TAI64_NULL", "");
+  function sleep(time) {
+    return new Promise((resolve) => {
+      setTimeout(() => {
+        resolve(true);
+      }, time);
+    });
+  }
   var chainConfig_default = {
     chain_name: "local_testnet",
     consensus_parameters: {
@@ -33687,9 +33694,9 @@ This unreleased fuel-core build may include features and updates not yet support
     da_block_height: 0
   };
   var defaultSnapshotConfigs = {
-    chainConfigJson: chainConfig_default,
-    metadataJson: metadata_default,
-    stateConfigJson: stateConfig_default
+    chainConfig: chainConfig_default,
+    metadata: metadata_default,
+    stateConfig: stateConfig_default
   };
   var defaultConsensusKey = "0xa449b1ffee0e2205fa924c6740cc48b3b473aa28587df6dab12abc245d1f5298";
   function isDefined(value) {
@@ -34429,11 +34436,72 @@ This unreleased fuel-core build may include features and updates not yet support
     };
   }
 
+  // ../../node_modules/.pnpm/ramda@0.29.0/node_modules/ramda/es/internal/_curry2.js
+  function _curry2(fn) {
+    return function f2(a, b) {
+      switch (arguments.length) {
+        case 0:
+          return f2;
+        case 1:
+          return _isPlaceholder(a) ? f2 : _curry1(function(_b) {
+            return fn(a, _b);
+          });
+        default:
+          return _isPlaceholder(a) && _isPlaceholder(b) ? f2 : _isPlaceholder(a) ? _curry1(function(_a) {
+            return fn(_a, b);
+          }) : _isPlaceholder(b) ? _curry1(function(_b) {
+            return fn(a, _b);
+          }) : fn(a, b);
+      }
+    };
+  }
+
+  // ../../node_modules/.pnpm/ramda@0.29.0/node_modules/ramda/es/internal/_curry3.js
+  function _curry3(fn) {
+    return function f3(a, b, c) {
+      switch (arguments.length) {
+        case 0:
+          return f3;
+        case 1:
+          return _isPlaceholder(a) ? f3 : _curry2(function(_b, _c) {
+            return fn(a, _b, _c);
+          });
+        case 2:
+          return _isPlaceholder(a) && _isPlaceholder(b) ? f3 : _isPlaceholder(a) ? _curry2(function(_a, _c) {
+            return fn(_a, b, _c);
+          }) : _isPlaceholder(b) ? _curry2(function(_b, _c) {
+            return fn(a, _b, _c);
+          }) : _curry1(function(_c) {
+            return fn(a, b, _c);
+          });
+        default:
+          return _isPlaceholder(a) && _isPlaceholder(b) && _isPlaceholder(c) ? f3 : _isPlaceholder(a) && _isPlaceholder(b) ? _curry2(function(_a, _b) {
+            return fn(_a, _b, c);
+          }) : _isPlaceholder(a) && _isPlaceholder(c) ? _curry2(function(_a, _c) {
+            return fn(_a, b, _c);
+          }) : _isPlaceholder(b) && _isPlaceholder(c) ? _curry2(function(_b, _c) {
+            return fn(a, _b, _c);
+          }) : _isPlaceholder(a) ? _curry1(function(_a) {
+            return fn(_a, b, c);
+          }) : _isPlaceholder(b) ? _curry1(function(_b) {
+            return fn(a, _b, c);
+          }) : _isPlaceholder(c) ? _curry1(function(_c) {
+            return fn(a, b, _c);
+          }) : fn(a, b, c);
+      }
+    };
+  }
+
   // ../../node_modules/.pnpm/ramda@0.29.0/node_modules/ramda/es/internal/_isArray.js
   var isArray_default = Array.isArray || function _isArray(val) {
     return val != null && val.length >= 0 && Object.prototype.toString.call(val) === "[object Array]";
   };
 
+  // ../../node_modules/.pnpm/ramda@0.29.0/node_modules/ramda/es/internal/_has.js
+  function _has(prop, obj) {
+    return Object.prototype.hasOwnProperty.call(obj, prop);
+  }
+
   // ../../node_modules/.pnpm/ramda@0.29.0/node_modules/ramda/es/type.js
   var type = /* @__PURE__ */ _curry1(function type2(val) {
     return val === null ? "Null" : val === void 0 ? "Undefined" : Object.prototype.toString.call(val).slice(8, -1);
@@ -34450,6 +34518,11 @@ This unreleased fuel-core build may include features and updates not yet support
     return d.getUTCFullYear() + "-" + pad(d.getUTCMonth() + 1) + "-" + pad(d.getUTCDate()) + "T" + pad(d.getUTCHours()) + ":" + pad(d.getUTCMinutes()) + ":" + pad(d.getUTCSeconds()) + "." + (d.getUTCMilliseconds() / 1e3).toFixed(3).slice(2, 5) + "Z";
   };
 
+  // ../../node_modules/.pnpm/ramda@0.29.0/node_modules/ramda/es/internal/_isObject.js
+  function _isObject(x) {
+    return Object.prototype.toString.call(x) === "[object Object]";
+  }
+
   // ../../node_modules/.pnpm/ramda@0.29.0/node_modules/ramda/es/internal/_isInteger.js
   var isInteger_default = Number.isInteger || function _isInteger(n) {
     return n << 0 === n;
@@ -34563,6 +34636,46 @@ This unreleased fuel-core build may include features and updates not yet support
   });
   var clone_default = clone;
 
+  // ../../node_modules/.pnpm/ramda@0.29.0/node_modules/ramda/es/mergeWithKey.js
+  var mergeWithKey = /* @__PURE__ */ _curry3(function mergeWithKey2(fn, l, r) {
+    var result = {};
+    var k;
+    l = l || {};
+    r = r || {};
+    for (k in l) {
+      if (_has(k, l)) {
+        result[k] = _has(k, r) ? fn(k, l[k], r[k]) : l[k];
+      }
+    }
+    for (k in r) {
+      if (_has(k, r) && !_has(k, result)) {
+        result[k] = r[k];
+      }
+    }
+    return result;
+  });
+  var mergeWithKey_default = mergeWithKey;
+
+  // ../../node_modules/.pnpm/ramda@0.29.0/node_modules/ramda/es/mergeDeepWithKey.js
+  var mergeDeepWithKey = /* @__PURE__ */ _curry3(function mergeDeepWithKey2(fn, lObj, rObj) {
+    return mergeWithKey_default(function(k, lVal, rVal) {
+      if (_isObject(lVal) && _isObject(rVal)) {
+        return mergeDeepWithKey2(fn, lVal, rVal);
+      } else {
+        return fn(k, lVal, rVal);
+      }
+    }, lObj, rObj);
+  });
+  var mergeDeepWithKey_default = mergeDeepWithKey;
+
+  // ../../node_modules/.pnpm/ramda@0.29.0/node_modules/ramda/es/mergeDeepRight.js
+  var mergeDeepRight = /* @__PURE__ */ _curry2(function mergeDeepRight2(lObj, rObj) {
+    return mergeDeepWithKey_default(function(k, lVal, rVal) {
+      return rVal;
+    }, lObj, rObj);
+  });
+  var mergeDeepRight_default = mergeDeepRight;
+
   // ../../node_modules/.pnpm/ramda@0.29.0/node_modules/ramda/es/trim.js
   var hasProtoTrim = typeof String.prototype.trim === "function";
 
@@ -41970,15 +42083,6 @@ ${MessageCoinFragmentDoc}`;
     return normalize2(clone_default(root));
   }
 
-  // src/providers/utils/sleep.ts
-  function sleep(time) {
-    return new Promise((resolve) => {
-      setTimeout(() => {
-        resolve(true);
-      }, time);
-    });
-  }
-
   // src/providers/utils/extract-tx-error.ts
   var assemblePanicError = (statusReason) => {
     let errorMessage = `The transaction reverted with reason: "${statusReason}".`;
@@ -50044,6 +50148,40 @@ Supported fuel-core version: ${supportedVersion}.`
       }
     }
   };
+  function getFinalStateConfigJSON({ stateConfig, chainConfig }) {
+    const defaultCoins = defaultSnapshotConfigs.stateConfig.coins.map((coin) => ({
+      ...coin,
+      amount: "18446744073709551615"
+    }));
+    const defaultMessages = defaultSnapshotConfigs.stateConfig.messages.map((message) => ({
+      ...message,
+      amount: "18446744073709551615"
+    }));
+    const coins = defaultCoins.concat(stateConfig.coins.map((coin) => ({ ...coin, amount: coin.amount.toString() }))).filter((coin, index, self2) => self2.findIndex((c) => c.tx_id === coin.tx_id) === index);
+    const messages = defaultMessages.concat(stateConfig.messages.map((msg) => ({ ...msg, amount: msg.amount.toString() }))).filter((msg, index, self2) => self2.findIndex((m) => m.nonce === msg.nonce) === index);
+    if (!process.env.GENESIS_SECRET) {
+      const pk = Signer.generatePrivateKey();
+      const signer = new Signer(pk);
+      process.env.GENESIS_SECRET = hexlify(pk);
+      coins.push({
+        tx_id: hexlify(randomBytes22(UTXO_ID_LEN)),
+        owner: signer.address.toHexString(),
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        amount: "18446744073709551615",
+        asset_id: chainConfig.consensus_parameters.V1.base_asset_id,
+        output_index: 0,
+        tx_pointer_block_height: 0,
+        tx_pointer_tx_idx: 0
+      });
+    }
+    const json = JSON.stringify({
+      ...stateConfig,
+      coins,
+      messages
+    });
+    const regexMakeNumber = /("amount":)"(\d+)"/gm;
+    return json.replace(regexMakeNumber, "$1$2");
+  }
   var launchNode = async ({
     ip,
     port,
@@ -50051,7 +50189,8 @@ Supported fuel-core version: ${supportedVersion}.`
     fuelCorePath = process.env.FUEL_CORE_PATH ?? void 0,
     loggingEnabled = true,
     debugEnabled = false,
-    basePath
+    basePath,
+    snapshotConfig = defaultSnapshotConfigs
   }) => (
     // eslint-disable-next-line no-async-promise-executor
     new Promise(async (resolve, reject) => {
@@ -50079,56 +50218,23 @@ Supported fuel-core version: ${supportedVersion}.`
       let snapshotDirToUse;
       const prefix = basePath || import_os.default.tmpdir();
       const suffix = basePath ? "" : (0, import_crypto19.randomUUID)();
-      const tempDirPath = import_path7.default.join(prefix, ".fuels", suffix, "snapshotDir");
+      const tempDir = import_path7.default.join(prefix, ".fuels", suffix, "snapshotDir");
       if (snapshotDir) {
         snapshotDirToUse = snapshotDir;
       } else {
-        if (!(0, import_fs.existsSync)(tempDirPath)) {
-          (0, import_fs.mkdirSync)(tempDirPath, { recursive: true });
-        }
-        let { stateConfigJson } = defaultSnapshotConfigs;
-        const { chainConfigJson, metadataJson } = defaultSnapshotConfigs;
-        stateConfigJson = {
-          ...stateConfigJson,
-          coins: [
-            ...stateConfigJson.coins.map((coin) => ({
-              ...coin,
-              amount: "18446744073709551615"
-            }))
-          ],
-          messages: stateConfigJson.messages.map((message) => ({
-            ...message,
-            amount: "18446744073709551615"
-          }))
-          // eslint-disable-next-line @typescript-eslint/no-explicit-any
-        };
-        if (!process.env.GENESIS_SECRET) {
-          const pk = Signer.generatePrivateKey();
-          const signer = new Signer(pk);
-          process.env.GENESIS_SECRET = hexlify(pk);
-          stateConfigJson.coins.push({
-            tx_id: hexlify(randomBytes22(UTXO_ID_LEN)),
-            owner: signer.address.toHexString(),
-            // eslint-disable-next-line @typescript-eslint/no-explicit-any
-            amount: "18446744073709551615",
-            asset_id: chainConfigJson.consensus_parameters.V1.base_asset_id,
-            output_index: 0,
-            tx_pointer_block_height: 0,
-            tx_pointer_tx_idx: 0
-          });
-        }
-        let fixedStateConfigJSON = JSON.stringify(stateConfigJson);
-        const regexMakeNumber = /("amount":)"(\d+)"/gm;
-        fixedStateConfigJSON = fixedStateConfigJSON.replace(regexMakeNumber, "$1$2");
-        const chainConfigWritePath = import_path7.default.join(tempDirPath, "chainConfig.json");
-        const stateConfigWritePath = import_path7.default.join(tempDirPath, "stateConfig.json");
-        const metadataWritePath = import_path7.default.join(tempDirPath, "metadata.json");
-        const stateTransitionWritePath = import_path7.default.join(tempDirPath, "state_transition_bytecode.wasm");
-        (0, import_fs.writeFileSync)(chainConfigWritePath, JSON.stringify(chainConfigJson), "utf8");
-        (0, import_fs.writeFileSync)(stateConfigWritePath, fixedStateConfigJSON, "utf8");
-        (0, import_fs.writeFileSync)(metadataWritePath, JSON.stringify(metadataJson), "utf8");
-        (0, import_fs.writeFileSync)(stateTransitionWritePath, JSON.stringify(""));
-        snapshotDirToUse = tempDirPath;
+        if (!(0, import_fs.existsSync)(tempDir)) {
+          (0, import_fs.mkdirSync)(tempDir, { recursive: true });
+        }
+        const { metadata } = snapshotConfig;
+        const metadataPath = import_path7.default.join(tempDir, "metadata.json");
+        const chainConfigPath = import_path7.default.join(tempDir, metadata.chain_config);
+        const stateConfigPath = import_path7.default.join(tempDir, metadata.table_encoding.Json.filepath);
+        const stateTransitionPath = import_path7.default.join(tempDir, "state_transition_bytecode.wasm");
+        (0, import_fs.writeFileSync)(chainConfigPath, JSON.stringify(snapshotConfig.chainConfig), "utf8");
+        (0, import_fs.writeFileSync)(stateConfigPath, getFinalStateConfigJSON(snapshotConfig), "utf8");
+        (0, import_fs.writeFileSync)(metadataPath, JSON.stringify(metadata), "utf8");
+        (0, import_fs.writeFileSync)(stateTransitionPath, JSON.stringify(""));
+        snapshotDirToUse = tempDir;
       }
       const child = (0, import_child_process.spawn)(
         command,
@@ -50136,7 +50242,7 @@ Supported fuel-core version: ${supportedVersion}.`
           "run",
           ["--ip", ipToUse],
           ["--port", portToUse],
-          useInMemoryDb ? ["--db-type", "in-memory"] : ["--db-path", tempDirPath],
+          useInMemoryDb ? ["--db-type", "in-memory"] : ["--db-path", tempDir],
           ["--min-gas-price", "1"],
           poaInstant ? ["--poa-instant", "true"] : [],
           ["--consensus-key", consensusKey],
@@ -50158,23 +50264,28 @@ Supported fuel-core version: ${supportedVersion}.`
       }
       const cleanupConfig = {
         child,
-        configPath: tempDirPath,
+        configPath: tempDir,
         killFn: import_tree_kill.default,
         state: {
           isDead: false
         }
       };
       child.stderr.on("data", (chunk) => {
-        if (chunk.indexOf(graphQLStartSubstring) !== -1) {
+        const text = typeof chunk === "string" ? chunk : chunk.toString();
+        if (text.indexOf(graphQLStartSubstring) !== -1) {
+          const rows = text.split("\n");
+          const rowWithUrl = rows.find((row) => row.indexOf(graphQLStartSubstring) !== -1);
+          const [realIp, realPort] = rowWithUrl.split(" ").at(-1).trim().split(":");
           resolve({
             cleanup: () => killNode(cleanupConfig),
-            ip: ipToUse,
-            port: portToUse,
+            ip: realIp,
+            port: realPort,
+            url: `http://${realIp}:${realPort}/v1/graphql`,
             snapshotDir: snapshotDirToUse
           });
         }
-        if (/error/i.test(chunk)) {
-          reject(chunk.toString());
+        if (/error/i.test(text)) {
+          reject(text.toString());
         }
       });
       process.on("exit", () => killNode(cleanupConfig));
@@ -50207,6 +50318,223 @@ Supported fuel-core version: ${supportedVersion}.`
     };
     return { wallets, stop: cleanup, provider };
   };
+
+  // src/test-utils/asset-id.ts
+  var _AssetId = class {
+    constructor(value) {
+      this.value = value;
+    }
+    static random(count = 1) {
+      const assetIds = [];
+      for (let i = 0; i < count; i++) {
+        assetIds.push(new _AssetId(hexlify(randomBytes22(32))));
+      }
+      return assetIds;
+    }
+  };
+  var AssetId = _AssetId;
+  __publicField(AssetId, "A", new _AssetId(
+    "0x0101010101010101010101010101010101010101010101010101010101010101"
+  ));
+  __publicField(AssetId, "B", new _AssetId(
+    "0x0202020202020202020202020202020202020202020202020202020202020202"
+  ));
+
+  // src/test-utils/wallet-config.ts
+  var WalletConfig = class {
+    initialState;
+    options;
+    wallets;
+    generateWallets = () => {
+      const generatedWallets = [];
+      for (let index = 1; index <= this.options.count; index++) {
+        generatedWallets.push(new WalletUnlocked(randomBytes22(32)));
+      }
+      return generatedWallets;
+    };
+    constructor(baseAssetId, config) {
+      WalletConfig.validate(config);
+      this.options = config;
+      const { assets: assets2, coinsPerAsset, amountPerCoin, messages } = this.options;
+      this.wallets = this.generateWallets();
+      this.initialState = {
+        messages: WalletConfig.createMessages(this.wallets, messages),
+        coins: WalletConfig.createCoins(
+          this.wallets,
+          baseAssetId,
+          assets2,
+          coinsPerAsset,
+          amountPerCoin
+        )
+      };
+    }
+    apply(snapshotConfig) {
+      return {
+        ...snapshotConfig,
+        stateConfig: {
+          ...snapshotConfig?.stateConfig ?? defaultSnapshotConfigs.stateConfig,
+          coins: this.initialState.coins.concat(snapshotConfig?.stateConfig?.coins || []),
+          messages: this.initialState.messages.concat(snapshotConfig?.stateConfig?.messages ?? [])
+        }
+      };
+    }
+    /**
+     * Create messages for the wallets in the format that the chain expects.
+     */
+    static createMessages(wallets, messages) {
+      return messages.map((msg) => wallets.map((wallet) => msg.toChainMessage(wallet.address))).flatMap((x) => x);
+    }
+    /**
+     * Create coins for the wallets in the format that the chain expects.
+     */
+    static createCoins(wallets, baseAssetId, assets2, coinsPerAsset, amountPerCoin) {
+      const coins = [];
+      let assetIds = [baseAssetId];
+      if (Array.isArray(assets2)) {
+        assetIds = assetIds.concat(assets2.map((a) => a.value));
+      } else {
+        assetIds = assetIds.concat(AssetId.random(assets2).map((a) => a.value));
+      }
+      console.log("assetIds", assets2, assetIds);
+      wallets.map((wallet) => wallet.address.toHexString()).forEach((walletAddress) => {
+        assetIds.forEach((assetId) => {
+          for (let index = 0; index < coinsPerAsset; index++) {
+            coins.push({
+              amount: amountPerCoin,
+              asset_id: assetId,
+              owner: walletAddress,
+              tx_pointer_block_height: 0,
+              tx_pointer_tx_idx: 0,
+              output_index: 0,
+              tx_id: hexlify(randomBytes22(32))
+            });
+          }
+        });
+      });
+      return coins;
+    }
+    static validate({
+      count: wallets,
+      assets: assets2,
+      coinsPerAsset,
+      amountPerCoin
+    }) {
+      if (Array.isArray(wallets) && wallets.length === 0 || typeof wallets === "number" && wallets <= 0) {
+        throw new FuelError(
+          FuelError.CODES.INVALID_INPUT_PARAMETERS,
+          "Number of wallets must be greater than zero."
+        );
+      }
+      if (Array.isArray(assets2) && assets2.length === 0 || typeof assets2 === "number" && assets2 <= 0) {
+        throw new FuelError(
+          FuelError.CODES.INVALID_INPUT_PARAMETERS,
+          "Number of assets per wallet must be greater than zero."
+        );
+      }
+      if (coinsPerAsset <= 0) {
+        throw new FuelError(
+          FuelError.CODES.INVALID_INPUT_PARAMETERS,
+          "Number of coins per asset must be greater than zero."
+        );
+      }
+      if (amountPerCoin <= 0) {
+        throw new FuelError(
+          FuelError.CODES.INVALID_INPUT_PARAMETERS,
+          "Amount per coin must be greater than zero."
+        );
+      }
+    }
+  };
+
+  // src/test-utils/setup-test-provider-and-wallets.ts
+  var defaultWalletConfigOptions = {
+    count: 2,
+    assets: [AssetId.A, AssetId.B],
+    coinsPerAsset: 1,
+    amountPerCoin: 1e10,
+    messages: []
+  };
+  async function setupTestProviderAndWallets({
+    walletConfig: walletConfigOptions = {},
+    providerOptions,
+    nodeOptions = {}
+  } = {}) {
+    Symbol.dispose ??= Symbol("Symbol.dispose");
+    const walletConfig = new WalletConfig(
+      nodeOptions.snapshotConfig?.chainConfig?.consensus_parameters?.V1?.base_asset_id ?? defaultSnapshotConfigs.chainConfig.consensus_parameters.V1.base_asset_id,
+      {
+        ...defaultWalletConfigOptions,
+        ...walletConfigOptions
+      }
+    );
+    const { cleanup, url } = await launchNode({
+      loggingEnabled: false,
+      ...nodeOptions,
+      snapshotConfig: mergeDeepRight_default(
+        defaultSnapshotConfigs,
+        walletConfig.apply(nodeOptions?.snapshotConfig)
+      ),
+      port: "0"
+    });
+    let provider;
+    try {
+      provider = await Provider.create(url, providerOptions);
+    } catch (err) {
+      cleanup();
+      throw err;
+    }
+    const wallets = walletConfig.wallets;
+    wallets.forEach((wallet) => {
+      wallet.connect(provider);
+    });
+    return {
+      provider,
+      wallets,
+      cleanup,
+      [Symbol.dispose]: cleanup
+    };
+  }
+
+  // src/test-utils/test-message.ts
+  var TestMessage = class {
+    sender;
+    recipient;
+    nonce;
+    amount;
+    data;
+    da_height;
+    /**
+     * A helper class to create messages for testing purposes.
+     *
+     * Used in tandem with `WalletConfig`.
+     * It can also be used standalone and passed into the initial state of a chain via the `.toChainMessage` method.
+     */
+    constructor({
+      sender = Address.fromRandom(),
+      recipient = Address.fromRandom(),
+      nonce = hexlify(randomBytes22(32)),
+      amount = 1e6,
+      data = "02",
+      da_height = 0
+    } = {}) {
+      this.sender = sender;
+      this.recipient = recipient;
+      this.nonce = nonce;
+      this.amount = amount;
+      this.data = data;
+      this.da_height = da_height;
+    }
+    toChainMessage(recipient) {
+      return {
+        sender: this.sender.toB256(),
+        recipient: recipient?.toB256() ?? this.recipient.toB256(),
+        nonce: this.nonce,
+        amount: bn(this.amount).toNumber(),
+        data: this.data,
+        da_height: this.da_height
+      };
+    }
+  };
 })();
 /*! Bundled license information: