@fuel-ts/account 0.0.0-rc-1815-20240322115843 → 0.0.0-rc-1356-20240322130951

package/dist/test-utils.mjs

@@ -1486,15 +1486,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/transaction-request/errors.ts
 var NoWitnessAtIndexError = class extends Error {
   constructor(index) {
@@ -1624,27 +1615,13 @@ var BaseTransactionRequest = class {
     this.outputs.push(output);
     return this.outputs.length - 1;
   }
-  /**
-   * @hidden
-   *
-   * Pushes a witness to the list and returns the index
-   *
-   * @param signature - The signature to add to the witness.
-   * @returns The index of the created witness.
-   */
-  addWitness(signature) {
-    this.witnesses.push(signature);
-    return this.witnesses.length - 1;
-  }
   /**
    * @hidden
    *
    * Creates an empty witness without any side effects and returns the index
-   *
-   * @returns The index of the created witness.
    */
-  addEmptyWitness() {
-    this.addWitness(concat([ZeroBytes324, ZeroBytes324]));
+  createWitness() {
+    this.witnesses.push(concat([ZeroBytes324, ZeroBytes324]));
     return this.witnesses.length - 1;
   }
   /**
@@ -1673,21 +1650,6 @@ var BaseTransactionRequest = class {
     }
     this.witnesses[index] = witness;
   }
-  /**
-   * Helper function to add an external signature to the transaction.
-   *
-   * @param account - The account/s to sign to the transaction.
-   * @returns The transaction with the signature witness added.
-   */
-  async addAccountWitnesses(account) {
-    const accounts = Array.isArray(account) ? account : [account];
-    await Promise.all(
-      accounts.map(async (acc) => {
-        this.addWitness(await acc.signTransaction(this));
-      })
-    );
-    return this;
-  }
   /**
    * Gets the coin inputs for a transaction.
    *
@@ -1753,7 +1715,7 @@ var BaseTransactionRequest = class {
     } else {
       witnessIndex = this.getCoinInputWitnessIndexByOwner(owner);
       if (typeof witnessIndex !== "number") {
-        witnessIndex = this.addEmptyWitness();
+        witnessIndex = this.createWitness();
       }
     }
     const input = {
@@ -1787,7 +1749,7 @@ var BaseTransactionRequest = class {
     } else {
       witnessIndex = this.getCoinInputWitnessIndexByOwner(recipient);
       if (typeof witnessIndex !== "number") {
-        witnessIndex = this.addEmptyWitness();
+        witnessIndex = this.createWitness();
       }
     }
     const input = {
@@ -3309,6 +3271,7 @@ import { BigNumberCoder } from "@fuel-ts/abi-coder";
 import { ReceiptType as ReceiptType5 } from "@fuel-ts/transactions";
 
 // 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) {
@@ -4714,21 +4677,6 @@ var Account = class extends AbstractAccount {
     }
     return this._connector.signMessage(this.address.toString(), message);
   }
-  /**
-   * Signs a transaction with the wallet's private key.
-   *
-   * @param transactionRequestLike - The transaction request to sign.
-   * @returns A promise that resolves to the signature of the transaction.
-   */
-  async signTransaction(transactionRequestLike) {
-    if (!this._connector) {
-      throw new FuelError14(
-        ErrorCode14.MISSING_CONNECTOR,
-        "A connector is required to sign transactions."
-      );
-    }
-    return this._connector.signTransaction(this.address.toString(), transactionRequestLike);
-  }
   /**
    * Sends a transaction to the network.
    *
@@ -5043,7 +4991,7 @@ var BaseWalletUnlocked = class extends Account {
    */
   async signTransaction(transactionRequestLike) {
     const transactionRequest = transactionRequestify(transactionRequestLike);
-    const chainId = this.provider.getChainId();
+    const chainId = this.provider.getChain().consensusParameters.chainId.toNumber();
     const hashedTransaction = transactionRequest.getTransactionId(chainId);
     const signature = await this.signer().sign(hashedTransaction);
     return hexlify15(signature);
@@ -7914,7 +7862,8 @@ var launchNode = async ({
   useSystemFuelCore = false,
   loggingEnabled = true,
   debugEnabled = false,
-  basePath
+  basePath,
+  chainConfig = defaultChainConfig
 }) => (
   // eslint-disable-next-line no-async-promise-executor
   new Promise(async (resolve, reject) => {
@@ -7951,17 +7900,17 @@ var launchNode = async ({
         mkdirSync(tempDirPath, { recursive: true });
       }
       const tempChainConfigFilePath = path.join(tempDirPath, "chainConfig.json");
-      let chainConfig = defaultChainConfig;
+      let generatedChainConfig = chainConfig;
       if (!process.env.GENESIS_SECRET) {
         const pk = Signer.generatePrivateKey();
         const signer = new Signer(pk);
         process.env.GENESIS_SECRET = hexlify18(pk);
-        chainConfig = {
-          ...defaultChainConfig,
+        generatedChainConfig = {
+          ...generatedChainConfig,
           initial_state: {
-            ...defaultChainConfig.initial_state,
+            ...generatedChainConfig.initial_state,
             coins: [
-              ...defaultChainConfig.initial_state.coins,
+              ...generatedChainConfig.initial_state.coins,
               {
                 owner: signer.address.toHexString(),
                 amount: toHex2(1e9),
@@ -7971,7 +7920,7 @@ var launchNode = async ({
           }
         };
       }
-      writeFileSync(tempChainConfigFilePath, JSON.stringify(chainConfig), "utf8");
+      writeFileSync(tempChainConfigFilePath, JSON.stringify(generatedChainConfig), "utf8");
       chainConfigPathToUse = tempChainConfigFilePath;
     }
     const child = spawn(
@@ -8009,16 +7958,21 @@ var launchNode = async ({
       }
     };
     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}/graphql`,
           chainConfigPath: chainConfigPathToUse
         });
       }
-      if (/error/i.test(chunk)) {
-        reject(chunk.toString());
+      if (/error/i.test(text)) {
+        reject(text.toString());
       }
     });
     process.on("exit", () => killNode(cleanupConfig));
@@ -8050,11 +8004,221 @@ var launchNodeAndGetWallets = async ({
   };
   return { wallets, stop: cleanup, provider };
 };
+
+// src/test-utils/setup-test-provider-and-wallets.ts
+import { defaultChainConfig as defaultChainConfig2 } from "@fuel-ts/utils";
+import { mergeDeepRight } from "ramda";
+
+// src/test-utils/asset-id.ts
+import { BaseAssetId as BaseAssetId5 } from "@fuel-ts/address/configs";
+import { randomBytes as randomBytes5 } from "@fuel-ts/crypto";
+import { hexlify as hexlify19 } from "@fuel-ts/utils";
+var _AssetId = class {
+  constructor(value) {
+    this.value = value;
+  }
+  static random() {
+    return new _AssetId(hexlify19(randomBytes5(32)));
+  }
+};
+var AssetId = _AssetId;
+__publicField(AssetId, "BaseAssetId", new _AssetId(BaseAssetId5));
+__publicField(AssetId, "A", new _AssetId(
+  "0x0101010101010101010101010101010101010101010101010101010101010101"
+));
+__publicField(AssetId, "B", new _AssetId(
+  "0x0202020202020202020202020202020202020202020202020202020202020202"
+));
+
+// src/test-utils/wallet-config.ts
+import { randomBytes as randomBytes6 } from "@fuel-ts/crypto";
+import { FuelError as FuelError19 } from "@fuel-ts/errors";
+import { toHex as toHex3 } from "@fuel-ts/math";
+var WalletConfig = class {
+  initialState;
+  options;
+  wallets;
+  generateWallets = () => {
+    const generatedWallets = [];
+    for (let index = 1; index <= this.options.count; index++) {
+      generatedWallets.push(new WalletUnlocked(randomBytes6(32)));
+    }
+    return generatedWallets;
+  };
+  constructor(config) {
+    WalletConfig.guard(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, assets2, coinsPerAsset, amountPerCoin)
+    };
+  }
+  apply(chainConfig) {
+    return {
+      ...chainConfig,
+      initial_state: {
+        ...chainConfig?.initial_state,
+        coins: this.initialState.coins.concat(chainConfig?.initial_state?.coins || []),
+        messages: this.initialState.messages.concat(chainConfig?.initial_state?.messages ?? [])
+      }
+    };
+  }
+  static createMessages(wallets, messages) {
+    return messages.map((msg) => wallets.map((wallet) => msg.toChainMessage(wallet.address))).flatMap((x) => x);
+  }
+  static createCoins(wallets, assets2, coinsPerAsset, amountPerCoin) {
+    const coins = [];
+    let assetIds = [AssetId.BaseAssetId.value];
+    if (Array.isArray(assets2)) {
+      assetIds = assetIds.concat(assets2.map((a) => a.value));
+    } else {
+      for (let index = 0; index < assets2 - 1; index++) {
+        assetIds.push(AssetId.random().value);
+      }
+    }
+    wallets.map((wallet) => wallet.address.toHexString()).forEach((walletAddress) => {
+      assetIds.forEach((assetId) => {
+        for (let index = 0; index < coinsPerAsset; index++) {
+          coins.push({
+            amount: toHex3(amountPerCoin, 8),
+            asset_id: assetId,
+            owner: walletAddress
+          });
+        }
+      });
+    });
+    return coins;
+  }
+  static guard({
+    count: wallets,
+    assets: assets2,
+    coinsPerAsset,
+    amountPerCoin
+  }) {
+    if (Array.isArray(wallets) && wallets.length === 0 || typeof wallets === "number" && wallets <= 0) {
+      throw new FuelError19(
+        FuelError19.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 FuelError19(
+        FuelError19.CODES.INVALID_INPUT_PARAMETERS,
+        "Number of assets per wallet must be greater than zero."
+      );
+    }
+    if (coinsPerAsset <= 0) {
+      throw new FuelError19(
+        FuelError19.CODES.INVALID_INPUT_PARAMETERS,
+        "Number of coins per asset must be greater than zero."
+      );
+    }
+    if (amountPerCoin <= 0) {
+      throw new FuelError19(
+        FuelError19.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({
+    ...defaultWalletConfigOptions,
+    ...walletConfigOptions
+  });
+  const { cleanup, url } = await launchNode({
+    ...nodeOptions,
+    chainConfig: mergeDeepRight(defaultChainConfig2, walletConfig.apply(nodeOptions?.chainConfig)),
+    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 randomBytes7 } from "@fuel-ts/crypto";
+import { BN as BN3 } from "@fuel-ts/math";
+import { hexlify as hexlify20 } 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 = hexlify20(randomBytes7(32)),
+    amount = 1e6,
+    data = "02",
+    da_height = "0x00"
+  } = {}) {
+    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: new BN3(this.amount).toHex(8),
+      data: this.data,
+      da_height: this.da_height
+    };
+  }
+};
 export {
+  AssetId,
+  TestMessage,
+  WalletConfig,
   generateTestWallet,
   killNode,
   launchNode,
   launchNodeAndGetWallets,
-  seedTestWallet
+  seedTestWallet,
+  setupTestProviderAndWallets
 };
 //# sourceMappingURL=test-utils.mjs.map