OLD | NEW |
| (Empty) |
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 #include "base/win/object_watcher.h" | |
6 | |
7 #include <process.h> | |
8 | |
9 #include "base/message_loop/message_loop.h" | |
10 #include "base/run_loop.h" | |
11 #include "testing/gtest/include/gtest/gtest.h" | |
12 | |
13 namespace base { | |
14 namespace win { | |
15 | |
16 namespace { | |
17 | |
18 class QuitDelegate : public ObjectWatcher::Delegate { | |
19 public: | |
20 void OnObjectSignaled(HANDLE object) override { | |
21 MessageLoop::current()->QuitWhenIdle(); | |
22 } | |
23 }; | |
24 | |
25 class DecrementCountDelegate : public ObjectWatcher::Delegate { | |
26 public: | |
27 explicit DecrementCountDelegate(int* counter) : counter_(counter) { | |
28 } | |
29 void OnObjectSignaled(HANDLE object) override { --(*counter_); } | |
30 | |
31 private: | |
32 int* counter_; | |
33 }; | |
34 | |
35 void RunTest_BasicSignal(MessageLoop::Type message_loop_type) { | |
36 MessageLoop message_loop(message_loop_type); | |
37 | |
38 ObjectWatcher watcher; | |
39 EXPECT_FALSE(watcher.IsWatching()); | |
40 | |
41 // A manual-reset event that is not yet signaled. | |
42 HANDLE event = CreateEvent(NULL, TRUE, FALSE, NULL); | |
43 | |
44 QuitDelegate delegate; | |
45 bool ok = watcher.StartWatching(event, &delegate); | |
46 EXPECT_TRUE(ok); | |
47 EXPECT_TRUE(watcher.IsWatching()); | |
48 EXPECT_EQ(event, watcher.GetWatchedObject()); | |
49 | |
50 SetEvent(event); | |
51 | |
52 MessageLoop::current()->Run(); | |
53 | |
54 EXPECT_FALSE(watcher.IsWatching()); | |
55 CloseHandle(event); | |
56 } | |
57 | |
58 void RunTest_BasicCancel(MessageLoop::Type message_loop_type) { | |
59 MessageLoop message_loop(message_loop_type); | |
60 | |
61 ObjectWatcher watcher; | |
62 | |
63 // A manual-reset event that is not yet signaled. | |
64 HANDLE event = CreateEvent(NULL, TRUE, FALSE, NULL); | |
65 | |
66 QuitDelegate delegate; | |
67 bool ok = watcher.StartWatching(event, &delegate); | |
68 EXPECT_TRUE(ok); | |
69 | |
70 watcher.StopWatching(); | |
71 | |
72 CloseHandle(event); | |
73 } | |
74 | |
75 void RunTest_CancelAfterSet(MessageLoop::Type message_loop_type) { | |
76 MessageLoop message_loop(message_loop_type); | |
77 | |
78 ObjectWatcher watcher; | |
79 | |
80 int counter = 1; | |
81 DecrementCountDelegate delegate(&counter); | |
82 | |
83 // A manual-reset event that is not yet signaled. | |
84 HANDLE event = CreateEvent(NULL, TRUE, FALSE, NULL); | |
85 | |
86 bool ok = watcher.StartWatching(event, &delegate); | |
87 EXPECT_TRUE(ok); | |
88 | |
89 SetEvent(event); | |
90 | |
91 // Let the background thread do its business | |
92 Sleep(30); | |
93 | |
94 watcher.StopWatching(); | |
95 | |
96 RunLoop().RunUntilIdle(); | |
97 | |
98 // Our delegate should not have fired. | |
99 EXPECT_EQ(1, counter); | |
100 | |
101 CloseHandle(event); | |
102 } | |
103 | |
104 void RunTest_SignalBeforeWatch(MessageLoop::Type message_loop_type) { | |
105 MessageLoop message_loop(message_loop_type); | |
106 | |
107 ObjectWatcher watcher; | |
108 | |
109 // A manual-reset event that is signaled before we begin watching. | |
110 HANDLE event = CreateEvent(NULL, TRUE, TRUE, NULL); | |
111 | |
112 QuitDelegate delegate; | |
113 bool ok = watcher.StartWatching(event, &delegate); | |
114 EXPECT_TRUE(ok); | |
115 | |
116 MessageLoop::current()->Run(); | |
117 | |
118 EXPECT_FALSE(watcher.IsWatching()); | |
119 CloseHandle(event); | |
120 } | |
121 | |
122 void RunTest_OutlivesMessageLoop(MessageLoop::Type message_loop_type) { | |
123 // Simulate a MessageLoop that dies before an ObjectWatcher. This ordinarily | |
124 // doesn't happen when people use the Thread class, but it can happen when | |
125 // people use the Singleton pattern or atexit. | |
126 HANDLE event = CreateEvent(NULL, TRUE, FALSE, NULL); // not signaled | |
127 { | |
128 ObjectWatcher watcher; | |
129 { | |
130 MessageLoop message_loop(message_loop_type); | |
131 | |
132 QuitDelegate delegate; | |
133 watcher.StartWatching(event, &delegate); | |
134 } | |
135 } | |
136 CloseHandle(event); | |
137 } | |
138 | |
139 } // namespace | |
140 | |
141 //----------------------------------------------------------------------------- | |
142 | |
143 TEST(ObjectWatcherTest, BasicSignal) { | |
144 RunTest_BasicSignal(MessageLoop::TYPE_DEFAULT); | |
145 RunTest_BasicSignal(MessageLoop::TYPE_IO); | |
146 RunTest_BasicSignal(MessageLoop::TYPE_UI); | |
147 } | |
148 | |
149 TEST(ObjectWatcherTest, BasicCancel) { | |
150 RunTest_BasicCancel(MessageLoop::TYPE_DEFAULT); | |
151 RunTest_BasicCancel(MessageLoop::TYPE_IO); | |
152 RunTest_BasicCancel(MessageLoop::TYPE_UI); | |
153 } | |
154 | |
155 TEST(ObjectWatcherTest, CancelAfterSet) { | |
156 RunTest_CancelAfterSet(MessageLoop::TYPE_DEFAULT); | |
157 RunTest_CancelAfterSet(MessageLoop::TYPE_IO); | |
158 RunTest_CancelAfterSet(MessageLoop::TYPE_UI); | |
159 } | |
160 | |
161 TEST(ObjectWatcherTest, SignalBeforeWatch) { | |
162 RunTest_SignalBeforeWatch(MessageLoop::TYPE_DEFAULT); | |
163 RunTest_SignalBeforeWatch(MessageLoop::TYPE_IO); | |
164 RunTest_SignalBeforeWatch(MessageLoop::TYPE_UI); | |
165 } | |
166 | |
167 TEST(ObjectWatcherTest, OutlivesMessageLoop) { | |
168 RunTest_OutlivesMessageLoop(MessageLoop::TYPE_DEFAULT); | |
169 RunTest_OutlivesMessageLoop(MessageLoop::TYPE_IO); | |
170 RunTest_OutlivesMessageLoop(MessageLoop::TYPE_UI); | |
171 } | |
172 | |
173 } // namespace win | |
174 } // namespace base | |
OLD | NEW |