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.
203 lines
4.7 KiB
203 lines
4.7 KiB
package dummy
|
|
|
|
import (
|
|
"bytes"
|
|
"io/ioutil"
|
|
"sort"
|
|
"testing"
|
|
|
|
. "github.com/tendermint/go-common"
|
|
"github.com/tendermint/go-crypto"
|
|
"github.com/tendermint/go-wire"
|
|
"github.com/tendermint/tmsp/types"
|
|
)
|
|
|
|
func testDummy(t *testing.T, dummy types.Application, tx []byte, key, value string) {
|
|
if r := dummy.DeliverTx(tx); r.IsErr() {
|
|
t.Fatal(r)
|
|
}
|
|
if r := dummy.DeliverTx(tx); r.IsErr() {
|
|
t.Fatal(r)
|
|
}
|
|
|
|
r := dummy.Query([]byte(key))
|
|
if r.IsErr() {
|
|
t.Fatal(r)
|
|
}
|
|
|
|
q := new(QueryResult)
|
|
if err := wire.ReadJSONBytes(r.Data, q); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if q.Value != value {
|
|
t.Fatalf("Got %s, expected %s", q.Value, value)
|
|
}
|
|
|
|
}
|
|
|
|
func TestDummyKV(t *testing.T) {
|
|
dummy := NewDummyApplication()
|
|
key := "abc"
|
|
value := key
|
|
tx := []byte(key)
|
|
testDummy(t, dummy, tx, key, value)
|
|
|
|
value = "def"
|
|
tx = []byte(key + "=" + value)
|
|
testDummy(t, dummy, tx, key, value)
|
|
}
|
|
|
|
func TestPersistentDummyKV(t *testing.T) {
|
|
dir, err := ioutil.TempDir("/tmp", "tmsp-dummy-test") // TODO
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
dummy := NewPersistentDummyApplication(dir)
|
|
key := "abc"
|
|
value := key
|
|
tx := []byte(key)
|
|
testDummy(t, dummy, tx, key, value)
|
|
|
|
value = "def"
|
|
tx = []byte(key + "=" + value)
|
|
testDummy(t, dummy, tx, key, value)
|
|
}
|
|
|
|
func TestPersistentDummyInfo(t *testing.T) {
|
|
dir, err := ioutil.TempDir("/tmp", "tmsp-dummy-test") // TODO
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
dummy := NewPersistentDummyApplication(dir)
|
|
height := uint64(0)
|
|
|
|
resInfo := dummy.Info()
|
|
if resInfo.LastBlockHeight != height {
|
|
t.Fatalf("expected height of %d, got %d", height, resInfo.LastBlockHeight)
|
|
}
|
|
|
|
// make and apply block
|
|
height = uint64(1)
|
|
hash := []byte("foo")
|
|
header := &types.Header{
|
|
Height: uint64(height),
|
|
}
|
|
dummy.BeginBlock(hash, header)
|
|
dummy.EndBlock(height)
|
|
dummy.Commit()
|
|
|
|
resInfo = dummy.Info()
|
|
if resInfo.LastBlockHeight != height {
|
|
t.Fatalf("expected height of %d, got %d", height, resInfo.LastBlockHeight)
|
|
}
|
|
|
|
}
|
|
|
|
// add a validator, remove a validator, update a validator
|
|
func TestValSetChanges(t *testing.T) {
|
|
dir, err := ioutil.TempDir("/tmp", "tmsp-dummy-test") // TODO
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
dummy := NewPersistentDummyApplication(dir)
|
|
|
|
// init with some validators
|
|
total := 10
|
|
nInit := 5
|
|
vals := make([]*types.Validator, total)
|
|
for i := 0; i < total; i++ {
|
|
pubkey := crypto.GenPrivKeyEd25519FromSecret([]byte(Fmt("test%d", i))).PubKey().Bytes()
|
|
power := RandInt()
|
|
vals[i] = &types.Validator{pubkey, uint64(power)}
|
|
}
|
|
// iniitalize with the first nInit
|
|
dummy.InitChain(vals[:nInit])
|
|
|
|
vals1, vals2 := vals[:nInit], dummy.Validators()
|
|
valsEqual(t, vals1, vals2)
|
|
|
|
var v1, v2, v3 *types.Validator
|
|
|
|
// add some validators
|
|
v1, v2 = vals[nInit], vals[nInit+1]
|
|
diff := []*types.Validator{v1, v2}
|
|
tx1 := MakeValSetChangeTx(v1.PubKey, v1.Power)
|
|
tx2 := MakeValSetChangeTx(v2.PubKey, v2.Power)
|
|
|
|
makeApplyBlock(t, dummy, 1, diff, tx1, tx2)
|
|
|
|
vals1, vals2 = vals[:nInit+2], dummy.Validators()
|
|
valsEqual(t, vals1, vals2)
|
|
|
|
// remove some validators
|
|
v1, v2, v3 = vals[nInit-2], vals[nInit-1], vals[nInit]
|
|
v1.Power = 0
|
|
v2.Power = 0
|
|
v3.Power = 0
|
|
diff = []*types.Validator{v1, v2, v3}
|
|
tx1 = MakeValSetChangeTx(v1.PubKey, v1.Power)
|
|
tx2 = MakeValSetChangeTx(v2.PubKey, v2.Power)
|
|
tx3 := MakeValSetChangeTx(v3.PubKey, v3.Power)
|
|
|
|
makeApplyBlock(t, dummy, 2, diff, tx1, tx2, tx3)
|
|
|
|
vals1 = append(vals[:nInit-2], vals[nInit+1])
|
|
vals2 = dummy.Validators()
|
|
valsEqual(t, vals1, vals2)
|
|
|
|
// update some validators
|
|
v1 = vals[0]
|
|
if v1.Power == 5 {
|
|
v1.Power = 6
|
|
} else {
|
|
v1.Power = 5
|
|
}
|
|
diff = []*types.Validator{v1}
|
|
tx1 = MakeValSetChangeTx(v1.PubKey, v1.Power)
|
|
|
|
makeApplyBlock(t, dummy, 3, diff, tx1)
|
|
|
|
vals1 = append([]*types.Validator{v1}, vals1[1:len(vals1)]...)
|
|
vals2 = dummy.Validators()
|
|
valsEqual(t, vals1, vals2)
|
|
|
|
}
|
|
|
|
func makeApplyBlock(t *testing.T, dummy types.Application, heightInt int, diff []*types.Validator, txs ...[]byte) {
|
|
// make and apply block
|
|
height := uint64(heightInt)
|
|
hash := []byte("foo")
|
|
header := &types.Header{
|
|
Height: height,
|
|
}
|
|
|
|
dummyChain := dummy.(types.BlockchainAware) // hmm...
|
|
dummyChain.BeginBlock(hash, header)
|
|
for _, tx := range txs {
|
|
if r := dummy.DeliverTx(tx); r.IsErr() {
|
|
t.Fatal(r)
|
|
}
|
|
}
|
|
resEndBlock := dummyChain.EndBlock(height)
|
|
dummy.Commit()
|
|
|
|
valsEqual(t, diff, resEndBlock.Diffs)
|
|
|
|
}
|
|
|
|
// order doesn't matter
|
|
func valsEqual(t *testing.T, vals1, vals2 []*types.Validator) {
|
|
if len(vals1) != len(vals2) {
|
|
t.Fatalf("vals dont match in len. got %d, expected %d", len(vals2), len(vals1))
|
|
}
|
|
sort.Sort(types.Validators(vals1))
|
|
sort.Sort(types.Validators(vals2))
|
|
for i, v1 := range vals1 {
|
|
v2 := vals2[i]
|
|
if !bytes.Equal(v1.PubKey, v2.PubKey) ||
|
|
v1.Power != v2.Power {
|
|
t.Fatalf("vals dont match at index %d. got %X/%d , expected %X/%d", i, v2.PubKey, v2.Power, v1.PubKey, v1.Power)
|
|
}
|
|
}
|
|
}
|