focus_tracker_unittest.mm revision ddb351dbec246cf1fab5ec20d2d5520909041de1 
1// Copyright (c) 2011 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 <Cocoa/Cocoa.h>
6
7#include "base/memory/scoped_nsobject.h"
8#import "chrome/browser/ui/cocoa/cocoa_test_helper.h"
9#import "chrome/browser/ui/cocoa/focus_tracker.h"
10#include "testing/gtest/include/gtest/gtest.h"
11#include "testing/platform_test.h"
12
13namespace {
14
15class FocusTrackerTest : public CocoaTest {
16 public:
17  virtual void SetUp() {
18    CocoaTest::SetUp();
19    scoped_nsobject<NSView> view([[NSView alloc] initWithFrame:NSZeroRect]);
20    viewA_ = view.get();
21    [[test_window() contentView] addSubview:viewA_];
22
23    view.reset([[NSView alloc] initWithFrame:NSZeroRect]);
24    viewB_ = view.get();
25    [[test_window() contentView] addSubview:viewB_];
26  }
27
28 protected:
29  NSView* viewA_;
30  NSView* viewB_;
31};
32
33TEST_F(FocusTrackerTest, SaveRestore) {
34  NSWindow* window = test_window();
35  ASSERT_TRUE([window makeFirstResponder:viewA_]);
36  scoped_nsobject<FocusTracker> tracker(
37      [[FocusTracker alloc] initWithWindow:window]);
38  // Give focus to |viewB_|, then try and restore it to view1.
39  ASSERT_TRUE([window makeFirstResponder:viewB_]);
40  EXPECT_TRUE([tracker restoreFocusInWindow:window]);
41  EXPECT_EQ(viewA_, [window firstResponder]);
42}
43
44TEST_F(FocusTrackerTest, SaveRestoreWithTextView) {
45  // Valgrind will complain if the text field has zero size.
46  NSRect frame = NSMakeRect(0, 0, 100, 20);
47  NSWindow* window = test_window();
48  scoped_nsobject<NSTextField> text([[NSTextField alloc] initWithFrame:frame]);
49  [[window contentView] addSubview:text];
50
51  ASSERT_TRUE([window makeFirstResponder:text]);
52  scoped_nsobject<FocusTracker> tracker([[FocusTracker alloc]
53                                         initWithWindow:window]);
54  // Give focus to |viewB_|, then try and restore it to the text field.
55  ASSERT_TRUE([window makeFirstResponder:viewB_]);
56  EXPECT_TRUE([tracker restoreFocusInWindow:window]);
57  EXPECT_TRUE([[window firstResponder] isKindOfClass:[NSTextView class]]);
58}
59
60TEST_F(FocusTrackerTest, DontRestoreToViewNotInWindow) {
61  NSWindow* window = test_window();
62  scoped_nsobject<NSView> viewC([[NSView alloc] initWithFrame:NSZeroRect]);
63  [[window contentView] addSubview:viewC];
64
65  ASSERT_TRUE([window makeFirstResponder:viewC]);
66  scoped_nsobject<FocusTracker> tracker(
67      [[FocusTracker alloc] initWithWindow:window]);
68
69  // Give focus to |viewB_|, then remove viewC from the hierarchy and try
70  // to restore focus.  The restore should fail.
71  ASSERT_TRUE([window makeFirstResponder:viewB_]);
72  [viewC removeFromSuperview];
73  EXPECT_FALSE([tracker restoreFocusInWindow:window]);
74}
75
76TEST_F(FocusTrackerTest, DontRestoreFocusToViewInDifferentWindow) {
77  NSWindow* window = test_window();
78  ASSERT_TRUE([window makeFirstResponder:viewA_]);
79  scoped_nsobject<FocusTracker> tracker(
80      [[FocusTracker alloc] initWithWindow:window]);
81
82  // Give focus to |viewB_|, then try and restore focus in a different
83  // window.  It is ok to pass a nil NSWindow here because we only use
84  // it for direct comparison.
85  ASSERT_TRUE([window makeFirstResponder:viewB_]);
86  EXPECT_FALSE([tracker restoreFocusInWindow:nil]);
87}
88
89
90}  // namespace
91