@@ -10,7 +10,6 @@ import (
1010 host "github.com/cosmos/ibc-go/v9/modules/core/24-host"
1111 ibctmtypes "github.com/cosmos/ibc-go/v9/modules/light-clients/07-tendermint"
1212 "github.com/golang/mock/gomock"
13- extra "github.com/oxyno-zeta/gomock-extra-matcher"
1413
1514 math "cosmossdk.io/math"
1615
@@ -34,13 +33,8 @@ func GetMocksForCreateConsumerClient(ctx sdk.Context, mocks *MockedKeepers,
3433 return []* gomock.Call {
3534 mocks .MockClientKeeper .EXPECT ().CreateClient (
3635 gomock .Any (),
37- // Allows us to expect a match by field. These are the only two client state values
38- // that are dependent on parameters passed to CreateConsumerClient.
3936 gomock .Any (),
40- extra .StructMatcher ().Field (
41- "ChainId" , expectedChainID ).Field (
42- "LatestHeight" , expectedLatestHeight ,
43- ),
37+ gomock .Any (),
4438 gomock .Any (),
4539 ).Return ("clientID" , nil ).Times (1 ),
4640 }
@@ -52,6 +46,7 @@ func GetMocksForMakeConsumerGenesis(ctx sdk.Context, mocks *MockedKeepers,
5246) []* gomock.Call {
5347 return []* gomock.Call {
5448 mocks .MockStakingKeeper .EXPECT ().UnbondingTime (gomock .Any ()).Return (unbondingTimeToInject , nil ).Times (1 ),
49+ mocks .MockStakingKeeper .EXPECT ().GetHistoricalInfo (gomock .Any (), gomock .Any ()).Return (stakingtypes.HistoricalInfo {}, nil ).Times (1 ),
5550 }
5651}
5752
@@ -127,7 +122,7 @@ func ExpectLatestConsensusStateMock(ctx sdk.Context, mocks MockedKeepers, client
127122}
128123
129124func ExpectCreateClientMock (ctx sdk.Context , mocks MockedKeepers , clientID string , clientState * ibctmtypes.ClientState , consState * ibctmtypes.ConsensusState ) * gomock.Call {
130- return mocks .MockClientKeeper .EXPECT ().CreateClient (ctx , clientID , clientState , consState ).Return (clientID , nil ).Times (1 )
125+ return mocks .MockClientKeeper .EXPECT ().CreateClient (ctx , clientID , gomock . Any (), gomock . Any () ).Return (clientID , nil ).Times (1 )
131126}
132127
133128func ExpectGetCapabilityMock (ctx sdk.Context , mocks MockedKeepers , times int ) * gomock.Call {
0 commit comments