Failing to upload to flakiness dashboard should not be a hard error
[WebKit-https.git] / Tools / WebKitTestRunner / WebNotificationProvider.h
index 46bf31b..d8724a9 100644 (file)
@@ -29,6 +29,7 @@
 #include <WebKit/WKNotificationManager.h>
 #include <WebKit/WKNotificationProvider.h>
 #include <WebKit/WKRetainPtr.h>
+#include <wtf/HashMap.h>
 #include <wtf/HashSet.h>
 
 namespace WTR {
@@ -49,8 +50,9 @@ public:
     void reset();
 
 private:
-    WKRetainPtr<WKNotificationManagerRef> m_currentNotificationManager;
-    HashSet<uint64_t> m_shownNotifications;
+    // Inverses of each other.
+    HashMap<WKRetainPtr<WKNotificationManagerRef>, HashSet<uint64_t>> m_ownedNotifications;
+    HashMap<uint64_t, WKNotificationManagerRef> m_owningManager;
 };
 
 }