Adding states/locale.py unit test cases

tests/unit/states/locale_test.py

@@ -0,0 +1,97 @@
+# -*- 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)