Add UpdateAccount to AccountWithWallet

iotaledger · May 15, 2024 · 8dd8f1f · 8dd8f1f
1 parent b1d1581
commit 8dd8f1f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pkg/testsuite/mock/wallet.go b/pkg/testsuite/mock/wallet.go
@@ -78,6 +78,11 @@ func (a *AccountWithWallet) Account() *AccountData {
 	return a.account
 }
 
+func (a *AccountWithWallet) UpdateAccount(updatedAccount *AccountData) {
+	a.account = updatedAccount
+	a.wallet.SetBlockIssuer(updatedAccount)
+}
+
 func (a *AccountWithWallet) Wallet() *Wallet {
 	return a.wallet
 }