Resolve staking endblock unit tests

KiraCore · Sep 22, 2023 · 8b1b166 · 8b1b166
1 parent afb5a54
commit 8b1b166
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/x/staking/abci_test.go b/x/staking/abci_test.go
@@ -33,15 +33,15 @@ func TestItUpdatesTheValidatorSetBasedOnPendingValidators(t *testing.T) {
 
 	// First checkings
 	validatorSet := app.CustomStakingKeeper.GetValidatorSet(ctx)
-	require.Len(t, validatorSet, 0)
+	require.Len(t, validatorSet, 1)
 	validatorSet = app.CustomStakingKeeper.GetPendingValidatorSet(ctx)
 	require.Len(t, validatorSet, 1)
 
 	updates := staking.EndBlocker(ctx, app.CustomStakingKeeper)
 	require.Len(t, updates, 1)
 
 	validatorSet = app.CustomStakingKeeper.GetValidatorSet(ctx)
-	require.Len(t, validatorSet, 1)
+	require.Len(t, validatorSet, 2)
 	validatorSet = app.CustomStakingKeeper.GetPendingValidatorSet(ctx)
 	require.Len(t, validatorSet, 0)
 }
@@ -52,15 +52,15 @@ func TestItDoesNotReturnUpdatesIfThereIsNoPending(t *testing.T) {
 
 	// First checkings
 	validatorSet := app.CustomStakingKeeper.GetValidatorSet(ctx)
-	require.Len(t, validatorSet, 0)
+	require.Len(t, validatorSet, 1)
 	validatorSet = app.CustomStakingKeeper.GetPendingValidatorSet(ctx)
 	require.Len(t, validatorSet, 0)
 
 	updates := staking.EndBlocker(ctx, app.CustomStakingKeeper)
 	require.Len(t, updates, 0)
 
 	validatorSet = app.CustomStakingKeeper.GetValidatorSet(ctx)
-	require.Len(t, validatorSet, 0)
+	require.Len(t, validatorSet, 1)
 	validatorSet = app.CustomStakingKeeper.GetPendingValidatorSet(ctx)
 	require.Len(t, validatorSet, 0)
 }