Fix broken tests

infinite-iOS · May 27, 2022 · 0e32216 · 0e32216
1 parent 153f6ff
commit 0e32216
Showing 1 changed file with 18 additions and 4 deletions.
diff --git a/WordPress/WordPressTest/PromptRemindersSchedulerTests.swift b/WordPress/WordPressTest/PromptRemindersSchedulerTests.swift
@@ -88,18 +88,32 @@ class PromptRemindersSchedulerTests: XCTestCase {
                     // prompt notifications
                     Expected(body: "Prompt text 1",
                              dateComponents: .init(year: 2022, month: 5, day: 21, hour: hour, minute: minute),
-                             userInfo: ["promptID": 101, "siteID": siteID]),
+                             userInfo: [
+                                PushNotificationsManager.Notification.typeKey: PushNotificationsManager.Notification.bloggingPrompts,
+                                PushNotificationsManager.BloggingPromptPayload.promptIDKey: 101,
+                                PushNotificationsManager.BloggingPromptPayload.siteIDKey: siteID
+                             ]),
                     Expected(body: "Prompt text 8",
                              dateComponents: .init(year: 2022, month: 5, day: 28, hour: hour, minute: minute),
-                             userInfo: ["promptID": 108, "siteID": siteID]),
+                             userInfo: [
+                                PushNotificationsManager.Notification.typeKey: PushNotificationsManager.Notification.bloggingPrompts,
+                                PushNotificationsManager.BloggingPromptPayload.promptIDKey: 108,
+                                PushNotificationsManager.BloggingPromptPayload.siteIDKey: siteID
+                             ]),
 
                     // static notifications
                     Expected(body: .staticNotificationContent,
                              dateComponents: .init(year: 2022, month: 6, day: 4, hour: hour, minute: minute),
-                             userInfo: ["siteID": siteID]),
+                             userInfo: [
+                                PushNotificationsManager.Notification.typeKey: PushNotificationsManager.Notification.bloggingPrompts,
+                                PushNotificationsManager.BloggingPromptPayload.siteIDKey: siteID
+                             ]),
                     Expected(body: .staticNotificationContent,
                              dateComponents: .init(year: 2022, month: 6, day: 11, hour: hour, minute: minute),
-                             userInfo: ["siteID": siteID])
+                             userInfo: [
+                                PushNotificationsManager.Notification.typeKey: PushNotificationsManager.Notification.bloggingPrompts,
+                                PushNotificationsManager.BloggingPromptPayload.siteIDKey: siteID
+                             ]),
                 ]
             }
         }