mirror of
https://github.com/valitydev/salt.git
synced 2024-11-09 01:36:48 +00:00
438 lines
18 KiB
Python
438 lines
18 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Import Python Libs
|
|
from __future__ import absolute_import
|
|
from tempfile import NamedTemporaryFile
|
|
import os
|
|
|
|
# Import Salt Testing Libs
|
|
from tests.support.mixins import LoaderModuleMockMixin
|
|
from tests.support.unit import TestCase, skipIf
|
|
from tests.support.mock import (
|
|
MagicMock,
|
|
NO_MOCK,
|
|
NO_MOCK_REASON,
|
|
patch,
|
|
mock_open
|
|
)
|
|
|
|
# Import Salt Libs
|
|
from salt.exceptions import CommandExecutionError, SaltInvocationError
|
|
import salt.modules.timezone as timezone
|
|
import salt.ext.six as six
|
|
import salt.utils
|
|
|
|
GET_ZONE_FILE = 'salt.modules.timezone._get_zone_file'
|
|
GET_ETC_LOCALTIME_PATH = 'salt.modules.timezone._get_etc_localtime_path'
|
|
|
|
|
|
@skipIf(NO_MOCK, NO_MOCK_REASON)
|
|
class TimezoneTestCase(TestCase, LoaderModuleMockMixin):
|
|
|
|
def setup_loader_modules(self):
|
|
return {timezone: {'__grains__': {'os_family': 'Ubuntu'}}}
|
|
|
|
def setUp(self):
|
|
self.tempfiles = []
|
|
|
|
def tearDown(self):
|
|
for tempfile in self.tempfiles:
|
|
try:
|
|
os.remove(tempfile.name)
|
|
except OSError:
|
|
pass
|
|
del self.tempfiles
|
|
|
|
def test_zone_compare_equal(self):
|
|
etc_localtime = self.create_tempfile_with_contents('a')
|
|
zone_path = self.create_tempfile_with_contents('a')
|
|
|
|
with patch(GET_ZONE_FILE, lambda p: zone_path.name):
|
|
with patch(GET_ETC_LOCALTIME_PATH, lambda: etc_localtime.name):
|
|
|
|
self.assertTrue(timezone.zone_compare('foo'))
|
|
|
|
def test_zone_compare_nonexistent(self):
|
|
etc_localtime = self.create_tempfile_with_contents('a')
|
|
|
|
with patch(GET_ZONE_FILE, lambda p: '/foopath/nonexistent'):
|
|
with patch(GET_ETC_LOCALTIME_PATH, lambda: etc_localtime.name):
|
|
|
|
self.assertRaises(SaltInvocationError, timezone.zone_compare, 'foo')
|
|
|
|
def test_zone_compare_unequal(self):
|
|
etc_localtime = self.create_tempfile_with_contents('a')
|
|
zone_path = self.create_tempfile_with_contents('b')
|
|
|
|
with patch(GET_ZONE_FILE, lambda p: zone_path.name):
|
|
with patch(GET_ETC_LOCALTIME_PATH, lambda: etc_localtime.name):
|
|
|
|
self.assertFalse(timezone.zone_compare('foo'))
|
|
|
|
def test_missing_localtime(self):
|
|
with patch(GET_ZONE_FILE, lambda p: '/nonexisting'):
|
|
with patch(GET_ETC_LOCALTIME_PATH, lambda: '/also-missing'):
|
|
self.assertRaises(CommandExecutionError, timezone.zone_compare, 'foo')
|
|
|
|
def create_tempfile_with_contents(self, contents):
|
|
temp = NamedTemporaryFile(delete=False)
|
|
if six.PY3:
|
|
temp.write(salt.utils.to_bytes(contents))
|
|
else:
|
|
temp.write(contents)
|
|
temp.close()
|
|
self.tempfiles.append(temp)
|
|
return temp
|
|
|
|
|
|
@skipIf(NO_MOCK, NO_MOCK_REASON)
|
|
class TimezoneModuleTestCase(TestCase, LoaderModuleMockMixin):
|
|
'''
|
|
Timezone test case
|
|
'''
|
|
TEST_TZ = 'UTC'
|
|
|
|
def setup_loader_modules(self):
|
|
return {timezone: {'__grains__': {'os': ''},
|
|
'__salt__': {'file.sed': MagicMock(),
|
|
'cmd.run': MagicMock(),
|
|
'cmd.retcode': MagicMock(return_value=0)}}}
|
|
|
|
@patch('salt.utils.which', MagicMock(return_value=False))
|
|
def test_get_zone_centos(self):
|
|
'''
|
|
Test CentOS is recognized
|
|
:return:
|
|
'''
|
|
with patch.dict(timezone.__grains__, {'os': 'centos'}):
|
|
with patch('salt.modules.timezone._get_zone_etc_localtime', MagicMock(return_value=self.TEST_TZ)):
|
|
assert timezone.get_zone() == self.TEST_TZ
|
|
|
|
@patch('salt.utils.which', MagicMock(return_value=False))
|
|
def test_get_zone_os_family_rh_suse(self):
|
|
'''
|
|
Test RedHat and Suse are recognized
|
|
:return:
|
|
'''
|
|
for osfamily in ['RedHat', 'Suse']:
|
|
with patch.dict(timezone.__grains__, {'os_family': [osfamily]}):
|
|
with patch('salt.modules.timezone._get_zone_sysconfig', MagicMock(return_value=self.TEST_TZ)):
|
|
assert timezone.get_zone() == self.TEST_TZ
|
|
|
|
@patch('salt.utils.which', MagicMock(return_value=False))
|
|
def test_get_zone_os_family_debian_gentoo(self):
|
|
'''
|
|
Test Debian and Gentoo are recognized
|
|
:return:
|
|
'''
|
|
for osfamily in ['Debian', 'Gentoo']:
|
|
with patch.dict(timezone.__grains__, {'os_family': [osfamily]}):
|
|
with patch('salt.modules.timezone._get_zone_etc_timezone', MagicMock(return_value=self.TEST_TZ)):
|
|
assert timezone.get_zone() == self.TEST_TZ
|
|
|
|
@patch('salt.utils.which', MagicMock(return_value=False))
|
|
def test_get_zone_os_family_allbsd_nilinuxrt(self):
|
|
'''
|
|
Test *BSD and NILinuxRT are recognized
|
|
:return:
|
|
'''
|
|
for osfamily in ['FreeBSD', 'OpenBSD', 'NetBSD', 'NILinuxRT']:
|
|
with patch.dict(timezone.__grains__, {'os_family': osfamily}):
|
|
with patch('salt.modules.timezone._get_zone_etc_localtime', MagicMock(return_value=self.TEST_TZ)):
|
|
assert timezone.get_zone() == self.TEST_TZ
|
|
|
|
@patch('salt.utils.which', MagicMock(return_value=False))
|
|
def test_get_zone_os_family_slowlaris(self):
|
|
'''
|
|
Test Slowlaris is recognized
|
|
:return:
|
|
'''
|
|
with patch.dict(timezone.__grains__, {'os_family': ['Solaris']}):
|
|
with patch('salt.modules.timezone._get_zone_solaris', MagicMock(return_value=self.TEST_TZ)):
|
|
assert timezone.get_zone() == self.TEST_TZ
|
|
|
|
@patch('salt.utils.which', MagicMock(return_value=False))
|
|
def test_get_zone_os_family_aix(self):
|
|
'''
|
|
Test IBM AIX is recognized
|
|
:return:
|
|
'''
|
|
with patch.dict(timezone.__grains__, {'os_family': ['AIX']}):
|
|
with patch('salt.modules.timezone._get_zone_aix', MagicMock(return_value=self.TEST_TZ)):
|
|
assert timezone.get_zone() == self.TEST_TZ
|
|
|
|
@patch('salt.utils.which', MagicMock(return_value=False))
|
|
@patch('os.path.exists', MagicMock(return_value=True))
|
|
@patch('os.unlink', MagicMock())
|
|
@patch('os.symlink', MagicMock())
|
|
def test_set_zone_redhat(self):
|
|
'''
|
|
Test zone set on RH series
|
|
:return:
|
|
'''
|
|
with patch.dict(timezone.__grains__, {'os_family': ['RedHat']}):
|
|
assert timezone.set_zone(self.TEST_TZ)
|
|
name, args, kwargs = timezone.__salt__['file.sed'].mock_calls[0]
|
|
assert args == ('/etc/sysconfig/clock', '^ZONE=.*', 'ZONE="UTC"')
|
|
|
|
@patch('salt.utils.which', MagicMock(return_value=False))
|
|
@patch('os.path.exists', MagicMock(return_value=True))
|
|
@patch('os.unlink', MagicMock())
|
|
@patch('os.symlink', MagicMock())
|
|
def test_set_zone_suse(self):
|
|
'''
|
|
Test zone set on SUSE series
|
|
:return:
|
|
'''
|
|
with patch.dict(timezone.__grains__, {'os_family': ['Suse']}):
|
|
assert timezone.set_zone(self.TEST_TZ)
|
|
name, args, kwargs = timezone.__salt__['file.sed'].mock_calls[0]
|
|
assert args == ('/etc/sysconfig/clock', '^TIMEZONE=.*', 'TIMEZONE="UTC"')
|
|
|
|
@patch('salt.utils.which', MagicMock(return_value=False))
|
|
@patch('os.path.exists', MagicMock(return_value=True))
|
|
@patch('os.unlink', MagicMock())
|
|
@patch('os.symlink', MagicMock())
|
|
def test_set_zone_gentoo(self):
|
|
'''
|
|
Test zone set on Gentoo series
|
|
:return:
|
|
'''
|
|
with patch.dict(timezone.__grains__, {'os_family': ['Gentoo']}):
|
|
_fopen = mock_open()
|
|
with patch('salt.utils.fopen', _fopen):
|
|
assert timezone.set_zone(self.TEST_TZ)
|
|
name, args, kwargs = _fopen.mock_calls[0]
|
|
assert args == ('/etc/timezone', 'w')
|
|
name, args, kwargs = _fopen.return_value.__enter__.return_value.write.mock_calls[0]
|
|
assert args == ('UTC',)
|
|
|
|
@patch('salt.utils.which', MagicMock(return_value=False))
|
|
@patch('os.path.exists', MagicMock(return_value=True))
|
|
@patch('os.unlink', MagicMock())
|
|
@patch('os.symlink', MagicMock())
|
|
def test_set_zone_debian(self):
|
|
'''
|
|
Test zone set on Debian series
|
|
:return:
|
|
'''
|
|
with patch.dict(timezone.__grains__, {'os_family': ['Debian']}):
|
|
_fopen = mock_open()
|
|
with patch('salt.utils.fopen', _fopen):
|
|
assert timezone.set_zone(self.TEST_TZ)
|
|
name, args, kwargs = _fopen.mock_calls[0]
|
|
assert args == ('/etc/timezone', 'w')
|
|
name, args, kwargs = _fopen.return_value.__enter__.return_value.write.mock_calls[0]
|
|
assert args == ('UTC',)
|
|
|
|
@patch('salt.utils.which', MagicMock(return_value=True))
|
|
@patch('os.path.exists', MagicMock(return_value=True))
|
|
@patch('os.unlink', MagicMock())
|
|
@patch('os.symlink', MagicMock())
|
|
def test_get_hwclock_timedate_utc(self):
|
|
'''
|
|
Test get hwclock UTC/localtime
|
|
:return:
|
|
'''
|
|
with patch('salt.modules.timezone._timedatectl', MagicMock(return_value={'stdout': 'rtc in local tz'})):
|
|
assert timezone.get_hwclock() == 'UTC'
|
|
with patch('salt.modules.timezone._timedatectl', MagicMock(return_value={'stdout': 'rtc in local tz:yes'})):
|
|
assert timezone.get_hwclock() == 'localtime'
|
|
|
|
@patch('salt.utils.which', MagicMock(return_value=False))
|
|
@patch('os.path.exists', MagicMock(return_value=True))
|
|
@patch('os.unlink', MagicMock())
|
|
@patch('os.symlink', MagicMock())
|
|
def test_get_hwclock_suse(self):
|
|
'''
|
|
Test get hwclock on SUSE
|
|
:return:
|
|
'''
|
|
with patch.dict(timezone.__grains__, {'os_family': ['Suse']}):
|
|
timezone.get_hwclock()
|
|
name, args, kwarg = timezone.__salt__['cmd.run'].mock_calls[0]
|
|
assert args == (['tail', '-n', '1', '/etc/adjtime'],)
|
|
assert kwarg == {'python_shell': False}
|
|
|
|
@patch('salt.utils.which', MagicMock(return_value=False))
|
|
@patch('os.path.exists', MagicMock(return_value=True))
|
|
@patch('os.unlink', MagicMock())
|
|
@patch('os.symlink', MagicMock())
|
|
def test_get_hwclock_redhat(self):
|
|
'''
|
|
Test get hwclock on RedHat
|
|
:return:
|
|
'''
|
|
with patch.dict(timezone.__grains__, {'os_family': ['RedHat']}):
|
|
timezone.get_hwclock()
|
|
name, args, kwarg = timezone.__salt__['cmd.run'].mock_calls[0]
|
|
assert args == (['tail', '-n', '1', '/etc/adjtime'],)
|
|
assert kwarg == {'python_shell': False}
|
|
|
|
def _test_get_hwclock_debian(self): # TODO: Enable this when testing environment is working properly
|
|
'''
|
|
Test get hwclock on Debian
|
|
:return:
|
|
'''
|
|
with patch('salt.utils.which', MagicMock(return_value=False)):
|
|
with patch('os.path.exists', MagicMock(return_value=True)):
|
|
with patch('os.unlink', MagicMock()):
|
|
with patch('os.symlink', MagicMock()):
|
|
with patch.dict(timezone.__grains__, {'os_family': ['Debian']}):
|
|
timezone.get_hwclock()
|
|
name, args, kwarg = timezone.__salt__['cmd.run'].mock_calls[0]
|
|
assert args == (['tail', '-n', '1', '/etc/adjtime'],)
|
|
assert kwarg == {'python_shell': False}
|
|
|
|
@patch('salt.utils.which', MagicMock(return_value=False))
|
|
@patch('os.path.exists', MagicMock(return_value=True))
|
|
@patch('os.unlink', MagicMock())
|
|
@patch('os.symlink', MagicMock())
|
|
def test_get_hwclock_solaris(self):
|
|
'''
|
|
Test get hwclock on Solaris
|
|
:return:
|
|
'''
|
|
# Incomplete
|
|
with patch.dict(timezone.__grains__, {'os_family': ['Solaris']}):
|
|
assert timezone.get_hwclock() == 'UTC'
|
|
with patch('salt.utils.fopen', mock_open()):
|
|
assert timezone.get_hwclock() == 'localtime'
|
|
|
|
@patch('salt.utils.which', MagicMock(return_value=False))
|
|
@patch('os.path.exists', MagicMock(return_value=True))
|
|
@patch('os.unlink', MagicMock())
|
|
@patch('os.symlink', MagicMock())
|
|
def test_get_hwclock_aix(self):
|
|
'''
|
|
Test get hwclock on AIX
|
|
:return:
|
|
'''
|
|
# Incomplete
|
|
hwclock = 'localtime'
|
|
if not os.path.isfile('/etc/environment'):
|
|
hwclock = 'UTC'
|
|
with patch.dict(timezone.__grains__, {'os_family': ['AIX']}):
|
|
assert timezone.get_hwclock() == hwclock
|
|
|
|
@patch('salt.utils.which', MagicMock(return_value=False))
|
|
@patch('os.path.exists', MagicMock(return_value=True))
|
|
@patch('os.unlink', MagicMock())
|
|
@patch('os.symlink', MagicMock())
|
|
def test_set_hwclock_aix_nilinuxrt(self):
|
|
'''
|
|
Test set hwclock on AIX and NILinuxRT
|
|
:return:
|
|
'''
|
|
for osfamily in ['AIX', 'NILinuxRT']:
|
|
with patch.dict(timezone.__grains__, {'os_family': osfamily}):
|
|
with self.assertRaises(SaltInvocationError):
|
|
assert timezone.set_hwclock('forty two')
|
|
assert timezone.set_hwclock('UTC')
|
|
|
|
@patch('salt.utils.which', MagicMock(return_value=False))
|
|
@patch('os.path.exists', MagicMock(return_value=True))
|
|
@patch('os.unlink', MagicMock())
|
|
@patch('os.symlink', MagicMock())
|
|
@patch('salt.modules.timezone.get_zone', MagicMock(return_value='TEST_TIMEZONE'))
|
|
def test_set_hwclock_solaris(self):
|
|
'''
|
|
Test set hwclock on Solaris
|
|
:return:
|
|
'''
|
|
with patch.dict(timezone.__grains__, {'os_family': ['Solaris'],
|
|
'cpuarch': 'x86'}):
|
|
with self.assertRaises(SaltInvocationError):
|
|
assert timezone.set_hwclock('forty two')
|
|
assert timezone.set_hwclock('UTC')
|
|
name, args, kwargs = timezone.__salt__['cmd.retcode'].mock_calls[0]
|
|
assert args == (['rtc', '-z', 'GMT'],)
|
|
assert kwargs == {'python_shell': False}
|
|
|
|
@patch('salt.utils.which', MagicMock(return_value=False))
|
|
@patch('os.path.exists', MagicMock(return_value=True))
|
|
@patch('os.unlink', MagicMock())
|
|
@patch('os.symlink', MagicMock())
|
|
@patch('salt.modules.timezone.get_zone', MagicMock(return_value='TEST_TIMEZONE'))
|
|
def test_set_hwclock_arch(self):
|
|
'''
|
|
Test set hwclock on arch
|
|
:return:
|
|
'''
|
|
with patch.dict(timezone.__grains__, {'os_family': ['Arch']}):
|
|
assert timezone.set_hwclock('UTC')
|
|
name, args, kwargs = timezone.__salt__['cmd.retcode'].mock_calls[0]
|
|
assert args == (['timezonectl', 'set-local-rtc', 'false'],)
|
|
assert kwargs == {'python_shell': False}
|
|
|
|
@patch('salt.utils.which', MagicMock(return_value=False))
|
|
@patch('os.path.exists', MagicMock(return_value=True))
|
|
@patch('os.unlink', MagicMock())
|
|
@patch('os.symlink', MagicMock())
|
|
@patch('salt.modules.timezone.get_zone', MagicMock(return_value='TEST_TIMEZONE'))
|
|
def test_set_hwclock_redhat(self):
|
|
'''
|
|
Test set hwclock on RedHat
|
|
:return:
|
|
'''
|
|
with patch.dict(timezone.__grains__, {'os_family': ['RedHat']}):
|
|
assert timezone.set_hwclock('UTC')
|
|
name, args, kwargs = timezone.__salt__['file.sed'].mock_calls[0]
|
|
assert args == ('/etc/sysconfig/clock', '^ZONE=.*', 'ZONE="TEST_TIMEZONE"')
|
|
|
|
@patch('salt.utils.which', MagicMock(return_value=False))
|
|
@patch('os.path.exists', MagicMock(return_value=True))
|
|
@patch('os.unlink', MagicMock())
|
|
@patch('os.symlink', MagicMock())
|
|
@patch('salt.modules.timezone.get_zone', MagicMock(return_value='TEST_TIMEZONE'))
|
|
def test_set_hwclock_suse(self):
|
|
'''
|
|
Test set hwclock on SUSE
|
|
:return:
|
|
'''
|
|
with patch.dict(timezone.__grains__, {'os_family': ['Suse']}):
|
|
assert timezone.set_hwclock('UTC')
|
|
name, args, kwargs = timezone.__salt__['file.sed'].mock_calls[0]
|
|
assert args == ('/etc/sysconfig/clock', '^TIMEZONE=.*', 'TIMEZONE="TEST_TIMEZONE"')
|
|
|
|
@patch('salt.utils.which', MagicMock(return_value=False))
|
|
@patch('os.path.exists', MagicMock(return_value=True))
|
|
@patch('os.unlink', MagicMock())
|
|
@patch('os.symlink', MagicMock())
|
|
@patch('salt.modules.timezone.get_zone', MagicMock(return_value='TEST_TIMEZONE'))
|
|
def test_set_hwclock_debian(self):
|
|
'''
|
|
Test set hwclock on Debian
|
|
:return:
|
|
'''
|
|
with patch.dict(timezone.__grains__, {'os_family': ['Debian']}):
|
|
assert timezone.set_hwclock('UTC')
|
|
name, args, kwargs = timezone.__salt__['file.sed'].mock_calls[0]
|
|
assert args == ('/etc/default/rcS', '^UTC=.*', 'UTC=yes')
|
|
|
|
assert timezone.set_hwclock('localtime')
|
|
name, args, kwargs = timezone.__salt__['file.sed'].mock_calls[1]
|
|
assert args == ('/etc/default/rcS', '^UTC=.*', 'UTC=no')
|
|
|
|
@patch('salt.utils.which', MagicMock(return_value=False))
|
|
@patch('os.path.exists', MagicMock(return_value=True))
|
|
@patch('os.unlink', MagicMock())
|
|
@patch('os.symlink', MagicMock())
|
|
@patch('salt.modules.timezone.get_zone', MagicMock(return_value='TEST_TIMEZONE'))
|
|
def test_set_hwclock_gentoo(self):
|
|
'''
|
|
Test set hwclock on Gentoo
|
|
:return:
|
|
'''
|
|
with patch.dict(timezone.__grains__, {'os_family': ['Gentoo']}):
|
|
with self.assertRaises(SaltInvocationError):
|
|
timezone.set_hwclock('forty two')
|
|
|
|
timezone.set_hwclock('UTC')
|
|
name, args, kwargs = timezone.__salt__['file.sed'].mock_calls[0]
|
|
assert args == ('/etc/conf.d/hwclock', '^clock=.*', 'clock="UTC"')
|
|
|
|
timezone.set_hwclock('localtime')
|
|
name, args, kwargs = timezone.__salt__['file.sed'].mock_calls[1]
|
|
assert args == ('/etc/conf.d/hwclock', '^clock=.*', 'clock="local"')
|