diff options
Diffstat (limited to 'tests/unit/test_utils.py')
| -rw-r--r-- | tests/unit/test_utils.py | 19 |
1 files changed, 18 insertions, 1 deletions
diff --git a/tests/unit/test_utils.py b/tests/unit/test_utils.py index 22597dca..41a06f63 100644 --- a/tests/unit/test_utils.py +++ b/tests/unit/test_utils.py @@ -7,11 +7,14 @@ # http://www.apache.org/licenses/LICENSE-2.0 ############################################################################## import os - +import mock import pytest import mock +from deploy import utils from deploy.utils import ( + err_exit, + check_sudo_privilege, check_file_exists, make_file_executable, confirm_dir_exists, @@ -19,6 +22,20 @@ from deploy.utils import ( ) +def test_err_exit(): + message = 'test error msg!' + with pytest.raises(SystemExit): + err_exit(message) + + +@mock.patch('deploy.utils.err_exit') +@mock.patch('deploy.utils.os.getuid') +def test_check_sudo_privilege(mock_getuid, mock_err_exit): + mock_getuid.return_value = 1 + check_sudo_privilege() + utils.err_exit.assert_called_once_with('You need run this script with sudo privilege') + + @pytest.mark.parametrize('test_file_name', [ ('no_exist_file'), ('exist_file')]) |