| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 "base/message_loop_proxy_impl.h" | 5 #include "base/message_loop/message_loop_proxy_impl.h" |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/memory/scoped_ptr.h" | 8 #include "base/memory/scoped_ptr.h" |
| 9 #include "base/message_loop.h" | 9 #include "base/message_loop.h" |
| 10 #include "base/message_loop_proxy.h" | 10 #include "base/message_loop/message_loop_proxy.h" |
| 11 #include "base/threading/thread.h" | 11 #include "base/threading/thread.h" |
| 12 #include "testing/gtest/include/gtest/gtest.h" | 12 #include "testing/gtest/include/gtest/gtest.h" |
| 13 #include "testing/platform_test.h" | 13 #include "testing/platform_test.h" |
| 14 | 14 |
| 15 namespace base { |
| 15 | 16 |
| 16 class MessageLoopProxyImplTest : public testing::Test { | 17 class MessageLoopProxyImplTest : public testing::Test { |
| 17 public: | 18 public: |
| 18 void Release() const { | 19 void Release() const { |
| 19 AssertOnIOThread(); | 20 AssertOnIOThread(); |
| 20 Quit(); | 21 Quit(); |
| 21 } | 22 } |
| 22 | 23 |
| 23 void Quit() const { | 24 void Quit() const { |
| 24 loop_.PostTask(FROM_HERE, MessageLoop::QuitWhenIdleClosure()); | 25 loop_.PostTask(FROM_HERE, MessageLoop::QuitWhenIdleClosure()); |
| 25 } | 26 } |
| 26 | 27 |
| 27 void AssertOnIOThread() const { | 28 void AssertOnIOThread() const { |
| 28 ASSERT_TRUE(io_thread_->message_loop_proxy()->BelongsToCurrentThread()); | 29 ASSERT_TRUE(io_thread_->message_loop_proxy()->BelongsToCurrentThread()); |
| 29 ASSERT_EQ(io_thread_->message_loop_proxy(), | 30 ASSERT_EQ(io_thread_->message_loop_proxy(), |
| 30 base::MessageLoopProxy::current()); | 31 MessageLoopProxy::current()); |
| 31 } | 32 } |
| 32 | 33 |
| 33 void AssertOnFileThread() const { | 34 void AssertOnFileThread() const { |
| 34 ASSERT_TRUE(file_thread_->message_loop_proxy()->BelongsToCurrentThread()); | 35 ASSERT_TRUE(file_thread_->message_loop_proxy()->BelongsToCurrentThread()); |
| 35 ASSERT_EQ(file_thread_->message_loop_proxy(), | 36 ASSERT_EQ(file_thread_->message_loop_proxy(), |
| 36 base::MessageLoopProxy::current()); | 37 MessageLoopProxy::current()); |
| 37 } | 38 } |
| 38 | 39 |
| 39 protected: | 40 protected: |
| 40 virtual void SetUp() OVERRIDE { | 41 virtual void SetUp() OVERRIDE { |
| 41 io_thread_.reset(new base::Thread("MessageLoopProxyImplTest_IO")); | 42 io_thread_.reset(new Thread("MessageLoopProxyImplTest_IO")); |
| 42 file_thread_.reset(new base::Thread("MessageLoopProxyImplTest_File")); | 43 file_thread_.reset(new Thread("MessageLoopProxyImplTest_File")); |
| 43 io_thread_->Start(); | 44 io_thread_->Start(); |
| 44 file_thread_->Start(); | 45 file_thread_->Start(); |
| 45 } | 46 } |
| 46 | 47 |
| 47 virtual void TearDown() OVERRIDE { | 48 virtual void TearDown() OVERRIDE { |
| 48 io_thread_->Stop(); | 49 io_thread_->Stop(); |
| 49 file_thread_->Stop(); | 50 file_thread_->Stop(); |
| 50 } | 51 } |
| 51 | 52 |
| 52 static void BasicFunction(MessageLoopProxyImplTest* test) { | 53 static void BasicFunction(MessageLoopProxyImplTest* test) { |
| (...skipping 11 matching lines...) Expand all Loading... |
| 64 | 65 |
| 65 ~DeletedOnFile() { | 66 ~DeletedOnFile() { |
| 66 test_->AssertOnFileThread(); | 67 test_->AssertOnFileThread(); |
| 67 test_->Quit(); | 68 test_->Quit(); |
| 68 } | 69 } |
| 69 | 70 |
| 70 private: | 71 private: |
| 71 MessageLoopProxyImplTest* test_; | 72 MessageLoopProxyImplTest* test_; |
| 72 }; | 73 }; |
| 73 | 74 |
| 74 scoped_ptr<base::Thread> io_thread_; | 75 scoped_ptr<Thread> io_thread_; |
| 75 scoped_ptr<base::Thread> file_thread_; | 76 scoped_ptr<Thread> file_thread_; |
| 76 | 77 |
| 77 private: | 78 private: |
| 78 mutable MessageLoop loop_; | 79 mutable MessageLoop loop_; |
| 79 }; | 80 }; |
| 80 | 81 |
| 81 TEST_F(MessageLoopProxyImplTest, Release) { | 82 TEST_F(MessageLoopProxyImplTest, Release) { |
| 82 EXPECT_TRUE(io_thread_->message_loop_proxy()->ReleaseSoon(FROM_HERE, this)); | 83 EXPECT_TRUE(io_thread_->message_loop_proxy()->ReleaseSoon(FROM_HERE, this)); |
| 83 MessageLoop::current()->Run(); | 84 MessageLoop::current()->Run(); |
| 84 } | 85 } |
| 85 | 86 |
| 86 TEST_F(MessageLoopProxyImplTest, Delete) { | 87 TEST_F(MessageLoopProxyImplTest, Delete) { |
| 87 DeletedOnFile* deleted_on_file = new DeletedOnFile(this); | 88 DeletedOnFile* deleted_on_file = new DeletedOnFile(this); |
| 88 EXPECT_TRUE(file_thread_->message_loop_proxy()->DeleteSoon( | 89 EXPECT_TRUE(file_thread_->message_loop_proxy()->DeleteSoon( |
| 89 FROM_HERE, deleted_on_file)); | 90 FROM_HERE, deleted_on_file)); |
| 90 MessageLoop::current()->Run(); | 91 MessageLoop::current()->Run(); |
| 91 } | 92 } |
| 92 | 93 |
| 93 TEST_F(MessageLoopProxyImplTest, PostTask) { | 94 TEST_F(MessageLoopProxyImplTest, PostTask) { |
| 94 EXPECT_TRUE(file_thread_->message_loop_proxy()->PostTask( | 95 EXPECT_TRUE(file_thread_->message_loop_proxy()->PostTask( |
| 95 FROM_HERE, base::Bind(&MessageLoopProxyImplTest::BasicFunction, | 96 FROM_HERE, Bind(&MessageLoopProxyImplTest::BasicFunction, |
| 96 base::Unretained(this)))); | 97 Unretained(this)))); |
| 97 MessageLoop::current()->Run(); | 98 MessageLoop::current()->Run(); |
| 98 } | 99 } |
| 99 | 100 |
| 100 TEST_F(MessageLoopProxyImplTest, PostTaskAfterThreadExits) { | 101 TEST_F(MessageLoopProxyImplTest, PostTaskAfterThreadExits) { |
| 101 scoped_ptr<base::Thread> test_thread( | 102 scoped_ptr<Thread> test_thread( |
| 102 new base::Thread("MessageLoopProxyImplTest_Dummy")); | 103 new Thread("MessageLoopProxyImplTest_Dummy")); |
| 103 test_thread->Start(); | 104 test_thread->Start(); |
| 104 scoped_refptr<base::MessageLoopProxy> message_loop_proxy = | 105 scoped_refptr<MessageLoopProxy> message_loop_proxy = |
| 105 test_thread->message_loop_proxy(); | 106 test_thread->message_loop_proxy(); |
| 106 test_thread->Stop(); | 107 test_thread->Stop(); |
| 107 | 108 |
| 108 bool ret = message_loop_proxy->PostTask( | 109 bool ret = message_loop_proxy->PostTask( |
| 109 FROM_HERE, | 110 FROM_HERE, |
| 110 base::Bind(&MessageLoopProxyImplTest::AssertNotRun)); | 111 Bind(&MessageLoopProxyImplTest::AssertNotRun)); |
| 111 EXPECT_FALSE(ret); | 112 EXPECT_FALSE(ret); |
| 112 } | 113 } |
| 113 | 114 |
| 114 TEST_F(MessageLoopProxyImplTest, PostTaskAfterThreadIsDeleted) { | 115 TEST_F(MessageLoopProxyImplTest, PostTaskAfterThreadIsDeleted) { |
| 115 scoped_refptr<base::MessageLoopProxy> message_loop_proxy; | 116 scoped_refptr<MessageLoopProxy> message_loop_proxy; |
| 116 { | 117 { |
| 117 scoped_ptr<base::Thread> test_thread( | 118 scoped_ptr<Thread> test_thread( |
| 118 new base::Thread("MessageLoopProxyImplTest_Dummy")); | 119 new Thread("MessageLoopProxyImplTest_Dummy")); |
| 119 test_thread->Start(); | 120 test_thread->Start(); |
| 120 message_loop_proxy = test_thread->message_loop_proxy(); | 121 message_loop_proxy = test_thread->message_loop_proxy(); |
| 121 } | 122 } |
| 122 bool ret = message_loop_proxy->PostTask( | 123 bool ret = message_loop_proxy->PostTask( |
| 123 FROM_HERE, | 124 FROM_HERE, |
| 124 base::Bind(&MessageLoopProxyImplTest::AssertNotRun)); | 125 Bind(&MessageLoopProxyImplTest::AssertNotRun)); |
| 125 EXPECT_FALSE(ret); | 126 EXPECT_FALSE(ret); |
| 126 } | 127 } |
| 128 |
| 129 } // namespace base |
| OLD | NEW |