mirror of https://github.com/certusone/wasmd.git
28 lines
714 B
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)
|
|
}
|