0g-chain/tests/e2e/testutil/suite.go
Robert Pirtle 0156b0e645
support chian upgrades in e2e tests (#1513)
* update kvtool

* add env vars for e2e automated upgrades

* run kvtool with upgrade flags when enabled

* add e2e test placeholder for upgrade handler tests

* allow override of kvtool's kava.configTemplate

* use the correct image tag when running e2e tests

* update kvtool

* always docker-build when running make test-e2e

* refactor skip shutdown to always skip shutdown

* add CtxAtHeight grpc helper

* add Community & Earn queriers

* expose UpgradeHeight in suite

* document e2e chain upgrades

* add reference to example branch
2023-03-28 15:32:36 -07:00

99 lines
2.4 KiB
Go

package testutil
import (
"fmt"
"github.com/stretchr/testify/suite"
"github.com/kava-labs/kava/app"
"github.com/kava-labs/kava/tests/e2e/runner"
)
const (
FundedAccountName = "whale"
// use coin type 60 so we are compatible with accounts from `kava add keys --eth <name>`
// these accounts use the ethsecp256k1 signing algorithm that allows the signing client
// to manage both sdk & evm txs.
Bip44CoinType = 60
IbcPort = "transfer"
IbcChannel = "channel-0"
)
type E2eTestSuite struct {
suite.Suite
config SuiteConfig
runner runner.NodeRunner
Kava *Chain
Ibc *Chain
UpgradeHeight int64
}
func (suite *E2eTestSuite) SetupSuite() {
var err error
fmt.Println("setting up test suite.")
app.SetSDKConfig()
suiteConfig := ParseSuiteConfig()
suite.config = suiteConfig
suite.UpgradeHeight = suiteConfig.KavaUpgradeHeight
runnerConfig := runner.Config{
KavaConfigTemplate: suiteConfig.KavaConfigTemplate,
IncludeIBC: suiteConfig.IncludeIbcTests,
ImageTag: "local",
EnableAutomatedUpgrade: suiteConfig.IncludeAutomatedUpgrade,
KavaUpgradeName: suiteConfig.KavaUpgradeName,
KavaUpgradeHeight: suiteConfig.KavaUpgradeHeight,
KavaUpgradeBaseImageTag: suiteConfig.KavaUpgradeBaseImageTag,
SkipShutdown: suiteConfig.SkipShutdown,
}
suite.runner = runner.NewKavaNode(runnerConfig)
chains := suite.runner.StartChains()
kavachain := chains.MustGetChain("kava")
suite.Kava, err = NewChain(suite.T(), kavachain, suiteConfig.FundedAccountMnemonic)
if err != nil {
suite.runner.Shutdown()
suite.T().Fatalf("failed to create kava chain querier: %s", err)
}
if suiteConfig.IncludeIbcTests {
ibcchain := chains.MustGetChain("ibc")
suite.Ibc, err = NewChain(suite.T(), ibcchain, suiteConfig.FundedAccountMnemonic)
if err != nil {
suite.runner.Shutdown()
suite.T().Fatalf("failed to create ibc chain querier: %s", err)
}
}
}
func (suite *E2eTestSuite) TearDownSuite() {
fmt.Println("tearing down test suite.")
// close all account request channels
suite.Kava.Shutdown()
if suite.Ibc != nil {
suite.Ibc.Shutdown()
}
// gracefully shutdown docker container(s)
suite.runner.Shutdown()
}
func (suite *E2eTestSuite) SkipIfIbcDisabled() {
if !suite.config.IncludeIbcTests {
suite.T().SkipNow()
}
}
func (suite *E2eTestSuite) SkipIfUpgradeDisabled() {
if !suite.config.IncludeAutomatedUpgrade {
suite.T().SkipNow()
}
}