test: adding proof len check

Author: KimiWu123

geth-utils/gethutil/mpt/witness/gen_witness_transactions_test.go

@@ -98,20 +98,36 @@ func transactionsStackTrieInsertionTemplate(t *testing.T, n int) {
 	rlp_last_tx, _ := txs[n-1].MarshalBinary()
 	last_proofC := proofs[len(proofs)-1].GetProofC()
 
+	tx_len := len(txs)
+	proof_len := len(proofs)
+
 	// Proof of the first tx is appended at the end of the proofs if len(tx) < 0x80
 	// That's why we minus 2 here.
-	if len(txs) > 1 && len(txs) < 256 {
-		last_proofC = proofs[len(proofs)-2].GetProofC()
+	if tx_len > 1 && tx_len < 256 {
+		last_proofC = proofs[proof_len-2].GetProofC()
 	}
 	last_leaf_proof := last_proofC[len(last_proofC)-1]
 
+	if tx_len != proof_len {
+		fmt.Println("Expected to have", tx_len, ", but only got", proof_len)
+		t.Fail()
+	}
+
 	if !bytes.Equal(last_leaf_proof, rlp_last_tx) {
 		fmt.Println("- last_tx ", rlp_last_tx)
 		fmt.Println("- last_proof ", last_leaf_proof)
 		t.Fail()
 	}
 }
 
+// func TestJson(t *testing.T, n int) {
+// 	txs := makeTransactions(n)
+// 	db := rawdb.NewMemoryDatabase()
+// 	stackTrie := trie.NewStackTrie(db)
+
+// 	proofs, _ := stackTrie.UpdateAndGetProofs(db, types.Transactions(txs))
+// }
+
 func TestStackTrieInsertion_1Tx(t *testing.T) {
 	// Only one leaf
 	transactionsStackTrieInsertionTemplate(t, 1)