mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-11-10 18:15:19 +00:00
0156b0e645
* 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
85 lines
2.7 KiB
Go
85 lines
2.7 KiB
Go
package testutil
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strconv"
|
|
|
|
"github.com/subosito/gotenv"
|
|
)
|
|
|
|
func init() {
|
|
// read the .env file, if present
|
|
gotenv.Load()
|
|
}
|
|
|
|
type SuiteConfig struct {
|
|
// A funded account used to fnd all other accounts.
|
|
FundedAccountMnemonic string
|
|
// The kava.configTemplate flag to be passed to kvtool, usually "master".
|
|
// This allows one to change the base genesis used to start the chain.
|
|
KavaConfigTemplate string
|
|
// Whether or not to start an IBC chain. Use `suite.SkipIfIbcDisabled()` in IBC tests in IBC tests.
|
|
IncludeIbcTests bool
|
|
|
|
// Whether or not to run a chain upgrade & run post-upgrade tests. Use `suite.SkipIfUpgradeDisabled()` in post-upgrade tests.
|
|
IncludeAutomatedUpgrade bool
|
|
// Name of the upgrade, if upgrade is enabled.
|
|
KavaUpgradeName string
|
|
// Height upgrade will be applied to the test chain, if upgrade is enabled.
|
|
KavaUpgradeHeight int64
|
|
// Tag of kava docker image that will be upgraded to the current image before tests are run, if upgrade is enabled.
|
|
KavaUpgradeBaseImageTag string
|
|
|
|
// When true, the chains will remain running after tests complete (pass or fail)
|
|
SkipShutdown bool
|
|
}
|
|
|
|
func ParseSuiteConfig() SuiteConfig {
|
|
config := SuiteConfig{
|
|
// this mnemonic is expected to be a funded account that can seed the funds for all
|
|
// new accounts created during tests. it will be available under Accounts["whale"]
|
|
FundedAccountMnemonic: nonemptyStringEnv("E2E_KAVA_FUNDED_ACCOUNT_MNEMONIC"),
|
|
KavaConfigTemplate: nonemptyStringEnv("E2E_KVTOOL_KAVA_CONFIG_TEMPLATE"),
|
|
IncludeIbcTests: mustParseBool("E2E_INCLUDE_IBC_TESTS"),
|
|
IncludeAutomatedUpgrade: mustParseBool("E2E_INCLUDE_AUTOMATED_UPGRADE"),
|
|
}
|
|
|
|
skipShutdownEnv := os.Getenv("E2E_SKIP_SHUTDOWN")
|
|
if skipShutdownEnv != "" {
|
|
config.SkipShutdown = mustParseBool("E2E_SKIP_SHUTDOWN")
|
|
}
|
|
|
|
if config.IncludeAutomatedUpgrade {
|
|
config.KavaUpgradeName = nonemptyStringEnv("E2E_KAVA_UPGRADE_NAME")
|
|
config.KavaUpgradeBaseImageTag = nonemptyStringEnv("E2E_KAVA_UPGRADE_BASE_IMAGE_TAG")
|
|
upgradeHeight, err := strconv.ParseInt(nonemptyStringEnv("E2E_KAVA_UPGRADE_HEIGHT"), 10, 64)
|
|
if err != nil {
|
|
panic(fmt.Sprintf("E2E_KAVA_UPGRADE_HEIGHT must be a number: %s", err))
|
|
}
|
|
config.KavaUpgradeHeight = upgradeHeight
|
|
}
|
|
|
|
return config
|
|
}
|
|
|
|
func mustParseBool(name string) bool {
|
|
envValue := os.Getenv(name)
|
|
if envValue == "" {
|
|
panic(fmt.Sprintf("%s is unset but expected a bool", name))
|
|
}
|
|
value, err := strconv.ParseBool(envValue)
|
|
if err != nil {
|
|
panic(fmt.Sprintf("%s (%s) cannot be parsed to a bool: %s", name, envValue, err))
|
|
}
|
|
return value
|
|
}
|
|
|
|
func nonemptyStringEnv(name string) string {
|
|
value := os.Getenv(name)
|
|
if value == "" {
|
|
panic(fmt.Sprintf("no %s env variable provided", name))
|
|
}
|
|
return value
|
|
}
|