Use a custom anvil sender during the recovery tests

Author: corbanbrook

packages/wallet/wdk/test/recovery.test.ts

@@ -1,6 +1,6 @@
 import { describe, expect, it } from 'vitest'
 import { QueuedRecoveryPayload, SignerReady, TransactionDefined } from '../src/sequence/index.js'
-import { Bytes, Hex, Mnemonic, Provider, RpcTransport } from 'ox'
+import { Address, Bytes, Hex, Mnemonic, Provider, RpcTransport, Secp256k1 } from 'ox'
 import { Network, Payload } from '@0xsequence/wallet-primitives'
 import { LOCAL_RPC_URL, newManager } from './constants.js'
 
@@ -102,11 +102,23 @@ describe('Recovery', () => {
     // Complete the recovery payload
     const { to, data } = await manager.recovery.completePayload(requestId2)
 
+    // Generate and fund a sender address to avoid race conditions with default Anvil account
+    const senderAddress = Address.fromPublicKey(Secp256k1.getPublicKey({ privateKey: Secp256k1.randomPrivateKey() }))
+    await provider.request({
+      method: 'anvil_setBalance',
+      params: [senderAddress, Hex.fromNumber(1000000000000000000n)],
+    })
+    await provider.request({
+      method: 'anvil_impersonateAccount',
+      params: [senderAddress],
+    })
+
     // Send this transaction to anvil so we queue the payload
     await provider.request({
       method: 'eth_sendTransaction',
       params: [
         {
+          from: senderAddress,
           to,
           data,
         },
@@ -318,9 +330,21 @@ describe('Recovery', () => {
 
     // Complete the recovery payload and send to blockchain
     const { to, data } = await manager.recovery.completePayload(requestId2)
+
+    // Generate and fund a sender address to avoid race conditions with default Anvil account
+    const senderAddress = Address.fromPublicKey(Secp256k1.getPublicKey({ privateKey: Secp256k1.randomPrivateKey() }))
+    await provider.request({
+      method: 'anvil_setBalance',
+      params: [senderAddress, Hex.fromNumber(1000000000000000000n)],
+    })
+    await provider.request({
+      method: 'anvil_impersonateAccount',
+      params: [senderAddress],
+    })
+
     await provider.request({
       method: 'eth_sendTransaction',
-      params: [{ to, data }],
+      params: [{ from: senderAddress, to, data }],
     })
 
     // Wait for payload to become valid
@@ -463,10 +487,21 @@ describe('Recovery', () => {
 
       const { to, data } = await manager.recovery.completePayload(requestId)
 
+      // Generate and fund a sender address to avoid race conditions with default Anvil account
+      const senderAddress = Address.fromPublicKey(Secp256k1.getPublicKey({ privateKey: Secp256k1.randomPrivateKey() }))
+      await provider.request({
+        method: 'anvil_setBalance',
+        params: [senderAddress, Hex.fromNumber(1000000000000000000n)],
+      })
+      await provider.request({
+        method: 'anvil_impersonateAccount',
+        params: [senderAddress],
+      })
+
       // Send transactions sequentially to avoid nonce conflicts
       await provider.request({
         method: 'eth_sendTransaction',
-        params: [{ to, data }],
+        params: [{ from: senderAddress, to, data }],
       })
 
       // Small delay to ensure transaction ordering