18 lines
515 B
Go
18 lines
515 B
Go
package wasmtesting
|
|
|
|
import (
|
|
wasmvmtypes "github.com/CosmWasm/wasmvm/types"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
)
|
|
|
|
type MockQueryHandler struct {
|
|
HandleQueryFn func(ctx sdk.Context, request wasmvmtypes.QueryRequest, caller sdk.AccAddress) ([]byte, error)
|
|
}
|
|
|
|
func (m *MockQueryHandler) HandleQuery(ctx sdk.Context, caller sdk.AccAddress, request wasmvmtypes.QueryRequest) ([]byte, error) {
|
|
if m.HandleQueryFn == nil {
|
|
panic("not expected to be called")
|
|
}
|
|
return m.HandleQueryFn(ctx, request, caller)
|
|
}
|