100 lines
3.5 KiB
Go
100 lines
3.5 KiB
Go
package counter
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
abci "github.com/tendermint/abci/types"
|
|
"github.com/tendermint/basecoin/app"
|
|
"github.com/tendermint/basecoin/testutils"
|
|
"github.com/tendermint/basecoin/types"
|
|
"github.com/tendermint/go-wire"
|
|
eyescli "github.com/tendermint/merkleeyes/client"
|
|
)
|
|
|
|
func TestCounterPlugin(t *testing.T) {
|
|
|
|
// Basecoin initialization
|
|
eyesCli := eyescli.NewLocalClient()
|
|
chainID := "test_chain_id"
|
|
bcApp := app.NewBasecoin(eyesCli)
|
|
bcApp.SetOption("base/chainID", chainID)
|
|
t.Log(bcApp.Info())
|
|
|
|
// Add Counter plugin
|
|
counterPluginName := "testcounter"
|
|
counterPlugin := NewCounterPlugin(counterPluginName)
|
|
bcApp.RegisterPlugin(counterPlugin)
|
|
|
|
// Account initialization
|
|
test1PrivAcc := testutils.PrivAccountFromSecret("test1")
|
|
|
|
// Seed Basecoin with account
|
|
test1Acc := test1PrivAcc.Account
|
|
test1Acc.Balance = types.Coins{{"", 1000}, {"gold", 1000}}
|
|
bcApp.SetOption("base/account", string(wire.JSONBytes(test1Acc)))
|
|
|
|
// Deliver a CounterTx
|
|
DeliverCounterTx := func(gas int64, fee types.Coin, inputCoins types.Coins, inputSequence int, cost types.Coins) abci.Result {
|
|
// Construct an AppTx signature
|
|
tx := &types.AppTx{
|
|
Gas: gas,
|
|
Fee: fee,
|
|
Name: counterPluginName,
|
|
Input: types.NewTxInput(test1Acc.PubKey, inputCoins, inputSequence),
|
|
Data: wire.BinaryBytes(CounterTx{Valid: true, Cost: cost}),
|
|
}
|
|
|
|
// Sign request
|
|
signBytes := tx.SignBytes(chainID)
|
|
t.Logf("Sign bytes: %X\n", signBytes)
|
|
sig := test1PrivAcc.PrivKey.Sign(signBytes)
|
|
tx.Input.Signature = sig
|
|
t.Logf("Signed TX bytes: %X\n", wire.BinaryBytes(struct{ types.Tx }{tx}))
|
|
|
|
// Write request
|
|
txBytes := wire.BinaryBytes(struct{ types.Tx }{tx})
|
|
return bcApp.DeliverTx(txBytes)
|
|
}
|
|
|
|
// REF: DeliverCounterTx(gas, fee, inputCoins, inputSequence, cost) {
|
|
|
|
// Test a basic send, no fee
|
|
res := DeliverCounterTx(0, types.Coin{}, types.Coins{{"", 1}}, 1, types.Coins{})
|
|
assert.True(t, res.IsOK(), res.String())
|
|
|
|
// Test fee prevented transaction
|
|
res = DeliverCounterTx(0, types.Coin{"", 2}, types.Coins{{"", 1}}, 2, types.Coins{})
|
|
assert.True(t, res.IsErr(), res.String())
|
|
|
|
// Test input equals fee
|
|
res = DeliverCounterTx(0, types.Coin{"", 2}, types.Coins{{"", 2}}, 2, types.Coins{})
|
|
assert.True(t, res.IsOK(), res.String())
|
|
|
|
// Test more input than fee
|
|
res = DeliverCounterTx(0, types.Coin{"", 2}, types.Coins{{"", 3}}, 3, types.Coins{})
|
|
assert.True(t, res.IsOK(), res.String())
|
|
|
|
// Test input equals fee+cost
|
|
res = DeliverCounterTx(0, types.Coin{"", 1}, types.Coins{{"", 3}, {"gold", 1}}, 4, types.Coins{{"", 2}, {"gold", 1}})
|
|
assert.True(t, res.IsOK(), res.String())
|
|
|
|
// Test fee+cost prevented transaction, not enough ""
|
|
res = DeliverCounterTx(0, types.Coin{"", 1}, types.Coins{{"", 2}, {"gold", 1}}, 5, types.Coins{{"", 2}, {"gold", 1}})
|
|
assert.True(t, res.IsErr(), res.String())
|
|
|
|
// Test fee+cost prevented transaction, not enough "gold"
|
|
res = DeliverCounterTx(0, types.Coin{"", 1}, types.Coins{{"", 3}, {"gold", 1}}, 5, types.Coins{{"", 2}, {"gold", 2}})
|
|
assert.True(t, res.IsErr(), res.String())
|
|
|
|
// Test more input than fee, more ""
|
|
res = DeliverCounterTx(0, types.Coin{"", 1}, types.Coins{{"", 4}, {"gold", 1}}, 6, types.Coins{{"", 2}, {"gold", 1}})
|
|
assert.True(t, res.IsOK(), res.String())
|
|
|
|
// Test more input than fee, more "gold"
|
|
res = DeliverCounterTx(0, types.Coin{"", 1}, types.Coins{{"", 3}, {"gold", 2}}, 7, types.Coins{{"", 2}, {"gold", 1}})
|
|
assert.True(t, res.IsOK(), res.String())
|
|
|
|
// REF: DeliverCounterTx(gas, fee, inputCoins, inputSequence, cost) {
|
|
}
|