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";
 }
 
 MockConnectionHandler::MockConnectionHandler() {}
 
 MockConnectionHandler::~MockConnectionHandler() {}
 
 void MockConnectionHandler::HandleConnection(
     std::unique_ptr<BlimpConnection> connection) {
@@ skipping 34 matching lines @@
 }
 
 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