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
|
// Copyright 2015 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#import "ui/base/test/windowed_nsnotification_observer.h"
#import <Cocoa/Cocoa.h>
#include "base/location.h"
#include "base/memory/raw_ptr.h"
#include "base/run_loop.h"
#import "base/task/single_thread_task_runner.h"
#include "base/task/single_thread_task_runner.h"
#include "base/test/test_timeouts.h"
@interface WindowedNSNotificationObserver ()
- (void)onNotification:(NSNotification*)notification;
@end
@implementation WindowedNSNotificationObserver {
NSString* __strong _bundleId;
int _notificationCount;
raw_ptr<base::RunLoop> _runLoop;
}
@synthesize notificationCount = _notificationCount;
- (instancetype)initForNotification:(NSString*)name {
return [self initForNotification:name object:nil];
}
- (instancetype)initForNotification:(NSString*)name object:(id)sender {
if ((self = [super init])) {
[NSNotificationCenter.defaultCenter addObserver:self
selector:@selector(onNotification:)
name:name
object:sender];
}
return self;
}
- (instancetype)initForWorkspaceNotification:(NSString*)name
bundleId:(NSString*)bundleId {
if ((self = [super init])) {
_bundleId = [bundleId copy];
[NSWorkspace.sharedWorkspace.notificationCenter
addObserver:self
selector:@selector(onNotification:)
name:name
object:nil];
}
return self;
}
- (void)dealloc {
if (_bundleId)
[NSWorkspace.sharedWorkspace.notificationCenter removeObserver:self];
else
[NSNotificationCenter.defaultCenter removeObserver:self];
}
- (void)onNotification:(NSNotification*)notification {
if (_bundleId) {
NSRunningApplication* application =
[notification userInfo][NSWorkspaceApplicationKey];
if (![application.bundleIdentifier isEqualToString:_bundleId]) {
return;
}
}
++_notificationCount;
if (_runLoop)
_runLoop->Quit();
}
- (BOOL)waitForCount:(int)minimumCount {
while (_notificationCount < minimumCount) {
const int oldCount = _notificationCount;
base::RunLoop runLoop;
base::SingleThreadTaskRunner::GetCurrentDefault()->PostDelayedTask(
FROM_HERE, runLoop.QuitClosure(), TestTimeouts::action_timeout());
_runLoop = &runLoop;
runLoop.Run();
_runLoop = nullptr;
// If there was no new notification, it must have been a timeout.
if (_notificationCount == oldCount)
break;
}
return _notificationCount >= minimumCount;
}
- (BOOL)wait {
return [self waitForCount:1];
}
@end
|
