Index: webrtc/pc/mediasession_unittest.cc |
diff --git a/webrtc/pc/mediasession_unittest.cc b/webrtc/pc/mediasession_unittest.cc |
index ca1e4b39394c4af9bb75a2fb03842662cbd32c75..4a3d7a7b9a4b54f486ca27fed5a031ec753913cc 100644 |
--- a/webrtc/pc/mediasession_unittest.cc |
+++ b/webrtc/pc/mediasession_unittest.cc |
@@ -190,10 +190,10 @@ GetMediaDirection(const ContentInfo* content) { |
return desc->direction(); |
} |
-static void AddRtxCodec(const VideoCodec& rtx_codec, |
- std::vector<VideoCodec>* codecs) { |
- ASSERT_FALSE(cricket::FindCodecById(*codecs, rtx_codec.id)); |
- codecs->push_back(rtx_codec); |
+static void AddSecondaryCodec(const VideoCodec& secondary_codec, |
perkj_webrtc
2016/11/17 20:41:12
Why not just AddCodec if you want to keep it at al
brandtr
2016/11/21 09:16:30
After a second thought, I don't think the the asse
|
+ std::vector<VideoCodec>* codecs) { |
+ ASSERT_FALSE(cricket::FindCodecById(*codecs, secondary_codec.id)); |
+ codecs->push_back(secondary_codec); |
} |
template <class T> |
@@ -1631,7 +1631,6 @@ TEST_F(MediaSessionDescriptionFactoryTest, TestCreateMultiStreamVideoAnswer) { |
EXPECT_TRUE(data_streams[0] == updated_data_streams[0]); |
} |
- |
// Create an updated offer after creating an answer to the original offer and |
// verify that the codecs that were part of the original answer are not changed |
// in the updated offer. |
@@ -1694,12 +1693,14 @@ TEST_F(MediaSessionDescriptionFactoryTest, |
opts.recv_audio = false; |
std::vector<VideoCodec> f1_codecs = MAKE_VECTOR(kVideoCodecs1); |
// This creates rtx for H264 with the payload type |f1_| uses. |
- AddRtxCodec(VideoCodec::CreateRtxCodec(126, kVideoCodecs1[1].id), &f1_codecs); |
+ AddSecondaryCodec(VideoCodec::CreateRtxCodec(126, kVideoCodecs1[1].id), |
+ &f1_codecs); |
f1_.set_video_codecs(f1_codecs); |
std::vector<VideoCodec> f2_codecs = MAKE_VECTOR(kVideoCodecs2); |
// This creates rtx for H264 with the payload type |f2_| uses. |
- AddRtxCodec(VideoCodec::CreateRtxCodec(125, kVideoCodecs2[0].id), &f2_codecs); |
+ AddSecondaryCodec(VideoCodec::CreateRtxCodec(125, kVideoCodecs2[0].id), |
+ &f2_codecs); |
f2_.set_video_codecs(f2_codecs); |
std::unique_ptr<SessionDescription> offer(f1_.CreateOffer(opts, NULL)); |
@@ -1711,8 +1712,8 @@ TEST_F(MediaSessionDescriptionFactoryTest, |
GetFirstVideoContentDescription(answer.get()); |
std::vector<VideoCodec> expected_codecs = MAKE_VECTOR(kVideoCodecsAnswer); |
- AddRtxCodec(VideoCodec::CreateRtxCodec(126, kVideoCodecs1[1].id), |
- &expected_codecs); |
+ AddSecondaryCodec(VideoCodec::CreateRtxCodec(126, kVideoCodecs1[1].id), |
+ &expected_codecs); |
EXPECT_EQ(expected_codecs, vcd->codecs()); |
@@ -1739,7 +1740,8 @@ TEST_F(MediaSessionDescriptionFactoryTest, |
RespondentCreatesOfferWithVideoAndRtxAfterCreatingAudioAnswer) { |
std::vector<VideoCodec> f1_codecs = MAKE_VECTOR(kVideoCodecs1); |
// This creates rtx for H264 with the payload type |f1_| uses. |
- AddRtxCodec(VideoCodec::CreateRtxCodec(126, kVideoCodecs1[1].id), &f1_codecs); |
+ AddSecondaryCodec(VideoCodec::CreateRtxCodec(126, kVideoCodecs1[1].id), |
+ &f1_codecs); |
f1_.set_video_codecs(f1_codecs); |
MediaSessionOptions opts; |
@@ -1763,7 +1765,7 @@ TEST_F(MediaSessionDescriptionFactoryTest, |
std::vector<VideoCodec> f2_codecs = MAKE_VECTOR(kVideoCodecs2); |
int used_pl_type = acd->codecs()[0].id; |
f2_codecs[0].id = used_pl_type; // Set the payload type for H264. |
- AddRtxCodec(VideoCodec::CreateRtxCodec(125, used_pl_type), &f2_codecs); |
+ AddSecondaryCodec(VideoCodec::CreateRtxCodec(125, used_pl_type), &f2_codecs); |
f2_.set_video_codecs(f2_codecs); |
std::unique_ptr<SessionDescription> updated_offer( |
@@ -1800,7 +1802,8 @@ TEST_F(MediaSessionDescriptionFactoryTest, |
std::vector<VideoCodec> f2_codecs = MAKE_VECTOR(kVideoCodecs2); |
// This creates rtx for H264 with the payload type |f2_| uses. |
- AddRtxCodec(VideoCodec::CreateRtxCodec(125, kVideoCodecs2[0].id), &f2_codecs); |
+ AddSecondaryCodec(VideoCodec::CreateRtxCodec(125, kVideoCodecs2[0].id), |
+ &f2_codecs); |
f2_.set_video_codecs(f2_codecs); |
std::unique_ptr<SessionDescription> offer(f1_.CreateOffer(opts, nullptr)); |
@@ -1826,8 +1829,8 @@ TEST_F(MediaSessionDescriptionFactoryTest, |
// New offer should attempt to add H263, and RTX for H264. |
expected_codecs.push_back(kVideoCodecs2[1]); |
- AddRtxCodec(VideoCodec::CreateRtxCodec(125, kVideoCodecs1[1].id), |
- &expected_codecs); |
+ AddSecondaryCodec(VideoCodec::CreateRtxCodec(125, kVideoCodecs1[1].id), |
+ &expected_codecs); |
EXPECT_EQ(expected_codecs, updated_vcd->codecs()); |
} |
@@ -1838,12 +1841,13 @@ TEST_F(MediaSessionDescriptionFactoryTest, RtxWithoutApt) { |
opts.recv_audio = false; |
std::vector<VideoCodec> f1_codecs = MAKE_VECTOR(kVideoCodecs1); |
// This creates RTX without associated payload type parameter. |
- AddRtxCodec(VideoCodec(126, cricket::kRtxCodecName), &f1_codecs); |
+ AddSecondaryCodec(VideoCodec(126, cricket::kRtxCodecName), &f1_codecs); |
f1_.set_video_codecs(f1_codecs); |
std::vector<VideoCodec> f2_codecs = MAKE_VECTOR(kVideoCodecs2); |
// This creates RTX for H264 with the payload type |f2_| uses. |
- AddRtxCodec(VideoCodec::CreateRtxCodec(125, kVideoCodecs2[0].id), &f2_codecs); |
+ AddSecondaryCodec(VideoCodec::CreateRtxCodec(125, kVideoCodecs2[0].id), |
+ &f2_codecs); |
f2_.set_video_codecs(f2_codecs); |
std::unique_ptr<SessionDescription> offer(f1_.CreateOffer(opts, NULL)); |
@@ -1882,12 +1886,14 @@ TEST_F(MediaSessionDescriptionFactoryTest, FilterOutRtxIfAptDoesntMatch) { |
opts.recv_audio = false; |
std::vector<VideoCodec> f1_codecs = MAKE_VECTOR(kVideoCodecs1); |
// This creates RTX for H264 in sender. |
- AddRtxCodec(VideoCodec::CreateRtxCodec(126, kVideoCodecs1[1].id), &f1_codecs); |
+ AddSecondaryCodec(VideoCodec::CreateRtxCodec(126, kVideoCodecs1[1].id), |
+ &f1_codecs); |
f1_.set_video_codecs(f1_codecs); |
std::vector<VideoCodec> f2_codecs = MAKE_VECTOR(kVideoCodecs2); |
// This creates RTX for H263 in receiver. |
- AddRtxCodec(VideoCodec::CreateRtxCodec(125, kVideoCodecs2[1].id), &f2_codecs); |
+ AddSecondaryCodec(VideoCodec::CreateRtxCodec(125, kVideoCodecs2[1].id), |
+ &f2_codecs); |
f2_.set_video_codecs(f2_codecs); |
std::unique_ptr<SessionDescription> offer(f1_.CreateOffer(opts, NULL)); |
@@ -1912,16 +1918,19 @@ TEST_F(MediaSessionDescriptionFactoryTest, |
opts.recv_audio = false; |
std::vector<VideoCodec> f1_codecs = MAKE_VECTOR(kVideoCodecs1); |
// This creates RTX for H264-SVC in sender. |
- AddRtxCodec(VideoCodec::CreateRtxCodec(125, kVideoCodecs1[0].id), &f1_codecs); |
+ AddSecondaryCodec(VideoCodec::CreateRtxCodec(125, kVideoCodecs1[0].id), |
+ &f1_codecs); |
f1_.set_video_codecs(f1_codecs); |
// This creates RTX for H264 in sender. |
- AddRtxCodec(VideoCodec::CreateRtxCodec(126, kVideoCodecs1[1].id), &f1_codecs); |
+ AddSecondaryCodec(VideoCodec::CreateRtxCodec(126, kVideoCodecs1[1].id), |
+ &f1_codecs); |
f1_.set_video_codecs(f1_codecs); |
std::vector<VideoCodec> f2_codecs = MAKE_VECTOR(kVideoCodecs2); |
// This creates RTX for H264 in receiver. |
- AddRtxCodec(VideoCodec::CreateRtxCodec(124, kVideoCodecs2[0].id), &f2_codecs); |
+ AddSecondaryCodec(VideoCodec::CreateRtxCodec(124, kVideoCodecs2[0].id), |
+ &f2_codecs); |
f2_.set_video_codecs(f2_codecs); |
// H264-SVC codec is removed in the answer, therefore, associated RTX codec |
@@ -1933,8 +1942,8 @@ TEST_F(MediaSessionDescriptionFactoryTest, |
const VideoContentDescription* vcd = |
GetFirstVideoContentDescription(answer.get()); |
std::vector<VideoCodec> expected_codecs = MAKE_VECTOR(kVideoCodecsAnswer); |
- AddRtxCodec(VideoCodec::CreateRtxCodec(126, kVideoCodecs1[1].id), |
- &expected_codecs); |
+ AddSecondaryCodec(VideoCodec::CreateRtxCodec(126, kVideoCodecs1[1].id), |
+ &expected_codecs); |
EXPECT_EQ(expected_codecs, vcd->codecs()); |
} |
@@ -1947,7 +1956,8 @@ TEST_F(MediaSessionDescriptionFactoryTest, AddSecondRtxInNewOffer) { |
opts.recv_audio = false; |
std::vector<VideoCodec> f1_codecs = MAKE_VECTOR(kVideoCodecs1); |
// This creates RTX for H264 for the offerer. |
- AddRtxCodec(VideoCodec::CreateRtxCodec(126, kVideoCodecs1[1].id), &f1_codecs); |
+ AddSecondaryCodec(VideoCodec::CreateRtxCodec(126, kVideoCodecs1[1].id), |
+ &f1_codecs); |
f1_.set_video_codecs(f1_codecs); |
std::unique_ptr<SessionDescription> offer(f1_.CreateOffer(opts, nullptr)); |
@@ -1956,12 +1966,13 @@ TEST_F(MediaSessionDescriptionFactoryTest, AddSecondRtxInNewOffer) { |
GetFirstVideoContentDescription(offer.get()); |
std::vector<VideoCodec> expected_codecs = MAKE_VECTOR(kVideoCodecs1); |
- AddRtxCodec(VideoCodec::CreateRtxCodec(126, kVideoCodecs1[1].id), |
- &expected_codecs); |
+ AddSecondaryCodec(VideoCodec::CreateRtxCodec(126, kVideoCodecs1[1].id), |
+ &expected_codecs); |
EXPECT_EQ(expected_codecs, vcd->codecs()); |
// Now, attempt to add RTX for H264-SVC. |
- AddRtxCodec(VideoCodec::CreateRtxCodec(125, kVideoCodecs1[0].id), &f1_codecs); |
+ AddSecondaryCodec(VideoCodec::CreateRtxCodec(125, kVideoCodecs1[0].id), |
+ &f1_codecs); |
f1_.set_video_codecs(f1_codecs); |
std::unique_ptr<SessionDescription> updated_offer( |
@@ -1969,8 +1980,8 @@ TEST_F(MediaSessionDescriptionFactoryTest, AddSecondRtxInNewOffer) { |
ASSERT_TRUE(updated_offer); |
vcd = GetFirstVideoContentDescription(updated_offer.get()); |
- AddRtxCodec(VideoCodec::CreateRtxCodec(125, kVideoCodecs1[0].id), |
- &expected_codecs); |
+ AddSecondaryCodec(VideoCodec::CreateRtxCodec(125, kVideoCodecs1[0].id), |
+ &expected_codecs); |
EXPECT_EQ(expected_codecs, vcd->codecs()); |
} |
@@ -1987,7 +1998,7 @@ TEST_F(MediaSessionDescriptionFactoryTest, SimSsrcsGenerateMultipleRtxSsrcs) { |
// Use a single real codec, and then add RTX for it. |
std::vector<VideoCodec> f1_codecs; |
f1_codecs.push_back(VideoCodec(97, "H264")); |
- AddRtxCodec(VideoCodec::CreateRtxCodec(125, 97), &f1_codecs); |
+ AddSecondaryCodec(VideoCodec::CreateRtxCodec(125, 97), &f1_codecs); |
f1_.set_video_codecs(f1_codecs); |
// Ensure that the offer has an RTX ssrc for each regular ssrc, and that there |
@@ -2015,6 +2026,88 @@ TEST_F(MediaSessionDescriptionFactoryTest, SimSsrcsGenerateMultipleRtxSsrcs) { |
EXPECT_EQ(3u, fid_ssrcs.size()); |
} |
+// Test that, when the FlexFEC codec is added, a FlexFEC ssrc is created |
+// together with a FEC-FR grouping. |
+TEST_F(MediaSessionDescriptionFactoryTest, GenerateFlexfecSsrc) { |
+ MediaSessionOptions opts; |
+ opts.recv_video = true; |
+ opts.recv_audio = false; |
+ |
+ // Add single stream. |
+ opts.AddSendVideoStream("stream1", "stream1label", 1); |
+ |
+ // Use a single real codec, and then add FlexFEC for it. |
+ std::vector<VideoCodec> f1_codecs; |
+ f1_codecs.push_back(VideoCodec(97, "H264")); |
+ AddSecondaryCodec(VideoCodec(118, "flexfec-03"), &f1_codecs); |
+ f1_.set_video_codecs(f1_codecs); |
+ |
+ // Ensure that the offer has a single FlexFEC ssrc and that |
+ // there is no FEC-FR ssrc + grouping for each. |
+ std::unique_ptr<SessionDescription> offer(f1_.CreateOffer(opts, nullptr)); |
+ ASSERT_TRUE(offer.get() != nullptr); |
+ VideoContentDescription* desc = static_cast<VideoContentDescription*>( |
+ offer->GetContentDescriptionByName(cricket::CN_VIDEO)); |
+ ASSERT_TRUE(desc != nullptr); |
+ EXPECT_TRUE(desc->multistream()); |
+ const StreamParamsVec& streams = desc->streams(); |
+ // Single stream. |
+ ASSERT_EQ(1u, streams.size()); |
+ // Stream should have 2 ssrcs: 1 for video, 1 for FlexFEC. |
+ EXPECT_EQ(2u, streams[0].ssrcs.size()); |
+ // And should a FEC-FR group for FlexFEC. |
perkj_webrtc
2016/11/17 20:41:12
and should have
brandtr
2016/11/21 09:16:30
Done.
|
+ EXPECT_TRUE(streams[0].has_ssrc_group("FEC-FR")); |
+ std::vector<uint32_t> primary_ssrcs; |
+ streams[0].GetPrimarySsrcs(&primary_ssrcs); |
+ ASSERT_EQ(1u, primary_ssrcs.size()); |
+ uint32_t flexfec_ssrc; |
+ EXPECT_TRUE(streams[0].GetFecFrSsrc(primary_ssrcs[0], &flexfec_ssrc)); |
+ EXPECT_NE(flexfec_ssrc, 0u); |
+} |
+ |
+// Test that FlexFEC is disabled for simulcast. |
+// TODO(brandtr): Remove this test when we support simulcast, either through |
+// multiple FlexfecSenders, or through multistream protection. |
+TEST_F(MediaSessionDescriptionFactoryTest, SimSsrcsGenerateNoFlexfecSsrcs) { |
+ MediaSessionOptions opts; |
+ opts.recv_video = true; |
+ opts.recv_audio = false; |
+ |
+ // Add simulcast streams. |
+ opts.AddSendVideoStream("stream1", "stream1label", 3); |
+ |
+ // Use a single real codec, and then add FlexFEC for it. |
+ std::vector<VideoCodec> f1_codecs; |
+ f1_codecs.push_back(VideoCodec(97, "H264")); |
+ AddSecondaryCodec(VideoCodec(118, "flexfec-03"), &f1_codecs); |
+ f1_.set_video_codecs(f1_codecs); |
+ |
+ // Ensure that the offer has no FlexFEC ssrcs for each regular ssrc, and that |
+ // there is no FEC-FR ssrc + grouping for each. |
+ std::unique_ptr<SessionDescription> offer(f1_.CreateOffer(opts, nullptr)); |
+ ASSERT_TRUE(offer.get() != nullptr); |
+ VideoContentDescription* desc = static_cast<VideoContentDescription*>( |
+ offer->GetContentDescriptionByName(cricket::CN_VIDEO)); |
+ ASSERT_TRUE(desc != nullptr); |
+ EXPECT_FALSE(desc->multistream()); |
+ const StreamParamsVec& streams = desc->streams(); |
+ // Single stream. |
+ ASSERT_EQ(1u, streams.size()); |
+ // Stream should have 3 ssrcs: 3 for video, 0 for FlexFEC. |
+ EXPECT_EQ(3u, streams[0].ssrcs.size()); |
+ // And should have a SIM group for the simulcast. |
+ EXPECT_TRUE(streams[0].has_ssrc_group("SIM")); |
+ // And not a FEC-FR group for FlexFEC. |
+ EXPECT_FALSE(streams[0].has_ssrc_group("FEC-FR")); |
+ std::vector<uint32_t> primary_ssrcs; |
+ streams[0].GetPrimarySsrcs(&primary_ssrcs); |
+ EXPECT_EQ(3u, primary_ssrcs.size()); |
+ for (uint32_t primary_ssrc : primary_ssrcs) { |
+ uint32_t flexfec_ssrc; |
+ EXPECT_FALSE(streams[0].GetFecFrSsrc(primary_ssrc, &flexfec_ssrc)); |
+ } |
+} |
+ |
// Create an updated offer after creating an answer to the original offer and |
// verify that the RTP header extensions that were part of the original answer |
// are not changed in the updated offer. |
@@ -2862,7 +2955,8 @@ void TestAudioCodecsAnswer(MediaContentDirection offer_direction, |
<< "Only inactive offers are allowed to not generate any audio content"; |
} |
} |
-} |
+ |
+} // namespace |
class AudioCodecsOfferTest |
: public ::testing::TestWithParam<std::tr1::tuple<MediaContentDirection, |