core: add BlockGen.GetBalance method (ethereum#22589)

METADIUM · Mar 29, 2021 · b6912c1 · b6912c1
1 parent 76700ac
commit b6912c1
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/core/chain_makers.go b/core/chain_makers.go
@@ -111,6 +111,11 @@ func (b *BlockGen) AddTxWithChain(bc *BlockChain, tx *types.Transaction) {
 	b.receipts = append(b.receipts, receipt)
 }
 
+// GetBalance returns the balance of the given address at the generated block.
+func (b *BlockGen) GetBalance(addr common.Address) *big.Int {
+	return b.statedb.GetBalance(addr)
+}
+
 // AddUncheckedTx forcefully adds a transaction to the block without any
 // validation.
 //