| 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) {
|
|
|