Index: blimp/net/thread_pipe_manager_unittest.cc |
diff --git a/blimp/net/thread_pipe_manager_unittest.cc b/blimp/net/thread_pipe_manager_unittest.cc |
deleted file mode 100644 |
index 2e0622058180d097b82534894879d6bf6cee2460..0000000000000000000000000000000000000000 |
--- a/blimp/net/thread_pipe_manager_unittest.cc |
+++ /dev/null |
@@ -1,179 +0,0 @@ |
-// Copyright 2016 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 "blimp/net/thread_pipe_manager.h" |
- |
-#include "base/location.h" |
-#include "base/memory/ptr_util.h" |
-#include "base/memory/ref_counted.h" |
-#include "base/message_loop/message_loop.h" |
-#include "base/single_thread_task_runner.h" |
-#include "base/threading/thread.h" |
-#include "base/threading/thread_task_runner_handle.h" |
-#include "blimp/common/create_blimp_message.h" |
-#include "blimp/net/blimp_message_thread_pipe.h" |
-#include "blimp/net/browser_connection_handler.h" |
-#include "blimp/net/test_common.h" |
-#include "net/base/net_errors.h" |
-#include "net/base/test_completion_callback.h" |
-#include "testing/gmock/include/gmock/gmock.h" |
-#include "testing/gtest/include/gtest/gtest.h" |
- |
-using testing::_; |
-using testing::SaveArg; |
- |
-namespace blimp { |
-namespace { |
- |
-// A feature that registers itself with ThreadPipeManager. |
-class FakeFeature { |
- public: |
- FakeFeature(BlimpMessage::FeatureCase feature_case, |
- ThreadPipeManager* pipe_manager_) { |
- outgoing_message_processor_ = pipe_manager_->RegisterFeature( |
- feature_case, &incoming_message_processor_); |
- } |
- |
- ~FakeFeature() {} |
- |
- BlimpMessageProcessor* outgoing_message_processor() { |
- return outgoing_message_processor_.get(); |
- } |
- |
- MockBlimpMessageProcessor* incoming_message_processor() { |
- return &incoming_message_processor_; |
- } |
- |
- private: |
- testing::StrictMock<MockBlimpMessageProcessor> incoming_message_processor_; |
- std::unique_ptr<BlimpMessageProcessor> outgoing_message_processor_; |
-}; |
- |
-// A feature peer on |thread_| that forwards incoming messages to |
-// |message_processor|. |
-class FakeFeaturePeer : public BlimpMessageProcessor { |
- public: |
- FakeFeaturePeer(BlimpMessage::FeatureCase feature_case, |
- BlimpMessageProcessor* message_processor, |
- const scoped_refptr<base::SequencedTaskRunner>& task_runner) |
- : feature_case_(feature_case), |
- message_processor_(message_processor), |
- task_runner_(task_runner) {} |
- |
- ~FakeFeaturePeer() override {} |
- |
- private: |
- void ForwardMessage(std::unique_ptr<BlimpMessage> message) { |
- DCHECK(task_runner_->RunsTasksOnCurrentThread()); |
- message_processor_->ProcessMessage(std::move(message), |
- net::CompletionCallback()); |
- } |
- |
- // BlimpMessageProcessor implementation. |
- void ProcessMessage(std::unique_ptr<BlimpMessage> message, |
- const net::CompletionCallback& callback) override { |
- DCHECK(task_runner_->RunsTasksOnCurrentThread()); |
- ASSERT_EQ(feature_case_, message->feature_case()); |
- base::ThreadTaskRunnerHandle::Get()->PostTask( |
- FROM_HERE, base::Bind(&FakeFeaturePeer::ForwardMessage, |
- base::Unretained(this), base::Passed(&message))); |
- if (!callback.is_null()) |
- callback.Run(net::OK); |
- } |
- |
- BlimpMessage::FeatureCase feature_case_; |
- BlimpMessageProcessor* message_processor_ = nullptr; |
- scoped_refptr<base::SequencedTaskRunner> task_runner_; |
-}; |
- |
-// A browser connection handler that returns FakeFeaturePeer to allow it |
-// forwarding message back so that FakeFeature can check message it receives |
-// with one it just sent. |
-class FakeBrowserConnectionHandler : public BrowserConnectionHandler { |
- public: |
- FakeBrowserConnectionHandler( |
- const scoped_refptr<base::SequencedTaskRunner>& task_runner) |
- : task_runner_(task_runner) {} |
- std::unique_ptr<BlimpMessageProcessor> RegisterFeature( |
- BlimpMessage::FeatureCase feature_case, |
- BlimpMessageProcessor* incoming_processor) override { |
- DCHECK(task_runner_->RunsTasksOnCurrentThread()); |
- return base::MakeUnique<FakeFeaturePeer>(feature_case, incoming_processor, |
- task_runner_); |
- } |
- |
- private: |
- scoped_refptr<base::SequencedTaskRunner> task_runner_; |
-}; |
- |
-} // namespace |
- |
-class ThreadPipeManagerTest : public testing::Test { |
- public: |
- ThreadPipeManagerTest() : thread_("IoThread") {} |
- |
- ~ThreadPipeManagerTest() override {} |
- |
- void SetUp() override { |
- ASSERT_TRUE(thread_.Start()); |
- connection_handler_ = |
- base::MakeUnique<FakeBrowserConnectionHandler>(thread_.task_runner()); |
- pipe_manager_ = base::MakeUnique<ThreadPipeManager>( |
- thread_.task_runner(), connection_handler_.get()); |
- |
- input_feature_.reset( |
- new FakeFeature(BlimpMessage::kInput, pipe_manager_.get())); |
- tab_control_feature_.reset( |
- new FakeFeature(BlimpMessage::kTabControl, pipe_manager_.get())); |
- } |
- |
- void TearDown() override { SynchronizeWithThread(); } |
- |
- // Synchronize with |thread_| to ensure that any pending work is done. |
- void SynchronizeWithThread() { |
- net::TestCompletionCallback cb; |
- thread_.task_runner()->PostTaskAndReply(FROM_HERE, |
- base::Bind(&base::DoNothing), |
- base::Bind(cb.callback(), net::OK)); |
- ASSERT_EQ(net::OK, cb.WaitForResult()); |
- } |
- |
- protected: |
- base::MessageLoop message_loop_; |
- std::unique_ptr<BrowserConnectionHandler> connection_handler_; |
- std::unique_ptr<ThreadPipeManager> pipe_manager_; |
- base::Thread thread_; |
- |
- std::unique_ptr<FakeFeature> input_feature_; |
- std::unique_ptr<FakeFeature> tab_control_feature_; |
-}; |
- |
-// Features send out message and receive the same message due to |
-// |FakeFeaturePeer| loops the message back on |thread_|. |
-TEST_F(ThreadPipeManagerTest, MessageSentIsReceived) { |
- InputMessage* input = nullptr; |
- std::unique_ptr<BlimpMessage> input_message = CreateBlimpMessage(&input); |
- TabControlMessage* tab_control = nullptr; |
- std::unique_ptr<BlimpMessage> tab_control_message = |
- CreateBlimpMessage(&tab_control); |
- |
- EXPECT_CALL(*(input_feature_->incoming_message_processor()), |
- MockableProcessMessage(EqualsProto(*input_message), _)) |
- .RetiresOnSaturation(); |
- EXPECT_CALL(*(tab_control_feature_->incoming_message_processor()), |
- MockableProcessMessage(EqualsProto(*tab_control_message), _)) |
- .RetiresOnSaturation(); |
- |
- net::TestCompletionCallback cb1; |
- input_feature_->outgoing_message_processor()->ProcessMessage( |
- std::move(input_message), cb1.callback()); |
- net::TestCompletionCallback cb2; |
- tab_control_feature_->outgoing_message_processor()->ProcessMessage( |
- std::move(tab_control_message), cb2.callback()); |
- |
- EXPECT_EQ(net::OK, cb1.WaitForResult()); |
- EXPECT_EQ(net::OK, cb2.WaitForResult()); |
-} |
- |
-} // namespace blimp |