OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 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/test_channel_endpoint_client.h" | 5 #include "mojo/edk/system/test_channel_endpoint_client.h" |
6 | 6 |
7 #include "base/synchronization/waitable_event.h" | 7 #include "base/synchronization/waitable_event.h" |
8 #include "mojo/edk/system/message_in_transit.h" | 8 #include "mojo/edk/system/message_in_transit.h" |
9 #include "testing/gtest/include/gtest/gtest.h" | 9 #include "testing/gtest/include/gtest/gtest.h" |
10 | 10 |
(...skipping 13 matching lines...) Expand all Loading... |
24 endpoint_ = endpoint; | 24 endpoint_ = endpoint; |
25 } | 25 } |
26 | 26 |
27 bool TestChannelEndpointClient::IsDetached() const { | 27 bool TestChannelEndpointClient::IsDetached() const { |
28 base::AutoLock locker(lock_); | 28 base::AutoLock locker(lock_); |
29 return !endpoint_; | 29 return !endpoint_; |
30 } | 30 } |
31 | 31 |
32 size_t TestChannelEndpointClient::NumMessages() const { | 32 size_t TestChannelEndpointClient::NumMessages() const { |
33 base::AutoLock locker(lock_); | 33 base::AutoLock locker(lock_); |
34 return messages_.size(); | 34 return messages_.Size(); |
35 } | 35 } |
36 | 36 |
37 scoped_ptr<MessageInTransit> TestChannelEndpointClient::PopMessage() { | 37 scoped_ptr<MessageInTransit> TestChannelEndpointClient::PopMessage() { |
38 base::AutoLock locker(lock_); | 38 base::AutoLock locker(lock_); |
39 if (!messages_.size()) | 39 if (messages_.IsEmpty()) |
40 return nullptr; | 40 return nullptr; |
41 scoped_ptr<MessageInTransit> rv(messages_.front()); | 41 return messages_.GetMessage(); |
42 messages_.pop_front(); | |
43 return rv; | |
44 } | 42 } |
45 | 43 |
46 void TestChannelEndpointClient::SetReadEvent(base::WaitableEvent* read_event) { | 44 void TestChannelEndpointClient::SetReadEvent(base::WaitableEvent* read_event) { |
47 base::AutoLock locker(lock_); | 45 base::AutoLock locker(lock_); |
48 read_event_ = read_event; | 46 read_event_ = read_event; |
49 } | 47 } |
50 | 48 |
51 bool TestChannelEndpointClient::OnReadMessage(unsigned port, | 49 bool TestChannelEndpointClient::OnReadMessage(unsigned port, |
52 MessageInTransit* message) { | 50 MessageInTransit* message) { |
53 base::AutoLock locker(lock_); | 51 base::AutoLock locker(lock_); |
54 | 52 |
55 EXPECT_EQ(port_, port); | 53 EXPECT_EQ(port_, port); |
56 EXPECT_TRUE(endpoint_); | 54 EXPECT_TRUE(endpoint_); |
57 messages_.push_back(message); | 55 messages_.AddMessage(make_scoped_ptr(message)); |
58 | 56 |
59 if (read_event_) | 57 if (read_event_) |
60 read_event_->Signal(); | 58 read_event_->Signal(); |
61 | 59 |
62 return true; | 60 return true; |
63 } | 61 } |
64 | 62 |
65 void TestChannelEndpointClient::OnDetachFromChannel(unsigned port) { | 63 void TestChannelEndpointClient::OnDetachFromChannel(unsigned port) { |
66 EXPECT_EQ(port_, port); | 64 EXPECT_EQ(port_, port); |
67 | 65 |
68 base::AutoLock locker(lock_); | 66 base::AutoLock locker(lock_); |
69 ASSERT_TRUE(endpoint_); | 67 ASSERT_TRUE(endpoint_); |
70 endpoint_->DetachFromClient(); | 68 endpoint_->DetachFromClient(); |
71 endpoint_ = nullptr; | 69 endpoint_ = nullptr; |
72 } | 70 } |
73 | 71 |
74 TestChannelEndpointClient::~TestChannelEndpointClient() { | 72 TestChannelEndpointClient::~TestChannelEndpointClient() { |
75 EXPECT_FALSE(endpoint_); | 73 EXPECT_FALSE(endpoint_); |
76 for (auto message : messages_) | |
77 delete message; | |
78 } | 74 } |
79 | 75 |
80 } // namespace test | 76 } // namespace test |
81 } // namespace system | 77 } // namespace system |
82 } // namespace mojo | 78 } // namespace mojo |
OLD | NEW |