summaryrefslogtreecommitdiffstats
path: root/tests
diff options
context:
space:
mode:
Diffstat (limited to 'tests')
-rw-r--r--tests/unit/test_utils.py19
1 files changed, 8 insertions, 11 deletions
diff --git a/tests/unit/test_utils.py b/tests/unit/test_utils.py
index e3b9dff7..4998a447 100644
--- a/tests/unit/test_utils.py
+++ b/tests/unit/test_utils.py
@@ -183,19 +183,16 @@ def test_ipmi_reboot_node(mock_getstatusoutput, mock_err_exit,
@pytest.mark.parametrize('cmd, check, expect', [
- ('cd /home', False, 0),
- ('cd /home', True, 0),
+ ('ls /home', False, 0),
+ ('ls /home', True, 0),
('test_command', False, 127),
('test_command', True, 127)])
-@mock.patch('deploy.utils.err_exit')
-def test_run_shell(mock_err_exit, cmd, check, expect):
- ret = run_shell(cmd, check=check)
- if check:
- if cmd == 'cd /home':
- mock_err_exit.assert_not_called()
- elif cmd == 'test_command':
- mock_err_exit.assert_called_once()
- assert ret == expect
+def test_run_shell(cmd, check, expect):
+ try:
+ ret = run_shell(cmd, check=check)
+ assert ret == expect
+ except OSError:
+ assert cmd == 'test_command'
@pytest.mark.parametrize('scenario', [