test_dbus.py 6.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163
  1. # systemctl-mqtt - MQTT client triggering & reporting shutdown on systemd-based systems
  2. #
  3. # Copyright (C) 2020 Fabian Peter Hammerle <fabian@hammerle.me>
  4. #
  5. # This program is free software: you can redistribute it and/or modify
  6. # it under the terms of the GNU General Public License as published by
  7. # the Free Software Foundation, either version 3 of the License, or
  8. # any later version.
  9. #
  10. # This program is distributed in the hope that it will be useful,
  11. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  12. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13. # GNU General Public License for more details.
  14. #
  15. # You should have received a copy of the GNU General Public License
  16. # along with this program. If not, see <https://www.gnu.org/licenses/>.
  17. import datetime
  18. import logging
  19. import unittest.mock
  20. import dbus
  21. import pytest
  22. import systemctl_mqtt._dbus
  23. _UTC = datetime.timezone(offset=datetime.timedelta(seconds=0))
  24. # pylint: disable=protected-access
  25. def test_get_login_manager():
  26. login_manager = systemctl_mqtt._dbus.get_login_manager()
  27. assert isinstance(login_manager, dbus.proxies.Interface)
  28. assert login_manager.dbus_interface == "org.freedesktop.login1.Manager"
  29. # https://freedesktop.org/wiki/Software/systemd/logind/
  30. assert isinstance(login_manager.CanPowerOff(), dbus.String)
  31. def test__log_shutdown_inhibitors_some(caplog):
  32. login_manager = unittest.mock.MagicMock()
  33. login_manager.ListInhibitors.return_value = dbus.Array(
  34. [
  35. dbus.Struct(
  36. (
  37. dbus.String("shutdown:sleep"),
  38. dbus.String("Developer"),
  39. dbus.String("Haven't pushed my commits yet"),
  40. dbus.String("delay"),
  41. dbus.UInt32(1000),
  42. dbus.UInt32(1234),
  43. ),
  44. signature=None,
  45. ),
  46. dbus.Struct(
  47. (
  48. dbus.String("shutdown"),
  49. dbus.String("Editor"),
  50. dbus.String(""),
  51. dbus.String("Unsafed files open"),
  52. dbus.UInt32(0),
  53. dbus.UInt32(42),
  54. ),
  55. signature=None,
  56. ),
  57. ],
  58. signature=dbus.Signature("(ssssuu)"),
  59. )
  60. with caplog.at_level(logging.DEBUG):
  61. systemctl_mqtt._dbus._log_shutdown_inhibitors(login_manager)
  62. assert len(caplog.records) == 2
  63. assert caplog.records[0].levelno == logging.DEBUG
  64. assert (
  65. caplog.records[0].message
  66. == "detected shutdown inhibitor Developer (pid=1234, uid=1000, mode=delay): "
  67. + "Haven't pushed my commits yet"
  68. )
  69. def test__log_shutdown_inhibitors_none(caplog):
  70. login_manager = unittest.mock.MagicMock()
  71. login_manager.ListInhibitors.return_value = dbus.Array([])
  72. with caplog.at_level(logging.DEBUG):
  73. systemctl_mqtt._dbus._log_shutdown_inhibitors(login_manager)
  74. assert len(caplog.records) == 1
  75. assert caplog.records[0].levelno == logging.DEBUG
  76. assert caplog.records[0].message == "no shutdown inhibitor locks found"
  77. def test__log_shutdown_inhibitors_fail(caplog):
  78. login_manager = unittest.mock.MagicMock()
  79. login_manager.ListInhibitors.side_effect = dbus.DBusException("mocked")
  80. with caplog.at_level(logging.DEBUG):
  81. systemctl_mqtt._dbus._log_shutdown_inhibitors(login_manager)
  82. assert len(caplog.records) == 1
  83. assert caplog.records[0].levelno == logging.WARNING
  84. assert caplog.records[0].message == "failed to fetch shutdown inhibitors: mocked"
  85. @pytest.mark.parametrize("action", ["poweroff", "reboot"])
  86. @pytest.mark.parametrize("delay", [datetime.timedelta(0), datetime.timedelta(hours=1)])
  87. def test__schedule_shutdown(action, delay):
  88. login_manager_mock = unittest.mock.MagicMock()
  89. with unittest.mock.patch(
  90. "systemctl_mqtt._dbus.get_login_manager", return_value=login_manager_mock,
  91. ):
  92. systemctl_mqtt._dbus.schedule_shutdown(action=action, delay=delay)
  93. assert login_manager_mock.ScheduleShutdown.call_count == 1
  94. schedule_args, schedule_kwargs = login_manager_mock.ScheduleShutdown.call_args
  95. assert len(schedule_args) == 2
  96. assert schedule_args[0] == action
  97. assert isinstance(schedule_args[1], dbus.UInt64)
  98. shutdown_datetime = datetime.datetime.fromtimestamp(
  99. schedule_args[1] / 10 ** 6, tz=_UTC
  100. )
  101. actual_delay = shutdown_datetime - datetime.datetime.now(tz=_UTC)
  102. assert actual_delay.total_seconds() == pytest.approx(delay.total_seconds(), abs=0.1)
  103. assert not schedule_kwargs
  104. @pytest.mark.parametrize("action", ["poweroff"])
  105. @pytest.mark.parametrize(
  106. ("exception_message", "log_message"),
  107. [
  108. ("test message", "test message"),
  109. (
  110. "Interactive authentication required.",
  111. "unauthorized; missing polkit authorization rules?",
  112. ),
  113. ],
  114. )
  115. def test__schedule_shutdown_fail(caplog, action, exception_message, log_message):
  116. login_manager_mock = unittest.mock.MagicMock()
  117. login_manager_mock.ScheduleShutdown.side_effect = dbus.DBusException(
  118. exception_message
  119. )
  120. with unittest.mock.patch(
  121. "systemctl_mqtt._dbus.get_login_manager", return_value=login_manager_mock,
  122. ), caplog.at_level(logging.DEBUG):
  123. systemctl_mqtt._dbus.schedule_shutdown(
  124. action=action, delay=datetime.timedelta(seconds=21)
  125. )
  126. assert login_manager_mock.ScheduleShutdown.call_count == 1
  127. assert len(caplog.records) == 3
  128. assert caplog.records[0].levelno == logging.INFO
  129. assert caplog.records[0].message.startswith("scheduling {} for ".format(action))
  130. assert caplog.records[1].levelno == logging.ERROR
  131. assert caplog.records[1].message == "failed to schedule {}: {}".format(
  132. action, log_message
  133. )
  134. assert "inhibitor" in caplog.records[2].message
  135. def test_lock_all_sessions(caplog):
  136. login_manager_mock = unittest.mock.MagicMock()
  137. with unittest.mock.patch(
  138. "systemctl_mqtt._dbus.get_login_manager", return_value=login_manager_mock
  139. ), caplog.at_level(logging.INFO):
  140. systemctl_mqtt._dbus.lock_all_sessions()
  141. login_manager_mock.LockSessions.assert_called_once_with()
  142. assert len(caplog.records) == 1
  143. assert caplog.records[0].levelno == logging.INFO
  144. assert caplog.records[0].message == "instruct all sessions to activate screen locks"