tests: improve unjail tests

cosmos · Dec 9, 2024 · 928baa5 · 928baa5
1 parent b65fc99
commit 928baa5
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/integration/democracy.go b/tests/integration/democracy.go
@@ -235,6 +235,7 @@ func (s *ConsumerDemocracyTestSuite) TestDemocracyMsgUpdateParams() {
 
 func (s *ConsumerDemocracyTestSuite) TestDemocracyValidatorUnjail() {
 	stakingKeeper := s.consumerApp.GetTestStakingKeeper()
+	consumerKeeper := s.consumerApp.GetConsumerKeeper()
 
 	validators, err := stakingKeeper.GetAllValidators(s.consumerCtx())
 	s.Require().NoError(err)
@@ -258,8 +259,9 @@ func (s *ConsumerDemocracyTestSuite) TestDemocracyValidatorUnjail() {
 		}
 	}
 
-	// the actual test if unjailing works
-	err = stakingKeeper.GetValidatorSet().Unjail(s.consumerCtx(), consAddr)
+	// confirm unjail will not error and properly unjail
+	// in case of a consumer chain without standalone staking the call is a no-op
+	err = consumerKeeper.Unjail(s.consumerCtx(), consAddr)
 	s.Require().NoError(err)
 	s.consumerChain.NextBlock()