| 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 <string> | 10 #include <string> |
| (...skipping 196 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 207 MockHelper(); | 207 MockHelper(); |
| 208 virtual ~MockHelper(); | 208 virtual ~MockHelper(); |
| 209 virtual const QuicClock* GetClock() const OVERRIDE; | 209 virtual const QuicClock* GetClock() const OVERRIDE; |
| 210 virtual QuicRandom* GetRandomGenerator() OVERRIDE; | 210 virtual QuicRandom* GetRandomGenerator() OVERRIDE; |
| 211 virtual QuicAlarm* CreateAlarm(QuicAlarm::Delegate* delegate) OVERRIDE; | 211 virtual QuicAlarm* CreateAlarm(QuicAlarm::Delegate* delegate) OVERRIDE; |
| 212 void AdvanceTime(QuicTime::Delta delta); | 212 void AdvanceTime(QuicTime::Delta delta); |
| 213 | 213 |
| 214 private: | 214 private: |
| 215 MockClock clock_; | 215 MockClock clock_; |
| 216 MockRandom random_generator_; | 216 MockRandom random_generator_; |
| 217 |
| 218 DISALLOW_COPY_AND_ASSIGN(MockHelper); |
| 217 }; | 219 }; |
| 218 | 220 |
| 219 class MockConnection : public QuicConnection { | 221 class MockConnection : public QuicConnection { |
| 220 public: | 222 public: |
| 221 // Uses a MockHelper, ConnectionId of 42, and 127.0.0.1:123. | 223 // Uses a MockHelper, ConnectionId of 42, and 127.0.0.1:123. |
| 222 explicit MockConnection(bool is_server); | 224 explicit MockConnection(bool is_server); |
| 223 | 225 |
| 224 // Uses a MockHelper, ConnectionId of 42. | 226 // Uses a MockHelper, ConnectionId of 42. |
| 225 MockConnection(IPEndPoint address, bool is_server); | 227 MockConnection(IPEndPoint address, bool is_server); |
| 226 | 228 |
| (...skipping 104 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 331 | 333 |
| 332 MOCK_METHOD1(CreateIncomingDataStream, QuicDataStream*(QuicStreamId id)); | 334 MOCK_METHOD1(CreateIncomingDataStream, QuicDataStream*(QuicStreamId id)); |
| 333 MOCK_METHOD0(CreateOutgoingDataStream, QuicDataStream*()); | 335 MOCK_METHOD0(CreateOutgoingDataStream, QuicDataStream*()); |
| 334 | 336 |
| 335 void SetCryptoStream(QuicCryptoStream* stream); | 337 void SetCryptoStream(QuicCryptoStream* stream); |
| 336 | 338 |
| 337 virtual QuicCryptoStream* GetCryptoStream() OVERRIDE; | 339 virtual QuicCryptoStream* GetCryptoStream() OVERRIDE; |
| 338 | 340 |
| 339 private: | 341 private: |
| 340 QuicCryptoStream* crypto_stream_; | 342 QuicCryptoStream* crypto_stream_; |
| 343 |
| 341 DISALLOW_COPY_AND_ASSIGN(TestSession); | 344 DISALLOW_COPY_AND_ASSIGN(TestSession); |
| 342 }; | 345 }; |
| 343 | 346 |
| 344 class TestClientSession : public QuicClientSessionBase { | 347 class TestClientSession : public QuicClientSessionBase { |
| 345 public: | 348 public: |
| 346 TestClientSession(QuicConnection* connection, const QuicConfig& config); | 349 TestClientSession(QuicConnection* connection, const QuicConfig& config); |
| 347 virtual ~TestClientSession(); | 350 virtual ~TestClientSession(); |
| 348 | 351 |
| 349 // QuicClientSessionBase | 352 // QuicClientSessionBase |
| 350 MOCK_METHOD1(OnProofValid, | 353 MOCK_METHOD1(OnProofValid, |
| 351 void(const QuicCryptoClientConfig::CachedState& cached)); | 354 void(const QuicCryptoClientConfig::CachedState& cached)); |
| 352 MOCK_METHOD1(OnProofVerifyDetailsAvailable, | 355 MOCK_METHOD1(OnProofVerifyDetailsAvailable, |
| 353 void(const ProofVerifyDetails& verify_details)); | 356 void(const ProofVerifyDetails& verify_details)); |
| 354 | 357 |
| 355 // TestClientSession | 358 // TestClientSession |
| 356 MOCK_METHOD1(CreateIncomingDataStream, QuicDataStream*(QuicStreamId id)); | 359 MOCK_METHOD1(CreateIncomingDataStream, QuicDataStream*(QuicStreamId id)); |
| 357 MOCK_METHOD0(CreateOutgoingDataStream, QuicDataStream*()); | 360 MOCK_METHOD0(CreateOutgoingDataStream, QuicDataStream*()); |
| 358 | 361 |
| 359 void SetCryptoStream(QuicCryptoStream* stream); | 362 void SetCryptoStream(QuicCryptoStream* stream); |
| 360 | 363 |
| 361 virtual QuicCryptoStream* GetCryptoStream() OVERRIDE; | 364 virtual QuicCryptoStream* GetCryptoStream() OVERRIDE; |
| 362 | 365 |
| 363 private: | 366 private: |
| 364 QuicCryptoStream* crypto_stream_; | 367 QuicCryptoStream* crypto_stream_; |
| 368 |
| 365 DISALLOW_COPY_AND_ASSIGN(TestClientSession); | 369 DISALLOW_COPY_AND_ASSIGN(TestClientSession); |
| 366 }; | 370 }; |
| 367 | 371 |
| 368 class MockPacketWriter : public QuicPacketWriter { | 372 class MockPacketWriter : public QuicPacketWriter { |
| 369 public: | 373 public: |
| 370 MockPacketWriter(); | 374 MockPacketWriter(); |
| 371 virtual ~MockPacketWriter(); | 375 virtual ~MockPacketWriter(); |
| 372 | 376 |
| 373 MOCK_METHOD4(WritePacket, | 377 MOCK_METHOD4(WritePacket, |
| 374 WriteResult(const char* buffer, | 378 WriteResult(const char* buffer, |
| 375 size_t buf_len, | 379 size_t buf_len, |
| 376 const IPAddressNumber& self_address, | 380 const IPAddressNumber& self_address, |
| 377 const IPEndPoint& peer_address)); | 381 const IPEndPoint& peer_address)); |
| 378 MOCK_CONST_METHOD0(IsWriteBlockedDataBuffered, bool()); | 382 MOCK_CONST_METHOD0(IsWriteBlockedDataBuffered, bool()); |
| 379 MOCK_CONST_METHOD0(IsWriteBlocked, bool()); | 383 MOCK_CONST_METHOD0(IsWriteBlocked, bool()); |
| 380 MOCK_METHOD0(SetWritable, void()); | 384 MOCK_METHOD0(SetWritable, void()); |
| 385 |
| 386 private: |
| 387 DISALLOW_COPY_AND_ASSIGN(MockPacketWriter); |
| 381 }; | 388 }; |
| 382 | 389 |
| 383 class MockSendAlgorithm : public SendAlgorithmInterface { | 390 class MockSendAlgorithm : public SendAlgorithmInterface { |
| 384 public: | 391 public: |
| 385 MockSendAlgorithm(); | 392 MockSendAlgorithm(); |
| 386 virtual ~MockSendAlgorithm(); | 393 virtual ~MockSendAlgorithm(); |
| 387 | 394 |
| 388 MOCK_METHOD2(SetFromConfig, void(const QuicConfig& config, bool is_server)); | 395 MOCK_METHOD2(SetFromConfig, void(const QuicConfig& config, bool is_server)); |
| 389 MOCK_METHOD1(SetMaxPacketSize, void(QuicByteCount max_packet_size)); | 396 MOCK_METHOD1(SetMaxPacketSize, void(QuicByteCount max_packet_size)); |
| 390 MOCK_METHOD2(OnIncomingQuicCongestionFeedbackFrame, | 397 MOCK_METHOD2(OnIncomingQuicCongestionFeedbackFrame, |
| (...skipping 24 matching lines...) Expand all Loading... |
| 415 MockLossAlgorithm(); | 422 MockLossAlgorithm(); |
| 416 virtual ~MockLossAlgorithm(); | 423 virtual ~MockLossAlgorithm(); |
| 417 | 424 |
| 418 MOCK_CONST_METHOD0(GetLossDetectionType, LossDetectionType()); | 425 MOCK_CONST_METHOD0(GetLossDetectionType, LossDetectionType()); |
| 419 MOCK_METHOD4(DetectLostPackets, | 426 MOCK_METHOD4(DetectLostPackets, |
| 420 SequenceNumberSet(const QuicUnackedPacketMap& unacked_packets, | 427 SequenceNumberSet(const QuicUnackedPacketMap& unacked_packets, |
| 421 const QuicTime& time, | 428 const QuicTime& time, |
| 422 QuicPacketSequenceNumber largest_observed, | 429 QuicPacketSequenceNumber largest_observed, |
| 423 const RttStats& rtt_stats)); | 430 const RttStats& rtt_stats)); |
| 424 MOCK_CONST_METHOD0(GetLossTimeout, QuicTime()); | 431 MOCK_CONST_METHOD0(GetLossTimeout, QuicTime()); |
| 432 |
| 433 private: |
| 434 DISALLOW_COPY_AND_ASSIGN(MockLossAlgorithm); |
| 425 }; | 435 }; |
| 426 | 436 |
| 427 class TestEntropyCalculator : | 437 class TestEntropyCalculator : |
| 428 public QuicReceivedEntropyHashCalculatorInterface { | 438 public QuicReceivedEntropyHashCalculatorInterface { |
| 429 public: | 439 public: |
| 430 TestEntropyCalculator(); | 440 TestEntropyCalculator(); |
| 431 virtual ~TestEntropyCalculator(); | 441 virtual ~TestEntropyCalculator(); |
| 432 | 442 |
| 433 virtual QuicPacketEntropyHash EntropyHash( | 443 virtual QuicPacketEntropyHash EntropyHash( |
| 434 QuicPacketSequenceNumber sequence_number) const OVERRIDE; | 444 QuicPacketSequenceNumber sequence_number) const OVERRIDE; |
| 445 |
| 446 private: |
| 447 DISALLOW_COPY_AND_ASSIGN(TestEntropyCalculator); |
| 435 }; | 448 }; |
| 436 | 449 |
| 437 class MockEntropyCalculator : public TestEntropyCalculator { | 450 class MockEntropyCalculator : public TestEntropyCalculator { |
| 438 public: | 451 public: |
| 439 MockEntropyCalculator(); | 452 MockEntropyCalculator(); |
| 440 virtual ~MockEntropyCalculator(); | 453 virtual ~MockEntropyCalculator(); |
| 441 | 454 |
| 442 MOCK_CONST_METHOD1( | 455 MOCK_CONST_METHOD1( |
| 443 EntropyHash, | 456 EntropyHash, |
| 444 QuicPacketEntropyHash(QuicPacketSequenceNumber sequence_number)); | 457 QuicPacketEntropyHash(QuicPacketSequenceNumber sequence_number)); |
| 458 |
| 459 private: |
| 460 DISALLOW_COPY_AND_ASSIGN(MockEntropyCalculator); |
| 445 }; | 461 }; |
| 446 | 462 |
| 447 class MockAckNotifierDelegate : public QuicAckNotifier::DelegateInterface { | 463 class MockAckNotifierDelegate : public QuicAckNotifier::DelegateInterface { |
| 448 public: | 464 public: |
| 449 MockAckNotifierDelegate(); | 465 MockAckNotifierDelegate(); |
| 450 | 466 |
| 451 MOCK_METHOD5(OnAckNotification, void(int num_original_packets, | 467 MOCK_METHOD5(OnAckNotification, void(int num_original_packets, |
| 452 int num_original_bytes, | 468 int num_original_bytes, |
| 453 int num_retransmitted_packets, | 469 int num_retransmitted_packets, |
| 454 int num_retransmitted_bytes, | 470 int num_retransmitted_bytes, |
| 455 QuicTime::Delta delta_largest_observed)); | 471 QuicTime::Delta delta_largest_observed)); |
| 456 | 472 |
| 457 protected: | 473 protected: |
| 458 // Object is ref counted. | 474 // Object is ref counted. |
| 459 virtual ~MockAckNotifierDelegate(); | 475 virtual ~MockAckNotifierDelegate(); |
| 476 |
| 477 private: |
| 478 DISALLOW_COPY_AND_ASSIGN(MockAckNotifierDelegate); |
| 460 }; | 479 }; |
| 461 | 480 |
| 462 } // namespace test | 481 } // namespace test |
| 463 } // namespace net | 482 } // namespace net |
| 464 | 483 |
| 465 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ | 484 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_UTILS_H_ |
| OLD | NEW |