| Index: remoting/protocol/fake_message_pipe.cc
|
| diff --git a/remoting/protocol/fake_message_pipe.cc b/remoting/protocol/fake_message_pipe.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..382978617d498117e51246d887aeefb40906e662
|
| --- /dev/null
|
| +++ b/remoting/protocol/fake_message_pipe.cc
|
| @@ -0,0 +1,124 @@
|
| +// Copyright 2017 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 "remoting/protocol/fake_message_pipe.h"
|
| +
|
| +#include "base/bind.h"
|
| +#include "base/location.h"
|
| +#include "base/memory/ptr_util.h"
|
| +#include "base/threading/thread_task_runner_handle.h"
|
| +#include "remoting/base/compound_buffer.h"
|
| +#include "remoting/protocol/fake_message_pipe_wrapper.h"
|
| +#include "testing/gtest/include/gtest/gtest.h"
|
| +
|
| +namespace remoting {
|
| +namespace protocol {
|
| +
|
| +FakeMessagePipe::FakeMessagePipe(bool asynchronous)
|
| + : asynchronous_(asynchronous) {}
|
| +
|
| +FakeMessagePipe::~FakeMessagePipe() = default;
|
| +
|
| +std::unique_ptr<FakeMessagePipeWrapper> FakeMessagePipe::Wrap() {
|
| + return base::MakeUnique<FakeMessagePipeWrapper>(this);
|
| +}
|
| +
|
| +void FakeMessagePipe::Start(EventHandler* event_handler) {
|
| + ASSERT_TRUE(event_handler_ == nullptr);
|
| + ASSERT_TRUE(event_handler != nullptr);
|
| + event_handler_ = event_handler;
|
| +}
|
| +
|
| +void FakeMessagePipe::Send(google::protobuf::MessageLite* message,
|
| + const base::Closure& done) {
|
| + if (asynchronous_) {
|
| + base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE,
|
| + base::Bind([](FakeMessagePipe* me,
|
| + google::protobuf::MessageLite* message,
|
| + const base::Closure& done) {
|
| + me->SendImpl(message, done);
|
| + },
|
| + base::Unretained(this),
|
| + base::Unretained(message),
|
| + done));
|
| + return;
|
| + }
|
| + SendImpl(message, done);
|
| +}
|
| +
|
| +void FakeMessagePipe::Receive(std::unique_ptr<CompoundBuffer> message) {
|
| + if (asynchronous_) {
|
| + base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE,
|
| + base::Bind([](FakeMessagePipe* me,
|
| + std::unique_ptr<CompoundBuffer> message) {
|
| + me->ReceiveImpl(std::move(message));
|
| + },
|
| + base::Unretained(this),
|
| + base::Passed(std::move(message))));
|
| + return;
|
| + }
|
| +
|
| + ReceiveImpl(std::move(message));
|
| +}
|
| +
|
| +void FakeMessagePipe::OpenPipe() {
|
| + if (asynchronous_) {
|
| + base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, base::Bind(
|
| + [](FakeMessagePipe* me) {
|
| + me->OpenPipeImpl();
|
| + },
|
| + base::Unretained(this)));
|
| + return;
|
| + }
|
| +
|
| + OpenPipeImpl();
|
| +}
|
| +
|
| +void FakeMessagePipe::ClosePipe() {
|
| + if (asynchronous_) {
|
| + base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, base::Bind(
|
| + [](FakeMessagePipe* me) {
|
| + me->ClosePipeImpl();
|
| + },
|
| + base::Unretained(this)));
|
| + return;
|
| + }
|
| +
|
| + ClosePipeImpl();
|
| +}
|
| +
|
| +void FakeMessagePipe::SendImpl(
|
| + google::protobuf::MessageLite* message,
|
| + const base::Closure& done) {
|
| + ASSERT_TRUE(pipe_opened_);
|
| + if (done) {
|
| + done.Run();
|
| + }
|
| +}
|
| +
|
| +void FakeMessagePipe::ReceiveImpl(std::unique_ptr<CompoundBuffer> message) {
|
| + ASSERT_TRUE(pipe_opened_);
|
| + ASSERT_TRUE(event_handler_ != nullptr);
|
| + if (message) {
|
| + message->Lock();
|
| + }
|
| + event_handler_->OnMessageReceived(std::move(message));
|
| +}
|
| +
|
| +void FakeMessagePipe::OpenPipeImpl() {
|
| + ASSERT_FALSE(pipe_opened_);
|
| + ASSERT_TRUE(event_handler_ != nullptr);
|
| + pipe_opened_ = true;
|
| + event_handler_->OnMessagePipeOpen();
|
| +}
|
| +
|
| +void FakeMessagePipe::ClosePipeImpl() {
|
| + ASSERT_TRUE(pipe_opened_);
|
| + ASSERT_TRUE(event_handler_ != nullptr);
|
| + pipe_opened_ = false;
|
| + event_handler_->OnMessagePipeClosed();
|
| +}
|
| +
|
| +} // namespace protocol
|
| +} // namespace remoting
|
|
|