chore: add panic on invalid staking keeper

cosmos · Apr 3, 2024 · be61348 · be61348
1 parent f1d2a0a
commit be61348
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/modules/core/keeper/keeper_test.go b/modules/core/keeper/keeper_test.go
@@ -5,8 +5,6 @@ import (
 	"testing"
 	"time"
 
-	ibctm "github.com/cosmos/ibc-go/v8/modules/light-clients/07-tendermint"
-
 	testifysuite "github.com/stretchr/testify/suite"
 
 	upgradekeeper "cosmossdk.io/x/upgrade/keeper"

diff --git a/modules/light-clients/07-tendermint/consensus_host.go b/modules/light-clients/07-tendermint/consensus_host.go
@@ -34,6 +34,10 @@ type StakingKeeper interface {
 
 // NewConsensusHost creates and returns a new ConsensusHost for tendermint consensus.
 func NewConsensusHost(stakingKeeper clienttypes.StakingKeeper) clienttypes.ConsensusHost {
+	if stakingKeeper == nil {
+		panic("staking keeper cannot be nil")
+	}
+
 	return &ConsensusHost{
 		stakingKeeper: stakingKeeper,
 	}