refactor(e2e): check wallet balance before running tests

input-output-hk · Mar 17, 2023 · 450dd33 · 450dd33
1 parent cdfdc15
commit 450dd33
Show file tree

Hide file tree

Showing 8 changed files with 29 additions and 15 deletions.
diff --git a/packages/e2e/test/wallet/SingleAddressWallet/delegation.test.ts b/packages/e2e/test/wallet/SingleAddressWallet/delegation.test.ts
@@ -87,14 +87,13 @@ describe('SingleAddressWallet/delegation', () => {
     // source wallet has the highest balance to begin with
     const [sourceWallet, destWallet] = await chooseWallets();
 
-    await walletReady(sourceWallet);
+    const tx1OutputCoins = 1_000_000n;
+    await walletReady(sourceWallet, tx1OutputCoins);
 
     const protocolParameters = await firstValueFrom(sourceWallet.protocolParameters$);
     const stakeKeyDeposit = BigInt(protocolParameters.stakeKeyDeposit);
     const initialState = await getWalletStateSnapshot(sourceWallet);
-    expect(initialState.balance.total.coins).toBeGreaterThan(0n);
     expect(initialState.balance.total.coins).toBe(initialState.balance.available.coins);
-    const tx1OutputCoins = 1_000_000n;
     const poolId = await chooseDifferentPoolIdRandomly(initialState.rewardAccount.delegatee?.nextNextEpoch?.id);
     expect(poolId).toBeDefined();
     const initialDeposit = initialState.isStakeKeyRegistered ? stakeKeyDeposit : 0n;

diff --git a/packages/e2e/test/wallet/SingleAddressWallet/metadata.test.ts b/packages/e2e/test/wallet/SingleAddressWallet/metadata.test.ts
@@ -26,6 +26,9 @@ describe('SingleAddressWallet/metadata', () => {
     const walletUtil = createWalletUtil(wallet);
     const { minimumCoin } = await walletUtil.validateValue({ coins: 0n });
 
+    // Make sure the wallet has sufficient funds to run this test
+    walletReady(wallet, minimumCoin);
+
     const builtTx = await buildTx({ logger, observableWallet: wallet })
       .addOutput({ address: ownAddress, value: { coins: minimumCoin } })
       .setMetadata(metadata)

diff --git a/packages/e2e/test/wallet/SingleAddressWallet/mint.test.ts b/packages/e2e/test/wallet/SingleAddressWallet/mint.test.ts
@@ -20,7 +20,8 @@ describe('SingleAddressWallet/mint', () => {
   it('can mint a token with no asset name', async () => {
     wallet = (await getWallet({ env, logger, name: 'Minting Wallet', polling: { interval: 50 } })).wallet;
 
-    await walletReady(wallet);
+    const coins = 3_000_000n;
+    await walletReady(wallet, coins);
 
     const genesis = await firstValueFrom(wallet.genesisParameters$);
 
@@ -65,7 +66,7 @@ describe('SingleAddressWallet/mint', () => {
           address: walletAddress,
           value: {
             assets: tokens,
-            coins: 3_000_000n
+            coins
           }
         }
       ]),

diff --git a/packages/e2e/test/wallet/SingleAddressWallet/multisignature.test.ts b/packages/e2e/test/wallet/SingleAddressWallet/multisignature.test.ts
@@ -21,7 +21,8 @@ describe('SingleAddressWallet/multisignature', () => {
   it('can create a transaction with multiple signatures to mint an asset', async () => {
     wallet = (await getWallet({ env, logger, name: 'Minting Wallet', polling: { interval: 50 } })).wallet;
 
-    await walletReady(wallet);
+    const coins = 3_000_000n;
+    await walletReady(wallet, coins);
 
     const genesis = await firstValueFrom(wallet.genesisParameters$);
 
@@ -85,7 +86,7 @@ describe('SingleAddressWallet/multisignature', () => {
           address: walletAddress,
           value: {
             assets: tokens,
-            coins: 3_000_000n
+            coins
           }
         }
       ]),

diff --git a/packages/e2e/test/wallet/SingleAddressWallet/nft.test.ts b/packages/e2e/test/wallet/SingleAddressWallet/nft.test.ts
@@ -24,11 +24,12 @@ describe('SingleAddressWallet.assets/nft', () => {
   const assetNames = ['4e46542d66696c6573', '4e46542d303031', '4e46542d303032'];
   let walletAddress: Cardano.PaymentAddress;
   let allMintedTokens: Cardano.TokenMap = new Map();
+  const coins = 10_000_000n; // number of coins to use in each transaction
 
   beforeAll(async () => {
     wallet = (await getWallet({ env, logger, name: 'Minting Wallet', polling: { interval: 50 } })).wallet;
 
-    await walletReady(wallet);
+    await walletReady(wallet, coins);
 
     const genesis = await firstValueFrom(wallet.genesisParameters$);
 
@@ -132,7 +133,7 @@ describe('SingleAddressWallet.assets/nft', () => {
           address: walletAddress,
           value: {
             assets: tokens,
-            coins: 50_000_000n
+            coins
           }
         }
       ]),
@@ -255,6 +256,9 @@ describe('SingleAddressWallet.assets/nft', () => {
   });
 
   it('supports burning tokens', async () => {
+    // Make sure the wallet has sufficient funds to run this test
+    walletReady(wallet, coins);
+
     // spend entire balance of test asset
     const availableBalance = await firstValueFrom(wallet.balance.utxo.available$);
     const assetBalance = availableBalance.assets!.get(assetIds[TOKEN_BURN_INDEX])!;
@@ -265,7 +269,7 @@ describe('SingleAddressWallet.assets/nft', () => {
         {
           address: walletAddress,
           value: {
-            coins: 50_000_000n
+            coins
           }
         }
       ]),
@@ -305,6 +309,9 @@ describe('SingleAddressWallet.assets/nft', () => {
     // eslint-disable-next-line unicorn/consistent-function-scoping
     const CIP0025Test = (testName: string, assetName: string, version: 1 | 2, encoding?: 'hex' | 'utf8') =>
       it(testName, async () => {
+        // Make sure the wallet has sufficient funds to run this test
+        walletReady(wallet, coins);
+
         const assetNameHex = Buffer.from(assetName).toString('hex');
         const assetId = Cardano.AssetId(`${policyId}${assetNameHex}`);
         const fingerprint = Cardano.AssetFingerprint.fromParts(policyId, Cardano.AssetName(assetNameHex));
@@ -336,7 +343,7 @@ describe('SingleAddressWallet.assets/nft', () => {
               address: walletAddress,
               value: {
                 assets: tokens,
-                coins: 50_000_000n
+                coins
               }
             }
           ]),

diff --git a/packages/e2e/test/wallet/SingleAddressWallet/txChainHistory.test.ts b/packages/e2e/test/wallet/SingleAddressWallet/txChainHistory.test.ts
@@ -5,7 +5,7 @@ import { filter, firstValueFrom, map, take } from 'rxjs';
 import { getEnv, getWallet, walletVariables } from '../../../src';
 import { isNotNil } from '@cardano-sdk/util';
 import { logger } from '@cardano-sdk/util-dev';
-import { normalizeTxBody } from '../../util';
+import { normalizeTxBody, walletReady } from '../../util';
 
 const env = getEnv(walletVariables);
 
@@ -22,6 +22,8 @@ describe('SingleAddressWallet/txChainHistory', () => {
 
   it('submit a transaction and find it in chain history', async () => {
     const tAdaToSend = 10_000_000n;
+    // Make sure the wallet has sufficient funds to run this test
+    walletReady(wallet, tAdaToSend);
 
     await firstValueFrom(wallet.syncStatus.isSettled$.pipe(filter((isSettled) => isSettled)));
 

diff --git a/packages/e2e/test/wallet/SingleAddressWallet/txChaining.test.ts b/packages/e2e/test/wallet/SingleAddressWallet/txChaining.test.ts
@@ -13,7 +13,7 @@ describe('SingleAddressWallet/txChaining', () => {
     jest.setTimeout(180_000);
     wallet = (await getWallet({ env, logger, name: 'Test Wallet' })).wallet;
 
-    await walletReady(wallet);
+    await walletReady(wallet, 1_000_000n);
   });
 
   afterAll(() => {

diff --git a/packages/e2e/test/wallet/SingleAddressWallet/unspendableUtxos.test.ts b/packages/e2e/test/wallet/SingleAddressWallet/unspendableUtxos.test.ts
@@ -25,8 +25,9 @@ describe('SingleAddressWallet/unspendableUtxos', () => {
     wallet1 = (await getWallet({ env, logger, name: 'Wallet 1', polling: { interval: 50 } })).wallet;
     wallet2 = (await getWallet({ env, logger, name: 'Wallet 2', polling: { interval: 50 } })).wallet;
 
-    await walletReady(wallet1);
-    await walletReady(wallet2);
+    const coins = 5_000_000n;
+    await walletReady(wallet1, coins);
+    await walletReady(wallet2, coins);
 
     const txBuilder1 = buildTx({ logger, observableWallet: wallet1 });
     const txBuilder2 = buildTx({ logger, observableWallet: wallet2 });