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

frozeman · Oct 15, 2021 · c6d3be1 · c6d3be1
1 parent 91520f7
commit c6d3be1
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.
 //