Index: chrome/browser/chromeos/status/power_menu_button_browsertest.cc |
diff --git a/chrome/browser/chromeos/status/power_menu_button_browsertest.cc b/chrome/browser/chromeos/status/power_menu_button_browsertest.cc |
index fe476f0d7dc1c8c32ef482a1fb9b7497ad686f96..d61a700989184da84a3709c213a089f658efae34 100644 |
--- a/chrome/browser/chromeos/status/power_menu_button_browsertest.cc |
+++ b/chrome/browser/chromeos/status/power_menu_button_browsertest.cc |
@@ -9,6 +9,12 @@ |
#include "chrome/browser/ui/browser.h" |
#include "chrome/browser/ui/browser_window.h" |
#include "chrome/test/base/in_process_browser_test.h" |
+#include "views/view.h" |
+ |
+#if defined(USE_AURA) |
+#include "ui/aura_shell/shell.h" |
+#include "views/widget/widget.h" |
+#endif |
namespace chromeos { |
@@ -21,9 +27,15 @@ class PowerMenuButtonTest : public InProcessBrowserTest { |
} |
PowerMenuButton* GetPowerMenuButton() { |
- BrowserView* view = static_cast<BrowserView*>(browser()->window()); |
- return static_cast<PowerMenuButton*>(view->GetViewByID( |
- VIEW_ID_STATUS_BUTTON_POWER)); |
+ views::View *view = |
+#if defined(USE_AURA) |
+ aura_shell::Shell::GetInstance()->GetStatusAreaWidget()-> |
+ GetContentsView(); |
+#else |
+ static_cast<BrowserView*>(browser()->window()); |
+#endif |
+ return static_cast<PowerMenuButton*>( |
+ view->GetViewByID(VIEW_ID_STATUS_BUTTON_POWER)); |
} |
string16 CallPowerChangedAndGetTooltipText(const PowerSupplyStatus& status) { |