Refactor BlimpConnection to TCPConnection

blimp/net/test_common.cc

 // Copyright 2015 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/test_common.h"
 
 #include <string>
+#include <utility>
 
 #include "base/memory/ptr_util.h"
 #include "base/sys_byteorder.h"
 #include "blimp/common/proto/blimp_message.pb.h"
 #include "blimp/net/blimp_connection.h"
 #include "blimp/net/common.h"
 #include "blimp/net/message_port.h"
 #include "net/base/io_buffer.h"
 
 namespace blimp {
 
 MockStreamSocket::MockStreamSocket() {}
 
 MockStreamSocket::~MockStreamSocket() {}
 
 MockTransport::MockTransport() {}
 
+std::unique_ptr<BlimpConnection> MockTransport::MakeConnection() {
+  return std::move(connection_);
+}
+
+void MockTransport::SetMockConnection(
+    std::unique_ptr<MockBlimpConnection> connection) {
+  connection_ = std::move(connection);
+}
+
 MockTransport::~MockTransport() {}
 
-std::unique_ptr<MessagePort> MockTransport::TakeMessagePort() {
-  return base::WrapUnique(TakeMessagePortPtr());
-}
-
-const char* MockTransport::GetName() const {
-  return "mock";
-}
+const char* MockTransport::GetName() const { return "mock"; }
 
 MockConnectionHandler::MockConnectionHandler() {}
 
 MockConnectionHandler::~MockConnectionHandler() {}
 
 void MockConnectionHandler::HandleConnection(
     std::unique_ptr<BlimpConnection> connection) {
   HandleConnectionPtr(connection.get());
 }
 
@@ skipping 23 matching lines @@
   MockableProcessMessage(*message, callback);
 }
 
 std::string EncodeHeader(size_t size) {
   std::unique_ptr<char[]> serialized(new char[kPacketHeaderSizeBytes]);
   uint32_t net_size = base::HostToNet32(size);
   memcpy(serialized.get(), &net_size, sizeof(net_size));
   return std::string(serialized.get(), kPacketHeaderSizeBytes);
 }
 
-bool BufferStartsWith(net::GrowableIOBuffer* buf,
-                      size_t buf_size,
+bool BufferStartsWith(net::GrowableIOBuffer* buf, size_t buf_size,
                       const std::string& str) {
   return (buf_size >= str.size() &&
           str == std::string(buf->data(), str.size()));
 }
 
 }  // namespace blimp