From a3a3d6ecfa181ab581844a718765c888c285ae85 Mon Sep 17 00:00:00 2001 From: RafilxTenfen Date: Thu, 25 Jul 2024 14:59:53 -0300 Subject: [PATCH] chore: add upgrade path as parameter in configurer --- test/e2e/configurer/factory.go | 5 ++--- test/e2e/software_upgrade_e2e_test.go | 3 ++- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/test/e2e/configurer/factory.go b/test/e2e/configurer/factory.go index b9823e17..a33b0be3 100644 --- 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 index 61a9f252..e8ede1e4 100644 --- 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)