| Index: chrome/browser/chromeos/status/caps_lock_menu_button_browsertest.cc
|
| diff --git a/chrome/browser/chromeos/status/caps_lock_menu_button_browsertest.cc b/chrome/browser/chromeos/status/caps_lock_menu_button_browsertest.cc
|
| index 55345f1ea4bd5900252470521e93d09691e45983..835a9c832b5d6fa7a0fc08b94f1c14a4e328b588 100644
|
| --- a/chrome/browser/chromeos/status/caps_lock_menu_button_browsertest.cc
|
| +++ b/chrome/browser/chromeos/status/caps_lock_menu_button_browsertest.cc
|
| @@ -12,6 +12,11 @@
|
| #include "chrome/browser/ui/browser_window.h"
|
| #include "grit/theme_resources.h"
|
|
|
| +#if defined(USE_AURA)
|
| +#include "ui/aura_shell/shell.h"
|
| +#include "views/widget/widget.h"
|
| +#endif
|
| +
|
| namespace chromeos {
|
|
|
| class CapsLockMenuButtonTest : public CrosInProcessBrowserTest {
|
| @@ -26,9 +31,15 @@ class CapsLockMenuButtonTest : public CrosInProcessBrowserTest {
|
| }
|
|
|
| CapsLockMenuButton* GetCapsLockMenuButton() {
|
| - BrowserView* view = static_cast<BrowserView*>(browser()->window());
|
| - return static_cast<CapsLockMenuButton*>(view->GetViewByID(
|
| - VIEW_ID_STATUS_BUTTON_CAPS_LOCK));
|
| + views::View *view =
|
| +#if defined(USE_AURA)
|
| + aura_shell::Shell::GetInstance()->GetStatusAreaWidget()->
|
| + GetContentsView();
|
| +#else
|
| + static_cast<BrowserView*>(browser()->window());
|
| +#endif
|
| + return static_cast<CapsLockMenuButton*>(
|
| + view->GetViewByID(VIEW_ID_STATUS_BUTTON_CAPS_LOCK));
|
| }
|
| };
|
|
|
|
|