Add test for countNumTransactionsInBlockchain()

aws-code/transaction_generator.go

@@ -122,7 +122,7 @@ func main() {
 
 	// Testing node to mirror the node data in consensus
 	dataNode := node.NewNode(&consensus.Consensus{})
-	dataNode.AddMoreFakeTransactions()
+	dataNode.AddMoreFakeTransactions(1000)
 
 	start := time.Now()
 	totalTime := 60.0

benchmark_main.go

@@ -88,7 +88,7 @@ func main() {
 	consensus.OnConsensusDone = node.AddNewBlockToBlockchain
 
 	// Temporary testing code, to be removed.
-	node.AddMoreFakeTransactions()
+	node.AddMoreFakeTransactions(1000)
 
 	if consensus.IsLeader {
 		// Let consensus run

node/node.go

@@ -73,8 +73,8 @@ func (node *Node) String() string {
 
 // [Testing code] Should be deleted for production
 // Create in genesis block 1000 transactions which assign 1000 token to each address in [1 - 1000]
-func (node *Node) AddMoreFakeTransactions() {
-	txs := make([]*blockchain.Transaction, 1000)
+func (node *Node) AddMoreFakeTransactions(numTxs int) {
+	txs := make([]*blockchain.Transaction, numTxs)
 	for i := range txs {
 		txs[i] = blockchain.NewCoinbaseTX(strconv.Itoa(i), "")
 	}

node/node_test.go

@@ -32,3 +32,15 @@ func TestNewNewNode(test *testing.T) {
 		test.Error("Utxo pool is not initialized for the node")
 	}
 }
+
+func TestCountNumTransactionsInBlockchain(test *testing.T) {
+	leader := p2p.Peer{Ip: "1", Port: "2"}
+	validator := p2p.Peer{Ip: "3", Port: "5"}
+	consensus := consensus.NewConsensus("1", "2", "0", []p2p.Peer{leader, validator}, leader)
+
+	node := NewNode(&consensus)
+	node.AddMoreFakeTransactions(1000)
+	if node.countNumTransactionsInBlockchain() != 1001 {
+		test.Error("Count of transactions in the blockchain is incorrect")
+	}
+}