diff options
Diffstat (limited to 'cast/streaming/receiver_unittest.cc')
-rw-r--r-- | cast/streaming/receiver_unittest.cc | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/cast/streaming/receiver_unittest.cc b/cast/streaming/receiver_unittest.cc index 0a6817e6..86fd95ff 100644 --- a/cast/streaming/receiver_unittest.cc +++ b/cast/streaming/receiver_unittest.cc @@ -26,6 +26,7 @@ #include "cast/streaming/sender_report_builder.h" #include "cast/streaming/session_config.h" #include "cast/streaming/ssrc.h" +#include "cast/streaming/testing/simple_socket_subscriber.h" #include "gmock/gmock.h" #include "gtest/gtest.h" #include "platform/api/time.h" @@ -249,7 +250,6 @@ class MockSender : public CompoundRtcpParser::Client { CompoundRtcpParser rtcp_parser_; FrameCrypto crypto_; RtpPacketizer rtp_packetizer_; - FrameId max_feedback_frame_id_ = FrameId::first() + kMaxUnackedFrames; EncryptedFrame frame_being_sent_; @@ -278,8 +278,7 @@ class ReceiverTest : public testing::Test { /* .aes_iv_mask = */ kCastIvMask, /* .is_pli_enabled = */ true}), sender_(&task_runner_, &env_) { - env_.set_socket_error_handler( - [](Error error) { ASSERT_TRUE(error.ok()) << error; }); + env_.SetSocketSubscriber(&socket_subscriber_); ON_CALL(env_, SendPacket(_)) .WillByDefault(Invoke([this](absl::Span<const uint8_t> packet) { task_runner_.PostTaskWithDelay( @@ -360,6 +359,7 @@ class ReceiverTest : public testing::Test { Receiver receiver_; testing::NiceMock<MockSender> sender_; testing::NiceMock<MockConsumer> consumer_; + SimpleSubscriber socket_subscriber_; }; // Tests that the Receiver processes RTCP packets correctly and sends RTCP |