Index: content/shell/browser/layout_test/layout_test_push_messaging_service.h |
diff --git a/content/shell/browser/layout_test/layout_test_push_messaging_service.h b/content/shell/browser/layout_test/layout_test_push_messaging_service.h |
index 1ff3dda8660953092651447e1be0002815d12379..9a1ec4fea82c838a4a5893eaba4703e1d9e73d36 100644 |
--- a/content/shell/browser/layout_test/layout_test_push_messaging_service.h |
+++ b/content/shell/browser/layout_test/layout_test_push_messaging_service.h |
@@ -6,6 +6,7 @@ |
#define CONTENT_SHELL_BROWSER_LAYOUT_TEST_LAYOUT_TEST_PUSH_MESSAGING_SERVICE_H_ |
#include <map> |
+#include <set> |
#include "content/public/browser/push_messaging_service.h" |
#include "content/public/common/push_messaging_status.h" |
@@ -45,10 +46,15 @@ class LayoutTestPushMessagingService : public PushMessagingService { |
blink::WebPushPermissionStatus GetPermissionStatus( |
const GURL& requesting_origin, |
const GURL& embedding_origin) override; |
+ void Unregister(const GURL& requesting_origin, |
+ int64 service_worker_registration_id, |
+ const UnregisterCallback& callback) override; |
private: |
// Map from origin to permission status. |
std::map<GURL, blink::WebPushPermissionStatus> permission_map_; |
+ // Set of currently registered origins. |
+ std::set<GURL> registrations_; |
DISALLOW_COPY_AND_ASSIGN(LayoutTestPushMessagingService); |
}; |