File: cocoa_mouse_capture_unittest.mm

package info (click to toggle)
chromium 138.0.7204.183-1
  • links: PTS, VCS
  • area: main
  • in suites: trixie
  • size: 6,071,908 kB
  • sloc: cpp: 34,937,088; ansic: 7,176,967; javascript: 4,110,704; python: 1,419,953; asm: 946,768; xml: 739,971; pascal: 187,324; sh: 89,623; perl: 88,663; objc: 79,944; sql: 50,304; cs: 41,786; fortran: 24,137; makefile: 21,806; php: 13,980; tcl: 13,166; yacc: 8,925; ruby: 7,485; awk: 3,720; lisp: 3,096; lex: 1,327; ada: 727; jsp: 228; sed: 36
file content (169 lines) | stat: -rw-r--r-- 5,246 bytes parent folder | download | duplicates (5)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
// Copyright 2014 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#import <Cocoa/Cocoa.h>

#import "components/remote_cocoa/app_shim/mouse_capture.h"
#import "components/remote_cocoa/app_shim/mouse_capture_delegate.h"
#import "ui/base/test/cocoa_helper.h"
#import "ui/events/test/cocoa_test_event_utils.h"

// Simple test view that counts calls to -[NSView mouseDown:].
@interface CocoaMouseCaptureTestView : NSView
@property(readonly, nonatomic) int mouseDownCount;
@end

@implementation CocoaMouseCaptureTestView {
  int _mouseDownCount;
}

@synthesize mouseDownCount = _mouseDownCount;

- (void)mouseDown:(NSEvent*)theEvent {
  ++_mouseDownCount;
}

@end

namespace remote_cocoa {
namespace {

// Simple capture delegate that just counts events forwarded.
class TestCaptureDelegate : public CocoaMouseCaptureDelegate {
 public:
  explicit TestCaptureDelegate(NSWindow* window) : window_(window) {}

  TestCaptureDelegate(const TestCaptureDelegate&) = delete;
  TestCaptureDelegate& operator=(const TestCaptureDelegate&) = delete;

  void Acquire() { mouse_capture_ = std::make_unique<CocoaMouseCapture>(this); }
  bool IsActive() { return mouse_capture_ && mouse_capture_->IsActive(); }
  void SimulateDestroy() { mouse_capture_.reset(); }
  void set_should_claim_event(bool should_claim_event) {
    should_claim_event_ = should_claim_event;
  }

  int event_count() { return event_count_; }
  int capture_lost_count() { return capture_lost_count_; }

  // CocoaMouseCaptureDelegate:
  bool PostCapturedEvent(NSEvent* event) override {
    ++event_count_;
    return should_claim_event_;
  }
  void OnMouseCaptureLost() override { ++capture_lost_count_; }
  NSWindow* GetWindow() const override { return window_; }

 private:
  std::unique_ptr<CocoaMouseCapture> mouse_capture_;
  bool should_claim_event_ = true;
  int event_count_ = 0;
  int capture_lost_count_ = 0;
  NSWindow* __strong window_;
};

}  // namespace

using CocoaMouseCaptureTest = ui::CocoaTest;

// Test that a new capture properly "steals" capture from an existing one.
TEST_F(CocoaMouseCaptureTest, OnCaptureLost) {
  TestCaptureDelegate capture(nil);

  capture.Acquire();
  EXPECT_TRUE(capture.IsActive());
  {
    TestCaptureDelegate capture2(nil);
    EXPECT_EQ(0, capture.capture_lost_count());

    // A second capture steals from the first.
    capture2.Acquire();
    EXPECT_TRUE(capture2.IsActive());
    EXPECT_FALSE(capture.IsActive());
    EXPECT_EQ(1, capture.capture_lost_count());
    EXPECT_EQ(0, capture2.capture_lost_count());

    // Simulate capture2 going out of scope. Inspect it.
    capture2.SimulateDestroy();
    EXPECT_FALSE(capture2.IsActive());
    EXPECT_EQ(1, capture2.capture_lost_count());
  }

  // Re-acquiring is fine (not stealing).
  EXPECT_FALSE(capture.IsActive());
  capture.Acquire();
  EXPECT_TRUE(capture.IsActive());

  // Having no CocoaMouseCapture instance is fine.
  capture.SimulateDestroy();
  EXPECT_FALSE(capture.IsActive());
  // Receives OnMouseCaptureLost again, since reacquired.
  EXPECT_EQ(2, capture.capture_lost_count());
}

// Test event capture.
TEST_F(CocoaMouseCaptureTest, CaptureEvents) {
  CocoaMouseCaptureTestView* view =
      [[CocoaMouseCaptureTestView alloc] initWithFrame:NSZeroRect];
  test_window().contentView = view;
  NSArray<NSEvent*>* click = cocoa_test_event_utils::MouseClickInView(view, 1);

  // First check that the view receives events normally.
  EXPECT_EQ(0, view.mouseDownCount);
  [NSApp sendEvent:click[0]];
  EXPECT_EQ(1, view.mouseDownCount);

  {
    TestCaptureDelegate capture(test_window());
    capture.Acquire();

    // Now check that the capture captures events.
    EXPECT_EQ(0, capture.event_count());
    [NSApp sendEvent:click[0]];
    EXPECT_EQ(1, view.mouseDownCount);
    EXPECT_EQ(1, capture.event_count());
  }

  // After the capture goes away, events should be received again.
  [NSApp sendEvent:click[0]];
  EXPECT_EQ(2, view.mouseDownCount);
}

// Test local events properly swallowed / propagated.
TEST_F(CocoaMouseCaptureTest, SwallowOrPropagateEvents) {
  CocoaMouseCaptureTestView* view =
      [[CocoaMouseCaptureTestView alloc] initWithFrame:NSZeroRect];
  test_window().contentView = view;
  NSArray<NSEvent*>* click = cocoa_test_event_utils::MouseClickInView(view, 1);

  // First check that the view receives events normally.
  EXPECT_EQ(0, view.mouseDownCount);
  [NSApp sendEvent:click[0]];
  EXPECT_EQ(1, view.mouseDownCount);

  {
    TestCaptureDelegate capture(test_window());
    capture.Acquire();

    // By default, the delegate should claim events.
    EXPECT_EQ(0, capture.event_count());
    [NSApp sendEvent:click[0]];
    EXPECT_EQ(1, view.mouseDownCount);
    EXPECT_EQ(1, capture.event_count());

    // Set the delegate not to claim events.
    capture.set_should_claim_event(false);
    [NSApp sendEvent:click[0]];
    EXPECT_EQ(2, view.mouseDownCount);
    EXPECT_EQ(2, capture.event_count());

    // Setting it back should restart the claiming of events.
    capture.set_should_claim_event(true);
    [NSApp sendEvent:click[0]];
    EXPECT_EQ(2, view.mouseDownCount);
    EXPECT_EQ(3, capture.event_count());
  }
}

}  // namespace remote_cocoa