Add browser tests for loading notification icons from blob and data URLs.

chrome/browser/notifications/platform_notification_service_browsertest.cc

 // 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.
 
 #include <string>
 
 #include "base/command_line.h"
 #include "base/files/file_path.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/path_service.h"
 #include "base/strings/utf_string_conversions.h"
 #include "chrome/browser/infobars/infobar_service.h"
+#include "chrome/browser/notifications/desktop_notification_profile_util.h"
 #include "chrome/browser/notifications/desktop_notification_service.h"
 #include "chrome/browser/notifications/desktop_notification_service_factory.h"
 #include "chrome/browser/notifications/notification_test_util.h"
 #include "chrome/browser/notifications/platform_notification_service_impl.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "chrome/test/base/ui_test_utils.h"
 #include "components/infobars/core/confirm_infobar_delegate.h"
 #include "components/infobars/core/infobar.h"
@@ skipping 53 matching lines @@
 
 void InfoBarResponder::Respond(ConfirmInfoBarDelegate* delegate) {
   if (accept_)
     delegate->Accept();
   else
     delegate->Cancel();
 }
 
 // -----------------------------------------------------------------------------
 
+// Dimensions of the icon.png resource in the notification test data directory.
+const int kIconWidth = 100;
+const int kIconHeight = 100;
+
 class PlatformNotificationServiceBrowserTest : public InProcessBrowserTest {
  public:
   PlatformNotificationServiceBrowserTest();
   ~PlatformNotificationServiceBrowserTest() override {}
 
   // InProcessBrowserTest overrides.
   void SetUp() override;
   void SetUpOnMainThread() override;
   void TearDown() override;
 
  protected:
   // Returns the Platform Notification Service these unit tests are for.
   PlatformNotificationServiceImpl* service() const {
     return PlatformNotificationServiceImpl::GetInstance();
   }
 
+  // Grants permission to display Web Notifications for origin of the test
+  // page that's being used in this browser test.
+  void GrantNotificationPermissionForTest() const;
+
   // Returns the UI Manager on which notifications will be displayed.
   StubNotificationUIManager* ui_manager() const { return ui_manager_.get(); }
 
   const base::FilePath& server_root() const { return server_root_; }
 
   // Navigates the browser to the test page indicated by |path|.
   void NavigateToTestPage(const std::string& path) const;
 
   // Executes |script| and stores the result as a string in |result|. A boolean
   // will be returned, indicating whether the script was executed successfully.
@@ skipping 38 matching lines @@
 void PlatformNotificationServiceBrowserTest::SetUpOnMainThread() {
   NavigateToTestPage(test_page_url_);
 
   InProcessBrowserTest::SetUpOnMainThread();
 }
 
 void PlatformNotificationServiceBrowserTest::TearDown() {
   service()->SetNotificationUIManagerForTesting(nullptr);
 }
 
+void PlatformNotificationServiceBrowserTest::
+    GrantNotificationPermissionForTest() const {
+  GURL origin = TestPageUrl().GetOrigin();
+
+  DesktopNotificationProfileUtil::GrantPermission(browser()->profile(), origin);
+  ASSERT_EQ(CONTENT_SETTING_ALLOW,
+            DesktopNotificationProfileUtil::GetContentSetting(
+                browser()->profile(), origin));
+}
+
 void PlatformNotificationServiceBrowserTest::NavigateToTestPage(
     const std::string& path) const {
   ui_test_utils::NavigateToURL(browser(), https_server_->GetURL(path));
 }
 
 bool PlatformNotificationServiceBrowserTest::RunScript(
     const std::string& script, std::string* result) const {
   return content::ExecuteScriptAndExtractString(
       browser()->tab_strip_model()->GetActiveWebContents()->GetMainFrame(),
       script,
@@ skipping 69 matching lines @@
 
   ASSERT_EQ(1u, ui_manager()->GetNotificationCount());
 
   // We don't use or check the notification's direction and language.
   const Notification& notification = ui_manager()->GetNotificationAt(0);
   EXPECT_EQ("Title", base::UTF16ToUTF8(notification.title()));
   EXPECT_EQ("Contents", base::UTF16ToUTF8(notification.message()));
   EXPECT_EQ("replace-id", notification.tag());
   EXPECT_FALSE(notification.icon().IsEmpty());
   EXPECT_TRUE(notification.silent());
+
+  EXPECT_EQ(kIconWidth, notification.icon().Width());
+  EXPECT_EQ(kIconHeight, notification.icon().Height());
 }
 
 IN_PROC_BROWSER_TEST_F(PlatformNotificationServiceBrowserTest,
                        CloseDisplayedPersistentNotification) {
   std::string script_result;
 
   InfoBarResponder accepting_responder(browser(), true);
   ASSERT_TRUE(RunScript("RequestPermission()", &script_result));
   EXPECT_EQ("granted", script_result);
 
@@ skipping 63 matching lines @@
 
   InfoBarResponder accepting_responder_file(browser(), true);
   ASSERT_TRUE(RunScript("RequestPermission()", &script_result));
   EXPECT_EQ("granted", script_result);
 
   EXPECT_FALSE(notification_service->IsNotifierEnabled(file_notifier))
       << "If this test fails, you may have fixed a bug preventing file origins "
       << "from sending their origin from Blink; if so you need to update the "
       << "display function for notification origins to show the file path.";
 }
+
+IN_PROC_BROWSER_TEST_F(PlatformNotificationServiceBrowserTest,
+                       DataUrlAsNotificationImage) {
+  ASSERT_NO_FATAL_FAILURE(GrantNotificationPermissionForTest());
+
+  std::string script_result;
+  ASSERT_TRUE(RunScript("DisplayPersistentNotificationDataUrlImage()",
+                        &script_result));
+  EXPECT_EQ("ok", script_result);
+
+  ASSERT_EQ(1u, ui_manager()->GetNotificationCount());
+
+  const Notification& notification = ui_manager()->GetNotificationAt(0);
+  EXPECT_FALSE(notification.icon().IsEmpty());
+
+  EXPECT_EQ("Data URL Title", base::UTF16ToUTF8(notification.title()));
+  EXPECT_EQ(kIconWidth, notification.icon().Width());
+  EXPECT_EQ(kIconHeight, notification.icon().Height());
+}
+
+IN_PROC_BROWSER_TEST_F(PlatformNotificationServiceBrowserTest,
+                       BlobAsNotificationImage) {
+  ASSERT_NO_FATAL_FAILURE(GrantNotificationPermissionForTest());
+
+  std::string script_result;
+  ASSERT_TRUE(RunScript("DisplayPersistentNotificationBlobImage()",
+                        &script_result));
+  EXPECT_EQ("ok", script_result);
+
+  ASSERT_EQ(1u, ui_manager()->GetNotificationCount());
+
+  const Notification& notification = ui_manager()->GetNotificationAt(0);
+  EXPECT_FALSE(notification.icon().IsEmpty());
+
+  EXPECT_EQ("Blob Title", base::UTF16ToUTF8(notification.title()));
+  EXPECT_EQ(kIconWidth, notification.icon().Width());
+  EXPECT_EQ(kIconHeight, notification.icon().Height());
+}