chore: add upgrade path as parameter in configurer

babylonchain · Jul 25, 2024 · a3a3d6e · a3a3d6e
1 parent 0a16ccb
commit a3a3d6e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/test/e2e/configurer/factory.go b/test/e2e/configurer/factory.go
@@ -4,7 +4,6 @@ import (
 	"testing"
 
 	"github.com/babylonchain/babylon/test/e2e/configurer/chain"
-	"github.com/babylonchain/babylon/test/e2e/configurer/config"
 	"github.com/babylonchain/babylon/test/e2e/containers"
 	"github.com/babylonchain/babylon/test/e2e/initialization"
 	zctypes "github.com/babylonchain/babylon/x/zoneconcierge/types"
@@ -198,7 +197,7 @@ func NewBTCStakingConfigurer(t *testing.T, isDebugLogEnabled bool) (Configurer,
 }
 
 // NewSoftwareUpgradeConfigurer returns a new Configurer for Software Upgrade testing
-func NewSoftwareUpgradeConfigurer(t *testing.T, isDebugLogEnabled bool) (Configurer, error) {
+func NewSoftwareUpgradeConfigurer(t *testing.T, isDebugLogEnabled bool, upgradePath string) (Configurer, error) {
 	containerManager, err := containers.NewManager(isDebugLogEnabled, false, true)
 	if err != nil {
 		return nil, err
@@ -211,7 +210,7 @@ func NewSoftwareUpgradeConfigurer(t *testing.T, isDebugLogEnabled bool) (Configu
 		},
 		withUpgrade(baseSetup), // base set up with upgrade
 		containerManager,
-		config.VanillaUpgradeFilePath,
+		upgradePath,
 		0,
 	), nil
 }
diff --git a/test/e2e/software_upgrade_e2e_test.go b/test/e2e/software_upgrade_e2e_test.go
@@ -4,6 +4,7 @@ import (
 	"github.com/stretchr/testify/suite"
 
 	"github.com/babylonchain/babylon/test/e2e/configurer"
+	"github.com/babylonchain/babylon/test/e2e/configurer/config"
 )
 
 type SoftwareUpgradeVanillaTestSuite struct {
@@ -16,7 +17,7 @@ func (s *SoftwareUpgradeVanillaTestSuite) SetupSuite() {
 	s.T().Log("setting up e2e integration test suite...")
 	var err error
 
-	s.configurer, err = configurer.NewSoftwareUpgradeConfigurer(s.T(), false)
+	s.configurer, err = configurer.NewSoftwareUpgradeConfigurer(s.T(), false, config.VanillaUpgradeFilePath)
 	s.NoError(err)
 	err = s.configurer.ConfigureChains()
 	s.NoError(err)