Index: net/quic/test_tools/quic_test_utils.h |
=================================================================== |
--- net/quic/test_tools/quic_test_utils.h (revision 165858) |
+++ net/quic/test_tools/quic_test_utils.h (working copy) |
@@ -10,8 +10,6 @@ |
#include "net/quic/congestion_control/quic_send_scheduler.h" |
#include "net/quic/quic_connection.h" |
#include "net/quic/quic_framer.h" |
-#include "net/quic/quic_session.h" |
-#include "net/quic/test_tools/mock_clock.h" |
#include "testing/gmock/include/gmock/gmock.h" |
namespace net { |
@@ -68,6 +66,7 @@ |
virtual void OnPacketComplete() OVERRIDE {} |
}; |
+ |
class FramerVisitorCapturingAcks : public NoOpFramerVisitor { |
public: |
// NoOpFramerVisitor |
@@ -84,7 +83,7 @@ |
class MockConnectionVisitor : public QuicConnectionVisitorInterface { |
public: |
MockConnectionVisitor(); |
- virtual ~MockConnectionVisitor(); |
+ ~MockConnectionVisitor(); |
MOCK_METHOD4(OnPacket, bool(const IPEndPoint& self_address, |
const IPEndPoint& peer_address, |
const QuicPacketHeader& header, |
@@ -104,78 +103,6 @@ |
MOCK_METHOD3(SentPacket, void(QuicPacketSequenceNumber, size_t, bool)); |
}; |
-class MockHelper : public QuicConnectionHelperInterface { |
- public: |
- MockHelper(); |
- virtual ~MockHelper(); |
- |
- MOCK_METHOD1(SetConnection, void(QuicConnection* connection)); |
- QuicClock* GetClock(); |
- MOCK_METHOD4(WritePacketToWire, int(QuicPacketSequenceNumber number, |
- const QuicEncryptedPacket& packet, |
- bool resend, |
- int* error)); |
- MOCK_METHOD2(SetResendAlarm, void(QuicPacketSequenceNumber sequence_number, |
- uint64 delay_in_us)); |
- MOCK_METHOD1(SetSendAlarm, void(uint64 delay_in_us)); |
- MOCK_METHOD1(SetTimeoutAlarm, void(uint64 delay_in_us)); |
- MOCK_METHOD0(IsSendAlarmSet, bool()); |
- MOCK_METHOD0(UnregisterSendAlarmIfRegistered, void()); |
- private: |
- MockClock clock_; |
-}; |
- |
-class MockConnection : public QuicConnection { |
- public: |
- MockConnection(QuicGuid guid, IPEndPoint address); |
- virtual ~MockConnection(); |
- |
- MOCK_METHOD3(ProcessUdpPacket, void(const IPEndPoint& self_address, |
- const IPEndPoint& peer_address, |
- const QuicEncryptedPacket& packet)); |
- MOCK_METHOD1(SendConnectionClose, void(QuicErrorCode error)); |
- |
- MOCK_METHOD3(SendRstStream, void(QuicStreamId id, |
- QuicErrorCode error, |
- QuicStreamOffset offset)); |
- |
- MOCK_METHOD0(OnCanWrite, bool()); |
-}; |
- |
-class PacketSavingConnection : public MockConnection { |
- public: |
- PacketSavingConnection(QuicGuid guid, IPEndPoint address); |
- virtual ~PacketSavingConnection(); |
- |
- virtual bool SendPacket(QuicPacketSequenceNumber number, |
- QuicPacket* packet, |
- bool resend, |
- bool force) OVERRIDE; |
- |
- std::vector<QuicPacket*> packets_; |
-}; |
- |
-class MockSession : public QuicSession { |
- public: |
- MockSession(QuicConnection* connection, bool is_server); |
- ~MockSession(); |
- |
- MOCK_METHOD4(OnPacket, bool(const IPEndPoint& seld_address, |
- const IPEndPoint& peer_address, |
- const QuicPacketHeader& header, |
- const std::vector<QuicStreamFrame>& frame)); |
- MOCK_METHOD2(ConnectionClose, void(QuicErrorCode error, bool from_peer)); |
- MOCK_METHOD1(CreateIncomingReliableStream, |
- ReliableQuicStream*(QuicStreamId id)); |
- MOCK_METHOD0(GetCryptoStream, QuicCryptoStream*()); |
- MOCK_METHOD0(CreateOutgoingReliableStream, ReliableQuicStream*()); |
- MOCK_METHOD3(WriteData, |
- void(QuicStreamId id, base::StringPiece data, bool fin)); |
- MOCK_METHOD4(WriteData, int(QuicStreamId id, base::StringPiece data, |
- QuicStreamOffset offset, bool fin)); |
- MOCK_METHOD0(IsHandshakeComplete, bool()); |
-}; |
- |
} // namespace test |
} // namespace net |