diff --git a/x/supplier/types/expected_keepers.go b/x/supplier/types/expected_keepers.go index d87cf9d1d..d4eae9871 100644 --- a/x/supplier/types/expected_keepers.go +++ b/x/supplier/types/expected_keepers.go @@ -1,13 +1,17 @@ package types +//go:generate mockgen -destination ../../../testutil/supplier/mocks/expected_keepers_mock.go -package mocks . AccountKeeper,BankKeeper,SessionKeeper + import ( "context" sdk "github.com/cosmos/cosmos-sdk/types" + + sessiontypes "github.com/pokt-network/poktroll/x/session/types" ) type SessionKeeper interface { - // TODO Add methods imported from session should be defined here + GetSession(context.Context, *sessiontypes.QueryGetSessionRequest) (*sessiontypes.QueryGetSessionResponse, error) } // AccountKeeper defines the expected interface for the Account module. @@ -18,7 +22,8 @@ type AccountKeeper interface { // BankKeeper defines the expected interface for the Bank module. type BankKeeper interface { - SpendableCoins(context.Context, sdk.AccAddress) sdk.Coins + DelegateCoinsFromAccountToModule(ctx sdk.Context, senderAddr sdk.AccAddress, recipientModule string, amt sdk.Coins) error + UndelegateCoinsFromModuleToAccount(ctx sdk.Context, senderModule string, recipientAddr sdk.AccAddress, amt sdk.Coins) error // Methods imported from bank should be defined here }