| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 // | 4 // |
| 5 // Common utilities for Quic tests | 5 // Common utilities for Quic tests |
| 6 | 6 |
| 7 #ifndef NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ | 7 #ifndef NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ |
| 8 #define NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ | 8 #define NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ |
| 9 | 9 |
| 10 #include "net/quic/congestion_control/quic_send_scheduler.h" | 10 #include "net/quic/congestion_control/quic_send_scheduler.h" |
| (...skipping 14 matching lines...) Expand all Loading... |
| 25 const char* expected, | 25 const char* expected, |
| 26 const int expected_len); | 26 const int expected_len); |
| 27 | 27 |
| 28 void CompareQuicDataWithHexError(const std::string& description, | 28 void CompareQuicDataWithHexError(const std::string& description, |
| 29 QuicData* actual, | 29 QuicData* actual, |
| 30 QuicData* expected); | 30 QuicData* expected); |
| 31 | 31 |
| 32 // Constructs a basic crypto handshake message | 32 // Constructs a basic crypto handshake message |
| 33 QuicPacket* ConstructHandshakePacket(QuicGuid guid, CryptoTag tag); | 33 QuicPacket* ConstructHandshakePacket(QuicGuid guid, CryptoTag tag); |
| 34 | 34 |
| 35 // Constructs a ClientHello crypto handshake message |
| 36 QuicPacket* ConstructClientHelloPacket(QuicGuid guid, |
| 37 const QuicClock* clock, |
| 38 QuicRandom* random_generator); |
| 39 |
| 35 class MockFramerVisitor : public QuicFramerVisitorInterface { | 40 class MockFramerVisitor : public QuicFramerVisitorInterface { |
| 36 public: | 41 public: |
| 37 MockFramerVisitor(); | 42 MockFramerVisitor(); |
| 38 ~MockFramerVisitor(); | 43 ~MockFramerVisitor(); |
| 39 | 44 |
| 40 MOCK_METHOD1(OnError, void(QuicFramer* framer)); | 45 MOCK_METHOD1(OnError, void(QuicFramer* framer)); |
| 41 MOCK_METHOD2(OnPacket, void(const IPEndPoint& self_address, | 46 MOCK_METHOD2(OnPacket, void(const IPEndPoint& self_address, |
| 42 const IPEndPoint& peer_address)); | 47 const IPEndPoint& peer_address)); |
| 43 MOCK_METHOD0(OnRevivedPacket, void()); | 48 MOCK_METHOD0(OnRevivedPacket, void()); |
| 44 // The constructor set this up to return true by default. | 49 // The constructor set this up to return true by default. |
| (...skipping 96 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 141 MOCK_METHOD0(IsSendAlarmSet, bool()); | 146 MOCK_METHOD0(IsSendAlarmSet, bool()); |
| 142 MOCK_METHOD0(UnregisterSendAlarmIfRegistered, void()); | 147 MOCK_METHOD0(UnregisterSendAlarmIfRegistered, void()); |
| 143 private: | 148 private: |
| 144 const MockClock clock_; | 149 const MockClock clock_; |
| 145 MockRandom random_generator_; | 150 MockRandom random_generator_; |
| 146 }; | 151 }; |
| 147 | 152 |
| 148 class MockConnection : public QuicConnection { | 153 class MockConnection : public QuicConnection { |
| 149 public: | 154 public: |
| 150 MockConnection(QuicGuid guid, IPEndPoint address); | 155 MockConnection(QuicGuid guid, IPEndPoint address); |
| 156 MockConnection(QuicGuid guid, |
| 157 IPEndPoint address, |
| 158 QuicConnectionHelperInterface* helper); |
| 151 virtual ~MockConnection(); | 159 virtual ~MockConnection(); |
| 152 | 160 |
| 153 MOCK_METHOD3(ProcessUdpPacket, void(const IPEndPoint& self_address, | 161 MOCK_METHOD3(ProcessUdpPacket, void(const IPEndPoint& self_address, |
| 154 const IPEndPoint& peer_address, | 162 const IPEndPoint& peer_address, |
| 155 const QuicEncryptedPacket& packet)); | 163 const QuicEncryptedPacket& packet)); |
| 156 MOCK_METHOD1(SendConnectionClose, void(QuicErrorCode error)); | 164 MOCK_METHOD1(SendConnectionClose, void(QuicErrorCode error)); |
| 157 | 165 |
| 158 MOCK_METHOD3(SendRstStream, void(QuicStreamId id, | 166 MOCK_METHOD3(SendRstStream, void(QuicStreamId id, |
| 159 QuicErrorCode error, | 167 QuicErrorCode error, |
| 160 QuicStreamOffset offset)); | 168 QuicStreamOffset offset)); |
| (...skipping 29 matching lines...) Expand all Loading... |
| 190 | 198 |
| 191 MOCK_METHOD4(OnPacket, bool(const IPEndPoint& self_address, | 199 MOCK_METHOD4(OnPacket, bool(const IPEndPoint& self_address, |
| 192 const IPEndPoint& peer_address, | 200 const IPEndPoint& peer_address, |
| 193 const QuicPacketHeader& header, | 201 const QuicPacketHeader& header, |
| 194 const std::vector<QuicStreamFrame>& frame)); | 202 const std::vector<QuicStreamFrame>& frame)); |
| 195 MOCK_METHOD2(ConnectionClose, void(QuicErrorCode error, bool from_peer)); | 203 MOCK_METHOD2(ConnectionClose, void(QuicErrorCode error, bool from_peer)); |
| 196 MOCK_METHOD1(CreateIncomingReliableStream, | 204 MOCK_METHOD1(CreateIncomingReliableStream, |
| 197 ReliableQuicStream*(QuicStreamId id)); | 205 ReliableQuicStream*(QuicStreamId id)); |
| 198 MOCK_METHOD0(GetCryptoStream, QuicCryptoStream*()); | 206 MOCK_METHOD0(GetCryptoStream, QuicCryptoStream*()); |
| 199 MOCK_METHOD0(CreateOutgoingReliableStream, ReliableQuicStream*()); | 207 MOCK_METHOD0(CreateOutgoingReliableStream, ReliableQuicStream*()); |
| 200 MOCK_METHOD3(WriteData, | |
| 201 void(QuicStreamId id, base::StringPiece data, bool fin)); | |
| 202 MOCK_METHOD4(WriteData, QuicConsumedData(QuicStreamId id, | 208 MOCK_METHOD4(WriteData, QuicConsumedData(QuicStreamId id, |
| 203 base::StringPiece data, | 209 base::StringPiece data, |
| 204 QuicStreamOffset offset, bool fin)); | 210 QuicStreamOffset offset, bool fin)); |
| 205 MOCK_METHOD0(IsHandshakeComplete, bool()); | 211 MOCK_METHOD0(IsHandshakeComplete, bool()); |
| 206 | 212 |
| 207 private: | 213 private: |
| 208 DISALLOW_COPY_AND_ASSIGN(MockSession); | 214 DISALLOW_COPY_AND_ASSIGN(MockSession); |
| 209 }; | 215 }; |
| 210 | 216 |
| 211 class MockScheduler : public QuicSendScheduler { | 217 class MockScheduler : public QuicSendScheduler { |
| 212 public: | 218 public: |
| 213 MockScheduler(); | 219 MockScheduler(); |
| 214 virtual ~MockScheduler(); | 220 virtual ~MockScheduler(); |
| 215 | 221 |
| 216 MOCK_METHOD1(TimeUntilSend, QuicTime::Delta(bool)); | 222 MOCK_METHOD1(TimeUntilSend, QuicTime::Delta(bool)); |
| 217 MOCK_METHOD1(OnIncomingAckFrame, void(const QuicAckFrame&)); | 223 MOCK_METHOD1(OnIncomingAckFrame, void(const QuicAckFrame&)); |
| 218 MOCK_METHOD3(SentPacket, void(QuicPacketSequenceNumber, size_t, bool)); | 224 MOCK_METHOD3(SentPacket, void(QuicPacketSequenceNumber, size_t, bool)); |
| 219 | 225 |
| 220 private: | 226 private: |
| 221 DISALLOW_COPY_AND_ASSIGN(MockScheduler); | 227 DISALLOW_COPY_AND_ASSIGN(MockScheduler); |
| 222 }; | 228 }; |
| 223 | 229 |
| 224 } // namespace test | 230 } // namespace test |
| 225 | 231 |
| 226 } // namespace net | 232 } // namespace net |
| 227 | 233 |
| 228 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ | 234 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ |
| OLD | NEW |