Chromium Code Reviews| 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) { |