| // Copyright (c) 2018 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_TEST_TOOLS_QPACK_QPACK_DECODER_TEST_UTILS_H_ |
| #define QUICHE_QUIC_TEST_TOOLS_QPACK_QPACK_DECODER_TEST_UTILS_H_ |
| |
| #include <string> |
| |
| #include "absl/strings/string_view.h" |
| #include "quiche/quic/core/qpack/qpack_decoder.h" |
| #include "quiche/quic/core/qpack/qpack_progressive_decoder.h" |
| #include "quiche/quic/core/quic_error_codes.h" |
| #include "quiche/quic/platform/api/quic_test.h" |
| #include "quiche/quic/test_tools/qpack/qpack_test_utils.h" |
| #include "quiche/spdy/core/http2_header_block.h" |
| |
| namespace quic { |
| namespace test { |
| |
| // Mock QpackDecoder::EncoderStreamErrorDelegate implementation. |
| class MockEncoderStreamErrorDelegate |
| : public QpackDecoder::EncoderStreamErrorDelegate { |
| public: |
| ~MockEncoderStreamErrorDelegate() override = default; |
| |
| MOCK_METHOD(void, OnEncoderStreamError, |
| (QuicErrorCode error_code, absl::string_view error_message), |
| (override)); |
| }; |
| |
| // HeadersHandlerInterface implementation that collects decoded headers |
| // into a Http2HeaderBlock. |
| class TestHeadersHandler |
| : public QpackProgressiveDecoder::HeadersHandlerInterface { |
| public: |
| TestHeadersHandler(); |
| ~TestHeadersHandler() override = default; |
| |
| // HeadersHandlerInterface implementation: |
| void OnHeaderDecoded(absl::string_view name, |
| absl::string_view value) override; |
| void OnDecodingCompleted() override; |
| void OnDecodingErrorDetected(QuicErrorCode error_code, |
| absl::string_view error_message) override; |
| |
| // Release decoded header list. Must only be called if decoding is complete |
| // and no errors have been detected. |
| spdy::Http2HeaderBlock ReleaseHeaderList(); |
| |
| bool decoding_completed() const; |
| bool decoding_error_detected() const; |
| const std::string& error_message() const; |
| |
| private: |
| spdy::Http2HeaderBlock header_list_; |
| bool decoding_completed_; |
| bool decoding_error_detected_; |
| std::string error_message_; |
| }; |
| |
| class MockHeadersHandler |
| : public QpackProgressiveDecoder::HeadersHandlerInterface { |
| public: |
| MockHeadersHandler() = default; |
| MockHeadersHandler(const MockHeadersHandler&) = delete; |
| MockHeadersHandler& operator=(const MockHeadersHandler&) = delete; |
| ~MockHeadersHandler() override = default; |
| |
| MOCK_METHOD(void, OnHeaderDecoded, |
| (absl::string_view name, absl::string_view value), (override)); |
| MOCK_METHOD(void, OnDecodingCompleted, (), (override)); |
| MOCK_METHOD(void, OnDecodingErrorDetected, |
| (QuicErrorCode error_code, absl::string_view error_message), |
| (override)); |
| }; |
| |
| class NoOpHeadersHandler |
| : public QpackProgressiveDecoder::HeadersHandlerInterface { |
| public: |
| ~NoOpHeadersHandler() override = default; |
| |
| void OnHeaderDecoded(absl::string_view /*name*/, |
| absl::string_view /*value*/) override {} |
| void OnDecodingCompleted() override {} |
| void OnDecodingErrorDetected(QuicErrorCode /*error_code*/, |
| absl::string_view /*error_message*/) override {} |
| }; |
| |
| void QpackDecode( |
| uint64_t maximum_dynamic_table_capacity, uint64_t maximum_blocked_streams, |
| QpackDecoder::EncoderStreamErrorDelegate* encoder_stream_error_delegate, |
| QpackStreamSenderDelegate* decoder_stream_sender_delegate, |
| QpackProgressiveDecoder::HeadersHandlerInterface* handler, |
| const FragmentSizeGenerator& fragment_size_generator, |
| absl::string_view data); |
| |
| } // namespace test |
| } // namespace quic |
| |
| #endif // QUICHE_QUIC_TEST_TOOLS_QPACK_QPACK_DECODER_TEST_UTILS_H_ |