Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "base/command_line.h" | |
| 6 #include "base/logging.h" | |
| 5 #include "base/process_util.h" | 7 #include "base/process_util.h" |
| 6 #include "chrome/browser/browser_process.h" | 8 #include "chrome/browser/browser_process.h" |
| 7 #include "chrome/browser/extensions/extension_browsertest.h" | 9 #include "chrome/browser/extensions/extension_browsertest.h" |
| 8 #include "chrome/browser/extensions/extension_host.h" | 10 #include "chrome/browser/extensions/extension_host.h" |
| 9 #include "chrome/browser/extensions/extension_process_manager.h" | 11 #include "chrome/browser/extensions/extension_process_manager.h" |
| 10 #include "chrome/browser/extensions/extension_service.h" | 12 #include "chrome/browser/extensions/extension_service.h" |
| 11 #include "chrome/browser/extensions/extension_system.h" | 13 #include "chrome/browser/extensions/extension_system.h" |
| 12 #include "chrome/browser/notifications/balloon.h" | 14 #include "chrome/browser/notifications/balloon.h" |
| 13 #include "chrome/browser/notifications/balloon_collection.h" | 15 #include "chrome/browser/notifications/balloon_collection.h" |
| 14 #include "chrome/browser/notifications/balloon_host.h" | 16 #include "chrome/browser/notifications/balloon_host.h" |
| 15 #include "chrome/browser/notifications/balloon_notification_ui_manager.h" | 17 #include "chrome/browser/notifications/balloon_notification_ui_manager.h" |
| 16 #include "chrome/browser/notifications/notification.h" | 18 #include "chrome/browser/notifications/notification.h" |
| 17 #include "chrome/browser/notifications/notification_delegate.h" | 19 #include "chrome/browser/notifications/notification_delegate.h" |
| 18 #include "chrome/browser/profiles/profile.h" | 20 #include "chrome/browser/profiles/profile.h" |
| 19 #include "chrome/browser/ui/browser.h" | 21 #include "chrome/browser/ui/browser.h" |
| 20 #include "chrome/browser/ui/browser_commands.h" | 22 #include "chrome/browser/ui/browser_commands.h" |
| 21 #include "chrome/browser/ui/tabs/tab_strip_model.h" | 23 #include "chrome/browser/ui/tabs/tab_strip_model.h" |
| 22 #include "chrome/test/base/ui_test_utils.h" | 24 #include "chrome/test/base/ui_test_utils.h" |
| 23 #include "content/public/browser/navigation_controller.h" | 25 #include "content/public/browser/navigation_controller.h" |
| 24 #include "content/public/browser/render_process_host.h" | 26 #include "content/public/browser/render_process_host.h" |
| 25 #include "content/public/browser/render_view_host.h" | 27 #include "content/public/browser/render_view_host.h" |
| 26 #include "content/public/browser/web_contents.h" | 28 #include "content/public/browser/web_contents.h" |
| 27 #include "content/public/common/result_codes.h" | 29 #include "content/public/common/result_codes.h" |
| 28 | |
| 29 #if defined(ENABLE_MESSAGE_CENTER) | |
| 30 #include "base/command_line.h" | |
| 31 #include "ui/message_center/message_center.h" | 30 #include "ui/message_center/message_center.h" |
| 32 #include "ui/message_center/message_center_switches.h" | 31 #include "ui/message_center/message_center_switches.h" |
| 32 #include "ui/message_center/message_center_util.h" | |
| 33 #include "ui/message_center/notification_list.h" | 33 #include "ui/message_center/notification_list.h" |
| 34 #endif | |
| 35 | 34 |
| 36 using content::NavigationController; | 35 using content::NavigationController; |
| 37 using content::WebContents; | 36 using content::WebContents; |
| 38 using extensions::Extension; | 37 using extensions::Extension; |
| 39 | 38 |
| 40 // Tests are timing out waiting for extension to crash. | 39 // Tests are timing out waiting for extension to crash. |
| 41 // http://crbug.com/174705 | 40 // http://crbug.com/174705 |
| 42 #if defined(OS_MACOSX) || defined(USE_AURA) | 41 #if defined(OS_MACOSX) || defined(USE_AURA) |
| 43 #define MAYBE_ExtensionCrashRecoveryTest DISABLED_ExtensionCrashRecoveryTest | 42 #define MAYBE_ExtensionCrashRecoveryTest DISABLED_ExtensionCrashRecoveryTest |
| 44 #else | 43 #else |
| (...skipping 65 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 110 ASSERT_TRUE(extension); | 109 ASSERT_TRUE(extension); |
| 111 second_extension_id_ = extension->id(); | 110 second_extension_id_ = extension->id(); |
| 112 CheckExtensionConsistency(second_extension_id_); | 111 CheckExtensionConsistency(second_extension_id_); |
| 113 } | 112 } |
| 114 | 113 |
| 115 std::string first_extension_id_; | 114 std::string first_extension_id_; |
| 116 std::string second_extension_id_; | 115 std::string second_extension_id_; |
| 117 | 116 |
| 118 }; | 117 }; |
| 119 | 118 |
| 120 // TODO(rsesek): Implement and enable these tests. http://crbug.com/179904 | 119 class MAYBE_ExtensionCrashRecoveryTest |
| 121 #if defined(ENABLE_MESSAGE_CENTER) && !defined(OS_MACOSX) | |
| 122 | |
| 123 class MessageCenterExtensionCrashRecoveryTest | |
| 124 : public ExtensionCrashRecoveryTestBase { | 120 : public ExtensionCrashRecoveryTestBase { |
| 125 protected: | 121 protected: |
| 126 virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE { | |
| 127 ExtensionCrashRecoveryTestBase::SetUpCommandLine(command_line); | |
| 128 command_line->AppendSwitch( | |
| 129 message_center::switches::kEnableRichNotifications); | |
| 130 } | |
| 131 | |
| 132 virtual void AcceptNotification(size_t index) OVERRIDE { | 122 virtual void AcceptNotification(size_t index) OVERRIDE { |
| 133 message_center::MessageCenter* message_center = | 123 if (message_center::IsRichNotificationEnabled()) { |
| 134 message_center::MessageCenter::Get(); | 124 message_center::MessageCenter* message_center = |
| 135 ASSERT_GT(message_center->NotificationCount(), index); | 125 message_center::MessageCenter::Get(); |
| 136 message_center::NotificationList::Notifications::reverse_iterator it = | 126 ASSERT_GT(message_center->NotificationCount(), index); |
| 137 message_center->GetNotifications().rbegin(); | 127 message_center::NotificationList::Notifications::reverse_iterator it = |
| 138 for (size_t i=0; i < index; ++i) | 128 message_center->GetNotifications().rbegin(); |
| 139 it++; | 129 for (size_t i=0; i < index; ++i) |
| 140 std::string id = (*it)->id(); | 130 it++; |
| 141 message_center->ClickOnNotification(id); | 131 std::string id = (*it)->id(); |
| 132 message_center->ClickOnNotification(id); | |
| 133 } else { | |
| 134 LOG(ERROR) << "AcceptNotification 1, index: " << index; | |
| 135 Balloon* balloon = GetNotificationDelegate(index); | |
| 136 LOG(ERROR) << "AcceptNotification 2"; | |
| 137 ASSERT_TRUE(balloon); | |
| 138 LOG(ERROR) << "AcceptNotification 3"; | |
| 139 balloon->OnClick(); | |
| 140 LOG(ERROR) << "AcceptNotification 4"; | |
| 141 } | |
| 142 LOG(ERROR) << "AcceptNotification 5"; | |
| 142 WaitForExtensionLoad(); | 143 WaitForExtensionLoad(); |
| 144 LOG(ERROR) << "AcceptNotification 6"; | |
| 143 } | 145 } |
| 144 | 146 |
| 145 virtual void CancelNotification(size_t index) OVERRIDE { | 147 virtual void CancelNotification(size_t index) OVERRIDE { |
| 146 message_center::MessageCenter* message_center = | 148 if (message_center::IsRichNotificationEnabled()) { |
| 147 message_center::MessageCenter::Get(); | 149 message_center::MessageCenter* message_center = |
| 148 ASSERT_GT(message_center->NotificationCount(), index); | 150 message_center::MessageCenter::Get(); |
| 149 message_center::NotificationList::Notifications::reverse_iterator it = | 151 ASSERT_GT(message_center->NotificationCount(), index); |
| 150 message_center->GetNotifications().rbegin(); | 152 message_center::NotificationList::Notifications::reverse_iterator it = |
| 151 for (size_t i=0; i < index; i++) { it++; } | 153 message_center->GetNotifications().rbegin(); |
| 152 ASSERT_TRUE( | 154 for (size_t i=0; i < index; i++) { it++; } |
| 153 g_browser_process->notification_ui_manager()->CancelById((*it)->id())); | 155 ASSERT_TRUE(g_browser_process->notification_ui_manager()-> |
| 156 CancelById((*it)->id())); | |
| 157 } else { | |
| 158 Balloon* balloon = GetNotificationDelegate(index); | |
| 159 ASSERT_TRUE(balloon); | |
| 160 std::string id = balloon->notification().notification_id(); | |
| 161 ASSERT_TRUE(g_browser_process->notification_ui_manager()->CancelById(id)); | |
| 162 } | |
| 154 } | 163 } |
| 155 | 164 |
| 156 virtual size_t CountBalloons() OVERRIDE { | 165 virtual size_t CountBalloons() OVERRIDE { |
| 157 message_center::MessageCenter* message_center = | 166 if (message_center::IsRichNotificationEnabled()) { |
| 158 message_center::MessageCenter::Get(); | 167 message_center::MessageCenter* message_center = |
| 159 return message_center->NotificationCount(); | 168 message_center::MessageCenter::Get(); |
| 160 } | 169 return message_center->NotificationCount(); |
| 161 }; | 170 } else { |
| 162 | 171 BalloonNotificationUIManager* manager = |
| 163 typedef MessageCenterExtensionCrashRecoveryTest | 172 BalloonNotificationUIManager::GetInstanceForTesting(); |
| 164 MAYBE_ExtensionCrashRecoveryTest; | 173 BalloonCollection::Balloons balloons = |
| 165 | 174 manager->balloon_collection()->GetActiveBalloons(); |
| 166 #else // defined(ENABLED_MESSAGE_CENTER) | 175 return balloons.size(); |
| 167 | 176 } |
| 168 class BalloonExtensionCrashRecoveryTest | |
| 169 : public ExtensionCrashRecoveryTestBase { | |
| 170 protected: | |
| 171 virtual void AcceptNotification(size_t index) OVERRIDE { | |
| 172 Balloon* balloon = GetNotificationDelegate(index); | |
| 173 ASSERT_TRUE(balloon); | |
| 174 balloon->OnClick(); | |
| 175 WaitForExtensionLoad(); | |
| 176 } | 177 } |
| 177 | 178 |
| 178 virtual void CancelNotification(size_t index) OVERRIDE { | 179 private: |
| 179 Balloon* balloon = GetNotificationDelegate(index); | 180 Balloon* GetNotificationDelegate(size_t index) { |
| 180 ASSERT_TRUE(balloon); | 181 BalloonNotificationUIManager* manager = |
| 181 std::string id = balloon->notification().notification_id(); | 182 BalloonNotificationUIManager::GetInstanceForTesting(); |
| 182 ASSERT_TRUE(g_browser_process->notification_ui_manager()->CancelById(id)); | 183 BalloonCollection::Balloons balloons = |
| 183 } | 184 manager->balloon_collection()->GetActiveBalloons(); |
| 184 | 185 return index < balloons.size() ? balloons.at(index) : NULL; |
| 185 virtual size_t CountBalloons() OVERRIDE { | 186 } |
| 186 BalloonNotificationUIManager* manager = | |
| 187 BalloonNotificationUIManager::GetInstanceForTesting(); | |
| 188 BalloonCollection::Balloons balloons = | |
| 189 manager->balloon_collection()->GetActiveBalloons(); | |
| 190 return balloons.size(); | |
| 191 } | |
| 192 private: | |
| 193 Balloon* GetNotificationDelegate(size_t index) { | |
| 194 BalloonNotificationUIManager* manager = | |
| 195 BalloonNotificationUIManager::GetInstanceForTesting(); | |
| 196 BalloonCollection::Balloons balloons = | |
| 197 manager->balloon_collection()->GetActiveBalloons(); | |
| 198 return index < balloons.size() ? balloons.at(index) : NULL; | |
| 199 } | |
| 200 }; | 187 }; |
| 201 | 188 |
| 202 typedef BalloonExtensionCrashRecoveryTest MAYBE_ExtensionCrashRecoveryTest; | |
| 203 #endif // defined(ENABLE_MESSAGE_CENTER) | |
| 204 | |
| 205 IN_PROC_BROWSER_TEST_F(MAYBE_ExtensionCrashRecoveryTest, Basic) { | 189 IN_PROC_BROWSER_TEST_F(MAYBE_ExtensionCrashRecoveryTest, Basic) { |
| 206 const size_t size_before = GetExtensionService()->extensions()->size(); | 190 const size_t size_before = GetExtensionService()->extensions()->size(); |
| 207 const size_t crash_size_before = | 191 const size_t crash_size_before = |
| 208 GetExtensionService()->terminated_extensions()->size(); | 192 GetExtensionService()->terminated_extensions()->size(); |
| 193 LOG(ERROR) << "Basic 1"; | |
|
dewittj
2013/05/08 18:53:01
These log statements should probably be removed.
Dmitry Titov
2013/05/08 23:03:26
Done.
| |
| 209 LoadTestExtension(); | 194 LoadTestExtension(); |
| 195 LOG(ERROR) << "Basic 2"; | |
| 210 CrashExtension(first_extension_id_); | 196 CrashExtension(first_extension_id_); |
| 197 LOG(ERROR) << "Basic 3"; | |
| 211 ASSERT_EQ(size_before, GetExtensionService()->extensions()->size()); | 198 ASSERT_EQ(size_before, GetExtensionService()->extensions()->size()); |
| 212 ASSERT_EQ(crash_size_before + 1, | 199 ASSERT_EQ(crash_size_before + 1, |
| 213 GetExtensionService()->terminated_extensions()->size()); | 200 GetExtensionService()->terminated_extensions()->size()); |
| 201 LOG(ERROR) << "Basic 4"; | |
| 214 ASSERT_NO_FATAL_FAILURE(AcceptNotification(0)); | 202 ASSERT_NO_FATAL_FAILURE(AcceptNotification(0)); |
| 215 | 203 |
| 216 SCOPED_TRACE("after clicking the balloon"); | 204 SCOPED_TRACE("after clicking the balloon"); |
| 205 LOG(ERROR) << "Basic 5"; | |
| 217 CheckExtensionConsistency(first_extension_id_); | 206 CheckExtensionConsistency(first_extension_id_); |
| 207 LOG(ERROR) << "Basic 6"; | |
| 218 ASSERT_EQ(crash_size_before, | 208 ASSERT_EQ(crash_size_before, |
| 219 GetExtensionService()->terminated_extensions()->size()); | 209 GetExtensionService()->terminated_extensions()->size()); |
| 220 } | 210 } |
| 221 | 211 |
| 222 IN_PROC_BROWSER_TEST_F(MAYBE_ExtensionCrashRecoveryTest, CloseAndReload) { | 212 IN_PROC_BROWSER_TEST_F(MAYBE_ExtensionCrashRecoveryTest, CloseAndReload) { |
| 223 const size_t size_before = GetExtensionService()->extensions()->size(); | 213 const size_t size_before = GetExtensionService()->extensions()->size(); |
| 224 const size_t crash_size_before = | 214 const size_t crash_size_before = |
| 225 GetExtensionService()->terminated_extensions()->size(); | 215 GetExtensionService()->terminated_extensions()->size(); |
| 226 LoadTestExtension(); | 216 LoadTestExtension(); |
| 227 CrashExtension(first_extension_id_); | 217 CrashExtension(first_extension_id_); |
| (...skipping 354 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 582 GetController())); | 572 GetController())); |
| 583 chrome::Reload(browser(), CURRENT_TAB); | 573 chrome::Reload(browser(), CURRENT_TAB); |
| 584 observer.Wait(); | 574 observer.Wait(); |
| 585 } | 575 } |
| 586 // Extension should now be loaded. | 576 // Extension should now be loaded. |
| 587 SCOPED_TRACE("after reloading the tab"); | 577 SCOPED_TRACE("after reloading the tab"); |
| 588 CheckExtensionConsistency(first_extension_id_); | 578 CheckExtensionConsistency(first_extension_id_); |
| 589 ASSERT_EQ(size_before + 1, GetExtensionService()->extensions()->size()); | 579 ASSERT_EQ(size_before + 1, GetExtensionService()->extensions()->size()); |
| 590 ASSERT_EQ(0U, CountBalloons()); | 580 ASSERT_EQ(0U, CountBalloons()); |
| 591 } | 581 } |
| OLD | NEW |