fix: Stress test refactor

changelog.md

@@ -81,6 +81,7 @@ Getting the correct TSS address for Bitcoin now requires proviidng the Bitcoin c
 * [1546](https://github.com/zeta-chain/node/pull/1546) - fix reset of pending nonces on genesis import
 * [1555](https://github.com/zeta-chain/node/pull/1555) - Reduce websocket message limit to 10MB
 * [1567](https://github.com/zeta-chain/node/pull/1567) - add bitcoin chain id to fetch the tss address rpc endpoint
+* [1501](https://github.com/zeta-chain/node/pull/1501) - fix stress test - use new refactored config file and smoketest runner
 * [1589](https://github.com/zeta-chain/node/pull/1589) - add bitcoin chain id to `get tss address` and `get tss address historical` cli query
 
 ### Refactoring

cmd/zetae2e/init.go

@@ -0,0 +1,39 @@
+package main
+
+import (
+	"fmt"
+
+	"github.com/zeta-chain/zetacore/cmd/zetae2e/local"
+
+	"github.com/spf13/cobra"
+	"github.com/zeta-chain/zetacore/contrib/localnet/orchestrator/smoketest/config"
+)
+
+var initConf = config.Config{}
+var configFile = ""
+
+func NewInitCmd() *cobra.Command {
+	var InitCmd = &cobra.Command{
+		Use:   "init",
+		Short: "initialize config file for e2e tests",
+		Run:   initConfig,
+	}
+
+	InitCmd.Flags().StringVar(&initConf.RPCs.EVM, "ethURL", "http://eth:8545", "--ethURL http://eth:8545")
+	InitCmd.Flags().StringVar(&initConf.RPCs.ZetaCoreGRPC, "grpcURL", "zetacore0:9090", "--grpcURL zetacore0:9090")
+	InitCmd.Flags().StringVar(&initConf.RPCs.ZetaCoreRPC, "rpcURL", "http://zetacore0:26657", "--rpcURL http://zetacore0:26657")
+	InitCmd.Flags().StringVar(&initConf.RPCs.Zevm, "zevmURL", "http://zetacore0:8545", "--zevmURL http://zetacore0:8545")
+	InitCmd.Flags().StringVar(&initConf.RPCs.Bitcoin, "btcURL", "bitcoin:18443", "--grpcURL bitcoin:18443")
+
+	InitCmd.Flags().StringVar(&initConf.ZetaChainID, "chainID", "athens_101-1", "--chainID athens_101-1")
+	InitCmd.Flags().StringVar(&configFile, local.FlagConfigFile, "smoketest.config", "--cfg ./smoketest.config")
+
+	return InitCmd
+}
+
+func initConfig(_ *cobra.Command, _ []string) {
+	err := config.WriteConfig(configFile, initConf)
+	if err != nil {
+		fmt.Printf("error writing config file: %s", err.Error())
+	}
+}

cmd/zetae2e/local/local.go

@@ -18,7 +18,7 @@ import (
 const (
 	flagContractsDeployed = "deployed"
 	flagWaitForHeight     = "wait-for"
-	flagConfigFile        = "config"
+	FlagConfigFile        = "config"
 	flagVerbose           = "verbose"
 	flagTestAdmin         = "test-admin"
 	flagTestCustom        = "test-custom"
@@ -51,7 +51,7 @@ func NewLocalCmd() *cobra.Command {
 		"block height for tests to begin, ex. --wait-for 100",
 	)
 	cmd.Flags().String(
-		flagConfigFile,
+		FlagConfigFile,
 		"",
 		"config file to use for the tests",
 	)
@@ -149,7 +149,7 @@ func localE2ETest(cmd *cobra.Command, _ []string) {
 	}()
 
 	// initialize tests config
-	conf, err := getConfig(cmd)
+	conf, err := GetConfig(cmd)
 	if err != nil {
 		panic(err)
 	}

cmd/zetae2e/local/utils.go

@@ -16,9 +16,9 @@ import (
 	crosschaintypes "github.com/zeta-chain/zetacore/x/crosschain/types"
 )
 
-// getConfig returns config from file from the command line flag
-func getConfig(cmd *cobra.Command) (config.Config, error) {
-	configFile, err := cmd.Flags().GetString(flagConfigFile)
+// GetConfig returns config from file from the command line flag
+func GetConfig(cmd *cobra.Command) (config.Config, error) {
+	configFile, err := cmd.Flags().GetString(FlagConfigFile)
 	if err != nil {
 		return config.Config{}, err
 	}

cmd/zetae2e/root.go

@@ -13,6 +13,8 @@ func NewRootCmd() *cobra.Command {
 	cmd.AddCommand(
 		NewRunCmd(),
 		local.NewLocalCmd(),
+		NewStressTestCmd(),
+		NewInitCmd(),
 	)
 
 	return cmd