net/tools/quic/test_tools/mock_quic_server_session_visitor.h

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef NET_TOOLS_QUIC_TEST_TOOLS_MOCK_QUIC_SERVER_SESSION_VISITOR_H_
   6 #define NET_TOOLS_QUIC_TEST_TOOLS_MOCK_QUIC_SERVER_SESSION_VISITOR_H_
   7
   8 #include "net/tools/quic/quic_server_session.h"
   9 #include "testing/gmock/include/gmock/gmock.h"
  10
  11 namespace net {
  12 namespace tools {
  13 namespace test {
  14
  15 class MockQuicServerSessionVisitor : public QuicServerSessionVisitor {
  16  public:
  17   MockQuicServerSessionVisitor();
  18   virtual ~MockQuicServerSessionVisitor();
  19   MOCK_METHOD2(OnConnectionClosed,
  20                void(QuicConnectionId connection_id, QuicErrorCode error));
  21   MOCK_METHOD1(OnWriteBlocked,
  22                void(QuicBlockedWriterInterface* blocked_writer));
  23   MOCK_METHOD1(OnConnectionAddedToTimeWaitList,
  24                void(QuicConnectionId connection_id));
  25   MOCK_METHOD1(OnConnectionRemovedFromTimeWaitList,
  26                void(QuicConnectionId connection_id));
  27
  28  private:
  29   DISALLOW_COPY_AND_ASSIGN(MockQuicServerSessionVisitor);
  30 };
  31
  32 }  // namespace test
  33 }  // namespace tools
  34 }  // namespace net
  35
  36 #endif  // NET_TOOLS_QUIC_TEST_TOOLS_MOCK_QUIC_SERVER_SESSION_VISITOR_H_