From f9a9d932313d6bbe8865c732ca5dbbcb690d2047 Mon Sep 17 00:00:00 2001
From: ascandone <alescandone@gmail.com>
Date: Fri, 30 Aug 2024 15:05:42 +0200
Subject: [PATCH] test: asserting requested balances

---
 .../internal/machine/vm/machine_test.go       | 23 ++++++++++++++++++-
 1 file changed, 22 insertions(+), 1 deletion(-)

diff --git a/components/ledger/internal/machine/vm/machine_test.go b/components/ledger/internal/machine/vm/machine_test.go
index 8d31ba0494..a78b93e2d8 100644
--- a/components/ledger/internal/machine/vm/machine_test.go
+++ b/components/ledger/internal/machine/vm/machine_test.go
@@ -1010,8 +1010,11 @@ func TestNeededBalances(t *testing.T) {
 	require.Equalf(t, []string{"c"}, readLockAccounts, "readlock")
 	require.Equalf(t, []string{"a", "b", "bounded"}, writeLockAccounts, "writelock")
 
-	err = m.ResolveBalances(context.Background(), EmptyStore)
+	store := mockStore{}
+	err = m.ResolveBalances(context.Background(), &store)
 	require.NoError(t, err)
+
+	require.Equal(t, []string{"a", "b", "bounded"}, store.RequestedAccounts)
 }
 
 func TestNeededBalances2(t *testing.T) {
@@ -1062,6 +1065,11 @@ send $balance (
 	require.NoError(t, err)
 	require.Equal(t, []string{"a"}, wlAccounts)
 	require.Equal(t, []string{"acc", "b"}, rlAccounts)
+
+	store := mockStore{}
+	err = m.ResolveBalances(context.Background(), &store)
+	require.NoError(t, err)
+	require.Equal(t, []string{"acc", "a"}, store.RequestedAccounts)
 }
 
 func TestSetTxMeta(t *testing.T) {
@@ -2295,3 +2303,16 @@ send [COIN 100] (
 	}
 	test(t, tc)
 }
+
+type mockStore struct {
+	RequestedAccounts []string
+}
+
+func (s *mockStore) GetBalance(ctx context.Context, address, asset string) (*big.Int, error) {
+	s.RequestedAccounts = append(s.RequestedAccounts, address)
+	return big.NewInt(0), nil
+}
+
+func (s *mockStore) GetAccount(ctx context.Context, address string) (*ledger.Account, error) {
+	panic("not implemented")
+}