Remove ENABLE_MESSAGE_CENTER

chrome/browser/extensions/extension_crash_recovery_browsertest.cc

 // Copyright (c) 2012 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/process_util.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/extensions/extension_browsertest.h"
 #include "chrome/browser/extensions/extension_host.h"
 #include "chrome/browser/extensions/extension_process_manager.h"
 #include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/extensions/extension_system.h"
 #include "chrome/browser/notifications/balloon.h"
 #include "chrome/browser/notifications/balloon_collection.h"
 #include "chrome/browser/notifications/balloon_host.h"
 #include "chrome/browser/notifications/balloon_notification_ui_manager.h"
 #include "chrome/browser/notifications/notification.h"
 #include "chrome/browser/notifications/notification_delegate.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_commands.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/test/base/ui_test_utils.h"
 #include "content/public/browser/navigation_controller.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/browser/render_view_host.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/common/result_codes.h"
-
-#if defined(ENABLE_MESSAGE_CENTER)
-#include "base/command_line.h"
 #include "ui/message_center/message_center.h"
 #include "ui/message_center/message_center_switches.h"
+#include "ui/message_center/message_center_util.h"
 #include "ui/message_center/notification_list.h"
-#endif
 
 using content::NavigationController;
 using content::WebContents;
 using extensions::Extension;
 
 // Tests are timing out waiting for extension to crash.
 // http://crbug.com/174705
 #if defined(OS_MACOSX) || defined(USE_AURA)
 #define MAYBE_ExtensionCrashRecoveryTest DISABLED_ExtensionCrashRecoveryTest
 #else
@@ skipping 65 matching lines @@
     ASSERT_TRUE(extension);
     second_extension_id_ = extension->id();
     CheckExtensionConsistency(second_extension_id_);
   }
 
   std::string first_extension_id_;
   std::string second_extension_id_;
 
 };
 
-// TODO(rsesek): Implement and enable these tests. http://crbug.com/179904
-#if defined(ENABLE_MESSAGE_CENTER) && !defined(OS_MACOSX)
-
-class MessageCenterExtensionCrashRecoveryTest
+class MAYBE_ExtensionCrashRecoveryTest
     : public ExtensionCrashRecoveryTestBase {
  protected:
-  virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
-    ExtensionCrashRecoveryTestBase::SetUpCommandLine(command_line);
-    command_line->AppendSwitch(
-        message_center::switches::kEnableRichNotifications);
-  }
-
   virtual void AcceptNotification(size_t index) OVERRIDE {
-    message_center::MessageCenter* message_center =
-        message_center::MessageCenter::Get();
-    ASSERT_GT(message_center->NotificationCount(), index);
-    message_center::NotificationList::Notifications::reverse_iterator it =
-        message_center->GetNotifications().rbegin();
-    for (size_t i=0; i < index; ++i)
-      it++;
-    std::string id = (*it)->id();
-    message_center->ClickOnNotification(id);
+    if (message_center::IsRichNotificationEnabled()) {
+      message_center::MessageCenter* message_center =
+          message_center::MessageCenter::Get();
+      ASSERT_GT(message_center->NotificationCount(), index);
+      message_center::NotificationList::Notifications::reverse_iterator it =
+          message_center->GetNotifications().rbegin();
+      for (size_t i=0; i < index; ++i)
+        it++;
+      std::string id = (*it)->id();
+      message_center->ClickOnNotification(id);
+    } else {
+      Balloon* balloon = GetNotificationDelegate(index);
+      ASSERT_TRUE(balloon);
+      balloon->OnClick();
+    }
     WaitForExtensionLoad();
   }
 
   virtual void CancelNotification(size_t index) OVERRIDE {
-    message_center::MessageCenter* message_center =
-        message_center::MessageCenter::Get();
-    ASSERT_GT(message_center->NotificationCount(), index);
-    message_center::NotificationList::Notifications::reverse_iterator it =
-        message_center->GetNotifications().rbegin();
-    for (size_t i=0; i < index; i++) { it++; }
-    ASSERT_TRUE(
-        g_browser_process->notification_ui_manager()->CancelById((*it)->id()));
+    if (message_center::IsRichNotificationEnabled()) {
+      message_center::MessageCenter* message_center =
+          message_center::MessageCenter::Get();
+      ASSERT_GT(message_center->NotificationCount(), index);
+      message_center::NotificationList::Notifications::reverse_iterator it =
+          message_center->GetNotifications().rbegin();
+      for (size_t i=0; i < index; i++) { it++; }
+      ASSERT_TRUE(g_browser_process->notification_ui_manager()->
+          CancelById((*it)->id()));
+    } else {
+      Balloon* balloon = GetNotificationDelegate(index);
+      ASSERT_TRUE(balloon);
+      std::string id = balloon->notification().notification_id();
+      ASSERT_TRUE(g_browser_process->notification_ui_manager()->CancelById(id));
+    }
   }
 
   virtual size_t CountBalloons() OVERRIDE {
-    message_center::MessageCenter* message_center =
-        message_center::MessageCenter::Get();
-    return message_center->NotificationCount();
-  }
-};
+    if (message_center::IsRichNotificationEnabled())
+      return message_center::MessageCenter::Get()->NotificationCount();
 
