Index: media/cast/transport/rtp_sender/rtp_packetizer/rtp_packetizer_unittest.cc |
diff --git a/media/cast/transport/rtp_sender/rtp_packetizer/rtp_packetizer_unittest.cc b/media/cast/transport/rtp_sender/rtp_packetizer/rtp_packetizer_unittest.cc |
index a2e19f555ef26d8d82b2b1d77d162469edcdcc0e..f5a1d0cd12dbf75162e517e6054c2ff65101fc47 100644 |
--- a/media/cast/transport/rtp_sender/rtp_packetizer/rtp_packetizer_unittest.cc |
+++ b/media/cast/transport/rtp_sender/rtp_packetizer/rtp_packetizer_unittest.cc |
@@ -6,6 +6,7 @@ |
#include "base/memory/scoped_ptr.h" |
#include "base/test/simple_test_tick_clock.h" |
+#include "media/cast/logging/simple_event_subscriber.h" |
#include "media/cast/test/fake_single_thread_task_runner.h" |
#include "media/cast/transport/pacing/paced_sender.h" |
#include "media/cast/transport/rtp_sender/packet_storage/packet_storage.h" |
@@ -99,16 +100,19 @@ class RtpPacketizerTest : public ::testing::Test { |
RtpPacketizerTest() |
: task_runner_(new test::FakeSingleThreadTaskRunner(&testing_clock_)), |
video_frame_(), |
- packet_storage_(&testing_clock_, kMaxPacketStorageTimeMs) { |
+ packet_storage_(&testing_clock_, kMaxPacketStorageTimeMs), |
+ logging_(GetLoggingConfigWithRawEventsAndStatsEnabled()) { |
+ logging_.AddRawEventSubscriber(&subscriber_); |
config_.sequence_number = kSeqNum; |
config_.ssrc = kSsrc; |
config_.payload_type = kPayload; |
config_.max_payload_length = kMaxPacketLength; |
transport_.reset(new TestRtpPacketTransport(config_)); |
- pacer_.reset( |
- new PacedSender(&testing_clock_, transport_.get(), task_runner_)); |
- rtp_packetizer_.reset( |
- new RtpPacketizer(pacer_.get(), &packet_storage_, config_)); |
+ pacer_.reset(new PacedSender( |
+ &testing_clock_, &logging_, transport_.get(), task_runner_)); |
+ pacer_->RegisterVideoSsrc(config_.ssrc); |
+ rtp_packetizer_.reset(new RtpPacketizer( |
+ pacer_.get(), &packet_storage_, config_, &testing_clock_, &logging_)); |
video_frame_.key_frame = false; |
video_frame_.frame_id = 0; |
video_frame_.last_referenced_frame_id = kStartFrameId; |
@@ -117,7 +121,9 @@ class RtpPacketizerTest : public ::testing::Test { |
GetVideoRtpTimestamp(testing_clock_.NowTicks()); |
} |
- virtual ~RtpPacketizerTest() {} |
+ virtual ~RtpPacketizerTest() { |
+ logging_.RemoveRawEventSubscriber(&subscriber_); |
+ } |
void RunTasks(int during_ms) { |
for (int i = 0; i < during_ms; ++i) { |
@@ -133,6 +139,8 @@ class RtpPacketizerTest : public ::testing::Test { |
PacketStorage packet_storage_; |
RtpPacketizerConfig config_; |
scoped_ptr<TestRtpPacketTransport> transport_; |
+ LoggingImpl logging_; |
+ SimpleEventSubscriber subscriber_; |
scoped_ptr<PacedSender> pacer_; |
scoped_ptr<RtpPacketizer> rtp_packetizer_; |
@@ -149,6 +157,18 @@ TEST_F(RtpPacketizerTest, SendStandardPackets) { |
rtp_packetizer_->IncomingEncodedVideoFrame(&video_frame_, time); |
RunTasks(33 + 1); |
EXPECT_EQ(expected_num_of_packets, transport_->number_of_packets_received()); |
+ std::vector<PacketEvent> packet_events; |
+ subscriber_.GetPacketEventsAndReset(&packet_events); |
+ int expected_num_video_sent_to_pacer_count = expected_num_of_packets; |
+ int num_video_sent_to_pacer_count = 0; |
+ for (std::vector<PacketEvent>::iterator it = packet_events.begin(); |
+ it != packet_events.end(); |
+ ++it) { |
+ if (it->type == kVideoPacketSentToPacer) |
+ num_video_sent_to_pacer_count++; |
+ } |
+ EXPECT_EQ(expected_num_video_sent_to_pacer_count, |
+ num_video_sent_to_pacer_count); |
} |
TEST_F(RtpPacketizerTest, Stats) { |