Support layout testing Web Notifications from Document and Worker contexts.

content/shell/browser/shell_content_browser_client.cc

 // Copyright 2013 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.
 
 #include "content/shell/browser/shell_content_browser_client.h"
 
 #include "base/base_switches.h"
 #include "base/command_line.h"
 #include "base/file_util.h"
 #include "base/files/file.h"
 #include "base/path_service.h"
 #include "base/strings/utf_string_conversions.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/browser/resource_dispatcher_host.h"
 #include "content/public/browser/storage_partition.h"
 #include "content/public/common/content_switches.h"
 #include "content/public/common/url_constants.h"
 #include "content/public/common/web_preferences.h"
 #include "content/shell/browser/shell.h"
 #include "content/shell/browser/shell_browser_context.h"
 #include "content/shell/browser/shell_browser_main_parts.h"
 #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"
 #include "content/shell/browser/webkit_test_controller.h"
 #include "content/shell/common/shell_messages.h"
 #include "content/shell/common/shell_switches.h"
 #include "content/shell/common/webkit_test_helpers.h"
 #include "content/shell/geolocation/shell_access_token_store.h"
 #include "net/url_request/url_request_context_getter.h"
 #include "url/gurl.h"
@@ skipping 95 matching lines @@
   g_browser_client = this;
   if (!CommandLine::ForCurrentProcess()->HasSwitch(switches::kDumpRenderTree))
     return;
   webkit_source_dir_ = GetWebKitRootDirFilePath();
 }
 
 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);
   return shell_browser_main_parts_;
 }
 
 void ShellContentBrowserClient::RenderProcessWillLaunch(
     RenderProcessHost* host) {
   if (!CommandLine::ForCurrentProcess()->HasSwitch(switches::kDumpRenderTree))
     return;
@@ skipping 116 matching lines @@
 #else
   return NULL;
 #endif
 }
 
 QuotaPermissionContext*
 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();
 }
 
 net::NetLog* ShellContentBrowserClient::GetNetLog() {
   return shell_browser_main_parts_->net_log();
 }
 
 bool ShellContentBrowserClient::ShouldSwapProcessesForRedirect(
@@ skipping 78 matching lines @@
 ShellBrowserContext*
 ShellContentBrowserClient::ShellBrowserContextForBrowserContext(
     BrowserContext* content_browser_context) {
   if (content_browser_context == browser_context())
     return browser_context();
   DCHECK_EQ(content_browser_context, off_the_record_browser_context());
   return off_the_record_browser_context();
 }
 
 }  // namespace content