blob: 0cd8eff237ac6f054f263cbad4a6e8104c65dcce (
plain)
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
|
// Copyright 2018 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.
#ifndef COMPONENTS_AUTOFILL_IOS_BROWSER_IOS_TEST_EVENT_WAITER_H_
#define COMPONENTS_AUTOFILL_IOS_BROWSER_IOS_TEST_EVENT_WAITER_H_
#include <list>
#import "base/test/ios/wait_util.h"
namespace autofill {
// IOSTestEventWaiter is used to wait on given events that may have occurred
// before call to Wait(), or after, in which case a |timeout| should be provided
// to wait for those events to occur until |timeout| expires.
//
// Usage:
// waiter_ = std::make_unique<IOSTestEventWaiter>({ ... });
//
// Do stuff, which (a)synchronously calls waiter_->OnEvent(...).
//
// waiter_->Wait();
template <typename Event>
class IOSTestEventWaiter {
public:
explicit IOSTestEventWaiter(std::list<Event> expected_events,
double timeout = 0);
~IOSTestEventWaiter() = default;
// Either returns true right away if all events were observed between this
// object's construction and this call to Wait(); or returns true if that
// condition is met before |timeout|; Otherwise returns false. If |timeout| is
// zero, a reasonable default is used. Returns false if the current NSRunLoop
// is already running.
bool Wait();
// Observes an event. Returns false if the event is unexpected and true
// Otherwise.
bool OnEvent(Event event);
private:
std::list<Event> expected_events_;
bool runloop_running_;
double timeout_;
DISALLOW_COPY_AND_ASSIGN(IOSTestEventWaiter);
};
template <typename Event>
IOSTestEventWaiter<Event>::IOSTestEventWaiter(std::list<Event> expected_events,
double timeout)
: expected_events_(std::move(expected_events)),
runloop_running_(false),
timeout_(timeout) {}
template <typename Event>
bool IOSTestEventWaiter<Event>::Wait() {
if (expected_events_.empty())
return true;
if (runloop_running_)
return false;
runloop_running_ = true;
bool result = base::test::ios::WaitUntilConditionOrTimeout(timeout_, ^{
return expected_events_.empty();
});
runloop_running_ = false;
return result;
}
template <typename Event>
bool IOSTestEventWaiter<Event>::OnEvent(Event event) {
if (expected_events_.empty() || expected_events_.front() != event)
return false;
expected_events_.pop_front();
return true;
}
} // namespace autofill
#endif // COMPONENTS_AUTOFILL_IOS_BROWSER_IOS_TEST_EVENT_WAITER_H_
|