diff options
Diffstat (limited to 'Tools/WebKitTestRunner/WebNotificationProvider.h')
-rw-r--r-- | Tools/WebKitTestRunner/WebNotificationProvider.h | 16 |
1 files changed, 10 insertions, 6 deletions
diff --git a/Tools/WebKitTestRunner/WebNotificationProvider.h b/Tools/WebKitTestRunner/WebNotificationProvider.h index fba2870fc..ccf8b1bce 100644 --- a/Tools/WebKitTestRunner/WebNotificationProvider.h +++ b/Tools/WebKitTestRunner/WebNotificationProvider.h @@ -26,9 +26,10 @@ #ifndef WebNotificationProvider_h #define WebNotificationProvider_h -#include <WebKit2/WKNotificationManager.h> -#include <WebKit2/WKNotificationProvider.h> -#include <WebKit2/WKRetainPtr.h> +#include <WebKit/WKNotificationManager.h> +#include <WebKit/WKNotificationProvider.h> +#include <WebKit/WKRetainPtr.h> +#include <wtf/HashMap.h> #include <wtf/HashSet.h> namespace WTR { @@ -45,12 +46,15 @@ public: void removeNotificationManager(WKNotificationManagerRef); WKDictionaryRef notificationPermissions(); - void simulateWebNotificationClick(uint64_t notificationID); + void simulateWebNotificationClick(WKPageRef, uint64_t notificationID); void reset(); private: - WKRetainPtr<WKNotificationManagerRef> m_notificationManager; - HashSet<uint64_t> m_shownNotifications; + // Inverses of each other. + HashMap<WKRetainPtr<WKNotificationManagerRef>, HashSet<uint64_t>> m_ownedNotifications; + HashMap<uint64_t, WKNotificationManagerRef> m_owningManager; + + HashMap<std::pair<WKPageRef, uint64_t>, uint64_t> m_localToGlobalNotificationIDMap; }; } |