1 // Copyright (c) 2018 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 #ifndef QUICHE_QUIC_TEST_TOOLS_QPACK_QPACK_TEST_UTILS_H_ 6 #define QUICHE_QUIC_TEST_TOOLS_QPACK_QPACK_TEST_UTILS_H_ 7 8 #include <cstddef> 9 #include <functional> 10 11 #include "absl/strings/string_view.h" 12 #include "quiche/quic/core/qpack/qpack_stream_sender_delegate.h" 13 #include "quiche/quic/platform/api/quic_test.h" 14 #include "quiche/common/quiche_callbacks.h" 15 16 namespace quic { 17 namespace test { 18 19 // Called repeatedly to determine the size of each fragment when encoding or 20 // decoding. Must return a positive value. 21 using FragmentSizeGenerator = quiche::MultiUseCallback<size_t()>; 22 23 enum class FragmentMode { 24 kSingleChunk, 25 kOctetByOctet, 26 }; 27 28 FragmentSizeGenerator FragmentModeToFragmentSizeGenerator( 29 FragmentMode fragment_mode); 30 31 // Mock QpackUnidirectionalStreamSenderDelegate implementation. 32 class MockQpackStreamSenderDelegate : public QpackStreamSenderDelegate { 33 public: 34 ~MockQpackStreamSenderDelegate() override = default; 35 36 MOCK_METHOD(void, WriteStreamData, (absl::string_view data), (override)); 37 MOCK_METHOD(uint64_t, NumBytesBuffered, (), (const, override)); 38 }; 39 40 class NoopQpackStreamSenderDelegate : public QpackStreamSenderDelegate { 41 public: 42 ~NoopQpackStreamSenderDelegate() override = default; 43 WriteStreamData(absl::string_view)44 void WriteStreamData(absl::string_view /*data*/) override {} 45 NumBytesBuffered()46 uint64_t NumBytesBuffered() const override { return 0; } 47 }; 48 49 } // namespace test 50 } // namespace quic 51 52 #endif // QUICHE_QUIC_TEST_TOOLS_QPACK_QPACK_TEST_UTILS_H_ 53