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) }