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 |