OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "mojo/edk/system/raw_channel.h" | 5 #include "mojo/edk/system/raw_channel.h" |
6 | 6 |
7 #include <stdint.h> | 7 #include <stdint.h> |
8 #include <stdio.h> | 8 #include <stdio.h> |
9 | 9 |
10 #include <memory> | 10 #include <memory> |
11 #include <utility> | 11 #include <utility> |
12 #include <vector> | 12 #include <vector> |
13 | 13 |
14 #include "base/logging.h" | 14 #include "base/logging.h" |
15 #include "mojo/edk/embedder/platform_channel_pair.h" | 15 #include "mojo/edk/embedder/platform_channel_pair.h" |
16 #include "mojo/edk/platform/platform_handle.h" | 16 #include "mojo/edk/platform/platform_handle.h" |
17 #include "mojo/edk/platform/scoped_platform_handle.h" | 17 #include "mojo/edk/platform/scoped_platform_handle.h" |
| 18 #include "mojo/edk/platform/thread_utils.h" |
18 #include "mojo/edk/system/message_in_transit.h" | 19 #include "mojo/edk/system/message_in_transit.h" |
19 #include "mojo/edk/system/test/random.h" | 20 #include "mojo/edk/system/test/random.h" |
20 #include "mojo/edk/system/test/scoped_test_dir.h" | 21 #include "mojo/edk/system/test/scoped_test_dir.h" |
21 #include "mojo/edk/system/test/simple_test_thread.h" | 22 #include "mojo/edk/system/test/simple_test_thread.h" |
22 #include "mojo/edk/system/test/sleep.h" | |
23 #include "mojo/edk/system/test/test_io_thread.h" | 23 #include "mojo/edk/system/test/test_io_thread.h" |
| 24 #include "mojo/edk/system/test/timeouts.h" |
24 #include "mojo/edk/system/transport_data.h" | 25 #include "mojo/edk/system/transport_data.h" |
25 #include "mojo/edk/test/test_utils.h" | 26 #include "mojo/edk/test/test_utils.h" |
26 #include "mojo/edk/util/make_unique.h" | 27 #include "mojo/edk/util/make_unique.h" |
27 #include "mojo/edk/util/mutex.h" | 28 #include "mojo/edk/util/mutex.h" |
28 #include "mojo/edk/util/scoped_file.h" | 29 #include "mojo/edk/util/scoped_file.h" |
29 #include "mojo/edk/util/waitable_event.h" | 30 #include "mojo/edk/util/waitable_event.h" |
30 #include "mojo/public/cpp/system/macros.h" | 31 #include "mojo/public/cpp/system/macros.h" |
31 #include "testing/gtest/include/gtest/gtest.h" | 32 #include "testing/gtest/include/gtest/gtest.h" |
32 | 33 |
33 using mojo::platform::PlatformHandle; | 34 using mojo::platform::PlatformHandle; |
34 using mojo::platform::ScopedPlatformHandle; | 35 using mojo::platform::ScopedPlatformHandle; |
| 36 using mojo::platform::ThreadSleep; |
35 using mojo::util::AutoResetWaitableEvent; | 37 using mojo::util::AutoResetWaitableEvent; |
36 using mojo::util::MakeUnique; | 38 using mojo::util::MakeUnique; |
37 using mojo::util::Mutex; | 39 using mojo::util::Mutex; |
38 using mojo::util::MutexLocker; | 40 using mojo::util::MutexLocker; |
39 | 41 |
40 namespace mojo { | 42 namespace mojo { |
41 namespace system { | 43 namespace system { |
42 namespace { | 44 namespace { |
43 | 45 |
44 std::unique_ptr<MessageInTransit> MakeTestMessage(uint32_t num_bytes) { | 46 std::unique_ptr<MessageInTransit> MakeTestMessage(uint32_t num_bytes) { |
(...skipping 121 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
166 | 168 |
167 // Erase message data. | 169 // Erase message data. |
168 bytes_.erase(bytes_.begin(), | 170 bytes_.erase(bytes_.begin(), |
169 bytes_.begin() + message_view.main_buffer_size()); | 171 bytes_.begin() + message_view.main_buffer_size()); |
170 return rv; | 172 return rv; |
171 } | 173 } |
172 } | 174 } |
173 | 175 |
174 if (static_cast<size_t>(read_size) < sizeof(buffer)) { | 176 if (static_cast<size_t>(read_size) < sizeof(buffer)) { |
175 i++; | 177 i++; |
176 test::SleepMilliseconds(kMessageReaderSleepMs); | 178 ThreadSleep(test::DeadlineFromMilliseconds(kMessageReaderSleepMs)); |
177 } | 179 } |
178 } | 180 } |
179 | 181 |
180 LOG(ERROR) << "Too many iterations."; | 182 LOG(ERROR) << "Too many iterations."; |
181 return false; | 183 return false; |
182 } | 184 } |
183 | 185 |
184 private: | 186 private: |
185 const PlatformHandle handle_; | 187 const PlatformHandle handle_; |
186 | 188 |
(...skipping 206 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
393 for (size_t i = 0; i < kNumWriterThreads; i++) { | 395 for (size_t i = 0; i < kNumWriterThreads; i++) { |
394 writer_threads.push_back(MakeUnique<RawChannelWriterThread>( | 396 writer_threads.push_back(MakeUnique<RawChannelWriterThread>( |
395 writer_rc.get(), kNumWriteMessagesPerThread)); | 397 writer_rc.get(), kNumWriteMessagesPerThread)); |
396 } | 398 } |
397 for (size_t i = 0; i < writer_threads.size(); i++) | 399 for (size_t i = 0; i < writer_threads.size(); i++) |
398 writer_threads[i]->Start(); | 400 writer_threads[i]->Start(); |
399 } // Joins all the writer threads. | 401 } // Joins all the writer threads. |
400 | 402 |
401 // Sleep a bit, to let any extraneous reads be processed. (There shouldn't be | 403 // Sleep a bit, to let any extraneous reads be processed. (There shouldn't be |
402 // any, but we want to know about them.) | 404 // any, but we want to know about them.) |
403 test::SleepMilliseconds(100u); | 405 ThreadSleep(test::DeadlineFromMilliseconds(100u)); |
404 | 406 |
405 // Wait for reading to finish. | 407 // Wait for reading to finish. |
406 reader_delegate.Wait(); | 408 reader_delegate.Wait(); |
407 | 409 |
408 io_thread()->PostTaskAndWait([&reader_rc]() { reader_rc->Shutdown(); }); | 410 io_thread()->PostTaskAndWait([&reader_rc]() { reader_rc->Shutdown(); }); |
409 | 411 |
410 io_thread()->PostTaskAndWait([&writer_rc]() { writer_rc->Shutdown(); }); | 412 io_thread()->PostTaskAndWait([&writer_rc]() { writer_rc->Shutdown(); }); |
411 } | 413 } |
412 | 414 |
413 // RawChannelTest.OnError ------------------------------------------------------ | 415 // RawChannelTest.OnError ------------------------------------------------------ |
(...skipping 67 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
481 // We should get a write error. | 483 // We should get a write error. |
482 delegate.WaitForWriteError(); | 484 delegate.WaitForWriteError(); |
483 | 485 |
484 // We should also get a read error. | 486 // We should also get a read error. |
485 delegate.WaitForReadError(); | 487 delegate.WaitForReadError(); |
486 | 488 |
487 EXPECT_FALSE(rc->WriteMessage(MakeTestMessage(2))); | 489 EXPECT_FALSE(rc->WriteMessage(MakeTestMessage(2))); |
488 | 490 |
489 // Sleep a bit, to make sure we don't get another |OnError()| | 491 // Sleep a bit, to make sure we don't get another |OnError()| |
490 // notification. (If we actually get another one, |OnError()| crashes.) | 492 // notification. (If we actually get another one, |OnError()| crashes.) |
491 test::SleepMilliseconds(20u); | 493 ThreadSleep(test::DeadlineFromMilliseconds(20u)); |
492 | 494 |
493 io_thread()->PostTaskAndWait([&rc]() { rc->Shutdown(); }); | 495 io_thread()->PostTaskAndWait([&rc]() { rc->Shutdown(); }); |
494 } | 496 } |
495 | 497 |
496 // RawChannelTest.ReadUnaffectedByWriteError ----------------------------------- | 498 // RawChannelTest.ReadUnaffectedByWriteError ----------------------------------- |
497 | 499 |
498 TEST_F(RawChannelTest, ReadUnaffectedByWriteError) { | 500 TEST_F(RawChannelTest, ReadUnaffectedByWriteError) { |
499 const size_t kMessageCount = 5; | 501 const size_t kMessageCount = 5; |
500 | 502 |
501 // Write a few messages into the other end. | 503 // Write a few messages into the other end. |
(...skipping 336 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
838 | 840 |
839 read_delegate.Wait(); | 841 read_delegate.Wait(); |
840 | 842 |
841 io_thread()->PostTaskAndWait([&rc_read]() { rc_read->Shutdown(); }); | 843 io_thread()->PostTaskAndWait([&rc_read]() { rc_read->Shutdown(); }); |
842 io_thread()->PostTaskAndWait([&rc_write]() { rc_write->Shutdown(); }); | 844 io_thread()->PostTaskAndWait([&rc_write]() { rc_write->Shutdown(); }); |
843 } | 845 } |
844 | 846 |
845 } // namespace | 847 } // namespace |
846 } // namespace system | 848 } // namespace system |
847 } // namespace mojo | 849 } // namespace mojo |
OLD | NEW |