zolfa
/
tendermint


								package mock_test


								import (

									"fmt"

									"testing"


									"github.com/pkg/errors"


									"github.com/stretchr/testify/assert"

									"github.com/stretchr/testify/require"


									"github.com/tendermint/abci/example/dummy"

									abci "github.com/tendermint/abci/types"

									data "github.com/tendermint/go-wire/data"

									"github.com/tendermint/tendermint/rpc/client/mock"

									ctypes "github.com/tendermint/tendermint/rpc/core/types"

									"github.com/tendermint/tendermint/types"

								)


								func TestABCIMock(t *testing.T) {

									assert, require := assert.New(t), require.New(t)


									key, value := []byte("foo"), []byte("bar")

									height := uint64(10)

									goodTx := types.Tx{0x01, 0xff}

									badTx := types.Tx{0x12, 0x21}


									m := mock.ABCIMock{

										Info: mock.Call{Error: errors.New("foobar")},

										Query: mock.Call{Response: abci.ResponseQuery{

											Key:    key,

											Value:  value,

											Height: height,

										}},

										// Broadcast commit depends on call

										BroadcastCommit: mock.Call{

											Args: goodTx,

											Response: &ctypes.ResultBroadcastTxCommit{

												CheckTx:   abci.Result{Data: data.Bytes("stand")},

												DeliverTx: abci.Result{Data: data.Bytes("deliver")},

											},

											Error: errors.New("bad tx"),

										},

										Broadcast: mock.Call{Error: errors.New("must commit")},

									}


									// now, let's try to make some calls

									_, err := m.ABCIInfo()

									require.NotNil(err)

									assert.Equal("foobar", err.Error())


									// query always returns the response

									query, err := m.ABCIQuery("/", nil, false)

									require.Nil(err)

									require.NotNil(query)

									assert.EqualValues(key, query.Key)

									assert.EqualValues(value, query.Value)

									assert.Equal(height, query.Height)


									// non-commit calls always return errors

									_, err = m.BroadcastTxSync(goodTx)

									require.NotNil(err)

									assert.Equal("must commit", err.Error())

									_, err = m.BroadcastTxAsync(goodTx)

									require.NotNil(err)

									assert.Equal("must commit", err.Error())


									// commit depends on the input

									_, err = m.BroadcastTxCommit(badTx)

									require.NotNil(err)

									assert.Equal("bad tx", err.Error())

									bres, err := m.BroadcastTxCommit(goodTx)

									require.Nil(err, "%+v", err)

									assert.EqualValues(0, bres.CheckTx.Code)

									assert.EqualValues("stand", bres.CheckTx.Data)

									assert.EqualValues("deliver", bres.DeliverTx.Data)

								}


								func TestABCIRecorder(t *testing.T) {

									assert, require := assert.New(t), require.New(t)

									m := mock.ABCIMock{

										Info: mock.Call{Response: abci.ResponseInfo{

											Data:    "data",

											Version: "v0.9.9",

										}},

										Query:           mock.Call{Error: errors.New("query")},

										Broadcast:       mock.Call{Error: errors.New("broadcast")},

										BroadcastCommit: mock.Call{Error: errors.New("broadcast_commit")},

									}

									r := mock.NewABCIRecorder(m)


									require.Equal(0, len(r.Calls))


									r.ABCIInfo()

									r.ABCIQuery("path", data.Bytes("data"), true)

									require.Equal(2, len(r.Calls))


									info := r.Calls[0]

									assert.Equal("abci_info", info.Name)

									assert.Nil(info.Error)

									assert.Nil(info.Args)

									require.NotNil(info.Response)

									ir, ok := info.Response.(*ctypes.ResultABCIInfo)

									require.True(ok)

									assert.Equal("data", ir.Response.Data)

									assert.Equal("v0.9.9", ir.Response.Version)


									query := r.Calls[1]

									assert.Equal("abci_query", query.Name)

									assert.Nil(query.Response)

									require.NotNil(query.Error)

									assert.Equal("query", query.Error.Error())

									require.NotNil(query.Args)

									qa, ok := query.Args.(mock.QueryArgs)

									require.True(ok)

									assert.Equal("path", qa.Path)

									assert.EqualValues("data", qa.Data)

									assert.True(qa.Prove)


									// now add some broadcasts

									txs := []types.Tx{{1}, {2}, {3}}

									r.BroadcastTxCommit(txs[0])

									r.BroadcastTxSync(txs[1])

									r.BroadcastTxAsync(txs[2])


									require.Equal(5, len(r.Calls))


									bc := r.Calls[2]

									assert.Equal("broadcast_tx_commit", bc.Name)

									assert.Nil(bc.Response)

									require.NotNil(bc.Error)

									assert.EqualValues(bc.Args, txs[0])


									bs := r.Calls[3]

									assert.Equal("broadcast_tx_sync", bs.Name)

									assert.Nil(bs.Response)

									require.NotNil(bs.Error)

									assert.EqualValues(bs.Args, txs[1])


									ba := r.Calls[4]

									assert.Equal("broadcast_tx_async", ba.Name)

									assert.Nil(ba.Response)

									require.NotNil(ba.Error)

									assert.EqualValues(ba.Args, txs[2])

								}


								func TestABCIApp(t *testing.T) {

									assert, require := assert.New(t), require.New(t)

									app := dummy.NewDummyApplication()

									m := mock.ABCIApp{app}


									// get some info

									info, err := m.ABCIInfo()

									require.Nil(err)

									assert.Equal(`{"size":0}`, info.Response.GetData())


									// add a key

									key, value := "foo", "bar"

									tx := fmt.Sprintf("%s=%s", key, value)

									res, err := m.BroadcastTxCommit(types.Tx(tx))

									require.Nil(err)

									assert.True(res.CheckTx.Code.IsOK())

									require.NotNil(res.DeliverTx)

									assert.True(res.DeliverTx.Code.IsOK())


									// check the key

									qres, err := m.ABCIQuery("/key", data.Bytes(key), false)

									require.Nil(err)

									assert.EqualValues(value, qres.Value)

								}