Notification settings for ephemeral apps

chrome/browser/apps/ephemeral_app_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 "chrome/browser/apps/app_browsertest_util.h"
 #include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/extensions/extension_test_message_listener.h"
+#include "chrome/browser/notifications/desktop_notification_service.h"
+#include "chrome/browser/notifications/desktop_notification_service_factory.h"
 #include "chrome/common/extensions/api/alarms.h"
 #include "content/public/test/browser_test.h"
 #include "content/public/test/test_utils.h"
 #include "extensions/browser/event_router.h"
 #include "extensions/browser/extension_system.h"
 #include "extensions/browser/process_manager.h"
 #include "extensions/common/switches.h"
+#include "ui/message_center/message_center.h"
+#include "ui/message_center/notifier_settings.h"
 
 using extensions::Event;
 using extensions::EventRouter;
 using extensions::Extension;
 using extensions::ExtensionSystem;
 using extensions::PlatformAppBrowserTest;
 
 namespace {
 
 namespace alarms = extensions::api::alarms;
 
 const char kDispatchEventTestApp[] =
     "platform_apps/ephemeral_apps/dispatch_event";
 
 const char kMessagingReceiverApp[] =
     "platform_apps/ephemeral_apps/messaging_receiver";
 
 const char kMessagingReceiverAppV2[] =
     "platform_apps/ephemeral_apps/messaging_receiver2";
 
+const char kNotificationsTestApp[] =
+    "platform_apps/ephemeral_apps/notification_settings";
+
+typedef std::vector<message_center::Notifier*> NotifierList;
+
+bool IsNotifierInList(const message_center::NotifierId& notifier_id,
+                      const NotifierList& notifiers) {
+  for (NotifierList::const_iterator it = notifiers.begin();
+       it != notifiers.end(); ++it) {
+    const message_center::Notifier* notifier = *it;
+    if (notifier->notifier_id == notifier_id)
+      return true;
+  }
+
+  return false;
+}
+
 class EphemeralAppBrowserTest : public PlatformAppBrowserTest {
  protected:
   virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
     // Skip PlatformAppBrowserTest, which sets different values for the switches
     // below.
     ExtensionBrowserTest::SetUpCommandLine(command_line);
 
     // Make event pages get suspended immediately.
     command_line->AppendSwitchASCII(
         extensions::switches::kEventPageIdleTime, "10");
@@ skipping 148 matching lines @@
       ExtensionSystem::Get(browser()->profile())->extension_service();
   EXPECT_TRUE(service->UpdateExtension(app_id, app_v2_path, true, GURL(),
                                        &crx_installer));
   windowed_observer.Wait();
 
   const Extension* app_v2 = service->GetExtensionById(app_id, false);
   ASSERT_TRUE(app_v2);
   EXPECT_TRUE(app_v2->version()->CompareTo(app_original_version) > 0);
   EXPECT_TRUE(app_v2->is_ephemeral());
 }
+
+// Verify that if notifications have been disabled for an ephemeral app, it will
+// remain disabled even after being evicted from the cache.
+IN_PROC_BROWSER_TEST_F(EphemeralAppBrowserTest, StickyNotificationSettings) {
+  const Extension* app = InstallEphemeralApp(kNotificationsTestApp);
+  ASSERT_TRUE(app);
+
+  // Disable notifications for this app.
+  DesktopNotificationService* notification_service =
+      DesktopNotificationServiceFactory::GetForProfile(browser()->profile());
+  ASSERT_TRUE(notification_service);
+
+  message_center::NotifierId notifier_id(
+      message_center::NotifierId::APPLICATION, app->id());
+  EXPECT_TRUE(notification_service->IsNotifierEnabled(notifier_id));
+  notification_service->SetNotifierEnabled(notifier_id, false);
+  EXPECT_FALSE(notification_service->IsNotifierEnabled(notifier_id));
+
+  // Uninstall the app, which is what happens when ephemeral apps get evicted
+  // from the cache.
+  content::WindowedNotificationObserver uninstalled_signal(
+      chrome::NOTIFICATION_EXTENSION_UNINSTALLED,
+      content::Source<Profile>(browser()->profile()));
+
+  ExtensionService* service =
+      ExtensionSystem::Get(browser()->profile())->extension_service();
+  ASSERT_TRUE(service);
+  service->UninstallExtension(app->id(), false, NULL);
+
+  uninstalled_signal.Wait();
+
+  // Reinstall the ephemeral app and verify that notifications remain disabled.
+  app = InstallEphemeralApp(kNotificationsTestApp);
+  ASSERT_TRUE(app);
+  message_center::NotifierId reinstalled_notifier_id(
+      message_center::NotifierId::APPLICATION, app->id());
+  EXPECT_FALSE(notification_service->IsNotifierEnabled(
+      reinstalled_notifier_id));
+}
+
+// Verify that only running ephemeral apps will appear in the Notification
+// Settings UI. Inactive, cached ephemeral apps should not appear.
+IN_PROC_BROWSER_TEST_F(EphemeralAppBrowserTest,
+                       IncludeRunningEphmeralAppsInNotifiers) {
+  message_center::NotifierSettingsProvider* settings_provider =
+      message_center::MessageCenter::Get()->GetNotifierSettingsProvider();
+  // TODO(tmdiep): Remove once notifications settings are supported across
+  // all platforms. This test will fail for Linux GTK.
+  if (!settings_provider)
+    return;
+
+  const Extension* app = InstallAndLaunchEphemeralApp(kNotificationsTestApp);
+  ASSERT_TRUE(app);
+  message_center::NotifierId notifier_id(
+      message_center::NotifierId::APPLICATION, app->id());
+
+  // Since the ephemeral app is running, it should be included in the list
+  // of notifiers to show in the UI.
+  NotifierList notifiers;
+  settings_provider->GetNotifierList(&notifiers);
+  EXPECT_TRUE(IsNotifierInList(notifier_id, notifiers));
+
+  // Close the ephemeral app.
+  CloseApp(app->id());
+
+  // Inactive ephemeral apps should not be included in the list of notifiers to
+  // show in the UI.
+  notifiers.clear();
+  settings_provider->GetNotifierList(&notifiers);
+  EXPECT_FALSE(IsNotifierInList(notifier_id, notifiers));
+}