Support persistent notifications in the PlatformNotificationServiceImpl.

chrome/browser/notifications/platform_notification_service_unittest.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 "base/strings/utf_string_conversions.h"
+#include "base/time/time.h"
 #include "chrome/browser/notifications/notification_test_util.h"
 #include "chrome/browser/notifications/platform_notification_service_impl.h"
 #include "chrome/test/base/testing_profile.h"
+#include "components/content_settings/core/browser/host_content_settings_map.h"
 #include "content/public/browser/desktop_notification_delegate.h"
 #include "content/public/test/test_browser_thread_bundle.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace {
 
 class MockDesktopNotificationDelegate
     : public content::DesktopNotificationDelegate {
  public:
   MockDesktopNotificationDelegate()
@@ skipping 39 matching lines @@
   MockDesktopNotificationDelegate* CreateSimplePageNotification() const {
     return CreateSimplePageNotificationWithCloseClosure(nullptr);
   }
 
   // Displays a simple, fake notification and returns a weak pointer to the
   // delegate receiving events for it (ownership is transferred to the service).
   // The close closure may be specified if so desired.
   MockDesktopNotificationDelegate* CreateSimplePageNotificationWithCloseClosure(
       base::Closure* close_closure) const {
     content::ShowDesktopNotificationHostMsgParams params;
-    params.origin = GURL("https://example.com/");
+    params.origin = GURL("https://chrome.com/");
     params.title = base::ASCIIToUTF16("My Notification");
     params.body = base::ASCIIToUTF16("Hello, world!");
 
     MockDesktopNotificationDelegate* delegate =
         new MockDesktopNotificationDelegate();
 
     service()->DisplayNotification(profile(),
                                    params,
                                    make_scoped_ptr(delegate),
                                    0 /* render_process_id */,
@@ skipping 35 matching lines @@
 
   ASSERT_FALSE(close_closure.is_null());
   close_closure.Run();
 
   EXPECT_EQ(0u, ui_manager()->GetNotificationCount());
 
   // Note that we cannot verify whether the closed event was called on the
   // delegate given that it'd result in a use-after-free.
 }
 
+TEST_F(PlatformNotificationServiceTest, PersistentNotificationDisplay) {
+  content::ShowDesktopNotificationHostMsgParams params;
+  params.origin = GURL("https://chrome.com/");
+  params.title = base::ASCIIToUTF16("My notification's title");
+  params.body = base::ASCIIToUTF16("Hello, world!");
+
+  service()->DisplayPersistentNotification(profile(),
+                                           42 /* sw_registration_id */,
+                                           params,
+                                           0 /* render_process_id */);
+
+  ASSERT_EQ(1u, ui_manager()->GetNotificationCount());
+
+  const Notification& notification = ui_manager()->GetNotificationAt(0);
+  EXPECT_EQ("https://chrome.com/", notification.origin_url().spec());
+  EXPECT_EQ("My notification's title",
+      base::UTF16ToUTF8(notification.title()));
+  EXPECT_EQ("Hello, world!",
+      base::UTF16ToUTF8(notification.message()));
+
+  service()->ClosePersistentNotification(profile(),
+                                         notification.delegate_id());
+  EXPECT_EQ(0u, ui_manager()->GetNotificationCount());
+}
+
 TEST_F(PlatformNotificationServiceTest, DisplayPageNotificationMatches) {
   content::ShowDesktopNotificationHostMsgParams params;
   params.origin = GURL("https://chrome.com/");
   params.title = base::ASCIIToUTF16("My notification's title");
   params.body = base::ASCIIToUTF16("Hello, world!");
 
   MockDesktopNotificationDelegate* delegate
       = new MockDesktopNotificationDelegate();
   service()->DisplayNotification(profile(),
                                  params,
@@ skipping 15 matching lines @@
   base::string16 display_name =
       service()->DisplayNameForOriginInProcessId(profile(),
                                                  GURL("https://chrome.com/"),
                                                  0 /* render_process_id */);
 
   EXPECT_EQ(base::ASCIIToUTF16("chrome.com"), display_name);
 
   // TODO(peter): Include unit tests for the extension-name translation
   // functionality of DisplayNameForOriginInProcessId.
 }
+
+TEST_F(PlatformNotificationServiceTest, NotificationPermissionLastUsage) {
+  // Both page and persistent notifications should update the last usage
+  // time of the notification permission for the origin.
+  GURL origin("https://chrome.com/");
+  base::Time begin_time;
+
+  CreateSimplePageNotification();
+
+  base::Time after_page_notification =
+      profile()->GetHostContentSettingsMap()->GetLastUsage(
+          origin, origin, CONTENT_SETTINGS_TYPE_NOTIFICATIONS);
+  EXPECT_GT(after_page_notification, begin_time);
+
+  content::ShowDesktopNotificationHostMsgParams params;
+  params.origin = origin;
+
+  service()->DisplayPersistentNotification(profile(),
+                                           42 /* sw_registration_id */,
+                                           params,
+                                           0 /* render_process_id */);
+
+  base::Time after_persistent_notification =
+      profile()->GetHostContentSettingsMap()->GetLastUsage(
+          origin, origin, CONTENT_SETTINGS_TYPE_NOTIFICATIONS);
+  EXPECT_GT(after_persistent_notification, after_page_notification);
+}