OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 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 | 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 <stdlib.h> | 5 #include <stdlib.h> |
6 #include <string.h> | 6 #include <string.h> |
7 | 7 |
8 #include "mojo/public/cpp/bindings/lib/connector.h" | 8 #include "mojo/public/cpp/bindings/lib/connector.h" |
9 #include "mojo/public/cpp/bindings/lib/message_builder.h" | 9 #include "mojo/public/cpp/bindings/lib/message_builder.h" |
10 #include "mojo/public/cpp/bindings/lib/message_queue.h" | 10 #include "mojo/public/cpp/bindings/lib/message_queue.h" |
11 #include "mojo/public/cpp/environment/environment.h" | 11 #include "mojo/public/cpp/environment/environment.h" |
12 #include "mojo/public/cpp/system/macros.h" | 12 #include "mojo/public/cpp/system/macros.h" |
13 #include "mojo/public/cpp/utility/run_loop.h" | 13 #include "mojo/public/cpp/utility/run_loop.h" |
14 #include "testing/gtest/include/gtest/gtest.h" | 14 #include "testing/gtest/include/gtest/gtest.h" |
15 | 15 |
16 namespace mojo { | 16 namespace mojo { |
17 namespace test { | 17 namespace test { |
18 namespace { | 18 namespace { |
19 | 19 |
20 class MessageAccumulator : public MessageReceiver { | 20 class MessageAccumulator : public MessageReceiver { |
21 public: | 21 public: |
22 MessageAccumulator() { | 22 MessageAccumulator() { |
23 } | 23 } |
24 | 24 |
25 virtual bool Accept(Message* message) MOJO_OVERRIDE { | 25 virtual bool Accept(Message* message) override { |
26 queue_.Push(message); | 26 queue_.Push(message); |
27 return true; | 27 return true; |
28 } | 28 } |
29 | 29 |
30 bool IsEmpty() const { | 30 bool IsEmpty() const { |
31 return queue_.IsEmpty(); | 31 return queue_.IsEmpty(); |
32 } | 32 } |
33 | 33 |
34 void Pop(Message* message) { | 34 void Pop(Message* message) { |
35 queue_.Pop(message); | 35 queue_.Pop(message); |
36 } | 36 } |
37 | 37 |
38 private: | 38 private: |
39 internal::MessageQueue queue_; | 39 internal::MessageQueue queue_; |
40 }; | 40 }; |
41 | 41 |
42 class ConnectorDeletingMessageAccumulator : public MessageAccumulator { | 42 class ConnectorDeletingMessageAccumulator : public MessageAccumulator { |
43 public: | 43 public: |
44 ConnectorDeletingMessageAccumulator(internal::Connector** connector) | 44 ConnectorDeletingMessageAccumulator(internal::Connector** connector) |
45 : connector_(connector) {} | 45 : connector_(connector) {} |
46 | 46 |
47 virtual bool Accept(Message* message) MOJO_OVERRIDE { | 47 virtual bool Accept(Message* message) override { |
48 delete *connector_; | 48 delete *connector_; |
49 *connector_ = 0; | 49 *connector_ = 0; |
50 return MessageAccumulator::Accept(message); | 50 return MessageAccumulator::Accept(message); |
51 } | 51 } |
52 | 52 |
53 private: | 53 private: |
54 internal::Connector** connector_; | 54 internal::Connector** connector_; |
55 }; | 55 }; |
56 | 56 |
57 class ReentrantMessageAccumulator : public MessageAccumulator { | 57 class ReentrantMessageAccumulator : public MessageAccumulator { |
58 public: | 58 public: |
59 ReentrantMessageAccumulator(internal::Connector* connector) | 59 ReentrantMessageAccumulator(internal::Connector* connector) |
60 : connector_(connector), number_of_calls_(0) {} | 60 : connector_(connector), number_of_calls_(0) {} |
61 | 61 |
62 virtual bool Accept(Message* message) MOJO_OVERRIDE { | 62 virtual bool Accept(Message* message) override { |
63 if (!MessageAccumulator::Accept(message)) | 63 if (!MessageAccumulator::Accept(message)) |
64 return false; | 64 return false; |
65 number_of_calls_++; | 65 number_of_calls_++; |
66 if (number_of_calls_ == 1) { | 66 if (number_of_calls_ == 1) { |
67 return connector_->WaitForIncomingMessage(); | 67 return connector_->WaitForIncomingMessage(); |
68 } | 68 } |
69 return true; | 69 return true; |
70 } | 70 } |
71 | 71 |
72 int number_of_calls() { return number_of_calls_; } | 72 int number_of_calls() { return number_of_calls_; } |
73 | 73 |
74 private: | 74 private: |
75 internal::Connector* connector_; | 75 internal::Connector* connector_; |
76 int number_of_calls_; | 76 int number_of_calls_; |
77 }; | 77 }; |
78 | 78 |
79 class ConnectorTest : public testing::Test { | 79 class ConnectorTest : public testing::Test { |
80 public: | 80 public: |
81 ConnectorTest() { | 81 ConnectorTest() { |
82 } | 82 } |
83 | 83 |
84 virtual void SetUp() MOJO_OVERRIDE { | 84 virtual void SetUp() override { |
85 CreateMessagePipe(NULL, &handle0_, &handle1_); | 85 CreateMessagePipe(NULL, &handle0_, &handle1_); |
86 } | 86 } |
87 | 87 |
88 virtual void TearDown() MOJO_OVERRIDE { | 88 virtual void TearDown() override {} |
89 } | |
90 | 89 |
91 void AllocMessage(const char* text, Message* message) { | 90 void AllocMessage(const char* text, Message* message) { |
92 size_t payload_size = strlen(text) + 1; // Plus null terminator. | 91 size_t payload_size = strlen(text) + 1; // Plus null terminator. |
93 internal::MessageBuilder builder(1, payload_size); | 92 internal::MessageBuilder builder(1, payload_size); |
94 memcpy(builder.buffer()->Allocate(payload_size), text, payload_size); | 93 memcpy(builder.buffer()->Allocate(payload_size), text, payload_size); |
95 builder.Finish(message); | 94 builder.Finish(message); |
96 } | 95 } |
97 | 96 |
98 void PumpMessages() { | 97 void PumpMessages() { |
99 loop_.RunUntilIdle(); | 98 loop_.RunUntilIdle(); |
(...skipping 294 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
394 std::string(kText[i]), | 393 std::string(kText[i]), |
395 std::string(reinterpret_cast<const char*>(message_received.payload()))); | 394 std::string(reinterpret_cast<const char*>(message_received.payload()))); |
396 } | 395 } |
397 | 396 |
398 ASSERT_EQ(2, accumulator.number_of_calls()); | 397 ASSERT_EQ(2, accumulator.number_of_calls()); |
399 } | 398 } |
400 | 399 |
401 } // namespace | 400 } // namespace |
402 } // namespace test | 401 } // namespace test |
403 } // namespace mojo | 402 } // namespace mojo |
OLD | NEW |