def setUp(self):
"""Set up the timezone module."""
# Set up the timezone module.
self.timezone_module = TimezoneService()
self.timezone_interface = TimezoneInterface(self.timezone_module)
# Connect to the properties changed signal.
self.callback = Mock()
self.timezone_interface.PropertiesChanged.connect(self.callback)
def setUp(self):
"""Set up the timezone module."""
# Set up the timezone module.
self.timezone_module = TimezoneService()
self.timezone_interface = TimezoneInterface(self.timezone_module)
class TimezoneInterfaceTestCase(unittest.TestCase):
"""Test DBus interface for the timezone module."""
def setUp(self):
"""Set up the timezone module."""
# Set up the timezone module.
self.timezone_module = TimezoneService()
self.timezone_interface = TimezoneInterface(self.timezone_module)
def _check_dbus_property(self, *args, **kwargs):
check_dbus_property(TIMEZONE, self.timezone_interface, *args, **kwargs)
def test_kickstart_properties(self):
"""Test kickstart properties."""
assert self.timezone_interface.KickstartCommands == [
"timezone", "timesource"
]
assert self.timezone_interface.KickstartSections == []
assert self.timezone_interface.KickstartAddons == []
def test_timezone_property(self):
"""Test the Timezone property."""
self._check_dbus_property("Timezone", "Europe/Prague")
def test_utc_property(self):
"""Test the IsUtc property."""
self._check_dbus_property("IsUTC", True)
def test_ntp_property(self):
"""Test the NTPEnabled property."""
self._check_dbus_property("NTPEnabled", False)
def test_time_sources_property(self):
"""Test the TimeSources property."""
server = {
"type": get_variant(Str, TIME_SOURCE_SERVER),
"hostname": get_variant(Str, "ntp.cesnet.cz"),
"options": get_variant(List[Str], ["iburst"]),
}
pool = {
"type": get_variant(Str, TIME_SOURCE_POOL),
"hostname": get_variant(Str, "0.fedora.pool.ntp.org"),
"options": get_variant(List[Str], []),
}
self._check_dbus_property("TimeSources", [server, pool])
def _test_kickstart(self, ks_in, ks_out):
check_kickstart_interface(self.timezone_interface, ks_in, ks_out)
def test_no_kickstart(self):
"""Test with no kickstart."""
ks_in = None
ks_out = """
# System timezone
timezone America/New_York
"""
self._test_kickstart(ks_in, ks_out)
def test_kickstart_empty(self):
"""Test with empty string."""
ks_in = ""
ks_out = ""
self._test_kickstart(ks_in, ks_out)
def test_kickstart(self):
"""Test the timezone command."""
ks_in = """
timezone Europe/Prague
"""
ks_out = """
# System timezone
timezone Europe/Prague
"""
self._test_kickstart(ks_in, ks_out)
def test_kickstart2(self):
"""Test the timezone command with flags."""
ks_in = """
timezone --utc --nontp Europe/Prague
"""
ks_out = """
timesource --ntp-disable
# System timezone
timezone Europe/Prague --utc
"""
self._test_kickstart(ks_in, ks_out)
def test_kickstart3(self):
"""Test the timezone command with ntp servers."""
ks_in = """
timezone --ntpservers ntp.cesnet.cz Europe/Prague
"""
ks_out = """
timesource --ntp-server=ntp.cesnet.cz
# System timezone
timezone Europe/Prague
"""
self._test_kickstart(ks_in, ks_out)
def test_kickstart_timesource_ntp_disabled(self):
"""Test the timesource command with ntp disabled."""
ks_in = """
timesource --ntp-disable
"""
ks_out = """
timesource --ntp-disable
"""
self._test_kickstart(ks_in, ks_out)
def test_kickstart_timesource_ntp_server(self):
"""Test the timesource command with ntp servers."""
ks_in = """
timesource --ntp-server ntp.cesnet.cz
"""
ks_out = """
timesource --ntp-server=ntp.cesnet.cz
"""
self._test_kickstart(ks_in, ks_out)
def test_kickstart_timesource_ntp_pool(self):
"""Test the timesource command with ntp pools."""
ks_in = """
timesource --ntp-pool ntp.cesnet.cz
"""
ks_out = """
timesource --ntp-pool=ntp.cesnet.cz
"""
self._test_kickstart(ks_in, ks_out)
def test_kickstart_timesource_nts(self):
"""Test the timesource command with the nts option."""
ks_in = """
timesource --ntp-pool ntp.cesnet.cz --nts
"""
ks_out = """
timesource --ntp-pool=ntp.cesnet.cz --nts
"""
self._test_kickstart(ks_in, ks_out)
def test_kickstart_timesource_all(self):
"""Test the timesource commands."""
ks_in = """
timesource --ntp-server ntp.cesnet.cz
timesource --ntp-pool 0.fedora.pool.ntp.org
"""
ks_out = """
timesource --ntp-server=ntp.cesnet.cz
timesource --ntp-pool=0.fedora.pool.ntp.org
"""
self._test_kickstart(ks_in, ks_out)
def test_kickstart_timezone_timesource(self):
"""Test the combination of timezone and timesource commands."""
ks_in = """
timezone --ntpservers ntp.cesnet.cz,0.fedora.pool.ntp.org Europe/Prague
timesource --ntp-server ntp.cesnet.cz --nts
timesource --ntp-pool 0.fedora.pool.ntp.org
"""
ks_out = """
timesource --ntp-server=ntp.cesnet.cz
timesource --ntp-server=0.fedora.pool.ntp.org
timesource --ntp-server=ntp.cesnet.cz --nts
timesource --ntp-pool=0.fedora.pool.ntp.org
# System timezone
timezone Europe/Prague
"""
self._test_kickstart(ks_in, ks_out)
def test_collect_requirements(self):
"""Test the requirements of the Timezone module."""
# Check the default requirements.
requirements = Requirement.from_structure_list(
self.timezone_interface.CollectRequirements())
assert len(requirements) == 1
assert requirements[0].type == "package"
assert requirements[0].name == "chrony"
# Check requirements with disabled NTP service.
self.timezone_interface.NTPEnabled = False
requirements = Requirement.from_structure_list(
self.timezone_interface.CollectRequirements())
assert len(requirements) == 0
@patch_dbus_publish_object
def test_install_with_tasks_default(self, publisher):
"""Test install tasks - module in default state."""
task_classes = [
ConfigureHardwareClockTask,
ConfigureTimezoneTask,
ConfigureNTPTask,
]
task_paths = self.timezone_interface.InstallWithTasks()
task_objs = check_task_creation_list(task_paths, publisher,
task_classes)
# ConfigureHardwareClockTask
obj = task_objs[0]
assert obj.implementation._is_utc is False
# ConfigureTimezoneTask
obj = task_objs[1]
assert obj.implementation._timezone == "America/New_York"
assert obj.implementation._is_utc is False
# ConfigureNTPTask
obj = task_objs[2]
assert obj.implementation._ntp_enabled is True
assert obj.implementation._ntp_servers == []
@patch_dbus_publish_object
def test_install_with_tasks_configured(self, publisher):
"""Test install tasks - module in configured state."""
self.timezone_interface.IsUTC = True
self.timezone_interface.Timezone = "Asia/Tokyo"
self.timezone_interface.NTPEnabled = False
# --nontp and --ntpservers are mutually exclusive in kicstart but
# there is no such enforcement in the module so for testing this is ok
server = TimeSourceData()
server.type = TIME_SOURCE_SERVER
server.hostname = "clock1.example.com"
server.options = ["iburst"]
pool = TimeSourceData()
pool.type = TIME_SOURCE_POOL
pool.hostname = "clock2.example.com"
self.timezone_interface.TimeSources = \
TimeSourceData.to_structure_list([server, pool])
task_classes = [
ConfigureHardwareClockTask,
ConfigureTimezoneTask,
ConfigureNTPTask,
]
task_paths = self.timezone_interface.InstallWithTasks()
task_objs = check_task_creation_list(task_paths, publisher,
task_classes)
# ConfigureHardwareClockTask
obj = task_objs[0]
assert obj.implementation._is_utc is True
# ConfigureTimezoneTask
obj = task_objs[1]
assert obj.implementation._timezone == "Asia/Tokyo"
assert obj.implementation._is_utc is True
# ConfigureNTPTask
obj = task_objs[2]
assert obj.implementation._ntp_enabled is False
assert len(obj.implementation._ntp_servers) == 2
assert compare_data(obj.implementation._ntp_servers[0], server)
assert compare_data(obj.implementation._ntp_servers[1], pool)
def test_deprecated_warnings(self):
response = self.timezone_interface.ReadKickstart(
"timezone --isUtc Europe/Bratislava")
report = KickstartReport.from_structure(response)
warning = "The option --isUtc will be deprecated in future releases. " \
"Please modify your kickstart file to replace this option with " \
"its preferred alias --utc."
assert len(report.warning_messages) == 1
assert report.warning_messages[0].message == warning
@patch("pyanaconda.modules.timezone.timezone.GeolocationTask")
def test_geoloc_interface(self, geoloc_mock):
"""Test geolocation-related interface and implementation of Timezone"""
# without any actions, we should get empty GeolocationData
new = GeolocationData.from_structure(
self.timezone_interface.GeolocationResult)
assert new.is_empty()
# let's "run" geolocation
self.timezone_interface.StartGeolocation()
# the task should have been instantiated, signal connected, and started
geoloc_mock.assert_called_once_with()
mock_task = geoloc_mock.return_value
mock_task.start.assert_called_once_with()
mock_task.succeeded_signal.connect.assert_called_once_with(
self.timezone_module._set_geolocation_result)
# running the callback for finishing should also save result
self.timezone_module._set_geolocation_result()
mock_task.get_result.assert_called_once_with()
assert self.timezone_module.geolocation_result == mock_task.get_result.return_value
# try re-running, same as earlier
geoloc_mock.reset_mock()
self.timezone_interface.StartGeolocation()
geoloc_mock.assert_called_once_with()
# # Timezone DBus service launcher. # # Copyright (C) 2020 Red Hat, Inc. # # This copyrighted material is made available to anyone wishing to use, # modify, copy, or redistribute it subject to the terms and conditions of # the GNU General Public License v.2, or (at your option) any later version. # This program is distributed in the hope that it will be useful, but WITHOUT # ANY WARRANTY expressed or implied, including the implied warranties of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General # Public License for more details. You should have received a copy of the # GNU General Public License along with this program; if not, write to the # Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA # 02110-1301, USA. Any Red Hat trademarks that are incorporated in the # source code or documentation are not subject to the GNU General Public # License and may only be used or replicated with the express permission of # Red Hat, Inc. # from pyanaconda.modules.common import init init() from pyanaconda.modules.timezone.timezone import TimezoneService service = TimezoneService() service.run()