You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
202 lines
5.7 KiB
202 lines
5.7 KiB
package mock
|
|
|
|
import (
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
|
cmn "github.com/tendermint/tendermint/libs/common"
|
|
"github.com/tendermint/tendermint/rpc/client"
|
|
ctypes "github.com/tendermint/tendermint/rpc/core/types"
|
|
"github.com/tendermint/tendermint/types"
|
|
"github.com/tendermint/tendermint/version"
|
|
)
|
|
|
|
// ABCIApp will send all abci related request to the named app,
|
|
// so you can test app behavior from a client without needing
|
|
// an entire tendermint node
|
|
type ABCIApp struct {
|
|
App abci.Application
|
|
}
|
|
|
|
var (
|
|
_ client.ABCIClient = ABCIApp{}
|
|
_ client.ABCIClient = ABCIMock{}
|
|
_ client.ABCIClient = (*ABCIRecorder)(nil)
|
|
)
|
|
|
|
func (a ABCIApp) ABCIInfo() (*ctypes.ResultABCIInfo, error) {
|
|
return &ctypes.ResultABCIInfo{a.App.Info(abci.RequestInfo{Version: version.Version})}, nil
|
|
}
|
|
|
|
func (a ABCIApp) ABCIQuery(path string, data cmn.HexBytes) (*ctypes.ResultABCIQuery, error) {
|
|
return a.ABCIQueryWithOptions(path, data, client.DefaultABCIQueryOptions)
|
|
}
|
|
|
|
func (a ABCIApp) ABCIQueryWithOptions(path string, data cmn.HexBytes, opts client.ABCIQueryOptions) (*ctypes.ResultABCIQuery, error) {
|
|
q := a.App.Query(abci.RequestQuery{Data: data, Path: path, Height: opts.Height, Prove: opts.Trusted})
|
|
return &ctypes.ResultABCIQuery{q}, nil
|
|
}
|
|
|
|
func (a ABCIApp) BroadcastTxCommit(tx types.Tx) (*ctypes.ResultBroadcastTxCommit, error) {
|
|
res := ctypes.ResultBroadcastTxCommit{}
|
|
res.CheckTx = a.App.CheckTx(tx)
|
|
if res.CheckTx.IsErr() {
|
|
return &res, nil
|
|
}
|
|
res.DeliverTx = a.App.DeliverTx(tx)
|
|
return &res, nil
|
|
}
|
|
|
|
func (a ABCIApp) BroadcastTxAsync(tx types.Tx) (*ctypes.ResultBroadcastTx, error) {
|
|
c := a.App.CheckTx(tx)
|
|
// and this gets written in a background thread...
|
|
if !c.IsErr() {
|
|
go func() { a.App.DeliverTx(tx) }() // nolint: errcheck
|
|
}
|
|
return &ctypes.ResultBroadcastTx{c.Code, c.Data, c.Log, tx.Hash()}, nil
|
|
}
|
|
|
|
func (a ABCIApp) BroadcastTxSync(tx types.Tx) (*ctypes.ResultBroadcastTx, error) {
|
|
c := a.App.CheckTx(tx)
|
|
// and this gets written in a background thread...
|
|
if !c.IsErr() {
|
|
go func() { a.App.DeliverTx(tx) }() // nolint: errcheck
|
|
}
|
|
return &ctypes.ResultBroadcastTx{c.Code, c.Data, c.Log, tx.Hash()}, nil
|
|
}
|
|
|
|
// ABCIMock will send all abci related request to the named app,
|
|
// so you can test app behavior from a client without needing
|
|
// an entire tendermint node
|
|
type ABCIMock struct {
|
|
Info Call
|
|
Query Call
|
|
BroadcastCommit Call
|
|
Broadcast Call
|
|
}
|
|
|
|
func (m ABCIMock) ABCIInfo() (*ctypes.ResultABCIInfo, error) {
|
|
res, err := m.Info.GetResponse(nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &ctypes.ResultABCIInfo{res.(abci.ResponseInfo)}, nil
|
|
}
|
|
|
|
func (m ABCIMock) ABCIQuery(path string, data cmn.HexBytes) (*ctypes.ResultABCIQuery, error) {
|
|
return m.ABCIQueryWithOptions(path, data, client.DefaultABCIQueryOptions)
|
|
}
|
|
|
|
func (m ABCIMock) ABCIQueryWithOptions(path string, data cmn.HexBytes, opts client.ABCIQueryOptions) (*ctypes.ResultABCIQuery, error) {
|
|
res, err := m.Query.GetResponse(QueryArgs{path, data, opts.Height, opts.Trusted})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
resQuery := res.(abci.ResponseQuery)
|
|
return &ctypes.ResultABCIQuery{resQuery}, nil
|
|
}
|
|
|
|
func (m ABCIMock) BroadcastTxCommit(tx types.Tx) (*ctypes.ResultBroadcastTxCommit, error) {
|
|
res, err := m.BroadcastCommit.GetResponse(tx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return res.(*ctypes.ResultBroadcastTxCommit), nil
|
|
}
|
|
|
|
func (m ABCIMock) BroadcastTxAsync(tx types.Tx) (*ctypes.ResultBroadcastTx, error) {
|
|
res, err := m.Broadcast.GetResponse(tx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return res.(*ctypes.ResultBroadcastTx), nil
|
|
}
|
|
|
|
func (m ABCIMock) BroadcastTxSync(tx types.Tx) (*ctypes.ResultBroadcastTx, error) {
|
|
res, err := m.Broadcast.GetResponse(tx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return res.(*ctypes.ResultBroadcastTx), nil
|
|
}
|
|
|
|
// ABCIRecorder can wrap another type (ABCIApp, ABCIMock, or Client)
|
|
// and record all ABCI related calls.
|
|
type ABCIRecorder struct {
|
|
Client client.ABCIClient
|
|
Calls []Call
|
|
}
|
|
|
|
func NewABCIRecorder(client client.ABCIClient) *ABCIRecorder {
|
|
return &ABCIRecorder{
|
|
Client: client,
|
|
Calls: []Call{},
|
|
}
|
|
}
|
|
|
|
type QueryArgs struct {
|
|
Path string
|
|
Data cmn.HexBytes
|
|
Height int64
|
|
Trusted bool
|
|
}
|
|
|
|
func (r *ABCIRecorder) addCall(call Call) {
|
|
r.Calls = append(r.Calls, call)
|
|
}
|
|
|
|
func (r *ABCIRecorder) ABCIInfo() (*ctypes.ResultABCIInfo, error) {
|
|
res, err := r.Client.ABCIInfo()
|
|
r.addCall(Call{
|
|
Name: "abci_info",
|
|
Response: res,
|
|
Error: err,
|
|
})
|
|
return res, err
|
|
}
|
|
|
|
func (r *ABCIRecorder) ABCIQuery(path string, data cmn.HexBytes) (*ctypes.ResultABCIQuery, error) {
|
|
return r.ABCIQueryWithOptions(path, data, client.DefaultABCIQueryOptions)
|
|
}
|
|
|
|
func (r *ABCIRecorder) ABCIQueryWithOptions(path string, data cmn.HexBytes, opts client.ABCIQueryOptions) (*ctypes.ResultABCIQuery, error) {
|
|
res, err := r.Client.ABCIQueryWithOptions(path, data, opts)
|
|
r.addCall(Call{
|
|
Name: "abci_query",
|
|
Args: QueryArgs{path, data, opts.Height, opts.Trusted},
|
|
Response: res,
|
|
Error: err,
|
|
})
|
|
return res, err
|
|
}
|
|
|
|
func (r *ABCIRecorder) BroadcastTxCommit(tx types.Tx) (*ctypes.ResultBroadcastTxCommit, error) {
|
|
res, err := r.Client.BroadcastTxCommit(tx)
|
|
r.addCall(Call{
|
|
Name: "broadcast_tx_commit",
|
|
Args: tx,
|
|
Response: res,
|
|
Error: err,
|
|
})
|
|
return res, err
|
|
}
|
|
|
|
func (r *ABCIRecorder) BroadcastTxAsync(tx types.Tx) (*ctypes.ResultBroadcastTx, error) {
|
|
res, err := r.Client.BroadcastTxAsync(tx)
|
|
r.addCall(Call{
|
|
Name: "broadcast_tx_async",
|
|
Args: tx,
|
|
Response: res,
|
|
Error: err,
|
|
})
|
|
return res, err
|
|
}
|
|
|
|
func (r *ABCIRecorder) BroadcastTxSync(tx types.Tx) (*ctypes.ResultBroadcastTx, error) {
|
|
res, err := r.Client.BroadcastTxSync(tx)
|
|
r.addCall(Call{
|
|
Name: "broadcast_tx_sync",
|
|
Args: tx,
|
|
Response: res,
|
|
Error: err,
|
|
})
|
|
return res, err
|
|
}
|