aura: Fix input method and a11y status area browser tests.

chrome/browser/chromeos/status/accessibility_menu_button_browsertest.cc

 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/chromeos/status/accessibility_menu_button.h"
 
 #include "chrome/browser/chromeos/accessibility_util.h"
 #include "chrome/browser/chromeos/frame/browser_view.h"
 #include "chrome/browser/chromeos/view_ids.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_window.h"
 
+#if defined(USE_AURA)
+#include "chrome/browser/ui/views/aura/chrome_shell_delegate.h"
+#endif
+
 namespace chromeos {
 
 class AccessibilityMenuButtonTest : public InProcessBrowserTest {
  protected:
   AccessibilityMenuButtonTest() : InProcessBrowserTest() {
   }
 
-  AccessibilityMenuButton* GetAccessibilityMenuButton() {
-    BrowserView* view = static_cast<BrowserView*>(browser()->window());
-    return static_cast<AccessibilityMenuButton*>(view->GetViewByID(
-        VIEW_ID_STATUS_BUTTON_ACCESSIBILITY));
+  const AccessibilityMenuButton* GetAccessibilityMenuButton() {
+    const views::View* view =
+#if defined(USE_AURA)
+        ChromeShellDelegate::instance()->GetStatusAreaForTest();
+#else
+        static_cast<BrowserView*>(browser()->window());
+#endif
+    return static_cast<const AccessibilityMenuButton*>(
+        view->GetViewByID(VIEW_ID_STATUS_BUTTON_ACCESSIBILITY));
   }
 };
 
 IN_PROC_BROWSER_TEST_F(AccessibilityMenuButtonTest,
                        VisibilityIsSyncedWithPreference) {
-  AccessibilityMenuButton* button = GetAccessibilityMenuButton();
+  const AccessibilityMenuButton* button = GetAccessibilityMenuButton();
   ASSERT_TRUE(button != NULL);
 
   accessibility::EnableAccessibility(true, NULL);
   EXPECT_TRUE(button->IsVisible());
 
   accessibility::EnableAccessibility(false, NULL);
   EXPECT_FALSE(button->IsVisible());
 }
 
 }  // namespace chromeos