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)); |
} |
}; |