diff options
| -rw-r--r-- | core/vm/oracle_contracts.go | 4 | ||||
| -rw-r--r-- | core/vm/oracle_contracts_test.go | 8 |
2 files changed, 11 insertions, 1 deletions
diff --git a/core/vm/oracle_contracts.go b/core/vm/oracle_contracts.go index 47a680455..eebfa9a3a 100644 --- a/core/vm/oracle_contracts.go +++ b/core/vm/oracle_contracts.go @@ -1919,7 +1919,9 @@ func (g *GovernanceContract) payFine(nodeAddr common.Address) ([]byte, error) { g.state.UpdateNode(nodeOffset, node) // Pay the fine to governance owner. - g.evm.StateDB.AddBalance(g.state.Owner(), g.contract.Value()) + if !g.transfer(GovernanceContractAddress, g.state.Owner(), g.contract.Value()) { + return nil, errExecutionReverted + } g.state.emitFinePaid(nodeAddr, g.contract.Value()) diff --git a/core/vm/oracle_contracts_test.go b/core/vm/oracle_contracts_test.go index 5be893f95..cbf08c273 100644 --- a/core/vm/oracle_contracts_test.go +++ b/core/vm/oracle_contracts_test.go @@ -612,11 +612,19 @@ func (g *OracleContractsTestSuite) TestFine() { g.Require().NotNil(err) // Pay the fine. + govBalance := g.stateDB.GetBalance(GovernanceContractAddress) + ownerBalance := g.stateDB.GetBalance(g.config.Owner) + input, err = GovernanceABI.ABI.Pack("payFine", addr) g.Require().NoError(err) _, err = g.call(GovernanceContractAddress, finePayer, input, amount) g.Require().NoError(err) + g.Require().Equal(g.stateDB.GetBalance(GovernanceContractAddress).String(), + govBalance.String()) + g.Require().Equal(g.stateDB.GetBalance(g.config.Owner).String(), + new(big.Int).Add(ownerBalance, amount).String()) + // Qualified. g.Require().Equal(1, len(g.s.QualifiedNodes())) } |