Fix unit test

SUSE · Jan 16, 2025 · 05fdf33 · 05fdf33
1 parent c4bcf7d
commit 05fdf33
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/jenkins_pipelines/scripts/test_environment_cleaner/tests/test_main_program.py b/jenkins_pipelines/scripts/test_environment_cleaner/tests/test_main_program.py
@@ -9,11 +9,13 @@
 from test_environment_cleaner_program.TestEnvironmentCleaner import main
 
 class TestMainProgram(unittest.TestCase):
+    def setUp(self):
+        self.product_version = "5.0"
 
     @patch("test_environment_cleaner_program.TestEnvironmentCleaner.ResourceManager")
     @patch("test_environment_cleaner_program.TestEnvironmentCleaner.SSHClientManager")
     def test_delete_users_mode(self, MockSSHClientManager, MockResourceManager):
-        test_args = ["test_environment_cleaner_program.TestEnvironmentCleaner", "--url", "http://test-url.com", "--mode", "delete_users", "--product_version", "5.0"]
+        test_args = ["test_environment_cleaner_program.TestEnvironmentCleaner", "--url", "http://test-url.com", "--mode", "delete_users"]
         with patch.object(sys, 'argv', test_args):
             mock_resource_manager = MockResourceManager.return_value
             mock_resource_manager.get_session_key.return_value = "session_key"
@@ -28,7 +30,7 @@ def test_delete_users_mode(self, MockSSHClientManager, MockResourceManager):
     @patch("test_environment_cleaner_program.TestEnvironmentCleaner.ResourceManager")
     @patch("test_environment_cleaner_program.TestEnvironmentCleaner.SSHClientManager")
     def test_delete_known_hosts_mode(self, MockSSHClientManager, MockResourceManager):
-        test_args = ["test_environment_cleaner_program.TestEnvironmentCleaner", "--url", "http://test-url.com", "--mode", "delete_known_hosts", "--product_version", "5.0"]
+        test_args = ["test_environment_cleaner_program.TestEnvironmentCleaner", "--url", "http://test-url.com", "--mode", "delete_known_hosts"]
         with patch.object(sys, 'argv', test_args):
             mock_ssh_manager = MockSSHClientManager.return_value
 
@@ -40,7 +42,7 @@ def test_delete_known_hosts_mode(self, MockSSHClientManager, MockResourceManager
     @patch("test_environment_cleaner_program.TestEnvironmentCleaner.ResourceManager")
     @patch("test_environment_cleaner_program.TestEnvironmentCleaner.SSHClientManager")
     def test_invalid_mode(self, MockSSHClientManager, MockResourceManager):
-        test_args = ["test_environment_cleaner_program.TestEnvironmentCleaner", "--url", "http://test-url.com", "--mode", "invalid", "--product_version", "5.0"]
+        test_args = ["test_environment_cleaner_program.TestEnvironmentCleaner", "--url", "http://test-url.com", "--mode", "invalid"]
         with patch.object(sys, 'argv', test_args):
             with patch("sys.stderr", new=StringIO()) as fake_err:
                 with self.assertRaises(SystemExit):
@@ -51,7 +53,7 @@ def test_invalid_mode(self, MockSSHClientManager, MockResourceManager):
                     "usage: test_environment_cleaner_program.TestEnvironmentCleaner [-h] --url URL --mode "
                     "{delete_users,delete_activation_keys,delete_config_projects,delete_software_channels,"
                     "delete_systems,delete_repositories,full_cleanup,delete_salt_keys,delete_known_hosts,update_custom_repositories,delete_distributions}"
-                    " [--default-resources-to-delete [{proxy,monitoring-server,retail} ...]] --product_version {head,5.1,5.0,4.3,uyuni} "
+                    " [--default-resources-to-delete [{proxy,monitoring-server,retail} ...]] "
                     "test_environment_cleaner_program.TestEnvironmentCleaner: error: argument --mode: invalid choice: 'invalid' (choose from 'delete_users',"
                     " 'delete_activation_keys', 'delete_config_projects', 'delete_software_channels', 'delete_systems',"
                     " 'delete_repositories', 'full_cleanup', 'delete_salt_keys', 'delete_known_hosts', 'update_custom_repositories',"
@@ -63,7 +65,7 @@ def test_invalid_mode(self, MockSSHClientManager, MockResourceManager):
     @patch("test_environment_cleaner_program.TestEnvironmentCleaner.ResourceManager")
     @patch("test_environment_cleaner_program.TestEnvironmentCleaner.SSHClientManager")
     def test_update_custom_repositories_mode(self, MockSSHClientManager, MockResourceManager):
-        test_args = ["test_environment_cleaner_program.TestEnvironmentCleaner", "--url", "http://test-url.com", "--mode", "update_custom_repositories", "--product_version", "5.0"]
+        test_args = ["test_environment_cleaner_program.TestEnvironmentCleaner", "--url", "http://test-url.com", "--mode", "update_custom_repositories"]
         with patch.object(sys, 'argv', test_args):
             mock_ssh_manager = MockSSHClientManager.return_value