125 lines
4.0 KiB
Python
125 lines
4.0 KiB
Python
import sys
|
|
import unittest
|
|
from pathlib import Path
|
|
from unittest import mock
|
|
|
|
sys.path.insert(0, str(Path(__file__).resolve().parents[1] / "src"))
|
|
|
|
from cthulhu import messages
|
|
from cthulhu.notification_presenter import NotificationPresenter
|
|
|
|
|
|
class _FakeMonitor:
|
|
def __init__(self):
|
|
self.dismissed = []
|
|
self.invoked = []
|
|
self.current_generation = 2
|
|
|
|
def refresh(self, announce_new=False):
|
|
return True
|
|
|
|
def is_current_entry(self, entry):
|
|
return entry.source == "mako" and entry.source_generation == self.current_generation
|
|
|
|
def dismiss_notification(self, notification_id):
|
|
self.dismissed.append(notification_id)
|
|
return True
|
|
|
|
def invoke_action(self, notification_id, action_key):
|
|
self.invoked.append((notification_id, action_key))
|
|
return True
|
|
|
|
|
|
class NotificationPresenterMakoTests(unittest.TestCase):
|
|
def setUp(self):
|
|
self.presenter = NotificationPresenter()
|
|
self.monitor = _FakeMonitor()
|
|
self.presenter.set_mako_monitor(self.monitor)
|
|
|
|
def test_sync_updates_current_generation_only(self):
|
|
old_entry = self.presenter.save_notification(
|
|
"Notification old",
|
|
source="mako",
|
|
source_generation=1,
|
|
notification_id=5,
|
|
live=False,
|
|
actions={"default": "Old"},
|
|
)
|
|
current_entry = self.presenter.save_notification(
|
|
"Notification current",
|
|
source="mako",
|
|
source_generation=2,
|
|
notification_id=6,
|
|
live=True,
|
|
actions={"default": "View"},
|
|
)
|
|
|
|
self.presenter.sync_live_notifications(
|
|
"mako",
|
|
{
|
|
5: {
|
|
"message": "Notification replaced",
|
|
"actions": {"default": "New"},
|
|
"app_name": "discord",
|
|
"summary": "replaced",
|
|
"body": "",
|
|
"urgency": 1,
|
|
"desktop_entry": "discord",
|
|
}
|
|
},
|
|
source_generation=2,
|
|
)
|
|
|
|
self.assertEqual(old_entry.message, "Notification old")
|
|
self.assertEqual(old_entry.actions, {"default": "Old"})
|
|
self.assertFalse(current_entry.live)
|
|
|
|
def test_can_control_entry_and_get_actions_require_current_monitor_generation(self):
|
|
entry = self.presenter.save_notification(
|
|
"Notification current",
|
|
source="mako",
|
|
source_generation=2,
|
|
notification_id=6,
|
|
live=True,
|
|
actions={"default": "View"},
|
|
)
|
|
|
|
self.assertTrue(self.presenter.can_control_entry(entry))
|
|
self.assertEqual(self.presenter.get_actions_for_entry(entry), {"default": "View"})
|
|
|
|
stale_entry = self.presenter.save_notification(
|
|
"Notification stale",
|
|
source="mako",
|
|
source_generation=1,
|
|
notification_id=7,
|
|
live=True,
|
|
actions={"default": "Open"},
|
|
)
|
|
|
|
self.assertFalse(self.presenter.can_control_entry(stale_entry))
|
|
self.assertEqual(self.presenter.get_actions_for_entry(stale_entry), {})
|
|
|
|
def test_dismiss_and_invoke_action_route_through_monitor(self):
|
|
entry = self.presenter.save_notification(
|
|
"Notification current",
|
|
source="mako",
|
|
source_generation=2,
|
|
notification_id=6,
|
|
live=True,
|
|
actions={"default": "View"},
|
|
)
|
|
script = mock.Mock()
|
|
|
|
self.assertTrue(self.presenter.dismiss_entry(script, entry))
|
|
self.assertEqual(self.monitor.dismissed, [6])
|
|
script.presentMessage.assert_called_with(messages.NOTIFICATION_DISMISSED)
|
|
|
|
script.reset_mock()
|
|
self.assertTrue(self.presenter.invoke_action_for_entry(script, entry, "default"))
|
|
self.assertEqual(self.monitor.invoked, [(6, "default")])
|
|
script.presentMessage.assert_called_with(messages.NOTIFICATION_ACTION_INVOKED)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|