| Index: remoting/protocol/connection_tester.cc
|
| diff --git a/remoting/protocol/connection_tester.cc b/remoting/protocol/connection_tester.cc
|
| index 111ebc104db8798d1cdd181fd27f7479f238fdce..2ef402885713456e35d39fcbb86ce105110f0933 100644
|
| --- a/remoting/protocol/connection_tester.cc
|
| +++ b/remoting/protocol/connection_tester.cc
|
| @@ -8,6 +8,9 @@
|
| #include "base/message_loop/message_loop.h"
|
| #include "net/base/io_buffer.h"
|
| #include "net/base/net_errors.h"
|
| +#include "remoting/proto/video.pb.h"
|
| +#include "remoting/protocol/message_pipe.h"
|
| +#include "remoting/protocol/message_serialization.h"
|
| #include "remoting/protocol/p2p_datagram_socket.h"
|
| #include "remoting/protocol/p2p_stream_socket.h"
|
| #include "testing/gtest/include/gtest/gtest.h"
|
| @@ -257,5 +260,46 @@ void DatagramConnectionTester::HandleReadResult(int result) {
|
| }
|
| }
|
|
|
| +MessagePipeConnectionTester::MessagePipeConnectionTester(
|
| + MessagePipe* client_pipe,
|
| + MessagePipe* host_pipe,
|
| + int message_size,
|
| + int message_count)
|
| + : host_pipe_(host_pipe),
|
| + client_pipe_(client_pipe),
|
| + message_size_(message_size),
|
| + message_count_(message_count) {}
|
| +MessagePipeConnectionTester::~MessagePipeConnectionTester() {}
|
| +
|
| +void MessagePipeConnectionTester::RunAndCheckResults() {
|
| + host_pipe_->StartReceiving(base::Bind(
|
| + &MessagePipeConnectionTester::OnMessageReceived, base::Unretained(this)));
|
| +
|
| + for (int i = 0; i < message_count_; ++i) {
|
| + scoped_ptr<VideoPacket> message(new VideoPacket());
|
| + message->mutable_data()->resize(message_size_);
|
| + for (int p = 0; p < message_size_; ++p) {
|
| + message->mutable_data()[0] = static_cast<char>(i + p);
|
| + }
|
| + client_pipe_->Send(message.get(), base::Closure());
|
| + sent_messages_.push_back(std::move(message));
|
| + }
|
| +
|
| + run_loop_.Run();
|
| +
|
| + ASSERT_EQ(sent_messages_.size(), received_messages_.size());
|
| + for (size_t i = 0; i < sent_messages_.size(); ++i) {
|
| + EXPECT_TRUE(sent_messages_[i]->data() == received_messages_[i]->data());
|
| + }
|
| +}
|
| +
|
| +void MessagePipeConnectionTester::OnMessageReceived(
|
| + scoped_ptr<CompoundBuffer> message) {
|
| + received_messages_.push_back(ParseMessage<VideoPacket>(message.get()));
|
| + if (received_messages_.size() >= sent_messages_.size()) {
|
| + run_loop_.Quit();
|
| + }
|
| +}
|
| +
|
| } // namespace protocol
|
| } // namespace remoting
|
|
|