src/chrome/browser/task_manager/task_manager_notification_browsertest.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/task_manager/task_manager.h"
   6
   7 #include "base/strings/utf_string_conversions.h"
   8 #include "chrome/browser/browser_process.h"
   9 #include "chrome/browser/extensions/extension_browsertest.h"
  10 #include "chrome/browser/notifications/desktop_notification_service.h"
  11 #include "chrome/browser/notifications/notification.h"
  12 #include "chrome/browser/notifications/notification_test_util.h"
  13 #include "chrome/browser/notifications/notification_ui_manager.h"
  14 #include "chrome/browser/task_manager/task_manager_browsertest_util.h"
  15 #include "chrome/browser/ui/browser.h"
  16 #include "chrome/browser/ui/browser_dialogs.h"
  17 #include "chrome/browser/ui/browser_window.h"
  18 #include "chrome/test/base/in_process_browser_test.h"
  19 #include "chrome/test/base/ui_test_utils.h"
  20 #include "content/public/common/content_switches.h"
  21 #include "testing/gtest/include/gtest/gtest.h"
  22 #include "ui/message_center/message_center_util.h"
  23
  24 class TaskManagerNotificationBrowserTest : public ExtensionBrowserTest {
  25  public:
  26   TaskManagerModel* model() const {
  27     return TaskManager::GetInstance()->model();
  28   }
  29  protected:
  30   virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
  31     ExtensionBrowserTest::SetUpCommandLine(command_line);
  32
  33     // Do not prelaunch the GPU process for these tests because it will show
  34     // up in task manager but whether it appears before or after the new tab
  35     // renderer process is not well defined.
  36     command_line->AppendSwitch(switches::kDisableGpuProcessPrelaunch);
  37   }
  38 };
  39
  40 // TODO(linux_aura) http://crbug.com/163931
  41 #if defined(OS_LINUX) && !defined(OS_CHROMEOS) && defined(USE_AURA)
  42 #define MAYBE_NoticeNotificationChanges DISABLED_NoticeNotificationChanges
  43 #else
  44 #define MAYBE_NoticeNotificationChanges NoticeNotificationChanges
  45 #endif
  46 IN_PROC_BROWSER_TEST_F(TaskManagerNotificationBrowserTest,
  47                        MAYBE_NoticeNotificationChanges) {
  48   // These tests do not apply with Message Center platforms
  49   // where notifications do not instantiate a new renderer.
  50   if (message_center::IsRichNotificationEnabled())
  51     return;
  52
  53   EXPECT_EQ(0, model()->ResourceCount());
  54
  55   // Show the task manager.
  56   chrome::ShowTaskManager(browser());
  57   // Expect to see the browser and the New Tab Page renderer.
  58   TaskManagerBrowserTestUtil::WaitForWebResourceChange(1);
  59
  60   // Show a notification.
  61   NotificationUIManager* notifications =
  62       g_browser_process->notification_ui_manager();
  63
  64   string16 content = DesktopNotificationService::CreateDataUrl(
  65       GURL(), ASCIIToUTF16("Hello World!"), string16(),
  66       WebKit::WebTextDirectionDefault);
  67
  68   scoped_refptr<NotificationDelegate> del1(new MockNotificationDelegate("n1"));
  69   Notification n1(
  70       GURL(), GURL(content), ASCIIToUTF16("Test 1"), string16(), del1.get());
  71   scoped_refptr<NotificationDelegate> del2(new MockNotificationDelegate("n2"));
  72   Notification n2(
  73       GURL(), GURL(content), ASCIIToUTF16("Test 2"), string16(), del2.get());
  74
  75   notifications->Add(n1, browser()->profile());
  76   TaskManagerBrowserTestUtil::WaitForWebResourceChange(2);
  77   notifications->Add(n2, browser()->profile());
  78   TaskManagerBrowserTestUtil::WaitForWebResourceChange(3);
  79   notifications->CancelById(n1.notification_id());
  80   TaskManagerBrowserTestUtil::WaitForWebResourceChange(2);
  81   notifications->CancelById(n2.notification_id());
  82   TaskManagerBrowserTestUtil::WaitForWebResourceChange(1);
  83 }