Refactor BlimpConnection to TCPConnection

blimp/net/test_common.cc

Index: blimp/net/test_common.cc
diff --git a/blimp/net/test_common.cc b/blimp/net/test_common.cc
index f72ed236b002ea7aa902abe657baa405bd63aee1..da3d8c6779382064bc8a8f5bd534fa1e00e13fff 100644
--- a/blimp/net/test_common.cc
+++ b/blimp/net/test_common.cc
@@ -5,6 +5,7 @@
 #include "blimp/net/test_common.h"
 
 #include <string>
+#include <utility>
 
 #include "base/memory/ptr_util.h"
 #include "base/sys_byteorder.h"
@@ -22,12 +23,17 @@ MockStreamSocket::~MockStreamSocket() {}
 
 MockTransport::MockTransport() {}
 
-MockTransport::~MockTransport() {}
+std::unique_ptr<BlimpConnection> MockTransport::MakeConnection() {
+  return std::move(connection_);
+}
 
-std::unique_ptr<MessagePort> MockTransport::TakeMessagePort() {
-  return base::WrapUnique(TakeMessagePortPtr());
+void MockTransport::SetMockConnection(
+    std::unique_ptr<MockBlimpConnection> connection) {
+  connection_ = std::move(connection);
 }
 
+MockTransport::~MockTransport() {}
+
 const char* MockTransport::GetName() const {
   return "mock";
 }