| 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 <stdlib.h> | 5 #include <stdlib.h> |
| 6 #include <string.h> | 6 #include <string.h> |
| 7 | 7 |
| 8 #include "mojo/public/cpp/bindings/lib/message_builder.h" | 8 #include "mojo/public/cpp/bindings/lib/message_builder.h" |
| 9 #include "mojo/public/cpp/bindings/lib/message_queue.h" | 9 #include "mojo/public/cpp/bindings/lib/message_queue.h" |
| 10 #include "mojo/public/cpp/bindings/lib/router.h" | 10 #include "mojo/public/cpp/bindings/lib/router.h" |
| (...skipping 19 matching lines...) Expand all Loading... |
| 30 internal::ResponseMessageBuilder builder(name, payload_size, request_id); | 30 internal::ResponseMessageBuilder builder(name, payload_size, request_id); |
| 31 memcpy(builder.buffer()->Allocate(payload_size), text, payload_size); | 31 memcpy(builder.buffer()->Allocate(payload_size), text, payload_size); |
| 32 builder.Finish(message); | 32 builder.Finish(message); |
| 33 } | 33 } |
| 34 | 34 |
| 35 class MessageAccumulator : public MessageReceiver { | 35 class MessageAccumulator : public MessageReceiver { |
| 36 public: | 36 public: |
| 37 explicit MessageAccumulator(internal::MessageQueue* queue) : queue_(queue) { | 37 explicit MessageAccumulator(internal::MessageQueue* queue) : queue_(queue) { |
| 38 } | 38 } |
| 39 | 39 |
| 40 virtual bool Accept(Message* message) MOJO_OVERRIDE { | 40 virtual bool Accept(Message* message) override { |
| 41 queue_->Push(message); | 41 queue_->Push(message); |
| 42 return true; | 42 return true; |
| 43 } | 43 } |
| 44 | 44 |
| 45 private: | 45 private: |
| 46 internal::MessageQueue* queue_; | 46 internal::MessageQueue* queue_; |
| 47 }; | 47 }; |
| 48 | 48 |
| 49 class ResponseGenerator : public MessageReceiverWithResponder { | 49 class ResponseGenerator : public MessageReceiverWithResponder { |
| 50 public: | 50 public: |
| 51 ResponseGenerator() { | 51 ResponseGenerator() { |
| 52 } | 52 } |
| 53 | 53 |
| 54 virtual bool Accept(Message* message) MOJO_OVERRIDE { | 54 virtual bool Accept(Message* message) override { return false; } |
| 55 return false; | |
| 56 } | |
| 57 | 55 |
| 58 virtual bool AcceptWithResponder(Message* message, MessageReceiver* responder) | 56 virtual bool AcceptWithResponder(Message* message, |
| 59 MOJO_OVERRIDE { | 57 MessageReceiver* responder) override { |
| 60 EXPECT_TRUE(message->has_flag(internal::kMessageExpectsResponse)); | 58 EXPECT_TRUE(message->has_flag(internal::kMessageExpectsResponse)); |
| 61 | 59 |
| 62 return SendResponse(message->name(), message->request_id(), responder); | 60 return SendResponse(message->name(), message->request_id(), responder); |
| 63 } | 61 } |
| 64 | 62 |
| 65 bool SendResponse(uint32_t name, uint64_t request_id, | 63 bool SendResponse(uint32_t name, uint64_t request_id, |
| 66 MessageReceiver* responder) { | 64 MessageReceiver* responder) { |
| 67 Message response; | 65 Message response; |
| 68 AllocResponseMessage(name, "world", request_id, &response); | 66 AllocResponseMessage(name, "world", request_id, &response); |
| 69 | 67 |
| 70 bool result = responder->Accept(&response); | 68 bool result = responder->Accept(&response); |
| 71 delete responder; | 69 delete responder; |
| 72 return result; | 70 return result; |
| 73 } | 71 } |
| 74 }; | 72 }; |
| 75 | 73 |
| 76 class LazyResponseGenerator : public ResponseGenerator { | 74 class LazyResponseGenerator : public ResponseGenerator { |
| 77 public: | 75 public: |
| 78 LazyResponseGenerator() : responder_(NULL), name_(0), request_id_(0) { | 76 LazyResponseGenerator() : responder_(NULL), name_(0), request_id_(0) { |
| 79 } | 77 } |
| 80 | 78 |
| 81 virtual ~LazyResponseGenerator() { | 79 virtual ~LazyResponseGenerator() { |
| 82 delete responder_; | 80 delete responder_; |
| 83 } | 81 } |
| 84 | 82 |
| 85 virtual bool AcceptWithResponder(Message* message, MessageReceiver* responder) | 83 virtual bool AcceptWithResponder(Message* message, |
| 86 MOJO_OVERRIDE { | 84 MessageReceiver* responder) override { |
| 87 name_ = message->name(); | 85 name_ = message->name(); |
| 88 request_id_ = message->request_id(); | 86 request_id_ = message->request_id(); |
| 89 responder_ = responder; | 87 responder_ = responder; |
| 90 return true; | 88 return true; |
| 91 } | 89 } |
| 92 | 90 |
| 93 bool has_responder() const { return !!responder_; } | 91 bool has_responder() const { return !!responder_; } |
| 94 | 92 |
| 95 void Complete() { | 93 void Complete() { |
| 96 SendResponse(name_, request_id_, responder_); | 94 SendResponse(name_, request_id_, responder_); |
| 97 responder_ = NULL; | 95 responder_ = NULL; |
| 98 } | 96 } |
| 99 | 97 |
| 100 private: | 98 private: |
| 101 MessageReceiver* responder_; | 99 MessageReceiver* responder_; |
| 102 uint32_t name_; | 100 uint32_t name_; |
| 103 uint32_t request_id_; | 101 uint32_t request_id_; |
| 104 }; | 102 }; |
| 105 | 103 |
| 106 class RouterTest : public testing::Test { | 104 class RouterTest : public testing::Test { |
| 107 public: | 105 public: |
| 108 RouterTest() { | 106 RouterTest() { |
| 109 } | 107 } |
| 110 | 108 |
| 111 virtual void SetUp() MOJO_OVERRIDE { | 109 virtual void SetUp() override { |
| 112 CreateMessagePipe(NULL, &handle0_, &handle1_); | 110 CreateMessagePipe(NULL, &handle0_, &handle1_); |
| 113 } | 111 } |
| 114 | 112 |
| 115 virtual void TearDown() MOJO_OVERRIDE { | 113 virtual void TearDown() override {} |
| 116 } | |
| 117 | 114 |
| 118 void PumpMessages() { | 115 void PumpMessages() { |
| 119 loop_.RunUntilIdle(); | 116 loop_.RunUntilIdle(); |
| 120 } | 117 } |
| 121 | 118 |
| 122 protected: | 119 protected: |
| 123 ScopedMessagePipeHandle handle0_; | 120 ScopedMessagePipeHandle handle0_; |
| 124 ScopedMessagePipeHandle handle1_; | 121 ScopedMessagePipeHandle handle1_; |
| 125 | 122 |
| 126 private: | 123 private: |
| (...skipping 94 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 221 EXPECT_TRUE(generator.has_responder()); | 218 EXPECT_TRUE(generator.has_responder()); |
| 222 | 219 |
| 223 } | 220 } |
| 224 | 221 |
| 225 generator.Complete(); // This should end up doing nothing. | 222 generator.Complete(); // This should end up doing nothing. |
| 226 } | 223 } |
| 227 | 224 |
| 228 } // namespace | 225 } // namespace |
| 229 } // namespace test | 226 } // namespace test |
| 230 } // namespace mojo | 227 } // namespace mojo |
| OLD | NEW |