chore: update vanilla file name and check the values are equal in btc…

… headers
babylonlabs-io · Jul 30, 2024 · 6f92d84 · 6f92d84
1 parent ef16043
commit 6f92d84
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/test/e2e/software_upgrade_e2e_signet_launch_test.go b/test/e2e/software_upgrade_e2e_signet_launch_test.go
@@ -43,7 +43,7 @@ func (s *SoftwareUpgradeSignetLaunchTestSuite) TestUpgradeSignetLaunch() {
 	expectedUpgradeHeight := int64(25)
 
 	// makes sure that the upgrade was actually executed
-	resp := n.QueryAppliedPlan("vanilla")
+	resp := n.QueryAppliedPlan(v1.Upgrade.UpgradeName)
 	s.EqualValues(expectedUpgradeHeight, resp.Height, "the plan should be applied at the height 25")
 
 	allBtcHeaders := n.QueryBtcLightClientMainchain()
@@ -58,5 +58,5 @@ func (s *SoftwareUpgradeSignetLaunchTestSuite) TestUpgradeSignetLaunch() {
 	s.Equal(newHeadersLen, oldHeadersLen+lenHeadersInserted)
 
 	// ensure the headers were inserted at the end
-	s.Equal(allBtcHeaders[newHeadersLen-lenHeadersInserted:], btcHeadersFromFile)
+	s.EqualValues(allBtcHeaders[newHeadersLen-lenHeadersInserted:], btcHeadersFromFile)
 }
diff --git a/test/e2e/software_upgrade_e2e_test.go → .../e2e/software_upgrade_e2e_vanilla_test.go b/test/e2e/software_upgrade_e2e_test.go → .../e2e/software_upgrade_e2e_vanilla_test.go
@@ -3,6 +3,7 @@ package e2e
 import (
 	"github.com/stretchr/testify/suite"
 
+	v1 "github.com/babylonlabs-io/babylon/app/upgrades/vanilla"
 	"github.com/babylonlabs-io/babylon/test/e2e/configurer"
 	"github.com/babylonlabs-io/babylon/test/e2e/configurer/config"
 )
@@ -42,7 +43,7 @@ func (s *SoftwareUpgradeVanillaTestSuite) TestUpgradeVanilla() {
 	expectedUpgradeHeight := int64(25)
 
 	// makes sure that the upgrade was actually executed
-	resp := n.QueryAppliedPlan("vanilla")
+	resp := n.QueryAppliedPlan(v1.Upgrade.UpgradeName)
 	s.EqualValues(expectedUpgradeHeight, resp.Height, "the plan should be applied at the height 25")
 
 	fps := n.QueryFinalityProviders()