[Blimp Net] Add EngineAuthHandler.

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 <string>
 
 #include "base/memory/scoped_ptr.h"
 #include "blimp/common/proto/blimp_message.pb.h"
+#include "blimp/net/blimp_connection.h"
 #include "blimp/net/blimp_message_processor.h"
 #include "blimp/net/blimp_transport.h"
 #include "blimp/net/connection_error_observer.h"
 #include "blimp/net/connection_handler.h"
 #include "blimp/net/packet_reader.h"
 #include "blimp/net/packet_writer.h"
 #include "net/socket/stream_socket.h"
 #include "testing/gmock/include/gmock/gmock.h"
 
 namespace net {
@@ skipping 136 matching lines @@
 class MockPacketWriter : public PacketWriter {
  public:
   MockPacketWriter();
   ~MockPacketWriter() override;
 
   MOCK_METHOD2(WritePacket,
                void(scoped_refptr<net::DrainableIOBuffer>,
                     const net::CompletionCallback&));
 };
 
+class MockBlimpConnection : public BlimpConnection {
+ public:
+  MockBlimpConnection();
+  ~MockBlimpConnection() override;
+
+  MOCK_METHOD1(SetConnectionErrorObserver,
+               void(ConnectionErrorObserver* observer));
+
+  MOCK_METHOD1(SetIncomingMessageProcessor,
+               void(BlimpMessageProcessor* processor));
+
+  MOCK_CONST_METHOD0(GetOutgoingMessageProcessor, BlimpMessageProcessor*());
+};
+
 class MockConnectionErrorObserver : public ConnectionErrorObserver {
  public:
   MockConnectionErrorObserver();
   ~MockConnectionErrorObserver() override;
 
   MOCK_METHOD1(OnConnectionError, void(int error));
 };
 
 class MockBlimpMessageProcessor : public BlimpMessageProcessor {
  public:
   MockBlimpMessageProcessor();
 
   ~MockBlimpMessageProcessor() override;
 
   // Adapts calls from ProcessMessage to MockableProcessMessage by
   // unboxing the |message| scoped_ptr for GMock compatibility.
   void ProcessMessage(scoped_ptr<BlimpMessage> message,
                       const net::CompletionCallback& callback) override;
 
   MOCK_METHOD2(MockableProcessMessage,
                void(const BlimpMessage& message,
                     const net::CompletionCallback& callback));
 };
 
 }  // namespace blimp
 
 #endif  // BLIMP_NET_TEST_COMMON_H_