61 lines
2.0 KiB
Go
61 lines
2.0 KiB
Go
package client_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
client "github.com/cosmos/cosmos-sdk/x/ibc/core/02-client"
|
|
"github.com/cosmos/cosmos-sdk/x/ibc/core/02-client/types"
|
|
"github.com/cosmos/cosmos-sdk/x/ibc/core/exported"
|
|
localhosttypes "github.com/cosmos/cosmos-sdk/x/ibc/light-clients/09-localhost/types"
|
|
ibctesting "github.com/cosmos/cosmos-sdk/x/ibc/testing"
|
|
)
|
|
|
|
type ClientTestSuite struct {
|
|
suite.Suite
|
|
|
|
coordinator *ibctesting.Coordinator
|
|
|
|
chainA *ibctesting.TestChain
|
|
chainB *ibctesting.TestChain
|
|
}
|
|
|
|
func (suite *ClientTestSuite) SetupTest() {
|
|
suite.coordinator = ibctesting.NewCoordinator(suite.T(), 2)
|
|
|
|
suite.chainA = suite.coordinator.GetChain(ibctesting.GetChainID(0))
|
|
suite.chainB = suite.coordinator.GetChain(ibctesting.GetChainID(1))
|
|
|
|
// set localhost client
|
|
version := types.ParseChainID(suite.chainA.GetContext().ChainID())
|
|
localHostClient := localhosttypes.NewClientState(
|
|
suite.chainA.GetContext().ChainID(), types.NewHeight(version, uint64(suite.chainA.GetContext().BlockHeight())),
|
|
)
|
|
suite.chainA.App.IBCKeeper.ClientKeeper.SetClientState(suite.chainA.GetContext(), exported.Localhost, localHostClient)
|
|
}
|
|
|
|
func TestClientTestSuite(t *testing.T) {
|
|
suite.Run(t, new(ClientTestSuite))
|
|
}
|
|
|
|
func (suite *ClientTestSuite) TestBeginBlocker() {
|
|
prevHeight := types.GetSelfHeight(suite.chainA.GetContext())
|
|
|
|
localHostClient := suite.chainA.GetClientState(exported.Localhost)
|
|
suite.Require().Equal(prevHeight, localHostClient.GetLatestHeight())
|
|
|
|
for i := 0; i < 10; i++ {
|
|
// increment height
|
|
suite.coordinator.CommitBlock(suite.chainA, suite.chainB)
|
|
|
|
suite.Require().NotPanics(func() {
|
|
client.BeginBlocker(suite.chainA.GetContext(), suite.chainA.App.IBCKeeper.ClientKeeper)
|
|
}, "BeginBlocker shouldn't panic")
|
|
|
|
localHostClient = suite.chainA.GetClientState(exported.Localhost)
|
|
suite.Require().Equal(prevHeight.Increment(), localHostClient.GetLatestHeight())
|
|
prevHeight = localHostClient.GetLatestHeight().(types.Height)
|
|
}
|
|
}
|