laconicd/x/evm/types/utils_test.go
2021-09-07 17:29:24 +00:00

88 lines
2.3 KiB
Go

package types_test
import (
"errors"
"math/big"
"testing"
"github.com/cosmos/cosmos-sdk/client"
sdk "github.com/cosmos/cosmos-sdk/types"
authtx "github.com/cosmos/cosmos-sdk/x/auth/tx"
proto "github.com/gogo/protobuf/proto"
"github.com/tharsis/ethermint/app"
"github.com/tharsis/ethermint/encoding"
evmtypes "github.com/tharsis/ethermint/x/evm/types"
"github.com/stretchr/testify/require"
"github.com/ethereum/go-ethereum/common"
)
func TestEvmDataEncoding(t *testing.T) {
ret := []byte{0x5, 0x8}
data := &evmtypes.MsgEthereumTxResponse{
Hash: common.BytesToHash([]byte("hash")).String(),
Logs: []*evmtypes.Log{{
Data: []byte{1, 2, 3, 4},
BlockNumber: 17,
}},
Ret: ret,
}
enc, err := proto.Marshal(data)
require.NoError(t, err)
txData := &sdk.TxMsgData{
Data: []*sdk.MsgData{{MsgType: evmtypes.TypeMsgEthereumTx, Data: enc}},
}
txDataBz, err := proto.Marshal(txData)
require.NoError(t, err)
res, err := evmtypes.DecodeTxResponse(txDataBz)
require.NoError(t, err)
require.NotNil(t, res)
require.Equal(t, data.Logs, res.Logs)
require.Equal(t, ret, res.Ret)
}
func TestUnwrapEthererumMsg(t *testing.T) {
_, err := evmtypes.UnwrapEthereumMsg(nil)
require.NotNil(t, err)
encodingConfig := encoding.MakeConfig(app.ModuleBasics)
clientCtx := client.Context{}.WithTxConfig(encodingConfig.TxConfig)
builder, _ := clientCtx.TxConfig.NewTxBuilder().(authtx.ExtensionOptionsTxBuilder)
tx := builder.GetTx().(sdk.Tx)
_, err = evmtypes.UnwrapEthereumMsg(&tx)
require.NotNil(t, err)
msg := evmtypes.NewTx(big.NewInt(1), 0, &common.Address{}, big.NewInt(0), 0, big.NewInt(0), []byte{}, nil)
err = builder.SetMsgs(msg)
tx = builder.GetTx().(sdk.Tx)
msg_, err := evmtypes.UnwrapEthereumMsg(&tx)
require.Nil(t, err)
require.Equal(t, msg_, msg)
}
func TestBinSearch(t *testing.T) {
success_executable := func(gas uint64) (bool, *evmtypes.MsgEthereumTxResponse, error) {
target := uint64(21000)
return gas < target, nil, nil
}
failed_executable := func(gas uint64) (bool, *evmtypes.MsgEthereumTxResponse, error) {
return true, nil, errors.New("contract failed")
}
gas, err := evmtypes.BinSearch(20000, 21001, success_executable)
require.NoError(t, err)
require.Equal(t, gas, uint64(21000))
gas, err = evmtypes.BinSearch(20000, 21001, failed_executable)
require.Error(t, err)
require.Equal(t, gas, uint64(0))
}