Fixed tests

astronomer · Dec 17, 2024 · dc52834 · dc52834
1 parent cbbf289
commit dc52834
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/cmd/software/deploy_test.go b/cmd/software/deploy_test.go
@@ -63,6 +63,9 @@ func (s *Suite) TestDeploy() {
 	err = execDeployCmd([]string{"test-deployment-id", "--force"}...)
 	s.NoError(err)
 
+	// Restore DagsOnlyDeploy to default behavior
+	DagsOnlyDeploy = deploy.DagsOnlyDeploy
+
 	s.Run("error should be returned for astro deploy, if DeployAirflowImage throws error", func() {
 		DeployAirflowImage = func(houstonClient houston.ClientInterface, path, deploymentID, wsID, byoRegistryDomain string, ignoreCacheDeploy, byoRegistryEnabled, prompt bool, description string, isImageOnlyDeploy bool, imageName string) (string, error) {
 			return deploymentID, deploy.ErrNoWorkspaceID

diff --git a/cmd/software/suite_test.go b/cmd/software/suite_test.go
@@ -23,6 +23,11 @@ func (s *Suite) SetupSuite() {
 func (s *Suite) SetupTest() {
 	// Reset the version once this is torn down
 	houstonVersion = "0.34.0"
+	DagsOnlyDeploy = deploy.DagsOnlyDeploy
+}
+
+func (s *Suite) SetupSubTest() {
+	DagsOnlyDeploy = deploy.DagsOnlyDeploy
 }
 
 func (s *Suite) TearDownSuite() {