Index: mojo/edk/system/channel_unittest.cc |
diff --git a/mojo/edk/system/channel_unittest.cc b/mojo/edk/system/channel_unittest.cc |
deleted file mode 100644 |
index 237c05cf22f7dd7aaa399a915a15b923556bb96f..0000000000000000000000000000000000000000 |
--- a/mojo/edk/system/channel_unittest.cc |
+++ /dev/null |
@@ -1,323 +0,0 @@ |
-// Copyright 2014 The Chromium Authors. All rights reserved. |
-// Use of this source code is governed by a BSD-style license that can be |
-// found in the LICENSE file. |
- |
-#include "mojo/edk/system/channel.h" |
- |
-#include "base/bind.h" |
-#include "base/location.h" |
-#include "base/message_loop/message_loop.h" |
-#include "base/test/test_io_thread.h" |
-#include "mojo/edk/embedder/platform_channel_pair.h" |
-#include "mojo/edk/embedder/simple_platform_support.h" |
-#include "mojo/edk/system/channel_endpoint.h" |
-#include "mojo/edk/system/message_in_transit.h" |
-#include "mojo/edk/system/message_pipe.h" |
-#include "mojo/edk/system/raw_channel.h" |
-#include "mojo/edk/system/test_utils.h" |
-#include "mojo/edk/system/waiter.h" |
-#include "testing/gtest/include/gtest/gtest.h" |
- |
-namespace mojo { |
-namespace system { |
-namespace { |
- |
-enum Tristate { TRISTATE_UNKNOWN = -1, TRISTATE_FALSE = 0, TRISTATE_TRUE = 1 }; |
- |
-Tristate BoolToTristate(bool b) { |
- return b ? TRISTATE_TRUE : TRISTATE_FALSE; |
-} |
- |
-class ChannelTest : public testing::Test { |
- public: |
- ChannelTest() |
- : io_thread_(base::TestIOThread::kAutoStart), |
- init_result_(TRISTATE_UNKNOWN) {} |
- virtual ~ChannelTest() {} |
- |
- virtual void SetUp() override { |
- io_thread_.PostTaskAndWait( |
- FROM_HERE, |
- base::Bind(&ChannelTest::SetUpOnIOThread, base::Unretained(this))); |
- } |
- |
- void CreateChannelOnIOThread() { |
- CHECK_EQ(base::MessageLoop::current(), io_thread()->message_loop()); |
- channel_ = new Channel(&platform_support_); |
- } |
- |
- void InitChannelOnIOThread() { |
- CHECK_EQ(base::MessageLoop::current(), io_thread()->message_loop()); |
- |
- CHECK(raw_channel_); |
- CHECK(channel_.get()); |
- CHECK_EQ(init_result_, TRISTATE_UNKNOWN); |
- |
- init_result_ = BoolToTristate(channel_->Init(raw_channel_.Pass())); |
- } |
- |
- void ShutdownChannelOnIOThread() { |
- CHECK_EQ(base::MessageLoop::current(), io_thread()->message_loop()); |
- |
- CHECK(channel_.get()); |
- channel_->Shutdown(); |
- } |
- |
- base::TestIOThread* io_thread() { return &io_thread_; } |
- RawChannel* raw_channel() { return raw_channel_.get(); } |
- scoped_ptr<RawChannel>* mutable_raw_channel() { return &raw_channel_; } |
- Channel* channel() { return channel_.get(); } |
- scoped_refptr<Channel>* mutable_channel() { return &channel_; } |
- Tristate init_result() const { return init_result_; } |
- |
- private: |
- void SetUpOnIOThread() { |
- CHECK_EQ(base::MessageLoop::current(), io_thread()->message_loop()); |
- |
- embedder::PlatformChannelPair channel_pair; |
- raw_channel_ = RawChannel::Create(channel_pair.PassServerHandle()).Pass(); |
- other_platform_handle_ = channel_pair.PassClientHandle(); |
- } |
- |
- embedder::SimplePlatformSupport platform_support_; |
- base::TestIOThread io_thread_; |
- scoped_ptr<RawChannel> raw_channel_; |
- embedder::ScopedPlatformHandle other_platform_handle_; |
- scoped_refptr<Channel> channel_; |
- |
- Tristate init_result_; |
- |
- DISALLOW_COPY_AND_ASSIGN(ChannelTest); |
-}; |
- |
-// ChannelTest.InitShutdown ---------------------------------------------------- |
- |
-TEST_F(ChannelTest, InitShutdown) { |
- io_thread()->PostTaskAndWait(FROM_HERE, |
- base::Bind(&ChannelTest::CreateChannelOnIOThread, |
- base::Unretained(this))); |
- ASSERT_TRUE(channel()); |
- |
- io_thread()->PostTaskAndWait( |
- FROM_HERE, |
- base::Bind(&ChannelTest::InitChannelOnIOThread, base::Unretained(this))); |
- EXPECT_EQ(TRISTATE_TRUE, init_result()); |
- |
- io_thread()->PostTaskAndWait( |
- FROM_HERE, |
- base::Bind(&ChannelTest::ShutdownChannelOnIOThread, |
- base::Unretained(this))); |
- |
- // Okay to destroy |Channel| on not-the-I/O-thread. |
- EXPECT_TRUE(channel()->HasOneRef()); |
- *mutable_channel() = nullptr; |
-} |
- |
-// ChannelTest.InitFails ------------------------------------------------------- |
- |
-class MockRawChannelOnInitFails : public RawChannel { |
- public: |
- MockRawChannelOnInitFails() : on_init_called_(false) {} |
- virtual ~MockRawChannelOnInitFails() {} |
- |
- // |RawChannel| public methods: |
- virtual size_t GetSerializedPlatformHandleSize() const override { return 0; } |
- |
- private: |
- // |RawChannel| protected methods: |
- virtual IOResult Read(size_t*) override { |
- CHECK(false); |
- return IO_FAILED_UNKNOWN; |
- } |
- virtual IOResult ScheduleRead() override { |
- CHECK(false); |
- return IO_FAILED_UNKNOWN; |
- } |
- virtual embedder::ScopedPlatformHandleVectorPtr GetReadPlatformHandles( |
- size_t, |
- const void*) override { |
- CHECK(false); |
- return embedder::ScopedPlatformHandleVectorPtr(); |
- } |
- virtual IOResult WriteNoLock(size_t*, size_t*) override { |
- CHECK(false); |
- return IO_FAILED_UNKNOWN; |
- } |
- virtual IOResult ScheduleWriteNoLock() override { |
- CHECK(false); |
- return IO_FAILED_UNKNOWN; |
- } |
- virtual bool OnInit() override { |
- EXPECT_FALSE(on_init_called_); |
- on_init_called_ = true; |
- return false; |
- } |
- virtual void OnShutdownNoLock(scoped_ptr<ReadBuffer>, |
- scoped_ptr<WriteBuffer>) override { |
- CHECK(false); |
- } |
- |
- bool on_init_called_; |
- |
- DISALLOW_COPY_AND_ASSIGN(MockRawChannelOnInitFails); |
-}; |
- |
-TEST_F(ChannelTest, InitFails) { |
- io_thread()->PostTaskAndWait(FROM_HERE, |
- base::Bind(&ChannelTest::CreateChannelOnIOThread, |
- base::Unretained(this))); |
- ASSERT_TRUE(channel()); |
- |
- ASSERT_TRUE(raw_channel()); |
- mutable_raw_channel()->reset(new MockRawChannelOnInitFails()); |
- |
- io_thread()->PostTaskAndWait( |
- FROM_HERE, |
- base::Bind(&ChannelTest::InitChannelOnIOThread, base::Unretained(this))); |
- EXPECT_EQ(TRISTATE_FALSE, init_result()); |
- |
- // Should destroy |Channel| with no |Shutdown()| (on not-the-I/O-thread). |
- EXPECT_TRUE(channel()->HasOneRef()); |
- *mutable_channel() = nullptr; |
-} |
- |
-// ChannelTest.CloseBeforeRun -------------------------------------------------- |
- |
-TEST_F(ChannelTest, CloseBeforeRun) { |
- io_thread()->PostTaskAndWait(FROM_HERE, |
- base::Bind(&ChannelTest::CreateChannelOnIOThread, |
- base::Unretained(this))); |
- ASSERT_TRUE(channel()); |
- |
- io_thread()->PostTaskAndWait( |
- FROM_HERE, |
- base::Bind(&ChannelTest::InitChannelOnIOThread, base::Unretained(this))); |
- EXPECT_EQ(TRISTATE_TRUE, init_result()); |
- |
- scoped_refptr<ChannelEndpoint> channel_endpoint; |
- scoped_refptr<MessagePipe> mp( |
- MessagePipe::CreateLocalProxy(&channel_endpoint)); |
- |
- MessageInTransit::EndpointId local_id = |
- channel()->AttachEndpoint(channel_endpoint); |
- EXPECT_EQ(Channel::kBootstrapEndpointId, local_id); |
- |
- mp->Close(0); |
- |
- // TODO(vtl): Currently, the |Close()| above won't detach (since it thinks |
- // we're still expecting a "run" message from the other side), so the |
- // |RunMessagePipeEndpoint()| below will return true. We need to refactor |
- // |AttachEndpoint()| to indicate whether |Run...()| will necessarily be |
- // called or not. (Then, in the case that it may not be called, this will |
- // return false.) |
- EXPECT_TRUE(channel()->RunMessagePipeEndpoint(local_id, |
- Channel::kBootstrapEndpointId)); |
- |
- io_thread()->PostTaskAndWait( |
- FROM_HERE, |
- base::Bind(&ChannelTest::ShutdownChannelOnIOThread, |
- base::Unretained(this))); |
- |
- EXPECT_TRUE(channel()->HasOneRef()); |
-} |
- |
-// ChannelTest.ShutdownAfterAttachAndRun --------------------------------------- |
- |
-TEST_F(ChannelTest, ShutdownAfterAttach) { |
- io_thread()->PostTaskAndWait(FROM_HERE, |
- base::Bind(&ChannelTest::CreateChannelOnIOThread, |
- base::Unretained(this))); |
- ASSERT_TRUE(channel()); |
- |
- io_thread()->PostTaskAndWait( |
- FROM_HERE, |
- base::Bind(&ChannelTest::InitChannelOnIOThread, base::Unretained(this))); |
- EXPECT_EQ(TRISTATE_TRUE, init_result()); |
- |
- scoped_refptr<ChannelEndpoint> channel_endpoint; |
- scoped_refptr<MessagePipe> mp( |
- MessagePipe::CreateLocalProxy(&channel_endpoint)); |
- |
- MessageInTransit::EndpointId local_id = |
- channel()->AttachEndpoint(channel_endpoint); |
- EXPECT_EQ(Channel::kBootstrapEndpointId, local_id); |
- |
- // TODO(vtl): Currently, we always "expect" a |RunMessagePipeEndpoint()| after |
- // an |AttachEndpoint()| (which is actually incorrect). We need to refactor |
- // |AttachEndpoint()| to indicate whether |Run...()| will necessarily be |
- // called or not. (Then, in the case that it may not be called, we should test |
- // a |Shutdown()| without the |Run...()|.) |
- EXPECT_TRUE(channel()->RunMessagePipeEndpoint(local_id, |
- Channel::kBootstrapEndpointId)); |
- |
- Waiter waiter; |
- waiter.Init(); |
- ASSERT_EQ( |
- MOJO_RESULT_OK, |
- mp->AddWaiter(0, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 123, nullptr)); |
- |
- // Don't wait for the shutdown to run ... |
- io_thread()->PostTask(FROM_HERE, |
- base::Bind(&ChannelTest::ShutdownChannelOnIOThread, |
- base::Unretained(this))); |
- |
- // ... since this |Wait()| should fail once the channel is shut down. |
- EXPECT_EQ(MOJO_RESULT_FAILED_PRECONDITION, |
- waiter.Wait(MOJO_DEADLINE_INDEFINITE, nullptr)); |
- HandleSignalsState hss; |
- mp->RemoveWaiter(0, &waiter, &hss); |
- EXPECT_EQ(0u, hss.satisfied_signals); |
- EXPECT_EQ(0u, hss.satisfiable_signals); |
- |
- mp->Close(0); |
- |
- EXPECT_TRUE(channel()->HasOneRef()); |
-} |
- |
-// ChannelTest.WaitAfterAttachRunAndShutdown ----------------------------------- |
- |
-TEST_F(ChannelTest, WaitAfterAttachRunAndShutdown) { |
- io_thread()->PostTaskAndWait(FROM_HERE, |
- base::Bind(&ChannelTest::CreateChannelOnIOThread, |
- base::Unretained(this))); |
- ASSERT_TRUE(channel()); |
- |
- io_thread()->PostTaskAndWait( |
- FROM_HERE, |
- base::Bind(&ChannelTest::InitChannelOnIOThread, base::Unretained(this))); |
- EXPECT_EQ(TRISTATE_TRUE, init_result()); |
- |
- scoped_refptr<ChannelEndpoint> channel_endpoint; |
- scoped_refptr<MessagePipe> mp( |
- MessagePipe::CreateLocalProxy(&channel_endpoint)); |
- |
- MessageInTransit::EndpointId local_id = |
- channel()->AttachEndpoint(channel_endpoint); |
- EXPECT_EQ(Channel::kBootstrapEndpointId, local_id); |
- |
- EXPECT_TRUE(channel()->RunMessagePipeEndpoint(local_id, |
- Channel::kBootstrapEndpointId)); |
- |
- io_thread()->PostTaskAndWait( |
- FROM_HERE, |
- base::Bind(&ChannelTest::ShutdownChannelOnIOThread, |
- base::Unretained(this))); |
- |
- Waiter waiter; |
- waiter.Init(); |
- HandleSignalsState hss; |
- EXPECT_EQ(MOJO_RESULT_FAILED_PRECONDITION, |
- mp->AddWaiter(0, &waiter, MOJO_HANDLE_SIGNAL_READABLE, 123, &hss)); |
- EXPECT_EQ(0u, hss.satisfied_signals); |
- EXPECT_EQ(0u, hss.satisfiable_signals); |
- |
- mp->Close(0); |
- |
- EXPECT_TRUE(channel()->HasOneRef()); |
-} |
- |
-// TODO(vtl): More. ------------------------------------------------------------ |
- |
-} // namespace |
-} // namespace system |
-} // namespace mojo |