Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(294)

Unified Diff: webrtc/modules/video_coding/h264_sps_pps_tracker_unittest.cc

Issue 2477343002: H264SpsPpsTracker now return PacketAction on CopyAndFixBitstream. (Closed)
Patch Set: Created 4 years, 1 month ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « webrtc/modules/video_coding/h264_sps_pps_tracker.cc ('k') | no next file » | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: webrtc/modules/video_coding/h264_sps_pps_tracker_unittest.cc
diff --git a/webrtc/modules/video_coding/h264_sps_pps_tracker_unittest.cc b/webrtc/modules/video_coding/h264_sps_pps_tracker_unittest.cc
index f467028f16d6d0668439e4e9f4032d3025797306..340f4bd28a84aee1672815b88075c110d7098f43 100644
--- a/webrtc/modules/video_coding/h264_sps_pps_tracker_unittest.cc
+++ b/webrtc/modules/video_coding/h264_sps_pps_tracker_unittest.cc
@@ -87,7 +87,7 @@ TEST_F(TestH264SpsPpsTracker, NoNalus) {
packet.dataPtr = data;
packet.sizeBytes = sizeof(data);
- EXPECT_TRUE(tracker_.CopyAndFixBitstream(&packet));
+ EXPECT_EQ(H264SpsPpsTracker::kInsert, tracker_.CopyAndFixBitstream(&packet));
EXPECT_EQ(memcmp(packet.dataPtr, data, sizeof(data)), 0);
delete[] packet.dataPtr;
}
@@ -100,7 +100,7 @@ TEST_F(TestH264SpsPpsTracker, FuAFirstPacket) {
packet.dataPtr = data;
packet.sizeBytes = sizeof(data);
- EXPECT_TRUE(tracker_.CopyAndFixBitstream(&packet));
+ EXPECT_EQ(H264SpsPpsTracker::kInsert, tracker_.CopyAndFixBitstream(&packet));
std::vector<uint8_t> expected;
expected.insert(expected.end(), start_code, start_code + sizeof(start_code));
expected.insert(expected.end(), {1, 2, 3});
@@ -116,7 +116,7 @@ TEST_F(TestH264SpsPpsTracker, StapAIncorrectSegmentLength) {
packet.dataPtr = data;
packet.sizeBytes = sizeof(data);
- EXPECT_FALSE(tracker_.CopyAndFixBitstream(&packet));
+ EXPECT_EQ(H264SpsPpsTracker::kDrop, tracker_.CopyAndFixBitstream(&packet));
}
TEST_F(TestH264SpsPpsTracker, NoNalusFirstPacket) {
@@ -126,7 +126,7 @@ TEST_F(TestH264SpsPpsTracker, NoNalusFirstPacket) {
packet.dataPtr = data;
packet.sizeBytes = sizeof(data);
- EXPECT_TRUE(tracker_.CopyAndFixBitstream(&packet));
+ EXPECT_EQ(H264SpsPpsTracker::kInsert, tracker_.CopyAndFixBitstream(&packet));
std::vector<uint8_t> expected;
expected.insert(expected.end(), start_code, start_code + sizeof(start_code));
expected.insert(expected.end(), {1, 2, 3});
@@ -143,7 +143,7 @@ TEST_F(TestH264SpsPpsTracker, IdrNoSpsPpsInserted) {
packet.dataPtr = data.data();
packet.sizeBytes = data.size();
- EXPECT_TRUE(tracker_.CopyAndFixBitstream(&packet));
+ EXPECT_EQ(H264SpsPpsTracker::kInsert, tracker_.CopyAndFixBitstream(&packet));
EXPECT_EQ(memcmp(packet.dataPtr, data.data(), data.size()), 0);
delete[] packet.dataPtr;
}
@@ -157,7 +157,8 @@ TEST_F(TestH264SpsPpsTracker, IdrFirstPacketNoSpsPpsInserted) {
packet.dataPtr = data.data();
packet.sizeBytes = data.size();
- EXPECT_FALSE(tracker_.CopyAndFixBitstream(&packet));
+ EXPECT_EQ(H264SpsPpsTracker::kRequestKeyframe,
+ tracker_.CopyAndFixBitstream(&packet));
}
TEST_F(TestH264SpsPpsTracker, IdrFirstPacketNoPpsInserted) {
@@ -170,7 +171,8 @@ TEST_F(TestH264SpsPpsTracker, IdrFirstPacketNoPpsInserted) {
packet.dataPtr = data.data();
packet.sizeBytes = data.size();
- EXPECT_FALSE(tracker_.CopyAndFixBitstream(&packet));
+ EXPECT_EQ(H264SpsPpsTracker::kRequestKeyframe,
+ tracker_.CopyAndFixBitstream(&packet));
}
TEST_F(TestH264SpsPpsTracker, IdrFirstPacketNoSpsInserted) {
@@ -183,7 +185,8 @@ TEST_F(TestH264SpsPpsTracker, IdrFirstPacketNoSpsInserted) {
packet.dataPtr = data.data();
packet.sizeBytes = data.size();
- EXPECT_FALSE(tracker_.CopyAndFixBitstream(&packet));
+ EXPECT_EQ(H264SpsPpsTracker::kRequestKeyframe,
+ tracker_.CopyAndFixBitstream(&packet));
}
TEST_F(TestH264SpsPpsTracker, SpsPpsPacketThenIdrFirstPacket) {
@@ -195,7 +198,8 @@ TEST_F(TestH264SpsPpsTracker, SpsPpsPacketThenIdrFirstPacket) {
AddPps(&sps_pps_packet, 0, 1, &data);
sps_pps_packet.dataPtr = data.data();
sps_pps_packet.sizeBytes = data.size();
- EXPECT_FALSE(tracker_.CopyAndFixBitstream(&sps_pps_packet));
+ EXPECT_EQ(H264SpsPpsTracker::kDrop,
+ tracker_.CopyAndFixBitstream(&sps_pps_packet));
data.clear();
// Insert first packet of the IDR
@@ -205,7 +209,8 @@ TEST_F(TestH264SpsPpsTracker, SpsPpsPacketThenIdrFirstPacket) {
data.insert(data.end(), {1, 2, 3});
idr_packet.dataPtr = data.data();
idr_packet.sizeBytes = data.size();
- EXPECT_TRUE(tracker_.CopyAndFixBitstream(&idr_packet));
+ EXPECT_EQ(H264SpsPpsTracker::kInsert,
+ tracker_.CopyAndFixBitstream(&idr_packet));
std::vector<uint8_t> expected;
expected.insert(expected.end(), start_code, start_code + sizeof(start_code));
@@ -235,7 +240,7 @@ TEST_F(TestH264SpsPpsTracker, SpsPpsIdrInStapA) {
packet.dataPtr = data.data();
packet.sizeBytes = data.size();
- EXPECT_TRUE(tracker_.CopyAndFixBitstream(&packet));
+ EXPECT_EQ(H264SpsPpsTracker::kInsert, tracker_.CopyAndFixBitstream(&packet));
std::vector<uint8_t> expected;
// The SPS/PPS is repeated because this packet both contains the SPS/PPS
« no previous file with comments | « webrtc/modules/video_coding/h264_sps_pps_tracker.cc ('k') | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698