Refactor BlimpConnection to TCPConnection

blimp/net/test_common.h

 // 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.
 
 #ifndef BLIMP_NET_TEST_COMMON_H_
 #define BLIMP_NET_TEST_COMMON_H_
 
 #include <stddef.h>
 #include <stdint.h>
 
@@ skipping 117 matching lines @@
   MOCK_CONST_METHOD0(GetConnectTimeMicros, base::TimeDelta());
   MOCK_CONST_METHOD0(WasNpnNegotiated, bool());
   MOCK_CONST_METHOD0(GetNegotiatedProtocol, net::NextProto());
   MOCK_METHOD1(GetSSLInfo, bool(net::SSLInfo*));
   MOCK_CONST_METHOD1(GetConnectionAttempts, void(net::ConnectionAttempts*));
   MOCK_METHOD0(ClearConnectionAttempts, void());
   MOCK_METHOD1(AddConnectionAttempts, void(const net::ConnectionAttempts&));
   MOCK_CONST_METHOD0(GetTotalReceivedBytes, int64_t());
 };
 
+class MockBlimpConnection;
+
 class MockTransport : public BlimpTransport {
  public:
   MockTransport();
   ~MockTransport() override;
 
   MOCK_METHOD1(Connect, void(const net::CompletionCallback& callback));
   MOCK_METHOD0(TakeMessagePortPtr, MessagePort*());
+  std::unique_ptr<BlimpConnection> MakeConnection() override;
 
-  std::unique_ptr<MessagePort> TakeMessagePort() override;
   const char* GetName() const override;
+
+  void SetMockConnection(std::unique_ptr<MockBlimpConnection> connection);
+
+ private:
+  std::unique_ptr<MockBlimpConnection> connection_;
 };
 
 class MockConnectionHandler : public ConnectionHandler {
  public:
   MockConnectionHandler();
   ~MockConnectionHandler() override;
 
   MOCK_METHOD1(HandleConnectionPtr, void(BlimpConnection* connection));
   void HandleConnection(std::unique_ptr<BlimpConnection> connection) override;
 };
@@ skipping 56 matching lines @@
                       const net::CompletionCallback& callback) override;
 
   MOCK_METHOD2(MockableProcessMessage,
                void(const BlimpMessage& message,
                     const net::CompletionCallback& callback));
 };
 
 }  // namespace blimp
 
 #endif  // BLIMP_NET_TEST_COMMON_H_