Chromium Code Reviews| 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, IPEndPoint address, | |
| 157 QuicConnectionHelperInterface* helper); | |
|
wtc
2013/01/15 23:11:03
This formatting is allowed by the current Style Gu
Ryan Hamilton
2013/01/16 00:29:36
Are you sure? jar complains about this every time
wtc
2013/01/16 00:43:49
Yes, I am sure. See the second code example at
htt
| |
| 151 virtual ~MockConnection(); | 158 virtual ~MockConnection(); |
| 152 | 159 |
| 153 MOCK_METHOD3(ProcessUdpPacket, void(const IPEndPoint& self_address, | 160 MOCK_METHOD3(ProcessUdpPacket, void(const IPEndPoint& self_address, |
| 154 const IPEndPoint& peer_address, | 161 const IPEndPoint& peer_address, |
| 155 const QuicEncryptedPacket& packet)); | 162 const QuicEncryptedPacket& packet)); |
| 156 MOCK_METHOD1(SendConnectionClose, void(QuicErrorCode error)); | 163 MOCK_METHOD1(SendConnectionClose, void(QuicErrorCode error)); |
| 157 | 164 |
| 158 MOCK_METHOD3(SendRstStream, void(QuicStreamId id, | 165 MOCK_METHOD3(SendRstStream, void(QuicStreamId id, |
| 159 QuicErrorCode error, | 166 QuicErrorCode error, |
| 160 QuicStreamOffset offset)); | 167 QuicStreamOffset offset)); |
| (...skipping 29 matching lines...) Expand all Loading... | |
| 190 | 197 |
| 191 MOCK_METHOD4(OnPacket, bool(const IPEndPoint& self_address, | 198 MOCK_METHOD4(OnPacket, bool(const IPEndPoint& self_address, |
| 192 const IPEndPoint& peer_address, | 199 const IPEndPoint& peer_address, |
| 193 const QuicPacketHeader& header, | 200 const QuicPacketHeader& header, |
| 194 const std::vector<QuicStreamFrame>& frame)); | 201 const std::vector<QuicStreamFrame>& frame)); |
| 195 MOCK_METHOD2(ConnectionClose, void(QuicErrorCode error, bool from_peer)); | 202 MOCK_METHOD2(ConnectionClose, void(QuicErrorCode error, bool from_peer)); |
| 196 MOCK_METHOD1(CreateIncomingReliableStream, | 203 MOCK_METHOD1(CreateIncomingReliableStream, |
| 197 ReliableQuicStream*(QuicStreamId id)); | 204 ReliableQuicStream*(QuicStreamId id)); |
| 198 MOCK_METHOD0(GetCryptoStream, QuicCryptoStream*()); | 205 MOCK_METHOD0(GetCryptoStream, QuicCryptoStream*()); |
| 199 MOCK_METHOD0(CreateOutgoingReliableStream, ReliableQuicStream*()); | 206 MOCK_METHOD0(CreateOutgoingReliableStream, ReliableQuicStream*()); |
| 200 MOCK_METHOD3(WriteData, | |
| 201 void(QuicStreamId id, base::StringPiece data, bool fin)); | |
| 202 MOCK_METHOD4(WriteData, int(QuicStreamId id, base::StringPiece data, | 207 MOCK_METHOD4(WriteData, int(QuicStreamId id, base::StringPiece data, |
| 203 QuicStreamOffset offset, bool fin)); | 208 QuicStreamOffset offset, bool fin)); |
| 204 MOCK_METHOD0(IsHandshakeComplete, bool()); | 209 MOCK_METHOD0(IsHandshakeComplete, bool()); |
| 205 | 210 |
| 206 private: | 211 private: |
| 207 DISALLOW_COPY_AND_ASSIGN(MockSession); | 212 DISALLOW_COPY_AND_ASSIGN(MockSession); |
| 208 }; | 213 }; |
| 209 | 214 |
| 210 class MockScheduler : public QuicSendScheduler { | 215 class MockScheduler : public QuicSendScheduler { |
| 211 public: | 216 public: |
| 212 MockScheduler(); | 217 MockScheduler(); |
| 213 virtual ~MockScheduler(); | 218 virtual ~MockScheduler(); |
| 214 | 219 |
| 215 MOCK_METHOD1(TimeUntilSend, QuicTime::Delta(bool)); | 220 MOCK_METHOD1(TimeUntilSend, QuicTime::Delta(bool)); |
| 216 MOCK_METHOD1(OnIncomingAckFrame, void(const QuicAckFrame&)); | 221 MOCK_METHOD1(OnIncomingAckFrame, void(const QuicAckFrame&)); |
| 217 MOCK_METHOD3(SentPacket, void(QuicPacketSequenceNumber, size_t, bool)); | 222 MOCK_METHOD3(SentPacket, void(QuicPacketSequenceNumber, size_t, bool)); |
| 218 | 223 |
| 219 private: | 224 private: |
| 220 DISALLOW_COPY_AND_ASSIGN(MockScheduler); | 225 DISALLOW_COPY_AND_ASSIGN(MockScheduler); |
| 221 }; | 226 }; |
| 222 | 227 |
| 223 } // namespace test | 228 } // namespace test |
| 224 | 229 |
| 225 } // namespace net | 230 } // namespace net |
| 226 | 231 |
| 227 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ | 232 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ |
| OLD | NEW |