Index: mojo/public/cpp/bindings/tests/connector_unittest.cc |
diff --git a/mojo/public/cpp/bindings/tests/connector_unittest.cc b/mojo/public/cpp/bindings/tests/connector_unittest.cc |
index 810ab8b88fcd138b89dc7fadb12a4403049652cf..0a81c9b85843f6b3bac8bb87de87e4a927663952 100644 |
--- a/mojo/public/cpp/bindings/tests/connector_unittest.cc |
+++ b/mojo/public/cpp/bindings/tests/connector_unittest.cc |
@@ -10,6 +10,8 @@ |
#include <utility> |
#include "base/bind.h" |
+#include "base/callback.h" |
+#include "base/callback_helpers.h" |
#include "base/message_loop/message_loop.h" |
#include "base/run_loop.h" |
#include "base/threading/thread_task_runner_handle.h" |
@@ -24,15 +26,13 @@ namespace { |
class MessageAccumulator : public MessageReceiver { |
public: |
MessageAccumulator() {} |
- explicit MessageAccumulator(const Closure& closure) : closure_(closure) {} |
+ explicit MessageAccumulator(const base::Closure& closure) |
+ : closure_(closure) {} |
bool Accept(Message* message) override { |
queue_.Push(message); |
- if (!closure_.is_null()) { |
- Closure closure = closure_; |
- closure_.Reset(); |
- closure.Run(); |
- } |
+ if (!closure_.is_null()) |
+ base::ResetAndReturn(&closure_).Run(); |
return true; |
} |
@@ -40,13 +40,13 @@ class MessageAccumulator : public MessageReceiver { |
void Pop(Message* message) { queue_.Pop(message); } |
- void set_closure(const Closure& closure) { closure_ = closure; } |
+ void set_closure(const base::Closure& closure) { closure_ = closure; } |
size_t size() const { return queue_.size(); } |
private: |
MessageQueue queue_; |
- Closure closure_; |
+ base::Closure closure_; |
}; |
class ConnectorDeletingMessageAccumulator : public MessageAccumulator { |