Index: webrtc/modules/rtp_rtcp/source/rtp_payload_registry_unittest.cc |
diff --git a/webrtc/modules/rtp_rtcp/source/rtp_payload_registry_unittest.cc b/webrtc/modules/rtp_rtcp/source/rtp_payload_registry_unittest.cc |
index abb2ae91c4eedbbfca564949ac6f2ee5bcab946b..9e8f4c44e82087c4bd771ac1c0fab2d36151ba37 100644 |
--- a/webrtc/modules/rtp_rtcp/source/rtp_payload_registry_unittest.cc |
+++ b/webrtc/modules/rtp_rtcp/source/rtp_payload_registry_unittest.cc |
@@ -35,6 +35,11 @@ class RtpPayloadRegistryTest : public ::testing::Test { |
// Note: the payload registry takes ownership of the strategy. |
mock_payload_strategy_ = new testing::NiceMock<MockRTPPayloadStrategy>(); |
rtp_payload_registry_.reset(new RTPPayloadRegistry(mock_payload_strategy_)); |
+ |
+ strcpy(typical_audio_codec_.plname, kTypicalPayloadName); |
+ typical_audio_codec_.plfreq = kTypicalFrequency; |
+ typical_audio_codec_.channels = kTypicalChannels; |
+ typical_audio_codec_.rate = kTypicalRate; |
} |
protected: |
@@ -51,15 +56,17 @@ class RtpPayloadRegistryTest : public ::testing::Test { |
// of the created object. |
RtpUtility::Payload* returned_payload_on_heap = |
new RtpUtility::Payload(returned_payload); |
- EXPECT_CALL(*mock_payload_strategy_, |
- CreatePayloadType(kTypicalPayloadName, payload_type, |
- kTypicalFrequency, kTypicalChannels, rate)) |
+ EXPECT_CALL( |
+ *mock_payload_strategy_, |
+ CreatePayloadType(testing::StrEq(kTypicalPayloadName), payload_type, |
danilchap
2016/11/23 15:19:23
probably nicer to add using ::testing::StrEq above
magjed_webrtc
2016/11/23 16:35:53
Done.
|
+ kTypicalFrequency, kTypicalChannels, rate)) |
.WillOnce(Return(returned_payload_on_heap)); |
return returned_payload_on_heap; |
} |
std::unique_ptr<RTPPayloadRegistry> rtp_payload_registry_; |
testing::NiceMock<MockRTPPayloadStrategy>* mock_payload_strategy_; |
+ CodecInst typical_audio_codec_; |
danilchap
2016/11/23 15:19:23
may be use constant outside of the fixture instead
magjed_webrtc
2016/11/23 16:35:53
Thanks, much better. I tried to do this but used k
|
}; |
TEST_F(RtpPayloadRegistryTest, RegistersAndRemembersPayloadsUntilDeregistered) { |
@@ -68,9 +75,10 @@ TEST_F(RtpPayloadRegistryTest, RegistersAndRemembersPayloadsUntilDeregistered) { |
ExpectReturnOfTypicalAudioPayload(payload_type, kTypicalRate); |
bool new_payload_created = false; |
+ CodecInst audio_codec = typical_audio_codec_; |
+ audio_codec.pltype = payload_type; |
EXPECT_EQ(0, rtp_payload_registry_->RegisterReceivePayload( |
- kTypicalPayloadName, payload_type, kTypicalFrequency, |
- kTypicalChannels, kTypicalRate, &new_payload_created)); |
+ audio_codec, &new_payload_created)); |
EXPECT_TRUE(new_payload_created) << "A new payload WAS created."; |
@@ -98,9 +106,14 @@ TEST_F(RtpPayloadRegistryTest, AudioRedWorkProperly) { |
new RTPPayloadRegistry(RTPPayloadStrategy::CreateStrategy(true))); |
bool new_payload_created = false; |
+ CodecInst red_audio_codec; |
+ strcpy(red_audio_codec.plname, "red"); |
+ red_audio_codec.pltype = kRedPayloadType; |
+ red_audio_codec.plfreq = kRedSampleRate; |
+ red_audio_codec.channels = kRedChannels; |
+ red_audio_codec.rate = kRedBitRate; |
EXPECT_EQ(0, rtp_payload_registry_->RegisterReceivePayload( |
- "red", kRedPayloadType, kRedSampleRate, kRedChannels, |
- kRedBitRate, &new_payload_created)); |
+ red_audio_codec, &new_payload_created)); |
EXPECT_TRUE(new_payload_created); |
EXPECT_EQ(kRedPayloadType, rtp_payload_registry_->red_payload_type()); |
@@ -123,20 +136,21 @@ TEST_F(RtpPayloadRegistryTest, |
bool ignored = false; |
RtpUtility::Payload* first_payload_on_heap = |
ExpectReturnOfTypicalAudioPayload(payload_type, kTypicalRate); |
- EXPECT_EQ(0, rtp_payload_registry_->RegisterReceivePayload( |
- kTypicalPayloadName, payload_type, kTypicalFrequency, |
- kTypicalChannels, kTypicalRate, &ignored)); |
+ CodecInst audio_codec = typical_audio_codec_; |
+ audio_codec.pltype = payload_type; |
+ EXPECT_EQ( |
+ 0, rtp_payload_registry_->RegisterReceivePayload(audio_codec, &ignored)); |
- EXPECT_EQ(-1, rtp_payload_registry_->RegisterReceivePayload( |
- kTypicalPayloadName, payload_type, kTypicalFrequency, |
- kTypicalChannels, kTypicalRate, &ignored)) |
+ EXPECT_EQ( |
+ -1, rtp_payload_registry_->RegisterReceivePayload(audio_codec, &ignored)) |
<< "Adding same codec twice = bad."; |
RtpUtility::Payload* second_payload_on_heap = |
ExpectReturnOfTypicalAudioPayload(payload_type - 1, kTypicalRate); |
- EXPECT_EQ(0, rtp_payload_registry_->RegisterReceivePayload( |
- kTypicalPayloadName, payload_type - 1, kTypicalFrequency, |
- kTypicalChannels, kTypicalRate, &ignored)) |
+ CodecInst audio_codec_2 = typical_audio_codec_; |
+ audio_codec_2.pltype = payload_type - 1; |
+ EXPECT_EQ( |
+ 0, rtp_payload_registry_->RegisterReceivePayload(audio_codec_2, &ignored)) |
<< "With a different payload type is fine though."; |
// Ensure both payloads are preserved. |
@@ -156,9 +170,8 @@ TEST_F(RtpPayloadRegistryTest, |
.WillByDefault(Return(true)); |
EXPECT_CALL(*mock_payload_strategy_, |
UpdatePayloadRate(first_payload_on_heap, kTypicalRate)); |
- EXPECT_EQ(0, rtp_payload_registry_->RegisterReceivePayload( |
- kTypicalPayloadName, payload_type, kTypicalFrequency, |
- kTypicalChannels, kTypicalRate, &ignored)); |
+ EXPECT_EQ( |
+ 0, rtp_payload_registry_->RegisterReceivePayload(audio_codec, &ignored)); |
} |
TEST_F(RtpPayloadRegistryTest, |
@@ -172,13 +185,15 @@ TEST_F(RtpPayloadRegistryTest, |
bool ignored = false; |
ExpectReturnOfTypicalAudioPayload(payload_type, kTypicalRate); |
- EXPECT_EQ(0, rtp_payload_registry_->RegisterReceivePayload( |
- kTypicalPayloadName, payload_type, kTypicalFrequency, |
- kTypicalChannels, kTypicalRate, &ignored)); |
+ CodecInst audio_codec = typical_audio_codec_; |
+ audio_codec.pltype = payload_type; |
+ EXPECT_EQ( |
+ 0, rtp_payload_registry_->RegisterReceivePayload(audio_codec, &ignored)); |
ExpectReturnOfTypicalAudioPayload(payload_type - 1, kTypicalRate); |
- EXPECT_EQ(0, rtp_payload_registry_->RegisterReceivePayload( |
- kTypicalPayloadName, payload_type - 1, kTypicalFrequency, |
- kTypicalChannels, kTypicalRate, &ignored)); |
+ CodecInst audio_codec_2 = typical_audio_codec_; |
+ audio_codec_2.pltype = payload_type - 1; |
+ EXPECT_EQ(0, rtp_payload_registry_->RegisterReceivePayload(audio_codec_2, |
+ &ignored)); |
EXPECT_FALSE(rtp_payload_registry_->PayloadTypeToPayload(payload_type)) |
<< "The first payload should be " |
@@ -190,9 +205,10 @@ TEST_F(RtpPayloadRegistryTest, |
ON_CALL(*mock_payload_strategy_, PayloadIsCompatible(_, _, _, _)) |
.WillByDefault(Return(false)); |
ExpectReturnOfTypicalAudioPayload(payload_type + 1, kTypicalRate); |
- EXPECT_EQ(0, rtp_payload_registry_->RegisterReceivePayload( |
- kTypicalPayloadName, payload_type + 1, kTypicalFrequency, |
- kTypicalChannels, kTypicalRate, &ignored)); |
+ CodecInst audio_codec_3 = typical_audio_codec_; |
+ audio_codec_3.pltype = payload_type + 1; |
+ EXPECT_EQ(0, rtp_payload_registry_->RegisterReceivePayload(audio_codec_3, |
+ &ignored)); |
EXPECT_TRUE(rtp_payload_registry_->PayloadTypeToPayload(payload_type - 1)) |
<< "Not compatible; both payloads should be kept."; |
@@ -212,9 +228,10 @@ TEST_F(RtpPayloadRegistryTest, |
bool ignored; |
ExpectReturnOfTypicalAudioPayload(34, kTypicalRate); |
- EXPECT_EQ(0, rtp_payload_registry_->RegisterReceivePayload( |
- kTypicalPayloadName, 34, kTypicalFrequency, kTypicalChannels, |
- kTypicalRate, &ignored)); |
+ CodecInst audio_codec = typical_audio_codec_; |
+ audio_codec.pltype = 34; |
+ EXPECT_EQ( |
+ 0, rtp_payload_registry_->RegisterReceivePayload(audio_codec, &ignored)); |
EXPECT_EQ(-1, rtp_payload_registry_->last_received_payload_type()); |
media_type_unchanged = rtp_payload_registry_->ReportMediaPayloadType(18); |
@@ -230,9 +247,14 @@ TEST_P(ParameterizedRtpPayloadRegistryTest, |
int payload_type = GetParam(); |
bool ignored; |
- EXPECT_EQ(-1, rtp_payload_registry_->RegisterReceivePayload( |
- "whatever", static_cast<uint8_t>(payload_type), 19, 1, 17, |
- &ignored)); |
+ CodecInst audio_codec; |
+ strcpy(audio_codec.plname, "whatever"); |
+ audio_codec.pltype = static_cast<uint8_t>(payload_type); |
+ audio_codec.plfreq = 19; |
+ audio_codec.channels = 1; |
+ audio_codec.rate = 17; |
+ EXPECT_EQ( |
+ -1, rtp_payload_registry_->RegisterReceivePayload(audio_codec, &ignored)); |
} |
INSTANTIATE_TEST_CASE_P(TestKnownBadPayloadTypes, |
@@ -248,9 +270,15 @@ TEST_P(RtpPayloadRegistryGenericTest, RegisterGenericReceivePayloadType) { |
bool ignored; |
- EXPECT_EQ(0, rtp_payload_registry_->RegisterReceivePayload( |
- "generic-codec", static_cast<int8_t>(payload_type), 19, 1, |
- 17, &ignored)); // dummy values, except for payload_type |
+ CodecInst audio_codec; |
+ strcpy(audio_codec.plname, "generic-codec"); |
+ audio_codec.pltype = static_cast<uint8_t>(payload_type); |
+ audio_codec.plfreq = 19; |
+ audio_codec.channels = 1; |
+ audio_codec.rate = 17; |
+ EXPECT_EQ( |
+ 0, rtp_payload_registry_->RegisterReceivePayload( |
+ audio_codec, &ignored)); // dummy values, except for payload_type |
danilchap
2016/11/23 15:19:23
May be fix comment as well: add dot in the end, st
magjed_webrtc
2016/11/23 16:35:53
Done.
|
} |
// Generates an RTX packet for the given length and original sequence number. |