mirror of
https://github.com/valitydev/salt.git
synced 2024-11-07 08:58:59 +00:00
98 lines
3.4 KiB
Python
98 lines
3.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
'''
|
|
:codeauthor: :email:`Rahul Handay <rahulha@saltstack.com>`
|
|
'''
|
|
|
|
# Import Python Libs
|
|
from __future__ import absolute_import
|
|
|
|
# Import Salt Testing Libs
|
|
from salttesting import TestCase, skipIf
|
|
from salttesting.helpers import ensure_in_syspath
|
|
from salttesting.mock import (
|
|
MagicMock,
|
|
patch,
|
|
NO_MOCK,
|
|
NO_MOCK_REASON
|
|
)
|
|
|
|
ensure_in_syspath('../../')
|
|
|
|
# Import Salt Libs
|
|
from salt.states import locale
|
|
|
|
# Globals
|
|
locale.__salt__ = {}
|
|
locale.__opts__ = {}
|
|
|
|
|
|
@skipIf(NO_MOCK, NO_MOCK_REASON)
|
|
class LocaleTestCase(TestCase):
|
|
'''
|
|
Validate the locale state
|
|
'''
|
|
def test_system(self):
|
|
'''
|
|
Test to set the locale for the system
|
|
'''
|
|
ret = [{'changes': {}, 'comment': 'System locale salt already set',
|
|
'name': 'salt', 'result': True},
|
|
{'changes': {},
|
|
'comment': 'System locale saltstack needs to be set',
|
|
'name': 'saltstack', 'result': None},
|
|
{'changes': {'locale': 'saltstack'},
|
|
'comment': 'Set system locale saltstack', 'name': 'saltstack',
|
|
'result': True},
|
|
{'changes': {},
|
|
'comment': 'Failed to set system locale to saltstack',
|
|
'name': 'saltstack', 'result': False}]
|
|
|
|
mock = MagicMock(return_value="salt")
|
|
with patch.dict(locale.__salt__, {"locale.get_locale": mock}):
|
|
self.assertDictEqual(locale.system("salt"), ret[0])
|
|
|
|
with patch.dict(locale.__opts__, {"test": True}):
|
|
self.assertDictEqual(locale.system("saltstack"), ret[1])
|
|
|
|
with patch.dict(locale.__opts__, {"test": False}):
|
|
mock = MagicMock(side_effect=[True, False])
|
|
with patch.dict(locale.__salt__, {"locale.set_locale": mock}):
|
|
self.assertDictEqual(locale.system("saltstack"), ret[2])
|
|
|
|
self.assertDictEqual(locale.system("saltstack"), ret[3])
|
|
|
|
def test_present(self):
|
|
'''
|
|
Test to generate a locale if it is not present
|
|
'''
|
|
ret = [{'changes': {},
|
|
'comment': 'Locale salt is already present', 'name': 'salt',
|
|
'result': True},
|
|
{'changes': {},
|
|
'comment': 'Locale salt needs to be generated', 'name': 'salt',
|
|
'result': None},
|
|
{'changes': {'locale': 'salt'},
|
|
'comment': 'Generated locale salt', 'name': 'salt',
|
|
'result': True},
|
|
{'changes': {}, 'comment': 'Failed to generate locale salt',
|
|
'name': 'salt', 'result': False}]
|
|
|
|
mock = MagicMock(side_effect=[True, False, False, False])
|
|
with patch.dict(locale.__salt__, {"locale.avail": mock}):
|
|
self.assertDictEqual(locale.present("salt"), ret[0])
|
|
|
|
with patch.dict(locale.__opts__, {"test": True}):
|
|
self.assertDictEqual(locale.present("salt"), ret[1])
|
|
|
|
with patch.dict(locale.__opts__, {"test": False}):
|
|
mock = MagicMock(side_effect=[True, False])
|
|
with patch.dict(locale.__salt__, {"locale.gen_locale": mock}):
|
|
self.assertDictEqual(locale.present("salt"), ret[2])
|
|
|
|
self.assertDictEqual(locale.present("salt"), ret[3])
|
|
|
|
|
|
if __name__ == '__main__':
|
|
from integration import run_tests
|
|
run_tests(LocaleTestCase, needs_daemon=False)
|