fix staking test affected by other test

Stride-Labs · Aug 7, 2023 · 243afe6 · 243afe6
1 parent 1869b86
commit 243afe6
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/tests/e2e/e2e_staking_test.go b/tests/e2e/e2e_staking_test.go
@@ -22,6 +22,12 @@ func (s *IntegrationTestSuite) testStaking() {
 
 	fees := sdk.NewCoin(uatomDenom, sdk.NewInt(1))
 
+	existingDelegation := sdk.ZeroDec()
+	res, err := queryDelegation(chainEndpoint, validatorAddressA, delegatorAddress)
+	if err == nil {
+		existingDelegation = res.GetDelegationResponse().GetDelegation().GetShares()
+	}
+
 	delegationAmount := sdk.NewInt(500000000)
 	delegation := sdk.NewCoin(uatomDenom, delegationAmount) // 500 atom
 
@@ -35,7 +41,7 @@ func (s *IntegrationTestSuite) testStaking() {
 			amt := res.GetDelegationResponse().GetDelegation().GetShares()
 			s.Require().NoError(err)
 
-			return amt.Equal(sdk.NewDecFromInt(delegationAmount))
+			return amt.Equal(existingDelegation.Add(sdk.NewDecFromInt(delegationAmount)))
 		},
 		20*time.Second,
 		5*time.Second,