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

package/dist/test-utils.mjs

@@ -1647,15 +1647,6 @@ function normalizeJSON(root) {
   return normalize(clone(root));
 }
 
-// src/providers/utils/sleep.ts
-function sleep(time) {
-  return new Promise((resolve) => {
-    setTimeout(() => {
-      resolve(true);
-    }, time);
-  });
-}
-
 // src/providers/utils/extract-tx-error.ts
 import { ErrorCode as ErrorCode7, FuelError as FuelError7 } from "@fuel-ts/errors";
 import { bn as bn6 } from "@fuel-ts/math";
@@ -3604,6 +3595,7 @@ var TransactionResponse = class {
 };
 
 // src/providers/utils/auto-retry-fetch.ts
+import { sleep } from "@fuel-ts/utils";
 function getWaitDelay(options, retryAttemptNum) {
   const duration = options.baseDelay ?? 150;
   switch (options.backoff) {
@@ -8482,7 +8474,7 @@ var generateTestWallet = async (provider, quantities) => {
 // src/test-utils/launchNode.ts
 import { UTXO_ID_LEN as UTXO_ID_LEN3 } from "@fuel-ts/abi-coder";
 import { randomBytes as randomBytes6 } from "@fuel-ts/crypto";
-import { defaultSnapshotConfigs, defaultConsensusKey, hexlify as hexlify18 } from "@fuel-ts/utils";
+import { defaultConsensusKey, hexlify as hexlify18, defaultSnapshotConfigs } from "@fuel-ts/utils";
 import { spawn } from "child_process";
 import { randomUUID } from "crypto";
 import { existsSync, mkdirSync, rmSync, writeFileSync } from "fs";
@@ -8521,6 +8513,40 @@ var killNode = (params) => {
     }
   }
 };
+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, self) => self.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, self) => self.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 = hexlify18(pk);
+    coins.push({
+      tx_id: hexlify18(randomBytes6(UTXO_ID_LEN3)),
+      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,
@@ -8528,7 +8554,8 @@ var launchNode = async ({
   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) => {
@@ -8556,56 +8583,23 @@ var launchNode = async ({
     let snapshotDirToUse;
     const prefix = basePath || os.tmpdir();
     const suffix = basePath ? "" : randomUUID();
-    const tempDirPath = path.join(prefix, ".fuels", suffix, "snapshotDir");
+    const tempDir = path.join(prefix, ".fuels", suffix, "snapshotDir");
     if (snapshotDir) {
       snapshotDirToUse = snapshotDir;
     } else {
-      if (!existsSync(tempDirPath)) {
-        mkdirSync(tempDirPath, { recursive: true });
+      if (!existsSync(tempDir)) {
+        mkdirSync(tempDir, { 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 = hexlify18(pk);
-        stateConfigJson.coins.push({
-          tx_id: hexlify18(randomBytes6(UTXO_ID_LEN3)),
-          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 = path.join(tempDirPath, "chainConfig.json");
-      const stateConfigWritePath = path.join(tempDirPath, "stateConfig.json");
-      const metadataWritePath = path.join(tempDirPath, "metadata.json");
-      const stateTransitionWritePath = path.join(tempDirPath, "state_transition_bytecode.wasm");
-      writeFileSync(chainConfigWritePath, JSON.stringify(chainConfigJson), "utf8");
-      writeFileSync(stateConfigWritePath, fixedStateConfigJSON, "utf8");
-      writeFileSync(metadataWritePath, JSON.stringify(metadataJson), "utf8");
-      writeFileSync(stateTransitionWritePath, JSON.stringify(""));
-      snapshotDirToUse = tempDirPath;
+      const { metadata } = snapshotConfig;
+      const metadataPath = path.join(tempDir, "metadata.json");
+      const chainConfigPath = path.join(tempDir, metadata.chain_config);
+      const stateConfigPath = path.join(tempDir, metadata.table_encoding.Json.filepath);
+      const stateTransitionPath = path.join(tempDir, "state_transition_bytecode.wasm");
+      writeFileSync(chainConfigPath, JSON.stringify(snapshotConfig.chainConfig), "utf8");
+      writeFileSync(stateConfigPath, getFinalStateConfigJSON(snapshotConfig), "utf8");
+      writeFileSync(metadataPath, JSON.stringify(metadata), "utf8");
+      writeFileSync(stateTransitionPath, JSON.stringify(""));
+      snapshotDirToUse = tempDir;
     }
     const child = spawn(
       command,
@@ -8613,7 +8607,7 @@ var launchNode = async ({
         "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],
@@ -8635,23 +8629,28 @@ var launchNode = async ({
     }
     const cleanupConfig = {
       child,
-      configPath: tempDirPath,
+      configPath: tempDir,
       killFn: treeKill,
       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));
@@ -8684,11 +8683,245 @@ var launchNodeAndGetWallets = async ({
   };
   return { wallets, stop: cleanup, provider };
 };
+
+// src/test-utils/setup-test-provider-and-wallets.ts
+import { defaultSnapshotConfigs as defaultSnapshotConfigs3 } from "@fuel-ts/utils";
+import { mergeDeepRight } from "ramda";
+
+// src/test-utils/asset-id.ts
+import { randomBytes as randomBytes7 } from "@fuel-ts/crypto";
+import { hexlify as hexlify19 } from "@fuel-ts/utils";
+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(hexlify19(randomBytes7(32))));
+    }
+    return assetIds;
+  }
+};
+var AssetId = _AssetId;
+__publicField(AssetId, "A", new _AssetId(
+  "0x0101010101010101010101010101010101010101010101010101010101010101"
+));
+__publicField(AssetId, "B", new _AssetId(
+  "0x0202020202020202020202020202020202020202020202020202020202020202"
+));
+
+// src/test-utils/wallet-config.ts
+import { randomBytes as randomBytes8 } from "@fuel-ts/crypto";
+import { FuelError as FuelError20 } from "@fuel-ts/errors";
+import { defaultSnapshotConfigs as defaultSnapshotConfigs2, hexlify as hexlify20 } from "@fuel-ts/utils";
+var WalletConfig = class {
+  initialState;
+  options;
+  wallets;
+  generateWallets = () => {
+    const generatedWallets = [];
+    for (let index = 1; index <= this.options.count; index++) {
+      generatedWallets.push(new WalletUnlocked(randomBytes8(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 ?? defaultSnapshotConfigs2.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: hexlify20(randomBytes8(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 FuelError20(
+        FuelError20.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 FuelError20(
+        FuelError20.CODES.INVALID_INPUT_PARAMETERS,
+        "Number of assets per wallet must be greater than zero."
+      );
+    }
+    if (coinsPerAsset <= 0) {
+      throw new FuelError20(
+        FuelError20.CODES.INVALID_INPUT_PARAMETERS,
+        "Number of coins per asset must be greater than zero."
+      );
+    }
+    if (amountPerCoin <= 0) {
+      throw new FuelError20(
+        FuelError20.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 ?? defaultSnapshotConfigs3.chainConfig.consensus_parameters.V1.base_asset_id,
+    {
+      ...defaultWalletConfigOptions,
+      ...walletConfigOptions
+    }
+  );
+  const { cleanup, url } = await launchNode({
+    loggingEnabled: false,
+    ...nodeOptions,
+    snapshotConfig: mergeDeepRight(
+      defaultSnapshotConfigs3,
+      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
+import { Address as Address6 } from "@fuel-ts/address";
+import { randomBytes as randomBytes9 } from "@fuel-ts/crypto";
+import { bn as bn21 } from "@fuel-ts/math";
+import { hexlify as hexlify21 } from "@fuel-ts/utils";
+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 = Address6.fromRandom(),
+    recipient = Address6.fromRandom(),
+    nonce = hexlify21(randomBytes9(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: bn21(this.amount).toNumber(),
+      data: this.data,
+      da_height: this.da_height
+    };
+  }
+};
 export {
+  AssetId,
+  TestMessage,
+  WalletConfig,
   generateTestWallet,
   killNode,
   launchNode,
   launchNodeAndGetWallets,
-  seedTestWallet
+  seedTestWallet,
+  setupTestProviderAndWallets
 };
 //# sourceMappingURL=test-utils.mjs.map