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
|
// Copyright 2014 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.
#include <string>
#include "mojo/public/cpp/environment/default_async_waiter.h"
#include "mojo/public/cpp/environment/environment.h"
#include "mojo/public/cpp/system/core.h"
#include "mojo/public/cpp/system/macros.h"
#include "mojo/public/cpp/utility/run_loop.h"
#include "mojo/public/tests/test_utils.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace mojo {
namespace {
class TestAsyncWaitCallback {
public:
TestAsyncWaitCallback() : result_count_(0), last_result_(MOJO_RESULT_OK) {
}
virtual ~TestAsyncWaitCallback() {}
int result_count() const { return result_count_; }
MojoResult last_result() const { return last_result_; }
// MojoAsyncWaitCallback:
static void OnHandleReady(void* closure, MojoResult result) {
TestAsyncWaitCallback* self = static_cast<TestAsyncWaitCallback*>(closure);
self->result_count_++;
self->last_result_ = result;
}
private:
int result_count_;
MojoResult last_result_;
MOJO_DISALLOW_COPY_AND_ASSIGN(TestAsyncWaitCallback);
};
MojoAsyncWaitID CallAsyncWait(const Handle& handle,
MojoWaitFlags flags,
TestAsyncWaitCallback* callback) {
MojoAsyncWaiter* waiter = GetDefaultAsyncWaiter();
return waiter->AsyncWait(waiter,
handle.value(),
flags,
MOJO_DEADLINE_INDEFINITE,
&TestAsyncWaitCallback::OnHandleReady,
callback);
}
void CallCancelWait(MojoAsyncWaitID wait_id) {
MojoAsyncWaiter* waiter = GetDefaultAsyncWaiter();
waiter->CancelWait(waiter, wait_id);
}
class AsyncWaiterTest : public testing::Test {
public:
AsyncWaiterTest() {}
private:
Environment environment_;
RunLoop run_loop_;
MOJO_DISALLOW_COPY_AND_ASSIGN(AsyncWaiterTest);
};
// Verifies AsyncWaitCallback is notified when pipe is ready.
TEST_F(AsyncWaiterTest, CallbackNotified) {
TestAsyncWaitCallback callback;
MessagePipe test_pipe;
EXPECT_TRUE(test::WriteTextMessage(test_pipe.handle1.get(), std::string()));
CallAsyncWait(test_pipe.handle0.get(),
MOJO_WAIT_FLAG_READABLE,
&callback);
RunLoop::current()->Run();
EXPECT_EQ(1, callback.result_count());
EXPECT_EQ(MOJO_RESULT_OK, callback.last_result());
}
// Verifies 2 AsyncWaitCallbacks are notified when there pipes are ready.
TEST_F(AsyncWaiterTest, TwoCallbacksNotified) {
TestAsyncWaitCallback callback1;
TestAsyncWaitCallback callback2;
MessagePipe test_pipe1;
MessagePipe test_pipe2;
EXPECT_TRUE(test::WriteTextMessage(test_pipe1.handle1.get(), std::string()));
EXPECT_TRUE(test::WriteTextMessage(test_pipe2.handle1.get(), std::string()));
CallAsyncWait(test_pipe1.handle0.get(), MOJO_WAIT_FLAG_READABLE, &callback1);
CallAsyncWait(test_pipe2.handle0.get(), MOJO_WAIT_FLAG_READABLE, &callback2);
RunLoop::current()->Run();
EXPECT_EQ(1, callback1.result_count());
EXPECT_EQ(MOJO_RESULT_OK, callback1.last_result());
EXPECT_EQ(1, callback2.result_count());
EXPECT_EQ(MOJO_RESULT_OK, callback2.last_result());
}
// Verifies cancel works.
TEST_F(AsyncWaiterTest, CancelCallback) {
TestAsyncWaitCallback callback;
MessagePipe test_pipe;
EXPECT_TRUE(test::WriteTextMessage(test_pipe.handle1.get(), std::string()));
CallCancelWait(
CallAsyncWait(test_pipe.handle0.get(),
MOJO_WAIT_FLAG_READABLE,
&callback));
RunLoop::current()->Run();
EXPECT_EQ(0, callback.result_count());
}
} // namespace
} // namespace mojo
|
