mirror of
https://github.com/valitydev/salt.git
synced 2024-11-09 01:36:48 +00:00
102 lines
3.2 KiB
Python
102 lines
3.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
'''
|
|
:codeauthor: :email:`Jayesh Kariya <jayeshk@saltstack.com>`
|
|
'''
|
|
|
|
# Import Python Libs
|
|
from __future__ import absolute_import
|
|
|
|
# Import Salt Testing Libs
|
|
from tests.support.mixins import LoaderModuleMockMixin
|
|
from tests.support.unit import TestCase
|
|
from tests.support.mock import (
|
|
MagicMock,
|
|
patch
|
|
)
|
|
|
|
# Import Salt Libs
|
|
import salt.modules.win_timezone as win_timezone
|
|
|
|
|
|
class WinTimezoneTestCase(TestCase, LoaderModuleMockMixin):
|
|
'''
|
|
Test cases for salt.modules.win_timezone
|
|
'''
|
|
def setup_loader_modules(self):
|
|
return {win_timezone: {}}
|
|
|
|
# 'get_zone' function tests: 1
|
|
|
|
def test_get_zone(self):
|
|
'''
|
|
Test if it get current timezone (i.e. Asia/Calcutta)
|
|
'''
|
|
mock_cmd = MagicMock(side_effect=['India Standard Time',
|
|
'Indian Standard Time'])
|
|
with patch.dict(win_timezone.__salt__, {'cmd.run': mock_cmd}):
|
|
self.assertEqual(win_timezone.get_zone(), 'Asia/Calcutta')
|
|
|
|
self.assertFalse(win_timezone.get_zone())
|
|
|
|
# 'get_offset' function tests: 1
|
|
|
|
def test_get_offset(self):
|
|
'''
|
|
Test if it get current numeric timezone offset from UCT (i.e. +0530)
|
|
'''
|
|
time = ('(UTC+05:30) Chennai, Kolkata, Mumbai, \
|
|
New Delhi\nIndia Standard Time')
|
|
mock_cmd = MagicMock(side_effect=['India Standard Time', time])
|
|
with patch.dict(win_timezone.__salt__, {'cmd.run': mock_cmd}):
|
|
self.assertEqual(win_timezone.get_offset(), '+0530')
|
|
|
|
mock_cmd = MagicMock(return_value='India Standard Time')
|
|
with patch.dict(win_timezone.__salt__, {'cmd.run': mock_cmd}):
|
|
self.assertFalse(win_timezone.get_offset())
|
|
|
|
# 'get_zonecode' function tests: 1
|
|
|
|
def test_get_zonecode(self):
|
|
'''
|
|
Test if it get current timezone (i.e. PST, MDT, etc)
|
|
'''
|
|
self.assertFalse(win_timezone.get_zonecode())
|
|
|
|
# 'set_zone' function tests: 1
|
|
|
|
def test_set_zone(self):
|
|
'''
|
|
Test if it unlinks, then symlinks /etc/localtime to the set timezone.
|
|
'''
|
|
mock_cmd = MagicMock(return_value=0)
|
|
with patch.dict(win_timezone.__salt__, {'cmd.retcode': mock_cmd}):
|
|
self.assertTrue(win_timezone.set_zone('Asia/Calcutta'))
|
|
|
|
# 'zone_compare' function tests: 1
|
|
|
|
def test_zone_compare(self):
|
|
'''
|
|
Test if it checks the md5sum between the given timezone, and
|
|
the one set in /etc/localtime. Returns True if they match,
|
|
and False if not. Mostly useful for running state checks.
|
|
'''
|
|
mock_cmd = MagicMock(return_value='India Standard Time')
|
|
with patch.dict(win_timezone.__salt__, {'cmd.run': mock_cmd}):
|
|
self.assertTrue(win_timezone.zone_compare('Asia/Calcutta'))
|
|
|
|
# 'get_hwclock' function tests: 1
|
|
|
|
def test_get_hwclock(self):
|
|
'''
|
|
Test if it get current hardware clock setting (UTC or localtime)
|
|
'''
|
|
self.assertEqual(win_timezone.get_hwclock(), 'localtime')
|
|
|
|
# 'set_hwclock' function tests: 1
|
|
|
|
def test_set_hwclock(self):
|
|
'''
|
|
Test if it sets the hardware clock to be either UTC or localtime
|
|
'''
|
|
self.assertFalse(win_timezone.set_hwclock('UTC'))
|