Move MockLoadBalancerEncoder into quic/test_tools for re-use. PiperOrigin-RevId: 545979300
diff --git a/build/source_list.bzl b/build/source_list.bzl index df1bda5..b86abf5 100644 --- a/build/source_list.bzl +++ b/build/source_list.bzl
@@ -790,6 +790,7 @@ "quic/test_tools/limited_mtu_test_writer.h", "quic/test_tools/mock_clock.h", "quic/test_tools/mock_connection_id_generator.h", + "quic/test_tools/mock_load_balancer_encoder.h", "quic/test_tools/mock_quic_client_promised_info.h", "quic/test_tools/mock_quic_dispatcher.h", "quic/test_tools/mock_quic_session_visitor.h",
diff --git a/build/source_list.gni b/build/source_list.gni index 0ca2990..c8faad9 100644 --- a/build/source_list.gni +++ b/build/source_list.gni
@@ -790,6 +790,7 @@ "src/quiche/quic/test_tools/limited_mtu_test_writer.h", "src/quiche/quic/test_tools/mock_clock.h", "src/quiche/quic/test_tools/mock_connection_id_generator.h", + "src/quiche/quic/test_tools/mock_load_balancer_encoder.h", "src/quiche/quic/test_tools/mock_quic_client_promised_info.h", "src/quiche/quic/test_tools/mock_quic_dispatcher.h", "src/quiche/quic/test_tools/mock_quic_session_visitor.h",
diff --git a/build/source_list.json b/build/source_list.json index 8423361..10480a1 100644 --- a/build/source_list.json +++ b/build/source_list.json
@@ -789,6 +789,7 @@ "quiche/quic/test_tools/limited_mtu_test_writer.h", "quiche/quic/test_tools/mock_clock.h", "quiche/quic/test_tools/mock_connection_id_generator.h", + "quiche/quic/test_tools/mock_load_balancer_encoder.h", "quiche/quic/test_tools/mock_quic_client_promised_info.h", "quiche/quic/test_tools/mock_quic_dispatcher.h", "quiche/quic/test_tools/mock_quic_session_visitor.h",
diff --git a/quiche/quic/load_balancer/load_balancer_encoder.h b/quiche/quic/load_balancer/load_balancer_encoder.h index 1099d7a..ad83850 100644 --- a/quiche/quic/load_balancer/load_balancer_encoder.h +++ b/quiche/quic/load_balancer/load_balancer_encoder.h
@@ -97,7 +97,7 @@ // Delete the current config and generate unroutable connection IDs from now // on. - void DeleteConfig(); + virtual void DeleteConfig(); // Returns the number of additional connection IDs that can be generated with // the current config, or 0 if there is no current config.
diff --git a/quiche/quic/test_tools/mock_load_balancer_encoder.h b/quiche/quic/test_tools/mock_load_balancer_encoder.h new file mode 100644 index 0000000..f40047e --- /dev/null +++ b/quiche/quic/test_tools/mock_load_balancer_encoder.h
@@ -0,0 +1,36 @@ +// Copyright (c) 2023 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_MOCK_LOAD_BALANCER_ENCODER_H_ +#define QUICHE_QUIC_TEST_TOOLS_MOCK_LOAD_BALANCER_ENCODER_H_ + +#include "quiche/quic/load_balancer/load_balancer_encoder.h" +#include "quiche/quic/platform/api/quic_test.h" + +namespace quic { +namespace test { + +class MockLoadBalancerEncoder : public LoadBalancerEncoder { + public: + MockLoadBalancerEncoder() + : LoadBalancerEncoder(*QuicRandom::GetInstance(), nullptr, false, + kLoadBalancerUnroutableLen) {} + MOCK_METHOD(bool, IsEncoding, (), (const, override)); + MOCK_METHOD(bool, IsEncrypted, (), (const, override)); + MOCK_METHOD(bool, len_self_encoded, (), (const, override)); + MOCK_METHOD(absl::optional<QuicConnectionId>, GenerateNextConnectionId, + (const QuicConnectionId& original), (override)); + MOCK_METHOD(absl::optional<QuicConnectionId>, MaybeReplaceConnectionId, + (const QuicConnectionId& original, + const ParsedQuicVersion& version), + (override)); + MOCK_METHOD(uint8_t, ConnectionIdLength, (uint8_t first_byte), + (const, override)); + MOCK_METHOD(void, DeleteConfig, (), (override)); +}; + +} // namespace test +} // namespace quic + +#endif // QUICHE_QUIC_TEST_TOOLS_MOCK_LOAD_BALANCER_ENCODER_H_