-typedef MessageCenterExtensionCrashRecoveryTest
-    MAYBE_ExtensionCrashRecoveryTest;
-
-#else  // defined(ENABLED_MESSAGE_CENTER)
-
-class BalloonExtensionCrashRecoveryTest
-    : public ExtensionCrashRecoveryTestBase {
- protected:
-  virtual void AcceptNotification(size_t index) OVERRIDE {
-    Balloon* balloon = GetNotificationDelegate(index);
-    ASSERT_TRUE(balloon);
-    balloon->OnClick();
-    WaitForExtensionLoad();
+    return BalloonNotificationUIManager::GetInstanceForTesting()->
+        balloon_collection()->GetActiveBalloons().size();
   }
 
-  virtual void CancelNotification(size_t index) OVERRIDE {
-    Balloon* balloon = GetNotificationDelegate(index);
-    ASSERT_TRUE(balloon);
-    std::string id = balloon->notification().notification_id();
-    ASSERT_TRUE(g_browser_process->notification_ui_manager()->CancelById(id));
-  }
-
-  virtual size_t CountBalloons() OVERRIDE {
-    BalloonNotificationUIManager* manager =
-        BalloonNotificationUIManager::GetInstanceForTesting();
-    BalloonCollection::Balloons balloons =
-        manager->balloon_collection()->GetActiveBalloons();
-    return balloons.size();
-  }
- private:
-  Balloon* GetNotificationDelegate(size_t index) {
-    BalloonNotificationUIManager* manager =
-        BalloonNotificationUIManager::GetInstanceForTesting();
-    BalloonCollection::Balloons balloons =
-        manager->balloon_collection()->GetActiveBalloons();
-    return index < balloons.size() ? balloons.at(index) : NULL;
-  }
+private:
+     Balloon* GetNotificationDelegate(size_t index) {
+       BalloonNotificationUIManager* manager =
+           BalloonNotificationUIManager::GetInstanceForTesting();
+       BalloonCollection::Balloons balloons =
+           manager->balloon_collection()->GetActiveBalloons();
+       return index < balloons.size() ? balloons.at(index) : NULL;
+     }
 };
 
-typedef BalloonExtensionCrashRecoveryTest MAYBE_ExtensionCrashRecoveryTest;
-#endif  // defined(ENABLE_MESSAGE_CENTER)
-
 IN_PROC_BROWSER_TEST_F(MAYBE_ExtensionCrashRecoveryTest, Basic) {
   const size_t size_before = GetExtensionService()->extensions()->size();
   const size_t crash_size_before =
       GetExtensionService()->terminated_extensions()->size();
   LoadTestExtension();
   CrashExtension(first_extension_id_);
   ASSERT_EQ(size_before, GetExtensionService()->extensions()->size());
   ASSERT_EQ(crash_size_before + 1,
             GetExtensionService()->terminated_extensions()->size());
   ASSERT_NO_FATAL_FAILURE(AcceptNotification(0));
@@ skipping 367 matching lines @@
                 GetController()));
     chrome::Reload(browser(), CURRENT_TAB);
     observer.Wait();
   }
   // Extension should now be loaded.
   SCOPED_TRACE("after reloading the tab");
   CheckExtensionConsistency(first_extension_id_);
   ASSERT_EQ(size_before + 1, GetExtensionService()->extensions()->size());
   ASSERT_EQ(0U, CountBalloons());
 }