| Index: mojo/public/cpp/bindings/tests/router_unittest.cc
|
| diff --git a/mojo/public/cpp/bindings/tests/router_unittest.cc b/mojo/public/cpp/bindings/tests/router_unittest.cc
|
| index ed3792f9c4387c483bdfe431f423d77dad79d1de..d3d8ccb8eae93327ab55e1a0f2a7f96db69d21c7 100644
|
| --- a/mojo/public/cpp/bindings/tests/router_unittest.cc
|
| +++ b/mojo/public/cpp/bindings/tests/router_unittest.cc
|
| @@ -47,19 +47,20 @@ class MessageAccumulator : public MessageReceiver {
|
| internal::MessageQueue* queue_;
|
| };
|
|
|
| -class ResponseGenerator : public MessageReceiverWithResponder {
|
| +class ResponseGenerator : public MessageReceiverWithResponderStatus {
|
| public:
|
| ResponseGenerator() {}
|
|
|
| bool Accept(Message* message) override { return false; }
|
|
|
| bool AcceptWithResponder(Message* message,
|
| - MessageReceiver* responder) override {
|
| + MessageReceiverWithStatus* responder) override {
|
| EXPECT_TRUE(message->has_flag(internal::kMessageExpectsResponse));
|
|
|
| bool result = SendResponse(
|
| message->name(), message->request_id(),
|
| reinterpret_cast<const char*>(message->payload()), responder);
|
| + EXPECT_TRUE(responder->IsValid());
|
| delete responder;
|
| return result;
|
| }
|
| @@ -84,7 +85,7 @@ class LazyResponseGenerator : public ResponseGenerator {
|
| ~LazyResponseGenerator() override { delete responder_; }
|
|
|
| bool AcceptWithResponder(Message* message,
|
| - MessageReceiver* responder) override {
|
| + MessageReceiverWithStatus* responder) override {
|
| name_ = message->name();
|
| request_id_ = message->request_id();
|
| request_string_ =
|
| @@ -95,6 +96,8 @@ class LazyResponseGenerator : public ResponseGenerator {
|
|
|
| bool has_responder() const { return !!responder_; }
|
|
|
| + bool responder_is_valid() const { return responder_->IsValid(); }
|
| +
|
| // Send the response and delete the responder.
|
| void CompleteWithResponse() { Complete(true); }
|
|
|
| @@ -112,7 +115,7 @@ class LazyResponseGenerator : public ResponseGenerator {
|
| responder_ = nullptr;
|
| }
|
|
|
| - MessageReceiver* responder_;
|
| + MessageReceiverWithStatus* responder_;
|
| uint32_t name_;
|
| uint64_t request_id_;
|
| std::string request_string_;
|
| @@ -261,6 +264,7 @@ TEST_F(RouterTest, LazyResponses) {
|
| EXPECT_TRUE(message_queue.IsEmpty());
|
|
|
| // Send the response.
|
| + EXPECT_TRUE(generator.responder_is_valid());
|
| generator.CompleteWithResponse();
|
| PumpMessages();
|
|
|
| @@ -283,6 +287,7 @@ TEST_F(RouterTest, LazyResponses) {
|
| EXPECT_TRUE(message_queue.IsEmpty());
|
|
|
| // Send the second response.
|
| + EXPECT_TRUE(generator.responder_is_valid());
|
| generator.CompleteWithResponse();
|
| PumpMessages();
|
|
|
| @@ -358,6 +363,7 @@ TEST_F(RouterTest, LateResponse) {
|
| EXPECT_TRUE(generator.has_responder());
|
| }
|
|
|
| + EXPECT_FALSE(generator.responder_is_valid());
|
| generator.CompleteWithResponse(); // This should end up doing nothing.
|
| }
|
|
|
|
|