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.
60 lines
1.7 KiB
60 lines
1.7 KiB
package node
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/harmony-one/harmony/crypto/bls"
|
|
|
|
"github.com/harmony-one/harmony/consensus"
|
|
"github.com/harmony-one/harmony/internal/utils"
|
|
"github.com/harmony-one/harmony/p2p"
|
|
"github.com/harmony-one/harmony/p2p/p2pimpl"
|
|
)
|
|
|
|
func TestAddNewBlock(t *testing.T) {
|
|
pubKey := bls.RandPrivateKey().GetPublicKey()
|
|
leader := p2p.Peer{IP: "127.0.0.1", Port: "9882", ConsensusPubKey: pubKey}
|
|
priKey, _, _ := utils.GenKeyP2P("127.0.0.1", "9902")
|
|
host, err := p2pimpl.NewHost(&leader, priKey)
|
|
if err != nil {
|
|
t.Fatalf("newhost failure: %v", err)
|
|
}
|
|
consensus, err := consensus.New(host, 0, leader, nil)
|
|
if err != nil {
|
|
t.Fatalf("Cannot craeate consensus: %v", err)
|
|
}
|
|
node := New(host, consensus, nil, false)
|
|
|
|
selectedTxs := node.getTransactionsForNewBlock(MaxNumberOfTransactionsPerBlock)
|
|
node.Worker.CommitTransactions(selectedTxs)
|
|
block, _ := node.Worker.Commit()
|
|
|
|
node.AddNewBlock(block)
|
|
|
|
if node.blockchain.CurrentBlock().NumberU64() != 1 {
|
|
t.Error("New block is not added successfully")
|
|
}
|
|
}
|
|
|
|
func TestVerifyNewBlock(t *testing.T) {
|
|
pubKey := bls.RandPrivateKey().GetPublicKey()
|
|
leader := p2p.Peer{IP: "127.0.0.1", Port: "8882", ConsensusPubKey: pubKey}
|
|
priKey, _, _ := utils.GenKeyP2P("127.0.0.1", "9902")
|
|
host, err := p2pimpl.NewHost(&leader, priKey)
|
|
if err != nil {
|
|
t.Fatalf("newhost failure: %v", err)
|
|
}
|
|
consensus, err := consensus.New(host, 0, leader, nil)
|
|
if err != nil {
|
|
t.Fatalf("Cannot craeate consensus: %v", err)
|
|
}
|
|
node := New(host, consensus, nil, false)
|
|
|
|
selectedTxs := node.getTransactionsForNewBlock(MaxNumberOfTransactionsPerBlock)
|
|
node.Worker.CommitTransactions(selectedTxs)
|
|
block, _ := node.Worker.Commit()
|
|
|
|
if !node.VerifyNewBlock(block) {
|
|
t.Error("New block is not verified successfully")
|
|
}
|
|
}
|
|
|