salt/tests/unit/states/test_locale.py
rallytime 3273bbdab7
Merge branch '2017.7' into '2018.3'
Conflicts:
  - doc/ref/configuration/master.rst
  - doc/ref/modules/all/index.rst
  - doc/topics/grains/index.rst
  - doc/topics/releases/2016.3.4.rst
  - doc/topics/spm/spm_formula.rst
  - doc/topics/tutorials/cron.rst
  - doc/topics/tutorials/index.rst
  - doc/topics/tutorials/stormpath.rst
  - salt/engines/slack.py
  - salt/log/handlers/fluent_mod.py
  - salt/modules/cyg.py
  - salt/modules/junos.py
  - salt/modules/namecheap_dns.py
  - salt/modules/namecheap_domains.py
  - salt/modules/namecheap_ns.py
  - salt/modules/namecheap_ssl.py
  - salt/modules/namecheap_users.py
  - salt/modules/reg.py
  - salt/modules/tomcat.py
  - salt/modules/vault.py
  - salt/modules/win_file.py
  - salt/modules/zpool.py
  - salt/output/highstate.py
  - salt/renderers/pass.py
  - salt/runners/cache.py
  - salt/states/boto_apigateway.py
  - salt/states/boto_iam.py
  - salt/states/boto_route53.py
  - salt/states/msteams.py
  - salt/states/reg.py
  - salt/states/win_iis.py
  - tests/integration/modules/test_cmdmod.py
  - tests/integration/states/test_user.py
  - tests/support/helpers.py
  - tests/unit/cloud/clouds/test_openstack.py
  - tests/unit/fileserver/test_gitfs.py
  - tests/unit/modules/test_junos.py
  - tests/unit/pillar/test_git.py
  - tests/unit/states/test_win_path.py
  - tests/unit/test_pillar.py
  - tests/unit/utils/test_format_call.py
  - tests/unit/utils/test_utils.py
  - tests/unit/utils/test_warnings.py
2018-06-01 14:54:12 -04:00

90 lines
3.3 KiB
Python

# -*- coding: utf-8 -*-
'''
:codeauthor: Rahul Handay <rahulha@saltstack.com>
'''
# Import Python Libs
from __future__ import absolute_import, unicode_literals, print_function
# Import Salt Testing Libs
from tests.support.mixins import LoaderModuleMockMixin
from tests.support.unit import TestCase, skipIf
from tests.support.mock import (
MagicMock,
patch,
NO_MOCK,
NO_MOCK_REASON
)
# Import Salt Libs
import salt.states.locale as locale
@skipIf(NO_MOCK, NO_MOCK_REASON)
class LocaleTestCase(TestCase, LoaderModuleMockMixin):
'''
Validate the locale state
'''
def setup_loader_modules(self):
return {locale: {}}
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])