wasmd/x/wasm/keeper/wasmtesting/message_router.go

28 lines
714 B
Go

package wasmtesting
import (
"github.com/cosmos/cosmos-sdk/baseapp"
sdk "github.com/cosmos/cosmos-sdk/types"
)
// MockMessageRouter mock for testing
type MockMessageRouter struct {
HandlerFn func(msg sdk.Msg) baseapp.MsgServiceHandler
}
// Handler is the entry point
func (m MockMessageRouter) Handler(msg sdk.Msg) baseapp.MsgServiceHandler {
if m.HandlerFn == nil {
panic("not expected to be called")
}
return m.HandlerFn(msg)
}
// MessageRouterFunc convenient type to match the keeper.MessageRouter interface
type MessageRouterFunc func(msg sdk.Msg) baseapp.MsgServiceHandler
// Handler is the entry point
func (m MessageRouterFunc) Handler(msg sdk.Msg) baseapp.MsgServiceHandler {
return m(msg)
}