|
|
- 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/abci/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", "abci-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", "abci-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", "abci-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)
- }
- }
- }
|