105 lines
3.6 KiB
Go
105 lines
3.6 KiB
Go
package counter
|
||
|
||
import (
|
||
"encoding/json"
|
||
"testing"
|
||
|
||
"github.com/stretchr/testify/assert"
|
||
"github.com/stretchr/testify/require"
|
||
abci "github.com/tendermint/abci/types"
|
||
"github.com/tendermint/basecoin/app"
|
||
"github.com/tendermint/basecoin/types"
|
||
crypto "github.com/tendermint/go-crypto"
|
||
"github.com/tendermint/go-wire"
|
||
eyescli "github.com/tendermint/merkleeyes/client"
|
||
)
|
||
|
||
func TestCounterPlugin(t *testing.T) {
|
||
assert := assert.New(t)
|
||
|
||
// Basecoin initialization
|
||
eyesCli := eyescli.NewLocalClient("", 0)
|
||
chainID := "test_chain_id"
|
||
bcApp := app.NewBasecoin(eyesCli)
|
||
bcApp.SetOption("base/chain_id", chainID)
|
||
// t.Log(bcApp.Info())
|
||
|
||
// Add Counter plugin
|
||
counterPlugin := New()
|
||
bcApp.RegisterPlugin(counterPlugin)
|
||
|
||
// Account initialization
|
||
test1PrivAcc := types.PrivAccountFromSecret("test1")
|
||
|
||
// Seed Basecoin with account
|
||
test1Acc := test1PrivAcc.Account
|
||
test1Acc.Balance = types.Coins{{"", 1000}, {"gold", 1000}}
|
||
accOpt, err := json.Marshal(test1Acc)
|
||
require.Nil(t, err)
|
||
bcApp.SetOption("base/account", string(accOpt))
|
||
|
||
// Deliver a CounterTx
|
||
DeliverCounterTx := func(gas int64, fee types.Coin, inputCoins types.Coins, inputSequence int, appFee types.Coins) abci.Result {
|
||
// Construct an AppTx signature
|
||
tx := &types.AppTx{
|
||
Gas: gas,
|
||
Fee: fee,
|
||
Name: counterPlugin.Name(),
|
||
Input: types.NewTxInput(test1Acc.PubKey, inputCoins, inputSequence),
|
||
Data: wire.BinaryBytes(CounterTx{Valid: true, Fee: appFee}),
|
||
}
|
||
|
||
// Sign request
|
||
signBytes := tx.SignBytes(chainID)
|
||
// t.Logf("Sign bytes: %X\n", signBytes)
|
||
sig := test1PrivAcc.Sign(signBytes)
|
||
tx.Input.Signature = crypto.SignatureS{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, appFee) {
|
||
|
||
// Test a basic send, no fee
|
||
res := DeliverCounterTx(0, types.Coin{}, types.Coins{{"", 1}}, 1, types.Coins{})
|
||
assert.True(res.IsOK(), res.String())
|
||
|
||
// Test fee prevented transaction
|
||
res = DeliverCounterTx(0, types.Coin{"", 2}, types.Coins{{"", 1}}, 2, types.Coins{})
|
||
assert.True(res.IsErr(), res.String())
|
||
|
||
// Test input equals fee
|
||
res = DeliverCounterTx(0, types.Coin{"", 2}, types.Coins{{"", 2}}, 2, types.Coins{})
|
||
assert.True(res.IsOK(), res.String())
|
||
|
||
// Test more input than fee
|
||
res = DeliverCounterTx(0, types.Coin{"", 2}, types.Coins{{"", 3}}, 3, types.Coins{})
|
||
assert.True(res.IsOK(), res.String())
|
||
|
||
// Test input equals fee+appFee
|
||
res = DeliverCounterTx(0, types.Coin{"", 1}, types.Coins{{"", 3}, {"gold", 1}}, 4, types.Coins{{"", 2}, {"gold", 1}})
|
||
assert.True(res.IsOK(), res.String())
|
||
|
||
// Test fee+appFee prevented transaction, not enough ""
|
||
res = DeliverCounterTx(0, types.Coin{"", 1}, types.Coins{{"", 2}, {"gold", 1}}, 5, types.Coins{{"", 2}, {"gold", 1}})
|
||
assert.True(res.IsErr(), res.String())
|
||
|
||
// Test fee+appFee prevented transaction, not enough "gold"
|
||
res = DeliverCounterTx(0, types.Coin{"", 1}, types.Coins{{"", 3}, {"gold", 1}}, 5, types.Coins{{"", 2}, {"gold", 2}})
|
||
assert.True(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(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(res.IsOK(), res.String())
|
||
|
||
// REF: DeliverCounterTx(gas, fee, inputCoins, inputSequence, appFee) {w
|
||
|
||
}
|