| OLD | NEW |
| (Empty) |
| 1 // Copyright 2013 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 "mojo/public/system/core_cpp.h" | |
| 6 | |
| 7 #include <map> | |
| 8 | |
| 9 #include "mojo/public/system/macros.h" | |
| 10 #include "testing/gtest/include/gtest/gtest.h" | |
| 11 | |
| 12 namespace mojo { | |
| 13 namespace { | |
| 14 | |
| 15 TEST(CoreCppTest, GetTimeTicksNow) { | |
| 16 const MojoTimeTicks start = GetTimeTicksNow(); | |
| 17 EXPECT_NE(static_cast<MojoTimeTicks>(0), start) | |
| 18 << "GetTimeTicksNow should return nonzero value"; | |
| 19 } | |
| 20 | |
| 21 TEST(CoreCppTest, Basic) { | |
| 22 // Basic |Handle| implementation: | |
| 23 { | |
| 24 EXPECT_EQ(MOJO_HANDLE_INVALID, kInvalidHandleValue); | |
| 25 | |
| 26 Handle h_0; | |
| 27 EXPECT_EQ(kInvalidHandleValue, h_0.value()); | |
| 28 EXPECT_EQ(kInvalidHandleValue, *h_0.mutable_value()); | |
| 29 EXPECT_FALSE(h_0.is_valid()); | |
| 30 | |
| 31 Handle h_1(static_cast<MojoHandle>(123)); | |
| 32 EXPECT_EQ(static_cast<MojoHandle>(123), h_1.value()); | |
| 33 EXPECT_EQ(static_cast<MojoHandle>(123), *h_1.mutable_value()); | |
| 34 EXPECT_TRUE(h_1.is_valid()); | |
| 35 *h_1.mutable_value() = static_cast<MojoHandle>(456); | |
| 36 EXPECT_EQ(static_cast<MojoHandle>(456), h_1.value()); | |
| 37 EXPECT_TRUE(h_1.is_valid()); | |
| 38 | |
| 39 h_1.swap(h_0); | |
| 40 EXPECT_EQ(static_cast<MojoHandle>(456), h_0.value()); | |
| 41 EXPECT_TRUE(h_0.is_valid()); | |
| 42 EXPECT_FALSE(h_1.is_valid()); | |
| 43 | |
| 44 h_1.set_value(static_cast<MojoHandle>(789)); | |
| 45 h_0.swap(h_1); | |
| 46 EXPECT_EQ(static_cast<MojoHandle>(789), h_0.value()); | |
| 47 EXPECT_TRUE(h_0.is_valid()); | |
| 48 EXPECT_EQ(static_cast<MojoHandle>(456), h_1.value()); | |
| 49 EXPECT_TRUE(h_1.is_valid()); | |
| 50 | |
| 51 // Make sure copy constructor works. | |
| 52 Handle h_2(h_0); | |
| 53 EXPECT_EQ(static_cast<MojoHandle>(789), h_2.value()); | |
| 54 // And assignment. | |
| 55 h_2 = h_1; | |
| 56 EXPECT_EQ(static_cast<MojoHandle>(456), h_2.value()); | |
| 57 | |
| 58 // Make sure that we can put |Handle|s into |std::map|s. | |
| 59 h_0 = Handle(static_cast<MojoHandle>(987)); | |
| 60 h_1 = Handle(static_cast<MojoHandle>(654)); | |
| 61 h_2 = Handle(static_cast<MojoHandle>(321)); | |
| 62 Handle h_3; | |
| 63 std::map<Handle, int> handle_to_int; | |
| 64 handle_to_int[h_0] = 0; | |
| 65 handle_to_int[h_1] = 1; | |
| 66 handle_to_int[h_2] = 2; | |
| 67 handle_to_int[h_3] = 3; | |
| 68 | |
| 69 EXPECT_EQ(4u, handle_to_int.size()); | |
| 70 EXPECT_FALSE(handle_to_int.find(h_0) == handle_to_int.end()); | |
| 71 EXPECT_EQ(0, handle_to_int[h_0]); | |
| 72 EXPECT_FALSE(handle_to_int.find(h_1) == handle_to_int.end()); | |
| 73 EXPECT_EQ(1, handle_to_int[h_1]); | |
| 74 EXPECT_FALSE(handle_to_int.find(h_2) == handle_to_int.end()); | |
| 75 EXPECT_EQ(2, handle_to_int[h_2]); | |
| 76 EXPECT_FALSE(handle_to_int.find(h_3) == handle_to_int.end()); | |
| 77 EXPECT_EQ(3, handle_to_int[h_3]); | |
| 78 EXPECT_TRUE(handle_to_int.find(Handle(static_cast<MojoHandle>(13579))) == | |
| 79 handle_to_int.end()); | |
| 80 | |
| 81 // TODO(vtl): With C++11, support |std::unordered_map|s, etc. (Or figure out | |
| 82 // how to support the variations of |hash_map|.) | |
| 83 } | |
| 84 | |
| 85 // |Handle|/|ScopedHandle| functions: | |
| 86 { | |
| 87 ScopedHandle h; | |
| 88 | |
| 89 EXPECT_EQ(kInvalidHandleValue, h.get().value()); | |
| 90 | |
| 91 // This should be a no-op. | |
| 92 Close(h.Pass()); | |
| 93 | |
| 94 // It should still be invalid. | |
| 95 EXPECT_EQ(kInvalidHandleValue, h.get().value()); | |
| 96 | |
| 97 EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT, | |
| 98 Wait(h.get(), MOJO_WAIT_FLAG_EVERYTHING, 1000000)); | |
| 99 | |
| 100 std::vector<Handle> wh; | |
| 101 wh.push_back(h.get()); | |
| 102 std::vector<MojoWaitFlags> wf; | |
| 103 wf.push_back(MOJO_WAIT_FLAG_EVERYTHING); | |
| 104 EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT, | |
| 105 WaitMany(wh, wf, MOJO_DEADLINE_INDEFINITE)); | |
| 106 } | |
| 107 | |
| 108 // |MessagePipeHandle|/|ScopedMessagePipeHandle| functions: | |
| 109 { | |
| 110 MessagePipeHandle h_invalid; | |
| 111 EXPECT_FALSE(h_invalid.is_valid()); | |
| 112 EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT, | |
| 113 WriteMessageRaw(h_invalid, | |
| 114 NULL, 0, | |
| 115 NULL, 0, | |
| 116 MOJO_WRITE_MESSAGE_FLAG_NONE)); | |
| 117 char buffer[10] = { 0 }; | |
| 118 EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT, | |
| 119 WriteMessageRaw(h_invalid, | |
| 120 buffer, sizeof(buffer), | |
| 121 NULL, 0, | |
| 122 MOJO_WRITE_MESSAGE_FLAG_NONE)); | |
| 123 EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT, | |
| 124 ReadMessageRaw(h_invalid, | |
| 125 NULL, NULL, | |
| 126 NULL, NULL, | |
| 127 MOJO_READ_MESSAGE_FLAG_NONE)); | |
| 128 uint32_t buffer_size = static_cast<uint32_t>(sizeof(buffer)); | |
| 129 EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT, | |
| 130 ReadMessageRaw(h_invalid, | |
| 131 buffer, &buffer_size, | |
| 132 NULL, NULL, | |
| 133 MOJO_READ_MESSAGE_FLAG_NONE)); | |
| 134 | |
| 135 // Basic tests of waiting and closing. | |
| 136 MojoHandle hv_0 = kInvalidHandleValue; | |
| 137 { | |
| 138 ScopedMessagePipeHandle h_0; | |
| 139 ScopedMessagePipeHandle h_1; | |
| 140 EXPECT_FALSE(h_0.get().is_valid()); | |
| 141 EXPECT_FALSE(h_1.get().is_valid()); | |
| 142 | |
| 143 CreateMessagePipe(&h_0, &h_1); | |
| 144 EXPECT_TRUE(h_0.get().is_valid()); | |
| 145 EXPECT_TRUE(h_1.get().is_valid()); | |
| 146 EXPECT_NE(h_0.get().value(), h_1.get().value()); | |
| 147 // Save the handle values, so we can check that things got closed | |
| 148 // correctly. | |
| 149 hv_0 = h_0.get().value(); | |
| 150 MojoHandle hv_1 = h_1.get().value(); | |
| 151 | |
| 152 EXPECT_EQ(MOJO_RESULT_DEADLINE_EXCEEDED, | |
| 153 Wait(h_0.get(), MOJO_WAIT_FLAG_READABLE, 0)); | |
| 154 std::vector<Handle> wh; | |
| 155 wh.push_back(h_0.get()); | |
| 156 wh.push_back(h_1.get()); | |
| 157 std::vector<MojoWaitFlags> wf; | |
| 158 wf.push_back(MOJO_WAIT_FLAG_READABLE); | |
| 159 wf.push_back(MOJO_WAIT_FLAG_WRITABLE); | |
| 160 EXPECT_EQ(1, WaitMany(wh, wf, 1000)); | |
| 161 | |
| 162 // Test closing |h_1| explicitly. | |
| 163 Close(h_1.Pass()); | |
| 164 EXPECT_FALSE(h_1.get().is_valid()); | |
| 165 | |
| 166 // Make sure |h_1| is closed. | |
| 167 EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT, | |
| 168 MojoWait(hv_1, | |
| 169 MOJO_WAIT_FLAG_EVERYTHING, | |
| 170 MOJO_DEADLINE_INDEFINITE)); | |
| 171 | |
| 172 EXPECT_EQ(MOJO_RESULT_FAILED_PRECONDITION, | |
| 173 Wait(h_0.get(), MOJO_WAIT_FLAG_READABLE, | |
| 174 MOJO_DEADLINE_INDEFINITE)); | |
| 175 } | |
| 176 // |hv_0| should have been closed when |h_0| went out of scope, so this | |
| 177 // close should fail. | |
| 178 EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT, MojoClose(hv_0)); | |
| 179 | |
| 180 // Actually test writing/reading messages. | |
| 181 { | |
| 182 ScopedMessagePipeHandle h_0; | |
| 183 ScopedMessagePipeHandle h_1; | |
| 184 CreateMessagePipe(&h_0, &h_1); | |
| 185 | |
| 186 const char kHello[] = "hello"; | |
| 187 const uint32_t kHelloSize = static_cast<uint32_t>(sizeof(kHello)); | |
| 188 EXPECT_EQ(MOJO_RESULT_OK, | |
| 189 WriteMessageRaw(h_0.get(), | |
| 190 kHello, kHelloSize, | |
| 191 NULL, 0, | |
| 192 MOJO_WRITE_MESSAGE_FLAG_NONE)); | |
| 193 EXPECT_EQ(MOJO_RESULT_OK, | |
| 194 Wait(h_1.get(), MOJO_WAIT_FLAG_READABLE, | |
| 195 MOJO_DEADLINE_INDEFINITE)); | |
| 196 char buffer[10] = { 0 }; | |
| 197 uint32_t buffer_size = static_cast<uint32_t>(sizeof(buffer)); | |
| 198 EXPECT_EQ(MOJO_RESULT_OK, | |
| 199 ReadMessageRaw(h_1.get(), | |
| 200 buffer, &buffer_size, | |
| 201 NULL, NULL, | |
| 202 MOJO_READ_MESSAGE_FLAG_NONE)); | |
| 203 EXPECT_EQ(kHelloSize, buffer_size); | |
| 204 EXPECT_STREQ(kHello, buffer); | |
| 205 | |
| 206 // Send a handle over the previously-establish |MessagePipe|. | |
| 207 ScopedMessagePipeHandle h_2; | |
| 208 ScopedMessagePipeHandle h_3; | |
| 209 CreateMessagePipe(&h_2, &h_3); | |
| 210 | |
| 211 // Write a message to |h_2|, before we send |h_3|. | |
| 212 const char kWorld[] = "world!"; | |
| 213 const uint32_t kWorldSize = static_cast<uint32_t>(sizeof(kWorld)); | |
| 214 EXPECT_EQ(MOJO_RESULT_OK, | |
| 215 WriteMessageRaw(h_2.get(), | |
| 216 kWorld, kWorldSize, | |
| 217 NULL, 0, | |
| 218 MOJO_WRITE_MESSAGE_FLAG_NONE)); | |
| 219 | |
| 220 // Send |h_3| over |h_1| to |h_0|. | |
| 221 MojoHandle handles[5]; | |
| 222 handles[0] = h_3.release().value(); | |
| 223 EXPECT_NE(kInvalidHandleValue, handles[0]); | |
| 224 EXPECT_FALSE(h_3.get().is_valid()); | |
| 225 uint32_t handles_count = 1; | |
| 226 EXPECT_EQ(MOJO_RESULT_OK, | |
| 227 WriteMessageRaw(h_1.get(), | |
| 228 kHello, kHelloSize, | |
| 229 handles, handles_count, | |
| 230 MOJO_WRITE_MESSAGE_FLAG_NONE)); | |
| 231 // |handles[0]| should actually be invalid now. | |
| 232 EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT, MojoClose(handles[0])); | |
| 233 | |
| 234 // Read "hello" and the sent handle. | |
| 235 EXPECT_EQ(MOJO_RESULT_OK, | |
| 236 Wait(h_0.get(), MOJO_WAIT_FLAG_READABLE, | |
| 237 MOJO_DEADLINE_INDEFINITE)); | |
| 238 memset(buffer, 0, sizeof(buffer)); | |
| 239 buffer_size = static_cast<uint32_t>(sizeof(buffer)); | |
| 240 for (size_t i = 0; i < MOJO_ARRAYSIZE(handles); i++) | |
| 241 handles[i] = kInvalidHandleValue; | |
| 242 handles_count = static_cast<uint32_t>(MOJO_ARRAYSIZE(handles)); | |
| 243 EXPECT_EQ(MOJO_RESULT_OK, | |
| 244 ReadMessageRaw(h_0.get(), | |
| 245 buffer, &buffer_size, | |
| 246 handles, &handles_count, | |
| 247 MOJO_READ_MESSAGE_FLAG_NONE)); | |
| 248 EXPECT_EQ(kHelloSize, buffer_size); | |
| 249 EXPECT_STREQ(kHello, buffer); | |
| 250 EXPECT_EQ(1u, handles_count); | |
| 251 EXPECT_NE(kInvalidHandleValue, handles[0]); | |
| 252 | |
| 253 // Read from the sent/received handle. | |
| 254 h_3.reset(MessagePipeHandle(handles[0])); | |
| 255 // Save |handles[0]| to check that it gets properly closed. | |
| 256 hv_0 = handles[0]; | |
| 257 EXPECT_EQ(MOJO_RESULT_OK, | |
| 258 Wait(h_3.get(), MOJO_WAIT_FLAG_READABLE, | |
| 259 MOJO_DEADLINE_INDEFINITE)); | |
| 260 memset(buffer, 0, sizeof(buffer)); | |
| 261 buffer_size = static_cast<uint32_t>(sizeof(buffer)); | |
| 262 for (size_t i = 0; i < MOJO_ARRAYSIZE(handles); i++) | |
| 263 handles[i] = kInvalidHandleValue; | |
| 264 handles_count = static_cast<uint32_t>(MOJO_ARRAYSIZE(handles)); | |
| 265 EXPECT_EQ(MOJO_RESULT_OK, | |
| 266 ReadMessageRaw(h_3.get(), | |
| 267 buffer, &buffer_size, | |
| 268 handles, &handles_count, | |
| 269 MOJO_READ_MESSAGE_FLAG_NONE)); | |
| 270 EXPECT_EQ(kWorldSize, buffer_size); | |
| 271 EXPECT_STREQ(kWorld, buffer); | |
| 272 EXPECT_EQ(0u, handles_count); | |
| 273 } | |
| 274 EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT, MojoClose(hv_0)); | |
| 275 } | |
| 276 | |
| 277 // TODO(vtl): Test |CloseRaw()|. | |
| 278 // TODO(vtl): Test |reset()| more thoroughly? | |
| 279 } | |
| 280 | |
| 281 TEST(CoreCppTest, TearDownWithMessagesEnqueued) { | |
| 282 // Tear down a |MessagePipe| which still has a message enqueued, with the | |
| 283 // message also having a valid |MessagePipe| handle. | |
| 284 { | |
| 285 ScopedMessagePipeHandle h_0; | |
| 286 ScopedMessagePipeHandle h_1; | |
| 287 CreateMessagePipe(&h_0, &h_1); | |
| 288 | |
| 289 // Send a handle over the previously-establish |MessagePipe|. | |
| 290 ScopedMessagePipeHandle h_2; | |
| 291 ScopedMessagePipeHandle h_3; | |
| 292 CreateMessagePipe(&h_2, &h_3); | |
| 293 | |
| 294 // Write a message to |h_2|, before we send |h_3|. | |
| 295 const char kWorld[] = "world!"; | |
| 296 const uint32_t kWorldSize = static_cast<uint32_t>(sizeof(kWorld)); | |
| 297 EXPECT_EQ(MOJO_RESULT_OK, | |
| 298 WriteMessageRaw(h_2.get(), | |
| 299 kWorld, kWorldSize, | |
| 300 NULL, 0, | |
| 301 MOJO_WRITE_MESSAGE_FLAG_NONE)); | |
| 302 // And also a message to |h_3|. | |
| 303 EXPECT_EQ(MOJO_RESULT_OK, | |
| 304 WriteMessageRaw(h_3.get(), | |
| 305 kWorld, kWorldSize, | |
| 306 NULL, 0, | |
| 307 MOJO_WRITE_MESSAGE_FLAG_NONE)); | |
| 308 | |
| 309 // Send |h_3| over |h_1| to |h_0|. | |
| 310 const char kHello[] = "hello"; | |
| 311 const uint32_t kHelloSize = static_cast<uint32_t>(sizeof(kHello)); | |
| 312 MojoHandle h_3_value; | |
| 313 h_3_value = h_3.release().value(); | |
| 314 EXPECT_NE(kInvalidHandleValue, h_3_value); | |
| 315 EXPECT_FALSE(h_3.get().is_valid()); | |
| 316 EXPECT_EQ(MOJO_RESULT_OK, | |
| 317 WriteMessageRaw(h_1.get(), | |
| 318 kHello, kHelloSize, | |
| 319 &h_3_value, 1, | |
| 320 MOJO_WRITE_MESSAGE_FLAG_NONE)); | |
| 321 // |h_3_value| should actually be invalid now. | |
| 322 EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT, MojoClose(h_3_value)); | |
| 323 | |
| 324 EXPECT_EQ(MOJO_RESULT_OK, MojoClose(h_0.release().value())); | |
| 325 EXPECT_EQ(MOJO_RESULT_OK, MojoClose(h_1.release().value())); | |
| 326 EXPECT_EQ(MOJO_RESULT_OK, MojoClose(h_2.release().value())); | |
| 327 } | |
| 328 | |
| 329 // Do this in a different order: make the enqueued |MessagePipe| handle only | |
| 330 // half-alive. | |
| 331 { | |
| 332 ScopedMessagePipeHandle h_0; | |
| 333 ScopedMessagePipeHandle h_1; | |
| 334 CreateMessagePipe(&h_0, &h_1); | |
| 335 | |
| 336 // Send a handle over the previously-establish |MessagePipe|. | |
| 337 ScopedMessagePipeHandle h_2; | |
| 338 ScopedMessagePipeHandle h_3; | |
| 339 CreateMessagePipe(&h_2, &h_3); | |
| 340 | |
| 341 // Write a message to |h_2|, before we send |h_3|. | |
| 342 const char kWorld[] = "world!"; | |
| 343 const uint32_t kWorldSize = static_cast<uint32_t>(sizeof(kWorld)); | |
| 344 EXPECT_EQ(MOJO_RESULT_OK, | |
| 345 WriteMessageRaw(h_2.get(), | |
| 346 kWorld, kWorldSize, | |
| 347 NULL, 0, | |
| 348 MOJO_WRITE_MESSAGE_FLAG_NONE)); | |
| 349 // And also a message to |h_3|. | |
| 350 EXPECT_EQ(MOJO_RESULT_OK, | |
| 351 WriteMessageRaw(h_3.get(), | |
| 352 kWorld, kWorldSize, | |
| 353 NULL, 0, | |
| 354 MOJO_WRITE_MESSAGE_FLAG_NONE)); | |
| 355 | |
| 356 // Send |h_3| over |h_1| to |h_0|. | |
| 357 const char kHello[] = "hello"; | |
| 358 const uint32_t kHelloSize = static_cast<uint32_t>(sizeof(kHello)); | |
| 359 MojoHandle h_3_value; | |
| 360 h_3_value = h_3.release().value(); | |
| 361 EXPECT_NE(kInvalidHandleValue, h_3_value); | |
| 362 EXPECT_FALSE(h_3.get().is_valid()); | |
| 363 EXPECT_EQ(MOJO_RESULT_OK, | |
| 364 WriteMessageRaw(h_1.get(), | |
| 365 kHello, kHelloSize, | |
| 366 &h_3_value, 1, | |
| 367 MOJO_WRITE_MESSAGE_FLAG_NONE)); | |
| 368 // |h_3_value| should actually be invalid now. | |
| 369 EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT, MojoClose(h_3_value)); | |
| 370 | |
| 371 EXPECT_EQ(MOJO_RESULT_OK, MojoClose(h_2.release().value())); | |
| 372 EXPECT_EQ(MOJO_RESULT_OK, MojoClose(h_0.release().value())); | |
| 373 EXPECT_EQ(MOJO_RESULT_OK, MojoClose(h_1.release().value())); | |
| 374 } | |
| 375 } | |
| 376 | |
| 377 // TODO(vtl): Write data pipe tests. | |
| 378 | |
| 379 } // namespace | |
| 380 } // namespace mojo | |
| OLD | NEW |