test: fix more failing tests

Author: greatertomi

packages/e2e/test/load-test-custom/wallet-init/wallet-init.test.ts

@@ -1,16 +1,8 @@
 /* eslint-disable import/imports-first */
 import * as dotenv from 'dotenv';
-import path from 'path';
-
-// This line must come before loading the env, to configure the location of the .env file
-dotenv.config({ path: path.join(__dirname, '../../../.env') });
-
 import { BaseWallet, createPersonalWallet } from '@cardano-sdk/wallet';
+import { Bip32Account, KeyPurpose, util } from '@cardano-sdk/key-management';
 import { Logger } from 'ts-log';
-import { bufferCount, bufferTime, from, mergeAll, tap } from 'rxjs';
-import { logger } from '@cardano-sdk/util-dev';
-
-import { Bip32Account, util } from '@cardano-sdk/key-management';
 import {
   MeasurementUtil,
   assetProviderFactory,
@@ -27,6 +19,12 @@ import {
   waitForWalletStateSettle,
   walletVariables
 } from '../../../src';
+import { bufferCount, bufferTime, from, mergeAll, tap } from 'rxjs';
+import { logger } from '@cardano-sdk/util-dev';
+import path from 'path';
+
+// This line must come before loading the env, to configure the location of the .env file
+dotenv.config({ path: path.join(__dirname, '../../../.env') });
 
 // Example call that creates 5000 wallets in 10 minutes:
 // VIRTUAL_USERS_GENERATE_DURATION=600 VIRTUAL_USERS_COUNT=5000 yarn load-test-custom:wallet-init
@@ -85,7 +83,7 @@ const createWallet = async (accountIndex: number): Promise<BaseWallet> => {
 
   measurementUtil.addStartMarker(MeasureTarget.wallet, accountIndex);
   return createPersonalWallet(
-    { name: `Wallet ${accountIndex}` },
+    { name: `Wallet ${accountIndex}`, purpose: KeyPurpose.STANDARD },
     {
       ...providers,
       bip32Account: await Bip32Account.fromAsyncKeyAgent(keyAgent),

packages/e2e/test/long-running/cache-invalidation.test.ts

@@ -1,5 +1,5 @@
 import * as Crypto from '@cardano-sdk/crypto';
-import { AddressType, KeyRole } from '@cardano-sdk/key-management';
+import { AddressType, KeyPurpose, KeyRole } from '@cardano-sdk/key-management';
 import { Cardano } from '@cardano-sdk/core';
 import {
   KeyAgentFactoryProps,
@@ -89,7 +89,8 @@ describe('cache invalidation', () => {
       idx: 0,
       logger,
       name: 'Pool Wallet 1',
-      polling: { interval: 500 }
+      polling: { interval: 500 },
+      purpose: KeyPurpose.STANDARD
     });
 
     await waitForWalletStateSettle(wallet1.wallet);

packages/e2e/test/long-running/multisig-wallet/MultiSigWallet.ts

@@ -217,7 +217,7 @@ export class MultiSigWallet {
         body: multiSigTx.getTransaction().body,
         hash: multiSigTx.getTransaction().id
       },
-      { knownAddresses: [this.#address], txInKeyPathMap: {} },
+      { knownAddresses: [this.#address], purpose: KeyPurpose.MULTI_SIG, txInKeyPathMap: {} },
       { additionalKeyPaths: [DERIVATION_PATH] }
     );
 
@@ -413,6 +413,7 @@ export class MultiSigWallet {
       address: baseAddress.toAddress().toBech32() as Cardano.PaymentAddress,
       index: 0,
       networkId,
+      purpose: KeyPurpose.MULTI_SIG,
       rewardAccount: Cardano.RewardAddress.fromCredentials(networkId, scriptCredential)
         .toAddress()
         .toBech32() as Cardano.RewardAccount,

packages/e2e/test/wallet_epoch_0/PersonalWallet/plutusTest.test.ts

@@ -1,6 +1,7 @@
 import { BaseWallet } from '@cardano-sdk/wallet';
 import { Cardano, Serialization, UtxoProvider } from '@cardano-sdk/core';
 import { HexBlob, isNotNil } from '@cardano-sdk/util';
+import { KeyPurpose } from '@cardano-sdk/key-management';
 import { Observable, filter, firstValueFrom, interval, map, switchMap, take } from 'rxjs';
 import { createLogger } from '@cardano-sdk/util-dev';
 import { getEnv, getWallet, utxoProviderFactory, walletReady, walletVariables } from '../../../src';
@@ -191,7 +192,9 @@ describe('PersonalWallet/plutus', () => {
   });
 
   it('can spend balance from a plutus script', async () => {
-    wallet = (await getWallet({ env, logger, name: 'Spending Wallet', polling: { interval: 50 } })).wallet;
+    wallet = (
+      await getWallet({ env, logger, name: 'Spending Wallet', polling: { interval: 50 }, purpose: KeyPurpose.STANDARD })
+    ).wallet;
     // UTXO provider can be use to fetch the UTXO set from the script address.
     const utxoProvider = await utxoProviderFactory.create(env.UTXO_PROVIDER, env.UTXO_PROVIDER_PARAMS, logger);
 
@@ -265,7 +268,9 @@ describe('PersonalWallet/plutus', () => {
   });
 
   it('can spend balance from a plutus script using a reference script', async () => {
-    wallet = (await getWallet({ env, logger, name: 'Spending Wallet', polling: { interval: 50 } })).wallet;
+    wallet = (
+      await getWallet({ env, logger, name: 'Spending Wallet', polling: { interval: 50 }, purpose: KeyPurpose.STANDARD })
+    ).wallet;
     // UTXO provider can be use to fetch the UTXO set from the script address.
     const utxoProvider = await utxoProviderFactory.create(env.UTXO_PROVIDER, env.UTXO_PROVIDER_PARAMS, logger);
 

packages/e2e/test/wallet_epoch_0/SharedWallet/utils.ts

@@ -2,6 +2,7 @@ import * as Crypto from '@cardano-sdk/crypto';
 import {
   AccountKeyDerivationPath,
   KeyAgent,
+  KeyPurpose,
   KeyRole,
   SignBlobResult,
   SignDataContext,
@@ -22,6 +23,7 @@ const randomPublicKey = () => Crypto.Ed25519PublicKeyHex(Array.from({ length: 64
 
 const DERIVATION_PATH = {
   index: 0,
+  purpose: KeyPurpose.STANDARD,
   role: KeyRole.External
 };
 
@@ -30,7 +32,12 @@ const getKeyAgent = async (
   genesisParameters: Cardano.CompactGenesis,
   bip32Ed25519: Crypto.Bip32Ed25519
 ) => {
-  const keyAgent = await createStandaloneKeyAgent(mnemonics.split(' '), genesisParameters, bip32Ed25519);
+  const keyAgent = await createStandaloneKeyAgent(
+    mnemonics.split(' '),
+    genesisParameters,
+    bip32Ed25519,
+    KeyPurpose.STANDARD
+  );
 
   const pubKey = await keyAgent.derivePublicKey(DERIVATION_PATH);
 
@@ -294,6 +301,7 @@ export const buildSharedWallets = async (env: any, genesisParameters: Cardano.Co
       name: 'Alice shared Wallet',
       paymentScript,
       polling: { interval: 50 },
+      purpose: KeyPurpose.STANDARD,
       stakingScript,
       witnesser: new SharedWalletWitnesser(alice.keyAgent, paymentScript, stakingScript)
     })
@@ -306,6 +314,7 @@ export const buildSharedWallets = async (env: any, genesisParameters: Cardano.Co
       name: 'Bob shared Wallet',
       paymentScript,
       polling: { interval: 50 },
+      purpose: KeyPurpose.STANDARD,
       stakingScript,
       witnesser: new SharedWalletWitnesser(bob.keyAgent, paymentScript, stakingScript)
     })
@@ -318,6 +327,7 @@ export const buildSharedWallets = async (env: any, genesisParameters: Cardano.Co
       name: 'Charlotte shared Wallet',
       paymentScript,
       polling: { interval: 50 },
+      purpose: KeyPurpose.STANDARD,
       stakingScript,
       witnesser: new SharedWalletWitnesser(charlotte.keyAgent, paymentScript, stakingScript)
     })

packages/wallet/test/hardware/ledger/LedgerKeyAgent.integration.test.ts

@@ -1,6 +1,5 @@
 import * as Crypto from '@cardano-sdk/crypto';
-
-import { Bip32Account, CommunicationType, KeyAgent, util } from '@cardano-sdk/key-management';
+import { Bip32Account, CommunicationType, KeyAgent, KeyPurpose, util } from '@cardano-sdk/key-management';
 import { Cardano } from '@cardano-sdk/core';
 import { LedgerKeyAgent } from '@cardano-sdk/hardware-ledger';
 import { ObservableWallet, createPersonalWallet, restoreKeyAgent } from '../../../src';
@@ -27,7 +26,7 @@ const createWallet = async (keyAgent: KeyAgent) => {
   const asyncKeyAgent = util.createAsyncKeyAgent(keyAgent);
   const chainHistoryProvider = mockChainHistoryProvider();
   return createPersonalWallet(
-    { name: 'Wallet1' },
+    { name: 'Wallet1', purpose: KeyPurpose.STANDARD },
     {
       assetProvider,
       bip32Account: await Bip32Account.fromAsyncKeyAgent(asyncKeyAgent),
@@ -52,7 +51,8 @@ describe('LedgerKeyAgent+BaseWallet', () => {
     const freshKeyAgent = await LedgerKeyAgent.createWithDevice(
       {
         chainId: Cardano.ChainIds.Preprod,
-        communicationType: CommunicationType.Node
+        communicationType: CommunicationType.Node,
+        purpose: KeyPurpose.STANDARD
       },
       keyAgentDependencies
     );

packages/wallet/test/hardware/trezor/TrezorKeyAgent.integration.test.ts

@@ -1,5 +1,5 @@
 import * as Crypto from '@cardano-sdk/crypto';
-import { Bip32Account, CommunicationType, KeyAgent, util } from '@cardano-sdk/key-management';
+import { Bip32Account, CommunicationType, KeyAgent, KeyPurpose, util } from '@cardano-sdk/key-management';
 import { Cardano } from '@cardano-sdk/core';
 import { ObservableWallet, createPersonalWallet, restoreKeyAgent } from '../../../src';
 import { TrezorKeyAgent } from '@cardano-sdk/hardware-trezor';
@@ -28,7 +28,7 @@ const createWallet = async (keyAgent: KeyAgent) => {
   const asyncKeyAgent = util.createAsyncKeyAgent(keyAgent);
   const chainHistoryProvider = mockChainHistoryProvider();
   return createPersonalWallet(
-    { name: 'Wallet1' },
+    { name: 'Wallet1', purpose: KeyPurpose.STANDARD },
     {
       assetProvider,
       bip32Account: await Bip32Account.fromAsyncKeyAgent(asyncKeyAgent),
@@ -51,6 +51,7 @@ describe('TrezorKeyAgent+BaseWallet', () => {
     const freshKeyAgent = await TrezorKeyAgent.createWithDevice(
       {
         chainId: Cardano.ChainIds.Preprod,
+        purpose: KeyPurpose.STANDARD,
         trezorConfig: {
           communicationType: CommunicationType.Node,
           manifest: {

packages/web-extension/test/walletManager/SigningCoordinator.test.ts

@@ -2,6 +2,7 @@ import {
   AccountKeyDerivationPath,
   CommunicationType,
   InMemoryKeyAgent,
+  KeyPurpose,
   KeyRole,
   SignBlobResult,
   SignDataContext,
@@ -33,6 +34,7 @@ describe('SigningCoordinator', () => {
   const requestContext = {
     accountIndex: 0,
     chainId: Cardano.ChainIds.Preprod,
+    purpose: KeyPurpose.STANDARD,
     wallet
   };
   let passphrase: Uint8Array;
@@ -69,7 +71,11 @@ describe('SigningCoordinator', () => {
       // eslint-disable-next-line max-len
       '84a60081825820260aed6e7a24044b1254a87a509468a649f522a4e54e830ac10f27ea7b5ec61f01018383581d70b429738bd6cc58b5c7932d001aa2bd05cfea47020a556c8c753d44361a004c4b40582007845f8f3841996e3d8157954e2f5e2fb90465f27112fc5fe9056d916fae245b82583900b1814238b0d287a8a46ce7348c6ad79ab8995b0e6d46010e2d9e1c68042f1946335c498d2e7556c5c647c4649c6a69d2b645cd1428a339ba1a0463676982583900b1814238b0d287a8a46ce7348c6ad79ab8995b0e6d46010e2d9e1c68042f1946335c498d2e7556c5c647c4649c6a69d2b645cd1428a339ba821a00177a6ea2581c648823ffdad1610b4162f4dbc87bd47f6f9cf45d772ddef661eff198a5447742544319271044774554481a0031f9194577444f47451a0056898d4577555344431a000fc589467753484942411a000103c2581c659ab0b5658687c2e74cd10dba8244015b713bf503b90557769d77a7a14a57696e675269646572731a02269552021a0002e665031a01353f84081a013531740b58204107eada931c72a600a6e3305bd22c7aeb9ada7c3f6823b155f4db85de36a69aa20081825820e686ade5bc97372f271fd2abc06cfd96c24b3d9170f9459de1d8e3dd8fd385575840653324a9dddad004f05a8ac99fa2d1811af5f00543591407fb5206cfe9ac91bb1412404323fa517e0e189684cd3592e7f74862e3f16afbc262519abec958180c0481d8799fd8799fd8799fd8799f581cb1814238b0d287a8a46ce7348c6ad79ab8995b0e6d46010e2d9e1c68ffd8799fd8799fd8799f581c042f1946335c498d2e7556c5c647c4649c6a69d2b645cd1428a339baffffffff581cb1814238b0d287a8a46ce7348c6ad79ab8995b0e6d46010e2d9e1c681b000001863784a12ed8799fd8799f4040ffd8799f581c648823ffdad1610b4162f4dbc87bd47f6f9cf45d772ddef661eff1984577444f4745ffffffd8799fd87980190c8efffff5f6'
     );
-    const signContext: SignTransactionContext = { knownAddresses: [], txInKeyPathMap: {} };
+    const signContext: SignTransactionContext = {
+      knownAddresses: [],
+      purpose: KeyPurpose.STANDARD,
+      txInKeyPathMap: {}
+    };
 
     it('rejects given invalid tx cbor', async () => {
       await expect(
@@ -155,7 +161,7 @@ describe('SigningCoordinator', () => {
       signature: '123' as Ed25519SignatureHex
     };
     const signContext: SignDataContext = {};
-    const derivationPath: AccountKeyDerivationPath = { index: 0, role: KeyRole.DRep };
+    const derivationPath: AccountKeyDerivationPath = { index: 0, purpose: KeyPurpose.STANDARD, role: KeyRole.DRep };
 
     it('rejects with AuthenticationError when there is no subscriber', async () => {
       keyAgent.signBlob.mockResolvedValueOnce(signResult);

packages/web-extension/test/walletManager/WalletRepository.test.ts

@@ -13,7 +13,7 @@ import {
 } from '../../src';
 import { Cardano, Serialization } from '@cardano-sdk/core';
 import { Hash28ByteBase16 } from '@cardano-sdk/crypto';
-import { KeyRole } from '@cardano-sdk/key-management';
+import { KeyPurpose, KeyRole } from '@cardano-sdk/key-management';
 import { firstValueFrom, of } from 'rxjs';
 import { logger } from '@cardano-sdk/util-dev';
 import pick from 'lodash/pick';
@@ -39,10 +39,12 @@ const createScriptWalletProps = {
       accountIndex: storedLedgerWallet.accounts[0].accountIndex,
       paymentScriptKeyPath: {
         index: 0,
+        purpose: KeyPurpose.STANDARD,
         role: KeyRole.External
       },
       stakingScriptKeyPath: {
         index: 0,
+        purpose: KeyPurpose.STANDARD,
         role: KeyRole.External
       },
       walletId: storedLedgerWallet.walletId
@@ -134,10 +136,12 @@ describe('WalletRepository', () => {
               accountIndex: createScriptWalletProps.ownSigners[0].accountIndex,
               paymentScriptKeyPath: {
                 index: 0,
+                purpose: KeyPurpose.STANDARD,
                 role: KeyRole.External
               },
               stakingScriptKeyPath: {
                 index: 0,
+                purpose: KeyPurpose.STANDARD,
                 role: KeyRole.External
               },
               walletId: 'does not exist' as WalletId
@@ -156,10 +160,12 @@ describe('WalletRepository', () => {
               accountIndex: 999_999_999,
               paymentScriptKeyPath: {
                 index: 0,
+                purpose: KeyPurpose.STANDARD,
                 role: KeyRole.External
               },
               stakingScriptKeyPath: {
                 index: 0,
+                purpose: KeyPurpose.STANDARD,
                 role: KeyRole.External
               },
               walletId: createScriptWalletProps.ownSigners[0].walletId