more checks

multiversx · Oct 3, 2024 · 77de626 · 77de626
1 parent bece3f3
commit 77de626
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/vm/systemSmartContracts/delegation_test.go b/vm/systemSmartContracts/delegation_test.go
@@ -5494,9 +5494,21 @@ func TestDelegationSystemSC_ExecuteMoveDelegationToDestination(t *testing.T) {
 	require.Equal(t, vmcommon.UserError, output)
 	require.Equal(t, eei.returnMessage, "total delegation cap reached")
 
+	_ = d.saveDelegatorData(vmInput.Arguments[0], &DelegatorData{
+		ActiveFund:            fundKey,
+		UnStakedFunds:         [][]byte{},
+		UnClaimedRewards:      big.NewInt(0),
+		TotalCumulatedRewards: big.NewInt(0),
+	})
+
 	_ = d.saveDelegationContractConfig(&DelegationConfig{InitialOwnerFunds: big.NewInt(100), MaxDelegationCap: big.NewInt(10000)})
 	eei.returnMessage = ""
 	output = d.Execute(vmInput)
 	require.Equal(t, vmcommon.Ok, output)
 
+	fund, _ := d.getFund(fundKey)
+	require.Equal(t, fund.Value, big.NewInt(200))
+
+	globalFund, _ := d.getGlobalFundData()
+	require.Equal(t, globalFund.TotalActive, big.NewInt(200))
 }