x/evm: state_transition test (#389)

* draft state_transition * working test * keeper test * Update x/evm/types/state_transition_test.go * update state_transition_test.go * failed Finalize test case Co-authored-by: Federico Kunze <31522760+fedekunze@users.noreply.github.com> Co-authored-by: Federico Kunze <federico.kunze94@gmail.com>
2020-07-15 06:36:55 -04:00
parent 90f39390bc
commit 6bc80c5357
3 changed files with 189 additions and 18 deletions
@@ -0,0 +1,163 @@
+package types_test
+
+import (
+	"math/big"
+
+	sdk "github.com/cosmos/cosmos-sdk/types"
+
+	"github.com/cosmos/ethermint/crypto"
+	ethermint "github.com/cosmos/ethermint/types"
+	"github.com/cosmos/ethermint/x/evm/types"
+
+	ethcmn "github.com/ethereum/go-ethereum/common"
+	ethcrypto "github.com/ethereum/go-ethereum/crypto"
+)
+
+func (suite *StateDBTestSuite) TestTransitionDb() {
+	suite.stateDB.SetNonce(suite.address, 123)
+
+	addr := sdk.AccAddress(suite.address.Bytes())
+	balance := sdk.NewCoin(ethermint.DenomDefault, sdk.NewInt(5000))
+	suite.app.BankKeeper.SetBalance(suite.ctx, addr, balance)
+
+	priv, err := crypto.GenerateKey()
+	suite.Require().NoError(err)
+	recipient := ethcrypto.PubkeyToAddress(priv.ToECDSA().PublicKey)
+
+	testCase := []struct {
+		name     string
+		malleate func()
+		state    types.StateTransition
+		expPass  bool
+	}{
+		{
+			"passing state transition",
+			func() {},
+			types.StateTransition{
+				AccountNonce: 123,
+				Price:        big.NewInt(10),
+				GasLimit:     11,
+				Recipient:    &recipient,
+				Amount:       big.NewInt(50),
+				Payload:      []byte("data"),
+				ChainID:      big.NewInt(1),
+				Csdb:         suite.stateDB,
+				TxHash:       &ethcmn.Hash{},
+				Sender:       suite.address,
+				Simulate:     suite.ctx.IsCheckTx(),
+			},
+			true,
+		},
+		{
+			"contract creation",
+			func() {},
+			types.StateTransition{
+				AccountNonce: 123,
+				Price:        big.NewInt(10),
+				GasLimit:     11,
+				Recipient:    nil,
+				Amount:       big.NewInt(10),
+				Payload:      []byte("data"),
+				ChainID:      big.NewInt(1),
+				Csdb:         suite.stateDB,
+				TxHash:       &ethcmn.Hash{},
+				Sender:       suite.address,
+				Simulate:     true,
+			},
+			true,
+		},
+		{
+			"state transition simulation",
+			func() {},
+			types.StateTransition{
+				AccountNonce: 123,
+				Price:        big.NewInt(10),
+				GasLimit:     11,
+				Recipient:    &recipient,
+				Amount:       big.NewInt(10),
+				Payload:      []byte("data"),
+				ChainID:      big.NewInt(1),
+				Csdb:         suite.stateDB,
+				TxHash:       &ethcmn.Hash{},
+				Sender:       suite.address,
+				Simulate:     true,
+			},
+			true,
+		},
+		{
+			"fail by sending more than balance",
+			func() {},
+			types.StateTransition{
+				AccountNonce: 123,
+				Price:        big.NewInt(10),
+				GasLimit:     11,
+				Recipient:    &recipient,
+				Amount:       big.NewInt(4951),
+				Payload:      []byte("data"),
+				ChainID:      big.NewInt(1),
+				Csdb:         suite.stateDB,
+				TxHash:       &ethcmn.Hash{},
+				Sender:       suite.address,
+				Simulate:     suite.ctx.IsCheckTx(),
+			},
+			false,
+		},
+		{
+			"failed to Finalize",
+			func() {},
+			types.StateTransition{
+				AccountNonce: 123,
+				Price:        big.NewInt(10),
+				GasLimit:     11,
+				Recipient:    &recipient,
+				Amount:       big.NewInt(-5000),
+				Payload:      []byte("data"),
+				ChainID:      big.NewInt(1),
+				Csdb:         suite.stateDB,
+				TxHash:       &ethcmn.Hash{},
+				Sender:       suite.address,
+				Simulate:     false,
+			},
+			false,
+		},
+		{
+			"nil gas price",
+			func() {
+				invalidGas := sdk.DecCoins{
+					{Denom: ethermint.DenomDefault},
+				}
+				suite.ctx = suite.ctx.WithMinGasPrices(invalidGas)
+			},
+			types.StateTransition{
+				AccountNonce: 123,
+				Price:        big.NewInt(10),
+				GasLimit:     11,
+				Recipient:    &recipient,
+				Amount:       big.NewInt(10),
+				Payload:      []byte("data"),
+				ChainID:      big.NewInt(1),
+				Csdb:         suite.stateDB,
+				TxHash:       &ethcmn.Hash{},
+				Sender:       suite.address,
+				Simulate:     suite.ctx.IsCheckTx(),
+			},
+			false,
+		},
+	}
+
+	for _, tc := range testCase {
+		tc.malleate()
+
+		_, err = tc.state.TransitionDb(suite.ctx)
+
+		if tc.expPass {
+			suite.Require().NoError(err, tc.name)
+			fromBalance := suite.app.EvmKeeper.GetBalance(suite.ctx, suite.address)
+			toBalance := suite.app.EvmKeeper.GetBalance(suite.ctx, recipient)
+			suite.Require().Equal(fromBalance, big.NewInt(4950), tc.name)
+			suite.Require().Equal(toBalance, big.NewInt(50), tc.name)
+		} else {
+			suite.Require().Error(err, tc.name)
+		}
+	}
+}