aboutsummaryrefslogtreecommitdiff
path: root/cast/streaming/receiver_unittest.cc
diff options
context:
space:
mode:
Diffstat (limited to 'cast/streaming/receiver_unittest.cc')
-rw-r--r--cast/streaming/receiver_unittest.cc6
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