99 lines
2.5 KiB
Go
99 lines
2.5 KiB
Go
package state
|
|
|
|
import (
|
|
. "github.com/tendermint/tendermint/blocks"
|
|
. "github.com/tendermint/tendermint/common"
|
|
. "github.com/tendermint/tendermint/config"
|
|
. "github.com/tendermint/tendermint/db"
|
|
|
|
"bytes"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func randAccountBalance(id uint64, status byte) *AccountBalance {
|
|
return &AccountBalance{
|
|
Account: Account{
|
|
Id: id,
|
|
PubKey: CRandBytes(32),
|
|
},
|
|
Balance: RandUInt64(),
|
|
Status: status,
|
|
}
|
|
}
|
|
|
|
// The first numValidators accounts are validators.
|
|
func randGenesisState(numAccounts int, numValidators int) *State {
|
|
db := NewMemDB()
|
|
accountBalances := make([]*AccountBalance, numAccounts)
|
|
for i := 0; i < numAccounts; i++ {
|
|
if i < numValidators {
|
|
accountBalances[i] = randAccountBalance(uint64(i), AccountBalanceStatusNominal)
|
|
} else {
|
|
accountBalances[i] = randAccountBalance(uint64(i), AccountBalanceStatusBonded)
|
|
}
|
|
}
|
|
s0 := GenesisState(db, time.Now(), accountBalances)
|
|
return s0
|
|
}
|
|
|
|
func TestGenesisSaveLoad(t *testing.T) {
|
|
|
|
// Generate a state, save & load it.
|
|
s0 := randGenesisState(10, 5)
|
|
// Figure out what the next state hashes should be.
|
|
s0ValsCopy := s0.Validators.Copy()
|
|
s0ValsCopy.IncrementAccum()
|
|
nextValidationStateHash := s0ValsCopy.Hash()
|
|
nextAccountStateHash := s0.AccountBalances.Tree.Hash()
|
|
// Mutate the state to append one empty block.
|
|
block := &Block{
|
|
Header: Header{
|
|
Network: Config.Network,
|
|
Height: 1,
|
|
ValidationStateHash: nextValidationStateHash,
|
|
AccountStateHash: nextAccountStateHash,
|
|
},
|
|
Data: Data{
|
|
Txs: []Tx{},
|
|
},
|
|
}
|
|
err := s0.AppendBlock(block)
|
|
if err != nil {
|
|
t.Error("Error appending initial block:", err)
|
|
}
|
|
|
|
// Save s0, load s1.
|
|
commitTime := time.Now()
|
|
s0.Save(commitTime)
|
|
//s0.DB.(*MemDB).Print()
|
|
s1 := LoadState(s0.DB)
|
|
|
|
// Compare CommitTime
|
|
if commitTime.Unix() != s1.CommitTime.Unix() {
|
|
t.Error("CommitTime was not the same")
|
|
}
|
|
// Compare height & blockHash
|
|
if s0.Height != 1 {
|
|
t.Error("s0 Height should be 1, got", s0.Height)
|
|
}
|
|
if s0.Height != s1.Height {
|
|
t.Error("Height mismatch")
|
|
}
|
|
if !bytes.Equal(s0.BlockHash, s1.BlockHash) {
|
|
t.Error("BlockHash mismatch")
|
|
}
|
|
// Compare Validators
|
|
if s0.Validators.Size() != s1.Validators.Size() {
|
|
t.Error("Validators Size changed")
|
|
}
|
|
if s0.Validators.TotalVotingPower() == 0 {
|
|
t.Error("s0 Validators TotalVotingPower should not be 0")
|
|
}
|
|
if s0.Validators.TotalVotingPower() != s1.Validators.TotalVotingPower() {
|
|
t.Error("Validators TotalVotingPower changed")
|
|
}
|
|
// TODO Compare accountBalances, height, blockHash
|
|
|
|
}
|