Index: remoting/protocol/message_reader_unittest.cc |
diff --git a/remoting/protocol/message_reader_unittest.cc b/remoting/protocol/message_reader_unittest.cc |
index 9aba90aec7a0c18527a82e42c26c229c76f46bfb..bbd57e408dab769e9ed9c8085bd91fefc7747e1b 100644 |
--- a/remoting/protocol/message_reader_unittest.cc |
+++ b/remoting/protocol/message_reader_unittest.cc |
@@ -6,6 +6,7 @@ |
#include "base/bind.h" |
#include "base/bind_helpers.h" |
+#include "base/callback.h" |
#include "base/message_loop.h" |
#include "base/synchronization/waitable_event.h" |
#include "base/threading/thread.h" |
@@ -29,14 +30,13 @@ const char kTestMessage1[] = "Message1"; |
const char kTestMessage2[] = "Message2"; |
ACTION(CallDoneTask) { |
- arg1->Run(); |
- delete arg1; |
+ arg1.Run(); |
} |
} |
class MockMessageReceivedCallback { |
public: |
- MOCK_METHOD2(OnMessage, void(CompoundBuffer*, Task*)); |
+ MOCK_METHOD2(OnMessage, void(CompoundBuffer*, const base::Closure&)); |
}; |
class MessageReaderTest : public testing::Test { |
@@ -46,12 +46,12 @@ class MessageReaderTest : public testing::Test { |
run_task_finished_(false, false) { |
} |
- void RunDoneTaskOnOtherThread(CompoundBuffer* buffer, Task* done_task) { |
+ void RunDoneTaskOnOtherThread(CompoundBuffer* buffer, |
+ const base::Closure& done_task) { |
other_thread_.message_loop()->PostTask( |
FROM_HERE, |
base::Bind(&MessageReaderTest::RunAndDeleteTask, |
- base::Unretained(this), |
- done_task)); |
+ base::Unretained(this), done_task)); |
} |
protected: |
@@ -60,8 +60,8 @@ class MessageReaderTest : public testing::Test { |
} |
void InitReader() { |
- reader_->Init(&socket_, NewCallback( |
- &callback_, &MockMessageReceivedCallback::OnMessage)); |
+ reader_->Init(&socket_, base::Bind( |
+ &MockMessageReceivedCallback::OnMessage, base::Unretained(&callback_))); |
} |
void AddMessage(const std::string& message) { |
@@ -77,9 +77,8 @@ class MessageReaderTest : public testing::Test { |
return result == expected; |
} |
- void RunAndDeleteTask(Task* task) { |
- task->Run(); |
- delete task; |
+ void RunAndDeleteTask(const base::Closure& task) { |
+ task.Run(); |
run_task_finished_.Signal(); |
} |
@@ -94,7 +93,7 @@ class MessageReaderTest : public testing::Test { |
// Receive one message and process it with delay |
TEST_F(MessageReaderTest, OneMessage_Delay) { |
CompoundBuffer* buffer; |
- Task* done_task; |
+ base::Closure done_task; |
AddMessage(kTestMessage1); |
@@ -135,9 +134,9 @@ TEST_F(MessageReaderTest, OneMessage_Instant) { |
// Receive two messages in one packet. |
TEST_F(MessageReaderTest, TwoMessages_Together) { |
CompoundBuffer* buffer1; |
- Task* done_task1; |
+ base::Closure done_task1; |
CompoundBuffer* buffer2; |
- Task* done_task2; |
+ base::Closure done_task2; |
AddMessage(kTestMessage1); |
AddMessage(kTestMessage2); |
@@ -174,7 +173,7 @@ TEST_F(MessageReaderTest, TwoMessages_Together) { |
// instantly. |
TEST_F(MessageReaderTest, TwoMessages_Instant) { |
CompoundBuffer* buffer2; |
- Task* done_task2; |
+ base::Closure done_task2; |
AddMessage(kTestMessage1); |
AddMessage(kTestMessage2); |
@@ -220,7 +219,7 @@ TEST_F(MessageReaderTest, TwoMessages_Instant2) { |
// Receive two messages in separate packets. |
TEST_F(MessageReaderTest, TwoMessages_Separately) { |
CompoundBuffer* buffer; |
- Task* done_task; |
+ base::Closure done_task; |
AddMessage(kTestMessage1); |
@@ -280,7 +279,7 @@ TEST_F(MessageReaderTest, UseSocketOnCorrectThread) { |
// Write another message and verify that we receive it. |
CompoundBuffer* buffer; |
- Task* done_task; |
+ base::Closure done_task; |
EXPECT_CALL(callback_, OnMessage(_, _)) |
.Times(1) |
.WillOnce(DoAll(SaveArg<0>(&buffer), |