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 9b485fae857c7f31a1f6b764ee4e5c9004b73757..f992a92479bbc52c466c392d6602896fad0d47cc 100644 |
--- a/chrome/browser/chromeos/status/power_menu_button_browsertest.cc |
+++ b/chrome/browser/chromeos/status/power_menu_button_browsertest.cc |
@@ -9,6 +9,11 @@ |
#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 "chrome/browser/ui/views/aura/chrome_shell_delegate.h" |
+#endif |
namespace chromeos { |
@@ -21,9 +26,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)); |
+ const views::View* view = |
+#if defined(USE_AURA) |
+ ChromeShellDelegate::instance()->GetStatusArea(); |
+#else |
+ static_cast<BrowserView*>(browser()->window()); |
+#endif |
+ return const_cast<PowerMenuButton*>( |
Daniel Erat
2011/11/19 01:49:11
This is somewhat gross. Scott, what's the plan fo
sky
2011/11/19 23:26:52
I was hoping we could keep it const as most tests
Daniel Erat
2011/11/21 16:45:51
I'm happy leaving it const if you're okay with the
|
+ static_cast<const PowerMenuButton*>( |
+ view->GetViewByID(VIEW_ID_STATUS_BUTTON_POWER))); |
} |
string16 CallPowerChangedAndGetTooltipText(const PowerSupplyStatus& status) { |