github.com/fibonacci-chain/fbc@v0.0.0-20231124064014-c7636198c1e9/x/wasm/keeper/wasmtesting/message_router.go (about)

     1  package wasmtesting
     2  
     3  import (
     4  	"github.com/fibonacci-chain/fbc/libs/cosmos-sdk/baseapp"
     5  	sdk "github.com/fibonacci-chain/fbc/libs/cosmos-sdk/types"
     6  )
     7  
     8  // MockMessageRouter mock for testing
     9  type MockMessageRouter struct {
    10  	HandlerFn func(msg sdk.Msg) baseapp.MsgServiceHandler
    11  }
    12  
    13  // Handler is the entry point
    14  func (m MockMessageRouter) Handler(msg sdk.Msg) baseapp.MsgServiceHandler {
    15  	if m.HandlerFn == nil {
    16  		panic("not expected to be called")
    17  	}
    18  	return m.HandlerFn(msg)
    19  }
    20  
    21  // MessageRouterFunc convenient type to match the keeper.MessageRouter interface
    22  type MessageRouterFunc func(methodName string) baseapp.MsgServiceHandler
    23  
    24  // Handler is the entry point
    25  func (m MessageRouterFunc) Handler(methodName string) baseapp.MsgServiceHandler {
    26  	return m(methodName)
    27  }