| // Copyright (c) 2019 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef QUICHE_QUIC_QBONE_QBONE_PACKET_PROCESSOR_TEST_TOOLS_H_ |
| #define QUICHE_QUIC_QBONE_QBONE_PACKET_PROCESSOR_TEST_TOOLS_H_ |
| |
| #include <cstdint> |
| |
| #include "absl/strings/string_view.h" |
| #include "quiche/quic/platform/api/quic_test.h" |
| #include "quiche/quic/qbone/qbone_packet_processor.h" |
| |
| namespace quic { |
| |
| class MockPacketProcessorOutput : public QbonePacketProcessor::OutputInterface { |
| public: |
| MockPacketProcessorOutput() {} |
| |
| MOCK_METHOD(void, SendPacketToClient, (absl::string_view), (override)); |
| MOCK_METHOD(void, SendPacketToNetwork, (absl::string_view), (override)); |
| }; |
| |
| class MockPacketProcessorStats : public QbonePacketProcessor::StatsInterface { |
| public: |
| MockPacketProcessorStats() {} |
| |
| MOCK_METHOD(void, OnPacketForwarded, |
| (QbonePacketProcessor::Direction, uint8_t), (override)); |
| MOCK_METHOD(void, OnPacketDroppedSilently, |
| (QbonePacketProcessor::Direction, uint8_t), (override)); |
| MOCK_METHOD(void, OnPacketDroppedWithIcmp, |
| (QbonePacketProcessor::Direction, uint8_t), (override)); |
| MOCK_METHOD(void, OnPacketDroppedWithTcpReset, |
| (QbonePacketProcessor::Direction, uint8_t), (override)); |
| MOCK_METHOD(void, RecordThroughput, |
| (size_t, QbonePacketProcessor::Direction, uint8_t), (override)); |
| }; |
| |
| std::string PrependIPv6HeaderForTest(const std::string& body, int hops); |
| |
| // Returns true if the hop limit was decremented. Returns false if the packet is |
| // too short, not IPv6, or already has a hop limit of zero. |
| bool DecrementIPv6HopLimit(std::string& packet); |
| |
| } // namespace quic |
| |
| #endif // QUICHE_QUIC_QBONE_QBONE_PACKET_PROCESSOR_TEST_TOOLS_H_ |