mirror of
https://github.com/valitydev/salt.git
synced 2024-11-08 01:18:58 +00:00
81 lines
2.8 KiB
Python
81 lines
2.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Import python libs
|
|
import os
|
|
|
|
# Import Salt Testing libs
|
|
from salttesting import skipIf
|
|
from salttesting.helpers import ensure_in_syspath
|
|
from salttesting.mock import NO_MOCK, NO_MOCK_REASON, patch
|
|
ensure_in_syspath('../../')
|
|
|
|
# Import salt libs
|
|
import integration
|
|
import salt.utils
|
|
|
|
|
|
@skipIf(NO_MOCK, NO_MOCK_REASON)
|
|
class TestWhich(integration.TestCase):
|
|
'''
|
|
Tests salt.utils.which function to ensure that it returns True as
|
|
expected.
|
|
'''
|
|
|
|
@patch('salt.utils.which', lambda exe: None)
|
|
def test_missing_binary_in_linux(self):
|
|
self.assertTrue(
|
|
salt.utils.which('this-binary-does-not-exist') is None
|
|
)
|
|
|
|
@patch('salt.utils.which', lambda exe: exe)
|
|
def test_existing_binary_in_linux(self):
|
|
self.assertTrue(salt.utils.which('this-binary-exists-under-linux'))
|
|
|
|
@patch('os.access')
|
|
def test_existing_binary_in_windows(self, osaccess):
|
|
osaccess.side_effect = [
|
|
# The first os.access should return False(the abspath one)
|
|
False,
|
|
# The second, iterating through $PATH, should also return False,
|
|
# still checking for Linux
|
|
False,
|
|
# Lastly return True, this is the windows check.
|
|
True
|
|
]
|
|
# Let's patch os.environ to provide a custom PATH variable
|
|
with patch.dict(os.environ, {'PATH': '/bin'}):
|
|
# Let's also patch is_widows to return True
|
|
with patch('salt.utils.is_windows', lambda: True):
|
|
self.assertEqual(
|
|
salt.utils.which('this-binary-exists-under-windows'),
|
|
# The returned path should return the .exe suffix
|
|
'/bin/this-binary-exists-under-windows.exe'
|
|
)
|
|
|
|
@patch('os.access')
|
|
def test_missing_binary_in_windows(self, osaccess):
|
|
osaccess.side_effect = [
|
|
# The first os.access should return False(the abspath one)
|
|
False,
|
|
# The second, iterating through $PATH, should also return False,
|
|
# still checking for Linux
|
|
False,
|
|
# Lastly return True, this is the windows check.
|
|
True
|
|
]
|
|
# Let's patch os.environ to provide a custom PATH variable
|
|
with patch.dict(os.environ, {'PATH': '/bin'}):
|
|
# Let's also patch is_widows to return True
|
|
with patch('salt.utils.is_windows', lambda: True):
|
|
self.assertEqual(
|
|
# Since we're passing the .exe suffix, the last True above
|
|
# will not matter. The result will be None
|
|
salt.utils.which('this-binary-is-missing-in-windows.exe'),
|
|
None
|
|
)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
from integration import run_tests
|
|
run_tests(TestWhich, needs_daemon=False)
|