2 files changed, 41 insertions, 38 deletions
diff --git a/third_party/libwebrtc/modules/audio_coding/neteq/mock/mock_packet_arrival_history.h b/third_party/libwebrtc/modules/audio_coding/neteq/mock/mock_packet_arrival_history.h
new file mode 100644
index 0000000000..1b2080cd94
--- /dev/null
+++ b/third_party/libwebrtc/modules/audio_coding/neteq/mock/mock_packet_arrival_history.h
@@ -0,0 +1,32 @@
+/*
+ *  Copyright 2023 The WebRTC Project Authors. All rights reserved.
+ *
+ *  Use of this source code is governed by a BSD-style license
+ *  that can be found in the LICENSE file in the root of the source
+ *  tree. An additional intellectual property rights grant can be found
+ *  in the file PATENTS.  All contributing project authors may
+ *  be found in the AUTHORS file in the root of the source tree.
+ */
+
+#ifndef MODULES_AUDIO_CODING_NETEQ_MOCK_MOCK_PACKET_ARRIVAL_HISTORY_H_
+#define MODULES_AUDIO_CODING_NETEQ_MOCK_MOCK_PACKET_ARRIVAL_HISTORY_H_
+
+#include "modules/audio_coding/neteq/packet_arrival_history.h"
+#include "test/gmock.h"
+
+namespace webrtc {
+
+class MockPacketArrivalHistory : public PacketArrivalHistory {
+ public:
+  MockPacketArrivalHistory() : PacketArrivalHistory(0) {}
+
+  MOCK_METHOD(int,
+              GetDelayMs,
+              (uint32_t rtp_timestamp, int64_t time_ms),
+              (const override));
+  MOCK_METHOD(int, GetMaxDelayMs, (), (const override));
+};
+
+}  // namespace webrtc
+
+#endif  // MODULES_AUDIO_CODING_NETEQ_MOCK_MOCK_PACKET_ARRIVAL_HISTORY_H_
diff --git a/third_party/libwebrtc/modules/audio_coding/neteq/mock/mock_packet_buffer.h b/third_party/libwebrtc/modules/audio_coding/neteq/mock/mock_packet_buffer.h
index 48357ea466..fa44f606fc 100644
--- a/third_party/libwebrtc/modules/audio_coding/neteq/mock/mock_packet_buffer.h
+++ b/third_party/libwebrtc/modules/audio_coding/neteq/mock/mock_packet_buffer.h
@@ -18,39 +18,15 @@ namespace webrtc {
 
 class MockPacketBuffer : public PacketBuffer {
  public:
-  MockPacketBuffer(size_t max_number_of_packets, const TickTimer* tick_timer)
-      : PacketBuffer(max_number_of_packets, tick_timer) {}
+  MockPacketBuffer(size_t max_number_of_packets,
+                   const TickTimer* tick_timer,
+                   StatisticsCalculator* stats)
+      : PacketBuffer(max_number_of_packets, tick_timer, stats) {}
   ~MockPacketBuffer() override { Die(); }
   MOCK_METHOD(void, Die, ());
-  MOCK_METHOD(void, Flush, (StatisticsCalculator * stats), (override));
-  MOCK_METHOD(void,
-              PartialFlush,
-              (int target_level_ms,
-               size_t sample_rate,
-               size_t last_decoded_length,
-               StatisticsCalculator* stats),
-              (override));
+  MOCK_METHOD(void, Flush, (), (override));
   MOCK_METHOD(bool, Empty, (), (const, override));
-  MOCK_METHOD(int,
-              InsertPacket,
-              (Packet && packet,
-               StatisticsCalculator* stats,
-               size_t last_decoded_length,
-               size_t sample_rate,
-               int target_level_ms,
-               const DecoderDatabase& decoder_database),
-              (override));
-  MOCK_METHOD(int,
-              InsertPacketList,
-              (PacketList * packet_list,
-               const DecoderDatabase& decoder_database,
-               absl::optional<uint8_t>* current_rtp_payload_type,
-               absl::optional<uint8_t>* current_cng_rtp_payload_type,
-               StatisticsCalculator* stats,
-               size_t last_decoded_length,
-               size_t sample_rate,
-               int target_level_ms),
-              (override));
+  MOCK_METHOD(int, InsertPacket, (Packet && packet), (override));
   MOCK_METHOD(int,
               NextTimestamp,
               (uint32_t * next_timestamp),
@@ -61,19 +37,14 @@ class MockPacketBuffer : public PacketBuffer {
               (const, override));
   MOCK_METHOD(const Packet*, PeekNextPacket, (), (const, override));
   MOCK_METHOD(absl::optional<Packet>, GetNextPacket, (), (override));
-  MOCK_METHOD(int,
-              DiscardNextPacket,
-              (StatisticsCalculator * stats),
-              (override));
+  MOCK_METHOD(int, DiscardNextPacket, (), (override));
   MOCK_METHOD(void,
               DiscardOldPackets,
-              (uint32_t timestamp_limit,
-               uint32_t horizon_samples,
-               StatisticsCalculator* stats),
+              (uint32_t timestamp_limit, uint32_t horizon_samples),
               (override));
   MOCK_METHOD(void,
               DiscardAllOldPackets,
-              (uint32_t timestamp_limit, StatisticsCalculator* stats),
+              (uint32_t timestamp_limit),
               (override));
   MOCK_METHOD(size_t, NumPacketsInBuffer, (), (const, override));
 };