[MacViews] Fix AccessiblePaneViewTest.SetPaneFocusAndRestore.

ui/base/test/scoped_fake_nswindow_focus.mm

 // Copyright 2015 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.
 
-#import "chrome/browser/ui/test/scoped_fake_nswindow_main_status.h"
+#import "ui/base/test/scoped_fake_nswindow_focus.h"
 
 #import <Cocoa/Cocoa.h>
 
 #import "base/mac/foundation_util.h"
 #import "base/mac/scoped_objc_class_swizzler.h"
 
+using base::mac::ScopedObjCClassSwizzler;
+
 namespace {
 
-NSWindow* g_fake_main_window = nil;
+NSWindow* g_fake_focused_window = nil;
 
+void SetFocus(NSWindow* window) {
+  g_fake_focused_window = window;
+  [[NSNotificationCenter defaultCenter]
+      postNotificationName:NSWindowDidBecomeMainNotification
+                    object:g_fake_focused_window];
+  [[NSNotificationCenter defaultCenter]
+      postNotificationName:NSWindowDidBecomeKeyNotification
+                    object:g_fake_focused_window];
 }
 
-// Donates a testing implementation of [NSWindow isMainWindow].
-@interface IsMainWindowDonorForWindow : NSObject
-@end
+void ClearFocus() {
+  NSWindow* window = g_fake_focused_window;
+  g_fake_focused_window = nil;
+  [[NSNotificationCenter defaultCenter]
+      postNotificationName:NSWindowDidResignKeyNotification
+                    object:window];
 
-@implementation IsMainWindowDonorForWindow
-- (BOOL)isMainWindow {
-  NSWindow* selfAsWindow = base::mac::ObjCCastStrict<NSWindow>(self);
-  return selfAsWindow == g_fake_main_window;
-}
-@end
-
-ScopedFakeNSWindowMainStatus::ScopedFakeNSWindowMainStatus(NSWindow* window)
-    : swizzler_(new base::mac::ScopedObjCClassSwizzler(
-          [NSWindow class],
-          [IsMainWindowDonorForWindow class],
-          @selector(isMainWindow))) {
-  DCHECK(!g_fake_main_window);
-  g_fake_main_window = window;
-  [[NSNotificationCenter defaultCenter]
-      postNotificationName:NSWindowDidBecomeMainNotification
-                    object:g_fake_main_window];
-}
-
-ScopedFakeNSWindowMainStatus::~ScopedFakeNSWindowMainStatus() {
-  NSWindow* window = g_fake_main_window;
-  g_fake_main_window = nil;
   [[NSNotificationCenter defaultCenter]
       postNotificationName:NSWindowDidResignMainNotification
                     object:window];
 }
+
+}  // namespace
+
+// Donates testing implementations of NSWindow methods.
+@interface FakeNSWindowFocusDonor : NSObject
+@end
+
+@implementation FakeNSWindowFocusDonor
+
+- (BOOL)isKeyWindow {
+  NSWindow* selfAsWindow = base::mac::ObjCCastStrict<NSWindow>(self);
+  return selfAsWindow == g_fake_focused_window;
+}
+
+- (BOOL)isMainWindow {
+  NSWindow* selfAsWindow = base::mac::ObjCCastStrict<NSWindow>(self);
+  return selfAsWindow == g_fake_focused_window;
+}
+
+- (void)makeKeyWindow {
+  NSWindow* selfAsWindow = base::mac::ObjCCastStrict<NSWindow>(self);
+  if (selfAsWindow == g_fake_focused_window ||
+      ![selfAsWindow canBecomeKeyWindow])
+    return;
+
+  ClearFocus();
+  SetFocus(selfAsWindow);
+}
+
+- (void)makeMainWindow {
+  [self makeKeyWindow];
+}
+
+@end
+
+namespace ui {
+namespace test {
+
+ScopedFakeNSWindowFocus::ScopedFakeNSWindowFocus()
+    : is_main_swizzler_(
+          new ScopedObjCClassSwizzler([NSWindow class],
+                                      [FakeNSWindowFocusDonor class],
+                                      @selector(isMainWindow))),
+      make_main_swizzler_(
+          new ScopedObjCClassSwizzler([NSWindow class],
+                                      [FakeNSWindowFocusDonor class],
+                                      @selector(makeMainWindow))),
+      is_key_swizzler_(
+          new ScopedObjCClassSwizzler([NSWindow class],
+                                      [FakeNSWindowFocusDonor class],
+                                      @selector(isKeyWindow))),
+      make_key_swizzler_(
+          new ScopedObjCClassSwizzler([NSWindow class],
+                                      [FakeNSWindowFocusDonor class],
+                                      @selector(makeKeyWindow))) {}
+
+ScopedFakeNSWindowFocus::~ScopedFakeNSWindowFocus() {
+  ClearFocus();
+}
+
+}  // namespace test
+}  // namespace ui