ui/base/test/scoped_fake_nswindow_focus.mm

   1 // Copyright 2015 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #import "ui/base/test/scoped_fake_nswindow_focus.h"
   6
   7 #import <Cocoa/Cocoa.h>
   8
   9 #import "base/mac/foundation_util.h"
  10 #import "base/mac/scoped_objc_class_swizzler.h"
  11
  12 using base::mac::ScopedObjCClassSwizzler;
  13
  14 namespace {
  15
  16 NSWindow* g_fake_focused_window = nil;
  17
  18 void SetFocus(NSWindow* window) {
  19   g_fake_focused_window = window;
  20   [[NSNotificationCenter defaultCenter]
  21       postNotificationName:NSWindowDidBecomeMainNotification
  22                     object:g_fake_focused_window];
  23   [[NSNotificationCenter defaultCenter]
  24       postNotificationName:NSWindowDidBecomeKeyNotification
  25                     object:g_fake_focused_window];
  26 }
  27
  28 void ClearFocus() {
  29   NSWindow* window = g_fake_focused_window;
  30   g_fake_focused_window = nil;
  31   [[NSNotificationCenter defaultCenter]
  32       postNotificationName:NSWindowDidResignKeyNotification
  33                     object:window];
  34
  35   [[NSNotificationCenter defaultCenter]
  36       postNotificationName:NSWindowDidResignMainNotification
  37                     object:window];
  38 }
  39
  40 }  // namespace
  41
  42 // Donates testing implementations of NSWindow methods.
  43 @interface FakeNSWindowFocusDonor : NSObject
  44 @end
  45
  46 @implementation FakeNSWindowFocusDonor
  47
  48 - (BOOL)isKeyWindow {
  49   NSWindow* selfAsWindow = base::mac::ObjCCastStrict<NSWindow>(self);
  50   return selfAsWindow == g_fake_focused_window;
  51 }
  52
  53 - (BOOL)isMainWindow {
  54   NSWindow* selfAsWindow = base::mac::ObjCCastStrict<NSWindow>(self);
  55   return selfAsWindow == g_fake_focused_window;
  56 }
  57
  58 - (void)makeKeyWindow {
  59   NSWindow* selfAsWindow = base::mac::ObjCCastStrict<NSWindow>(self);
  60   if (selfAsWindow == g_fake_focused_window ||
  61       ![selfAsWindow canBecomeKeyWindow])
  62     return;
  63
  64   ClearFocus();
  65   SetFocus(selfAsWindow);
  66 }
  67
  68 - (void)makeMainWindow {
  69   [self makeKeyWindow];
  70 }
  71
  72 @end
  73
  74 namespace ui {
  75 namespace test {
  76
  77 ScopedFakeNSWindowFocus::ScopedFakeNSWindowFocus()
  78     : is_main_swizzler_(
  79           new ScopedObjCClassSwizzler([NSWindow class],
  80                                       [FakeNSWindowFocusDonor class],
  81                                       @selector(isMainWindow))),
  82       make_main_swizzler_(
  83           new ScopedObjCClassSwizzler([NSWindow class],
  84                                       [FakeNSWindowFocusDonor class],
  85                                       @selector(makeMainWindow))),
  86       is_key_swizzler_(
  87           new ScopedObjCClassSwizzler([NSWindow class],
  88                                       [FakeNSWindowFocusDonor class],
  89                                       @selector(isKeyWindow))),
  90       make_key_swizzler_(
  91           new ScopedObjCClassSwizzler([NSWindow class],
  92                                       [FakeNSWindowFocusDonor class],
  93                                       @selector(makeKeyWindow))) {}
  94
  95 ScopedFakeNSWindowFocus::~ScopedFakeNSWindowFocus() {
  96   ClearFocus();
  97 }
  98
  99 }  // namespace test
 100 }  // namespace ui