From c24f0d8451a1a2350bcad444db374939ff0ba7d1 Mon Sep 17 00:00:00 2001
From: rodrigo <77309055+RodrigoVillar@users.noreply.github.com>
Date: Tue, 1 Oct 2024 09:25:18 -0400
Subject: [PATCH] Fix VM-With-Contracts Unit Tests (#1602)

---
 .../vmwithcontracts/actions/transfer_test.go  | 22 +++++++++++++++----
 1 file changed, 18 insertions(+), 4 deletions(-)

diff --git a/examples/vmwithcontracts/actions/transfer_test.go b/examples/vmwithcontracts/actions/transfer_test.go
index dfdecbf54b..2b016fa3a1 100644
--- a/examples/vmwithcontracts/actions/transfer_test.go
+++ b/examples/vmwithcontracts/actions/transfer_test.go
@@ -52,10 +52,16 @@ func TestTransferAction(t *testing.T) {
 				Value: 1,
 			},
 			State: func() state.Mutable {
-				keys := make(state.Keys)
-				keys.Add(string(emptyBalanceKey), state.Read)
-				tsv := ts.NewView(keys, map[string][]byte{})
-				return tsv
+				s := chaintest.NewInMemoryStore()
+				_, err := storage.AddBalance(
+					context.Background(),
+					s,
+					codec.EmptyAddress,
+					0,
+					true,
+				)
+				require.NoError(t, err)
+				return s
 			}(),
 			ExpectedErr: storage.ErrInvalidBalance,
 		},
@@ -79,6 +85,10 @@ func TestTransferAction(t *testing.T) {
 				require.NoError(err)
 				require.Equal(balance, uint64(1))
 			},
+			ExpectedOutputs: &TransferResult{
+				SenderBalance:   0,
+				ReceiverBalance: 1,
+			},
 		},
 		{
 			Name:  "OverflowBalance",
@@ -120,6 +130,10 @@ func TestTransferAction(t *testing.T) {
 				require.NoError(err)
 				require.Equal(senderBalance, uint64(0))
 			},
+			ExpectedOutputs: &TransferResult{
+				SenderBalance:   0,
+				ReceiverBalance: 1,
+			},
 		},
 	}