diff --git a/tests/test_enclosure_service.py b/tests/test_enclosure_service.py index 7eb4b8a..2162ab3 100644 --- a/tests/test_enclosure_service.py +++ b/tests/test_enclosure_service.py @@ -101,29 +101,24 @@ def test_enclosure_service(self): class TestCLI(unittest.TestCase): runner = CliRunner() - @patch("neon_enclosure.cli.init_config_dir") @patch("neon_enclosure.__main__.main") - def test_run(self, main, init_config): + def test_run(self, main): from neon_enclosure.cli import run self.runner.invoke(run) - init_config.assert_called_once() main.assert_called_once() @patch("os.geteuid") - @patch("neon_enclosure.cli.init_config_dir") @patch("neon_enclosure.admin.__main__.main") - def test_run_admin(self, main, init_config, get_id): + def test_run_admin(self, main, get_id): from neon_enclosure.cli import run_admin # Non-root get_id.return_value = 100 self.runner.invoke(run_admin) - init_config.assert_not_called() main.assert_not_called() # Root get_id.return_value = 0 self.runner.invoke(run_admin) - init_config.assert_called_once() main.assert_called_once()