summaryrefslogtreecommitdiff
path: root/tests/integration.rs
diff options
context:
space:
mode:
Diffstat (limited to 'tests/integration.rs')
-rw-r--r--tests/integration.rs275
1 files changed, 275 insertions, 0 deletions
diff --git a/tests/integration.rs b/tests/integration.rs
new file mode 100644
index 0000000..a63daa6
--- /dev/null
+++ b/tests/integration.rs
@@ -0,0 +1,275 @@
+//! Integration test: simulate a full notification lifecycle using AppMap + DbusParser together,
+//! feeding realistic dbus-monitor output through the parser and processing results with AppMap.
+
+use notification_badge::{AppMap, DbusMessage, DbusParser};
+use std::collections::HashMap;
+
+fn setup_app_map() -> AppMap {
+ let mut map = AppMap::new();
+ let mut patterns = HashMap::new();
+ patterns.insert(
+ "firefox".into(),
+ "application://org.mozilla.firefox.desktop".into(),
+ );
+ patterns.insert(
+ "org.mozilla.firefox".into(),
+ "application://org.mozilla.firefox.desktop".into(),
+ );
+ patterns.insert(
+ "slack".into(),
+ "application://com.slack.Slack.desktop".into(),
+ );
+ patterns.insert(
+ "com.slack.slack".into(),
+ "application://com.slack.Slack.desktop".into(),
+ );
+ map.update_patterns(patterns);
+ map
+}
+
+fn process_stream(input: &str, map: &mut AppMap) -> Vec<(String, usize)> {
+ let mut parser = DbusParser::new();
+ let mut badge_updates = Vec::new();
+
+ let mut handle = |msg: DbusMessage, map: &mut AppMap| match msg {
+ DbusMessage::Notify {
+ serial,
+ app_name,
+ replaces_id,
+ desktop_entry_hint,
+ } => {
+ let hint_ref = desktop_entry_hint.as_deref().unwrap_or("");
+ if let Some(desktop_id) = map.match_app(&[&app_name, hint_ref]) {
+ map.record_notify(serial, replaces_id, &desktop_id);
+ }
+ }
+ DbusMessage::NotifyReply {
+ reply_serial,
+ notification_id,
+ } => {
+ if let Some((desktop_id, count)) = map.resolve_reply(reply_serial, notification_id) {
+ badge_updates.push((desktop_id, count));
+ }
+ }
+ DbusMessage::NotificationClosed { notification_id } => {
+ if let Some((desktop_id, count)) = map.notification_closed(notification_id) {
+ badge_updates.push((desktop_id, count));
+ }
+ }
+ };
+
+ for line in input.lines() {
+ if let Some(msg) = parser.feed_line(line) {
+ handle(msg, map);
+ }
+ }
+ if let Some(msg) = parser.flush() {
+ handle(msg, map);
+ }
+
+ badge_updates
+}
+
+/// Full lifecycle: Firefox notification arrives, badge set to 1, then dismissed, badge set to 0.
+#[test]
+fn test_full_lifecycle_notify_and_dismiss() {
+ let stream = r#"method call time=1700000000.000 sender=:1.100 -> destination=:1.5 serial=500 path=/org/freedesktop/Notifications; interface=org.freedesktop.Notifications; member=Notify
+ string "firefox"
+ uint32 0
+ string "firefox"
+ string "New tab notification"
+ string "You have updates"
+ array [
+ ]
+ array [
+ ]
+ int32 -1
+method return time=1700000000.100 sender=:1.5 -> destination=:1.100 serial=600 reply_serial=500
+ uint32 42
+signal time=1700000001.000 sender=:1.5 -> destination=(null destination) serial=700 path=/org/freedesktop/Notifications; interface=org.freedesktop.Notifications; member=NotificationClosed
+ uint32 42
+ uint32 2
+"#;
+
+ let mut map = setup_app_map();
+ let updates = process_stream(stream, &mut map);
+
+ assert_eq!(updates.len(), 2);
+ assert_eq!(
+ updates[0],
+ ("application://org.mozilla.firefox.desktop".into(), 1)
+ );
+ assert_eq!(
+ updates[1],
+ ("application://org.mozilla.firefox.desktop".into(), 0)
+ );
+}
+
+/// Multiple notifications from different apps accumulate independently.
+#[test]
+fn test_multiple_apps_independent_counts() {
+ let stream = r#"method call time=1700000000.000 sender=:1.100 -> destination=:1.5 serial=500 path=/org/freedesktop/Notifications; interface=org.freedesktop.Notifications; member=Notify
+ string "firefox"
+ uint32 0
+ string ""
+ string "Firefox notification"
+ string ""
+method return time=1700000000.100 sender=:1.5 -> destination=:1.100 serial=600 reply_serial=500
+ uint32 42
+method call time=1700000000.200 sender=:1.200 -> destination=:1.5 serial=501 path=/org/freedesktop/Notifications; interface=org.freedesktop.Notifications; member=Notify
+ string "Slack"
+ uint32 0
+ string ""
+ string "Slack message"
+ string ""
+method return time=1700000000.300 sender=:1.5 -> destination=:1.200 serial=601 reply_serial=501
+ uint32 43
+method call time=1700000000.400 sender=:1.100 -> destination=:1.5 serial=502 path=/org/freedesktop/Notifications; interface=org.freedesktop.Notifications; member=Notify
+ string "firefox"
+ uint32 0
+ string ""
+ string "Another Firefox notification"
+ string ""
+method return time=1700000000.500 sender=:1.5 -> destination=:1.100 serial=602 reply_serial=502
+ uint32 44
+"#;
+
+ let mut map = setup_app_map();
+ let updates = process_stream(stream, &mut map);
+
+ assert_eq!(updates.len(), 3);
+ // Firefox: 1
+ assert_eq!(
+ updates[0],
+ ("application://org.mozilla.firefox.desktop".into(), 1)
+ );
+ // Slack: 1
+ assert_eq!(
+ updates[1],
+ ("application://com.slack.Slack.desktop".into(), 1)
+ );
+ // Firefox: 2
+ assert_eq!(
+ updates[2],
+ ("application://org.mozilla.firefox.desktop".into(), 2)
+ );
+
+ assert_eq!(map.count("application://org.mozilla.firefox.desktop"), 2);
+ assert_eq!(map.count("application://com.slack.Slack.desktop"), 1);
+}
+
+/// Flatpak app sends notification via portal with desktop-entry hint.
+#[test]
+fn test_flatpak_desktop_entry_hint() {
+ let stream = r#"method call time=1700000000.000 sender=:1.100 -> destination=:1.5 serial=500 path=/org/freedesktop/Notifications; interface=org.freedesktop.Notifications; member=Notify
+ string "System Notifications"
+ uint32 0
+ string ""
+ string "New message from channel"
+ string "Message body"
+ array [
+ ]
+ dict entry(
+ string "desktop-entry"
+ variant string "com.slack.Slack"
+ )
+ int32 -1
+method return time=1700000000.100 sender=:1.5 -> destination=:1.100 serial=600 reply_serial=500
+ uint32 50
+"#;
+
+ let mut map = setup_app_map();
+ let updates = process_stream(stream, &mut map);
+
+ assert_eq!(updates.len(), 1);
+ assert_eq!(
+ updates[0],
+ ("application://com.slack.Slack.desktop".into(), 1)
+ );
+}
+
+/// Replacement notification: new notification replaces an existing one, count stays at 1.
+#[test]
+fn test_replacement_notification() {
+ let stream = r#"method call time=1700000000.000 sender=:1.100 -> destination=:1.5 serial=500 path=/org/freedesktop/Notifications; interface=org.freedesktop.Notifications; member=Notify
+ string "firefox"
+ uint32 0
+ string ""
+ string "First"
+ string ""
+method return time=1700000000.100 sender=:1.5 -> destination=:1.100 serial=600 reply_serial=500
+ uint32 42
+method call time=1700000000.200 sender=:1.100 -> destination=:1.5 serial=501 path=/org/freedesktop/Notifications; interface=org.freedesktop.Notifications; member=Notify
+ string "firefox"
+ uint32 42
+ string ""
+ string "Replaced"
+ string ""
+method return time=1700000000.300 sender=:1.5 -> destination=:1.100 serial=601 reply_serial=501
+ uint32 43
+"#;
+
+ let mut map = setup_app_map();
+ let updates = process_stream(stream, &mut map);
+
+ assert_eq!(updates.len(), 2);
+ // First notification: count=1
+ assert_eq!(updates[0].1, 1);
+ // Replacement: old removed, new added, still count=1
+ assert_eq!(updates[1].1, 1);
+ assert_eq!(map.count("application://org.mozilla.firefox.desktop"), 1);
+}
+
+/// Unknown app notifications are ignored entirely.
+#[test]
+fn test_unknown_app_ignored() {
+ let stream = r#"method call time=1700000000.000 sender=:1.100 -> destination=:1.5 serial=500 path=/org/freedesktop/Notifications; interface=org.freedesktop.Notifications; member=Notify
+ string "unknown-app"
+ uint32 0
+ string ""
+ string "Some notification"
+ string ""
+method return time=1700000000.100 sender=:1.5 -> destination=:1.100 serial=600 reply_serial=500
+ uint32 99
+"#;
+
+ let mut map = setup_app_map();
+ let updates = process_stream(stream, &mut map);
+
+ assert_eq!(updates.len(), 0);
+}
+
+/// Interleaved traffic: unrelated D-Bus messages mixed with notifications.
+#[test]
+fn test_interleaved_unrelated_traffic() {
+ let stream = r#"signal time=1700000000.000 sender=:1.1 -> destination=(null destination) serial=100 path=/org/freedesktop/DBus; interface=org.freedesktop.DBus; member=NameOwnerChanged
+ string ":1.200"
+ string ""
+ string ":1.200"
+method call time=1700000000.100 sender=:1.50 -> destination=:1.5 serial=200 path=/org/kde/StatusNotifierWatcher; interface=org.kde.StatusNotifierWatcher; member=RegisterStatusNotifierItem
+ string "/StatusNotifierItem"
+method call time=1700000000.200 sender=:1.100 -> destination=:1.5 serial=500 path=/org/freedesktop/Notifications; interface=org.freedesktop.Notifications; member=Notify
+ string "firefox"
+ uint32 0
+ string ""
+ string "Real notification"
+ string ""
+method return time=1700000000.250 sender=:1.5 -> destination=:1.50 serial=201 reply_serial=200
+ string "ok"
+method return time=1700000000.300 sender=:1.5 -> destination=:1.100 serial=600 reply_serial=500
+ uint32 42
+signal time=1700000000.400 sender=:1.1 -> destination=(null destination) serial=101 path=/org/freedesktop/DBus; interface=org.freedesktop.DBus; member=NameOwnerChanged
+ string ":1.300"
+ string ":1.300"
+ string ""
+"#;
+
+ let mut map = setup_app_map();
+ let updates = process_stream(stream, &mut map);
+
+ assert_eq!(updates.len(), 1);
+ assert_eq!(
+ updates[0],
+ ("application://org.mozilla.firefox.desktop".into(), 1)
+ );
+}
generated by cgit v1.2.3 (git 2.39.1) at 2026-06-07 01:14:30 +0000