[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"
 
 namespace {
 
 NSWindow* g_fake_main_window = nil;
+NSWindow* g_fake_key_window = nil;
 
+void SetAndNotify(NSWindow** state,

[tapted, 2015/08/05 07:07:17]

I'm undecided about these methods.

It might be be nicer if it returned |window|, and only do the DCHECK in the
ScopedFakeNSWindowMainStatus constructor. then do calls like

g_fake_main_window = SetAndNotify(window, NSWindowDidBecomeMainNotification);

g_fake_main_window = ClearAndNotify(g_fake_main_window,
NSWindowDidResignMainNotification);


Another option might be to lose the genericity and do

SetMain(NSWindow* window) {
  DCHECK_NE(!window, !g_fake_main_window);
  NSString* name = window ?
      NSWindowDidBecomeMainNotification : NSWindowDidResignMainNotification;
  [[NSNotificationCenter defaultCenter] postNotificationName:name
      object:window ? window : g_fake_main_window];
  g_fake_main_window = window;
}


But! all this becomes irrelevant if we can combine the two classes, and just
have one key+main window

[jackhou1, 2015/08/06 00:22:30]

Done.

+                  NSWindow* window,
+                  NSString* notification_name) {
+  DCHECK(!*state);
+  *state = window;
+  [[NSNotificationCenter defaultCenter] postNotificationName:notification_name
+                                                      object:*state];
 }
 
-// Donates a testing implementation of [NSWindow isMainWindow].
+void ClearAndNotify(NSWindow** state, NSString* notification_name) {
+  if (!*state)
+    return;
+
+  NSWindow* window = *state;
+  *state = nil;
+  [[NSNotificationCenter defaultCenter] postNotificationName:notification_name
+                                                      object:window];
+}
+}

[tapted, 2015/08/05 07:07:17]

blank line +
}  // namespace

[jackhou1, 2015/08/06 00:22:30]

Done.

+
+// Donates a testing implementation of -[NSWindow isMainWindow].
 @interface IsMainWindowDonorForWindow : NSObject
 @end
 
 @implementation IsMainWindowDonorForWindow
 - (BOOL)isMainWindow {
   NSWindow* selfAsWindow = base::mac::ObjCCastStrict<NSWindow>(self);
   return selfAsWindow == g_fake_main_window;
 }
 @end
 
+// Donates testing implementations of -[NSWindow isKeyWindow] and
+// -[NSWindow makeKeyWindow].
+@interface IsKeyWindowDonorForWindow : NSObject

[tapted, 2015/08/05 07:07:17]

even if we can't merge the C++ classes, it should be fine to merge the donor
classes -- the swizzler doesn't care where they come from

[jackhou1, 2015/08/06 00:22:30]

Done.

+@end
+
+@implementation IsKeyWindowDonorForWindow
+- (void)makeKeyWindow {
+  NSWindow* window = base::mac::ObjCCastStrict<NSWindow>(self);
+  if (window == g_fake_key_window)
+    return;
+
+  ClearAndNotify(&g_fake_key_window, NSWindowDidResignKeyNotification);

[tapted, 2015/08/05 07:07:17]

Before this, it should do

if (![window canBecomeKeyWindow])
  return;

But I think that will cause the test to fail.

[jackhou1, 2015/08/06 00:22:30]

Yeah, that returns NO. If we really want to support this, we might have to
swizzle -[NSWindow canBecomeKeyWindow] to return YES for classes that don't
override it. But I don't think that's necessary.

+  SetAndNotify(&g_fake_key_window, window, NSWindowDidBecomeKeyNotification);
+}
+- (BOOL)isKeyWindow {
+  NSWindow* selfAsWindow = base::mac::ObjCCastStrict<NSWindow>(self);
+  return selfAsWindow == g_fake_key_window;
+}
+@end
+
+namespace ui {
+namespace test {
+
 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];
+  SetAndNotify(&g_fake_main_window, window, NSWindowDidBecomeMainNotification);
 }
 
 ScopedFakeNSWindowMainStatus::~ScopedFakeNSWindowMainStatus() {
-  NSWindow* window = g_fake_main_window;
-  g_fake_main_window = nil;
-  [[NSNotificationCenter defaultCenter]
-      postNotificationName:NSWindowDidResignMainNotification
-                    object:window];
+  ClearAndNotify(&g_fake_main_window, NSWindowDidResignMainNotification);
 }
+
+ScopedFakeNSWindowKeyStatus::ScopedFakeNSWindowKeyStatus()
+    : is_key_swizzler_(new base::mac::ScopedObjCClassSwizzler(
+          [NSWindow class],
+          [IsKeyWindowDonorForWindow class],
+          @selector(isKeyWindow))),
+      make_key_swizzler_(new base::mac::ScopedObjCClassSwizzler(
+          [NSWindow class],
+          [IsKeyWindowDonorForWindow class],
+          @selector(makeKeyWindow))) {}
+
+ScopedFakeNSWindowKeyStatus::~ScopedFakeNSWindowKeyStatus() {
+  ClearAndNotify(&g_fake_key_window, NSWindowDidResignKeyNotification);
+}
+
+}  // namespace test
+}  // namespace ui