diff options
Diffstat (limited to 'dex/app_test.go')
| -rw-r--r-- | dex/app_test.go | 207 |
1 files changed, 207 insertions, 0 deletions
diff --git a/dex/app_test.go b/dex/app_test.go index 7fc4933de..f4cc2fd9a 100644 --- a/dex/app_test.go +++ b/dex/app_test.go @@ -4,12 +4,14 @@ import ( "crypto/ecdsa" "fmt" "math/big" + "strings" "testing" "time" coreCommon "github.com/dexon-foundation/dexon-consensus/common" coreTypes "github.com/dexon-foundation/dexon-consensus/core/types" + "github.com/dexon-foundation/dexon/accounts/abi" "github.com/dexon-foundation/dexon/common" "github.com/dexon-foundation/dexon/consensus/dexcon" "github.com/dexon-foundation/dexon/core" @@ -158,6 +160,7 @@ func TestVerifyBlock(t *testing.T) { block.Hash = coreCommon.NewRandomHash() block.Position.ChainID = uint32(chainID.Uint64()) block.Position.Height = 1 + block.ProposerID = coreTypes.NodeID{coreCommon.Hash{1, 2, 3}} block.Payload, block.Witness, err = prepareDataWithoutTxPool(dex, key, 0, 100) if err != nil { t.Errorf("prepare data error: %v", err) @@ -174,6 +177,7 @@ func TestVerifyBlock(t *testing.T) { block.Hash = coreCommon.NewRandomHash() block.Position.ChainID = uint32(chainID.Uint64()) block.Position.Height = 1 + block.ProposerID = coreTypes.NodeID{coreCommon.Hash{1, 2, 3}} block.Payload, block.Witness, err = prepareDataWithoutTxPool(dex, key, 1, 100) if err != nil { t.Errorf("prepare data error: %v", err) @@ -190,6 +194,7 @@ func TestVerifyBlock(t *testing.T) { block.Hash = coreCommon.NewRandomHash() block.Position.ChainID = uint32(chainID.Uint64()) block.Position.Height = 2 + block.ProposerID = coreTypes.NodeID{coreCommon.Hash{1, 2, 3}} block.Payload, block.Witness, err = prepareDataWithoutTxPool(dex, key, 0, 100) if err != nil { t.Errorf("prepare data error: %v", err) @@ -206,6 +211,7 @@ func TestVerifyBlock(t *testing.T) { block.Hash = coreCommon.NewRandomHash() block.Position.ChainID = uint32(chainID.Uint64()) block.Position.Height = 1 + block.ProposerID = coreTypes.NodeID{coreCommon.Hash{1, 2, 3}} block.Payload, block.Witness, err = prepareDataWithoutTxPool(dex, key, 0, 10000) if err != nil { t.Errorf("prepare data error: %v", err) @@ -222,6 +228,7 @@ func TestVerifyBlock(t *testing.T) { block.Hash = coreCommon.NewRandomHash() block.Position.ChainID = uint32(chainID.Uint64()) block.Position.Height = 1 + block.ProposerID = coreTypes.NodeID{coreCommon.Hash{1, 2, 3}} _, block.Witness, err = prepareDataWithoutTxPool(dex, key, 0, 0) if err != nil { t.Errorf("prepare data error: %v", err) @@ -256,6 +263,7 @@ func TestVerifyBlock(t *testing.T) { block.Hash = coreCommon.NewRandomHash() block.Position.ChainID = uint32(chainID.Uint64()) block.Position.Height = 1 + block.ProposerID = coreTypes.NodeID{coreCommon.Hash{1, 2, 3}} _, block.Witness, err = prepareDataWithoutTxPool(dex, key, 0, 0) if err != nil { t.Errorf("prepare data error: %v", err) @@ -290,6 +298,7 @@ func TestVerifyBlock(t *testing.T) { block.Hash = coreCommon.NewRandomHash() block.Position.ChainID = uint32(chainID.Uint64()) block.Position.Height = 1 + block.ProposerID = coreTypes.NodeID{coreCommon.Hash{1, 2, 3}} _, block.Witness, err = prepareDataWithoutTxPool(dex, key, 0, 0) if err != nil { t.Errorf("prepare data error: %v", err) @@ -368,6 +377,7 @@ func TestBlockConfirmed(t *testing.T) { block.Hash = coreCommon.NewRandomHash() block.Witness = witness block.Payload = payload + block.ProposerID = coreTypes.NodeID{coreCommon.Hash{1, 2, 3}} block.Position.ChainID = uint32(chainID.Uint64()) dex.app.BlockConfirmed(*block) @@ -463,6 +473,147 @@ func TestBlockDelivered(t *testing.T) { } } +func TestNumChainsChange(t *testing.T) { + key, err := crypto.GenerateKey() + if err != nil { + t.Errorf("hex to ecdsa error: %v", err) + } + + params.TestnetChainConfig.Dexcon.Owner = crypto.PubkeyToAddress(key.PublicKey) + + dex, err := newTestDexonWithGenesis(key) + if err != nil { + t.Errorf("new test dexon error: %v", err) + } + + abiObject, err := abi.JSON(strings.NewReader(vm.GovernanceABIJSON)) + if err != nil { + t.Errorf("get abi object fail: %v", err) + } + + // Update config in round 1 and height 1. + // Config will affect in round 3. + input, err := abiObject.Pack("updateConfiguration", + new(big.Int).Mul(big.NewInt(1e18), big.NewInt(1e4)), big.NewInt(2000), + big.NewInt(1e17), big.NewInt(9000000), big.NewInt(3), big.NewInt(500), big.NewInt(5000), + big.NewInt(1), big.NewInt(700000), big.NewInt(5), big.NewInt(5), big.NewInt(700000), big.NewInt(1000), + []*big.Int{big.NewInt(1), big.NewInt(1), big.NewInt(1)}) + if err != nil { + t.Errorf("updateConfiguration abiObject pack error: %v", err) + } + + block, err := prepareConfirmedBlockWithTxAndData(dex, key, [][]byte{input}, 1) + if err != nil { + t.Errorf("prepare block error: %v", err) + } + dex.app.BlockDelivered(block.Hash, block.Position, + coreTypes.FinalizationResult{ + Timestamp: time.Now(), + Height: 1, + }) + + // Snapshot round on round 1 and height 2. + input, err = abiObject.Pack("snapshotRound", big.NewInt(1), big.NewInt(1)) + if err != nil { + t.Errorf("abiObject pack error: %v", err) + } + + block, err = prepareConfirmedBlockWithTxAndData(dex, key, [][]byte{input}, 1) + if err != nil { + t.Errorf("prepare block error: %v", err) + } + dex.app.BlockDelivered(block.Hash, block.Position, + coreTypes.FinalizationResult{ + Timestamp: time.Now(), + Height: 2, + }) + + // Make round 1 height 2 block write into chain. + block, err = prepareConfirmedBlockWithTxAndData(dex, key, nil, 1) + if err != nil { + t.Errorf("prepare block error: %v", err) + } + dex.app.BlockDelivered(block.Hash, block.Position, + coreTypes.FinalizationResult{ + Timestamp: time.Now(), + Height: 3, + }) + + // Round 2 will keep prepare tx as usual. + block, err = prepareConfirmedBlockWithTxAndData(dex, key, [][]byte{{1}}, 2) + if err != nil { + t.Errorf("prepare block error: %v", err) + } + if block.Payload == nil { + t.Errorf("payload should not be nil") + } + dex.app.BlockDelivered(block.Hash, block.Position, + coreTypes.FinalizationResult{ + Timestamp: time.Now(), + Height: 4, + }) + + // It will prepare empty payload until witness any block in round 3. + block, err = prepareConfirmedBlockWithTxAndData(dex, key, [][]byte{{1}}, 3) + if err != nil { + t.Errorf("prepare block error: %v", err) + } + if block.Payload != nil { + t.Errorf("payload should be nil but %v", block.Payload) + } + + // Test non-empty payload. + block.Payload = []byte{1} + if status := dex.app.VerifyBlock(block); status != coreTypes.VerifyInvalidBlock { + t.Errorf("unexpected verify status %v", status) + } + block.Payload = nil + + dex.app.BlockDelivered(block.Hash, block.Position, + coreTypes.FinalizationResult{ + Timestamp: time.Now(), + Height: 5, + }) + + // Still empty payload because round 3 is not witness yet. + // This block just for witness round 3 height 5. + block, err = prepareConfirmedBlockWithTxAndData(dex, key, [][]byte{{1}}, 3) + if err != nil { + t.Errorf("prepare block error: %v", err) + } + if block.Payload != nil { + t.Errorf("payload should be nil but %v", block.Payload) + } + + // Test non-empty payload. + block.Payload = []byte{1} + if status := dex.app.VerifyBlock(block); status != coreTypes.VerifyInvalidBlock { + t.Errorf("unexpected verify status %v", status) + } + block.Payload = nil + + dex.app.BlockDelivered(block.Hash, block.Position, + coreTypes.FinalizationResult{ + Timestamp: time.Now(), + Height: 6, + }) + + // Empty block in round 3 has been delivered. + // Now can prepare payload as usual. + block, err = prepareConfirmedBlockWithTxAndData(dex, key, [][]byte{{1}}, 3) + if err != nil { + t.Errorf("prepare block error: %v", err) + } + if block.Payload == nil { + t.Errorf("payload should not be nil") + } + dex.app.BlockDelivered(block.Hash, block.Position, + coreTypes.FinalizationResult{ + Timestamp: time.Now(), + Height: 7, + }) +} + func BenchmarkBlockDeliveredFlow(b *testing.B) { key, err := crypto.GenerateKey() if err != nil { @@ -594,6 +745,61 @@ func prepareData(dex *Dexon, key *ecdsa.PrivateKey, startNonce, txNum int) ( return } +func prepareConfirmedBlockWithTxAndData(dex *Dexon, key *ecdsa.PrivateKey, data [][]byte, round uint64) ( + Block *coreTypes.Block, err error) { + address := crypto.PubkeyToAddress(key.PublicKey) + numChains := dex.governance.GetConfigHelper(round).Configuration().NumChains + chainID := new(big.Int).Mod(address.Big(), big.NewInt(int64(numChains))) + + for _, d := range data { + // Prepare one block for pending. + nonce := dex.txPool.State().GetNonce(address) + signer := types.NewEIP155Signer(dex.chainConfig.ChainID) + tx := types.NewTransaction(uint64(nonce), vm.GovernanceContractAddress, big.NewInt(0), params.TxGas*2, + big.NewInt(1), d) + tx, err = types.SignTx(tx, signer, key) + if err != nil { + return nil, err + } + + dex.txPool.AddRemote(tx) + if err != nil { + return nil, err + } + } + var ( + payload []byte + witness coreTypes.Witness + ) + + payload, err = dex.app.PreparePayload(coreTypes.Position{Round: round, ChainID: uint32(chainID.Uint64())}) + if err != nil { + return + } + + witness, err = dex.app.PrepareWitness(0) + if err != nil { + return nil, err + } + + block := &coreTypes.Block{} + block.Hash = coreCommon.NewRandomHash() + block.Witness = witness + block.Payload = payload + block.Position.ChainID = uint32(chainID.Uint64()) + block.Position.Round = round + block.ProposerID = coreTypes.NodeID{coreCommon.Hash{1, 2, 3}} + + status := dex.app.VerifyBlock(block) + if status != coreTypes.VerifyOK { + err = fmt.Errorf("verify fail: %v", status) + return nil, err + } + + dex.app.BlockConfirmed(*block) + return block, nil +} + func prepareDataWithoutTxPool(dex *Dexon, key *ecdsa.PrivateKey, startNonce, txNum int) ( payload []byte, witness coreTypes.Witness, err error) { signer := types.NewEIP155Signer(dex.chainConfig.ChainID) @@ -656,6 +862,7 @@ func prepareConfirmedBlocks(dex *Dexon, keys []*ecdsa.PrivateKey, txNum int) (bl block.Witness = witness block.Payload = payload block.Position.ChainID = uint32(chainID.Uint64()) + block.ProposerID = coreTypes.NodeID{coreCommon.Hash{1, 2, 3}} status := dex.app.VerifyBlock(block) if status != coreTypes.VerifyOK { |