From 928baa5317b84e0cc7df5ca0a55e6de5f84067be Mon Sep 17 00:00:00 2001 From: MSalopek Date: Mon, 9 Dec 2024 18:54:48 +0100 Subject: [PATCH] tests: improve unjail tests --- tests/integration/democracy.go | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/integration/democracy.go b/tests/integration/democracy.go index e2d198ce2f..69619d2aa5 100644 --- 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()