Implement the ability to layout test Web Worker-based Notifications.

content/shell/browser/layout_test/layout_test_notification_manager.h

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef CONTENT_SHELL_BROWSER_LAYOUT_TEST_LAYOUT_TEST_NOTIFICATION_MANAGER_H_
 #define CONTENT_SHELL_BROWSER_LAYOUT_TEST_LAYOUT_TEST_NOTIFICATION_MANAGER_H_
 
 #include <map>
+#include <string>
 
 #include "base/callback.h"
+#include "base/memory/weak_ptr.h"
 #include "third_party/WebKit/public/platform/WebNotificationPermission.h"
 #include "url/gurl.h"
 
 namespace content {
 
+struct ShowDesktopNotificationHostMsgParams;
+class DesktopNotificationDelegate;
+
 // Responsible for tracking active notifications and allowed origins for the
-// Web Notification API when running layout tests. The methods in this class
-// should only be used on the IO thread.
+// Web Notification API when running layout tests.
 class LayoutTestNotificationManager {
  public:
   LayoutTestNotificationManager();
   ~LayoutTestNotificationManager();
 
   // Checks whether |origin| has permission to display notifications in tests.
+  // Must be called on the IO thread.
   blink::WebNotificationPermission CheckPermission(
       const GURL& origin);
 
   // Requests permission for |origin| to display notifications in layout tests.
+  // Must be called on the IO thread.
   void RequestPermission(
       const GURL& origin,
       const base::Callback<void(blink::WebNotificationPermission)>& callback);
 
   // Sets the permission to display notifications for |origin| to |permission|.
+  // Must be called on the IO thread.
   void SetPermission(const GURL& origin,
                      blink::WebNotificationPermission permission);
 
-  // Clears the currently granted permissions.
+  // Clears the currently granted permissions. Must be called on the IO thread.
   void ClearPermissions();
 
+  // Pretends to show the given notification for testing, storing a delegate so
+  // that interaction with the notification can be simulated later on. Must
+  // be called on the UI thread.
+  void Show(const ShowDesktopNotificationHostMsgParams& params,
+            scoped_ptr<DesktopNotificationDelegate> delegate,
+            base::Closure* cancel_callback);
+
+  // Simulates a click on the notification titled |title|. Must be called on the
+  // UI thread.
+  void SimulateClick(const std::string& title);
+
  private:
+  // Closes the notification titled |title|. Must be called on the UI thread.
+  void Close(const std::string& title);
+
   typedef std::map<GURL, blink::WebNotificationPermission>
       NotificationPermissionMap;
   NotificationPermissionMap permission_map_;
 
+  std::map<std::string, DesktopNotificationDelegate*> notifications_;
+  std::map<std::string, std::string> replacements_;
+
+  base::WeakPtrFactory<LayoutTestNotificationManager> weak_factory_;
+
   DISALLOW_COPY_AND_ASSIGN(LayoutTestNotificationManager);
 };
 
 }  // content
 
 #endif  // CONTENT_SHELL_BROWSER_LAYOUT_TEST_LAYOUT_TEST_NOTIFICATION_MANAGER_H_