Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(2751)

Unified Diff: blimp/net/thread_pipe_manager_unittest.cc

Issue 2632803002: Remove all blimp network code. (Closed)
Patch Set: merge from origin/master for good measure Created 3 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « blimp/net/thread_pipe_manager.cc ('k') | chrome/browser/extensions/api/socket/mock_tcp_client_socket.h » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
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
« no previous file with comments | « blimp/net/thread_pipe_manager.cc ('k') | chrome/browser/extensions/api/socket/mock_tcp_client_socket.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698