Support layout testing Web Notifications from Document and Worker contexts.

content/shell/browser/shell_content_browser_client.cc

Index: content/shell/browser/shell_content_browser_client.cc
diff --git a/content/shell/browser/shell_content_browser_client.cc b/content/shell/browser/shell_content_browser_client.cc
index 560cef1dfa14251596a4f3e23c1e04bf92320947..ffb25d39b455b205c56203dc27a5d5859fbf8951 100644
--- a/content/shell/browser/shell_content_browser_client.cc
+++ b/content/shell/browser/shell_content_browser_client.cc
@@ -22,6 +22,7 @@
 #include "content/shell/browser/shell_devtools_delegate.h"
 #include "content/shell/browser/shell_message_filter.h"
 #include "content/shell/browser/shell_net_log.h"
+#include "content/shell/browser/shell_notification_manager.h"
 #include "content/shell/browser/shell_quota_permission_context.h"
 #include "content/shell/browser/shell_resource_dispatcher_host_delegate.h"
 #include "content/shell/browser/shell_web_contents_view_delegate_creator.h"
@@ -137,6 +138,17 @@ ShellContentBrowserClient::~ShellContentBrowserClient() {
   g_browser_client = NULL;
 }
 
+ShellNotificationManager*
+ShellContentBrowserClient::GetShellNotificationManager() {

[jochen (gone - plz use gerrit), 2014/07/17 09:29:45]

add a DCHECK() that we're on the IO thread?

[Peter Beverloo, 2014/07/17 12:22:52]

Ugh, fantastic. CBC::RequestDesktopNotificationPermission is expected to be
called on the IO thread, whereas CBC::CheckDesktopNotificationPermission is
expected to be called on the UI thread. Thanks for pointing this out. I'll jump
to IO for Request*.

+  if (!CommandLine::ForCurrentProcess()->HasSwitch(switches::kDumpRenderTree))
+    return NULL;
+
+  if (!shell_notification_manager_)
+    shell_notification_manager_.reset(new ShellNotificationManager());
+
+  return shell_notification_manager_.get();
+}
+
 BrowserMainParts* ShellContentBrowserClient::CreateBrowserMainParts(
     const MainFunctionParams& parameters) {
   shell_browser_main_parts_ = new ShellBrowserMainParts(parameters);
@@ -273,6 +285,27 @@ ShellContentBrowserClient::CreateQuotaPermissionContext() {
   return new ShellQuotaPermissionContext();
 }
 
+void ShellContentBrowserClient::RequestDesktopNotificationPermission(
+    const GURL& source_origin,
+    RenderFrameHost* render_frame_host,
+    const base::Closure& callback) {
+  ShellNotificationManager* manager = GetShellNotificationManager();
+  if (manager)
+    manager->RequestPermission(source_origin, callback);

[jochen (gone - plz use gerrit), 2014/07/17 09:29:45]

invoke the callback with some default if no manager is present?

[Peter Beverloo, 2014/07/17 12:22:52]

Done.

+}
+
+blink::WebNotificationPresenter::Permission
+ShellContentBrowserClient::CheckDesktopNotificationPermission(
+    const GURL& source_url,
+    ResourceContext* context,
+    int render_process_id) {
+  ShellNotificationManager* manager = GetShellNotificationManager();
+  if (manager)
+    return manager->CheckPermission(source_url);
+
+  return blink::WebNotificationPresenter::PermissionAllowed;
+}
+
 SpeechRecognitionManagerDelegate*
     ShellContentBrowserClient::GetSpeechRecognitionManagerDelegate() {
   return new ShellSpeechRecognitionManagerDelegate();