mirror of
https://github.com/planetmint/planetmint-go.git
synced 2025-11-24 06:25:47 +00:00
2534828fc6
* feature: add ante handler to block non validators from sending restricted txs * fix: add missing msg types and linter errors * fix: staticcheck error * refactor: use fallthrough on ante handle switch * fix: set val address on machine attest test --------- Signed-off-by: Lorenz Herzberger <lorenzherzberger@gmail.com>
89 lines
2.6 KiB
Go
89 lines
2.6 KiB
Go
package dao
|
|
|
|
import (
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
banktypes "github.com/cosmos/cosmos-sdk/x/bank/types"
|
|
"github.com/planetmint/planetmint-go/config"
|
|
"github.com/planetmint/planetmint-go/lib"
|
|
clitestutil "github.com/planetmint/planetmint-go/testutil/cli"
|
|
e2etestutil "github.com/planetmint/planetmint-go/testutil/e2e"
|
|
"github.com/planetmint/planetmint-go/testutil/network"
|
|
"github.com/planetmint/planetmint-go/testutil/sample"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/suite"
|
|
)
|
|
|
|
type GasConsumptionE2ETestSuite struct {
|
|
suite.Suite
|
|
|
|
cfg network.Config
|
|
network *network.Network
|
|
}
|
|
|
|
func NewGasConsumptionE2ETestSuite(cfg network.Config) *GasConsumptionE2ETestSuite {
|
|
return &GasConsumptionE2ETestSuite{cfg: cfg}
|
|
}
|
|
|
|
func (s *GasConsumptionE2ETestSuite) SetupSuite() {
|
|
s.T().Log("setting up e2e test suite")
|
|
conf := config.GetConfig()
|
|
conf.FeeDenom = sample.FeeDenom
|
|
s.network = network.New(s.T(), s.cfg)
|
|
account, err := e2etestutil.CreateAccount(s.network, sample.Name, sample.Mnemonic)
|
|
s.Require().NoError(err)
|
|
err = e2etestutil.FundAccount(s.network, account)
|
|
s.Require().NoError(err)
|
|
}
|
|
|
|
func (s *GasConsumptionE2ETestSuite) TearDownSuite() {
|
|
s.T().Log("tearing down e2e test suites")
|
|
}
|
|
|
|
func (s *GasConsumptionE2ETestSuite) TestValidatorConsumption() {
|
|
val := s.network.Validators[0]
|
|
|
|
k, err := val.ClientCtx.Keyring.Key(sample.Name)
|
|
s.Require().NoError(err)
|
|
addr, _ := k.GetAddress()
|
|
|
|
// send huge tx but as val and with no gas kv costs
|
|
msgs := createMsgs(val.Address, addr, 10)
|
|
|
|
out, err := lib.BroadcastTxWithFileLock(val.Address, msgs...)
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().NoError(s.network.WaitForNextBlock())
|
|
|
|
_, err = clitestutil.GetRawLogFromTxOut(val, out)
|
|
s.Require().NoError(err)
|
|
}
|
|
|
|
func (s *GasConsumptionE2ETestSuite) TestNonValidatorConsumptionOverflow() {
|
|
val := s.network.Validators[0]
|
|
|
|
k, err := val.ClientCtx.Keyring.Key(sample.Name)
|
|
s.Require().NoError(err)
|
|
addr, _ := k.GetAddress()
|
|
|
|
// exceed gas limit with too many msgs as non validator
|
|
msgs := createMsgs(addr, val.Address, 10)
|
|
|
|
out, err := lib.BroadcastTxWithFileLock(addr, msgs...)
|
|
s.Require().NoError(err)
|
|
|
|
s.Require().NoError(s.network.WaitForNextBlock())
|
|
|
|
_, err = clitestutil.GetRawLogFromTxOut(val, out)
|
|
s.Require().Error(err)
|
|
assert.Equal(s.T(), err.Error(), "out of gas in location: Has; gasWanted: 200000, gasUsed: 200241: out of gas")
|
|
}
|
|
|
|
func createMsgs(from sdk.AccAddress, to sdk.AccAddress, n int) (msgs []sdk.Msg) {
|
|
coins := sdk.NewCoins(sdk.NewInt64Coin("stake", 10))
|
|
for i := 0; i < n; i++ {
|
|
msg := banktypes.NewMsgSend(from, to, coins)
|
|
msgs = append(msgs, msg)
|
|
}
|
|
return
|
|
}
|