Index: webrtc/modules/audio_coding/neteq/decoder_database_unittest.cc |
diff --git a/webrtc/modules/audio_coding/neteq/decoder_database_unittest.cc b/webrtc/modules/audio_coding/neteq/decoder_database_unittest.cc |
index e85d8d32fb8da439a3aa2b1a48041553486181f9..f41a17ae3a8ec5164d6e7e1d2fde395de791cee7 100644 |
--- a/webrtc/modules/audio_coding/neteq/decoder_database_unittest.cc |
+++ b/webrtc/modules/audio_coding/neteq/decoder_database_unittest.cc |
@@ -32,8 +32,10 @@ TEST(DecoderDatabase, CreateAndDestroy) { |
TEST(DecoderDatabase, InsertAndRemove) { |
DecoderDatabase db; |
const uint8_t kPayloadType = 0; |
- EXPECT_EQ(DecoderDatabase::kOK, |
- db.RegisterPayload(kPayloadType, NetEqDecoder::kDecoderPCMu)); |
+ const std::string kCodecName = "Robert\'); DROP TABLE Students;"; |
+ EXPECT_EQ( |
+ DecoderDatabase::kOK, |
+ db.RegisterPayload(kPayloadType, NetEqDecoder::kDecoderPCMu, kCodecName)); |
EXPECT_EQ(1, db.Size()); |
EXPECT_FALSE(db.Empty()); |
EXPECT_EQ(DecoderDatabase::kOK, db.Remove(kPayloadType)); |
@@ -44,14 +46,17 @@ TEST(DecoderDatabase, InsertAndRemove) { |
TEST(DecoderDatabase, GetDecoderInfo) { |
DecoderDatabase db; |
const uint8_t kPayloadType = 0; |
- EXPECT_EQ(DecoderDatabase::kOK, |
- db.RegisterPayload(kPayloadType, NetEqDecoder::kDecoderPCMu)); |
+ const std::string kCodecName = "Robert\'); DROP TABLE Students;"; |
+ EXPECT_EQ( |
+ DecoderDatabase::kOK, |
+ db.RegisterPayload(kPayloadType, NetEqDecoder::kDecoderPCMu, kCodecName)); |
const DecoderDatabase::DecoderInfo* info; |
info = db.GetDecoderInfo(kPayloadType); |
ASSERT_TRUE(info != NULL); |
EXPECT_EQ(NetEqDecoder::kDecoderPCMu, info->codec_type); |
EXPECT_EQ(NULL, info->decoder); |
EXPECT_EQ(8000, info->fs_hz); |
+ EXPECT_EQ(kCodecName, info->name); |
EXPECT_FALSE(info->external); |
info = db.GetDecoderInfo(kPayloadType + 1); // Other payload type. |
EXPECT_TRUE(info == NULL); // Should not be found. |
@@ -60,8 +65,10 @@ TEST(DecoderDatabase, GetDecoderInfo) { |
TEST(DecoderDatabase, GetRtpPayloadType) { |
DecoderDatabase db; |
const uint8_t kPayloadType = 0; |
- EXPECT_EQ(DecoderDatabase::kOK, |
- db.RegisterPayload(kPayloadType, NetEqDecoder::kDecoderPCMu)); |
+ const std::string kCodecName = "Robert\'); DROP TABLE Students;"; |
+ EXPECT_EQ( |
+ DecoderDatabase::kOK, |
+ db.RegisterPayload(kPayloadType, NetEqDecoder::kDecoderPCMu, kCodecName)); |
EXPECT_EQ(kPayloadType, db.GetRtpPayloadType(NetEqDecoder::kDecoderPCMu)); |
const uint8_t expected_value = DecoderDatabase::kRtpPayloadTypeError; |
EXPECT_EQ(expected_value, |
@@ -72,8 +79,10 @@ TEST(DecoderDatabase, GetRtpPayloadType) { |
TEST(DecoderDatabase, GetDecoder) { |
DecoderDatabase db; |
const uint8_t kPayloadType = 0; |
+ const std::string kCodecName = "Robert\'); DROP TABLE Students;"; |
EXPECT_EQ(DecoderDatabase::kOK, |
- db.RegisterPayload(kPayloadType, NetEqDecoder::kDecoderPCM16B)); |
+ db.RegisterPayload(kPayloadType, NetEqDecoder::kDecoderPCM16B, |
+ kCodecName)); |
AudioDecoder* dec = db.GetDecoder(kPayloadType); |
ASSERT_TRUE(dec != NULL); |
} |
@@ -86,14 +95,18 @@ TEST(DecoderDatabase, TypeTests) { |
const uint8_t kPayloadTypeRed = 101; |
const uint8_t kPayloadNotUsed = 102; |
// Load into database. |
+ EXPECT_EQ( |
+ DecoderDatabase::kOK, |
+ db.RegisterPayload(kPayloadTypePcmU, NetEqDecoder::kDecoderPCMu, "pcmu")); |
EXPECT_EQ(DecoderDatabase::kOK, |
- db.RegisterPayload(kPayloadTypePcmU, NetEqDecoder::kDecoderPCMu)); |
- EXPECT_EQ(DecoderDatabase::kOK, |
- db.RegisterPayload(kPayloadTypeCng, NetEqDecoder::kDecoderCNGnb)); |
- EXPECT_EQ(DecoderDatabase::kOK, |
- db.RegisterPayload(kPayloadTypeDtmf, NetEqDecoder::kDecoderAVT)); |
- EXPECT_EQ(DecoderDatabase::kOK, |
- db.RegisterPayload(kPayloadTypeRed, NetEqDecoder::kDecoderRED)); |
+ db.RegisterPayload(kPayloadTypeCng, NetEqDecoder::kDecoderCNGnb, |
+ "cng-nb")); |
+ EXPECT_EQ( |
+ DecoderDatabase::kOK, |
+ db.RegisterPayload(kPayloadTypeDtmf, NetEqDecoder::kDecoderAVT, "avt")); |
+ EXPECT_EQ( |
+ DecoderDatabase::kOK, |
+ db.RegisterPayload(kPayloadTypeRed, NetEqDecoder::kDecoderRED, "red")); |
EXPECT_EQ(4, db.Size()); |
// Test. |
EXPECT_FALSE(db.IsComfortNoise(kPayloadNotUsed)); |
@@ -112,11 +125,12 @@ TEST(DecoderDatabase, TypeTests) { |
TEST(DecoderDatabase, ExternalDecoder) { |
DecoderDatabase db; |
const uint8_t kPayloadType = 0; |
+ const std::string kCodecName = "Robert\'); DROP TABLE Students;"; |
MockAudioDecoder decoder; |
// Load into database. |
EXPECT_EQ(DecoderDatabase::kOK, |
- db.InsertExternal(kPayloadType, NetEqDecoder::kDecoderPCMu, 8000, |
- &decoder)); |
+ db.InsertExternal(kPayloadType, NetEqDecoder::kDecoderPCMu, |
+ kCodecName, 8000, &decoder)); |
EXPECT_EQ(1, db.Size()); |
// Get decoder and make sure we get the external one. |
EXPECT_EQ(&decoder, db.GetDecoder(kPayloadType)); |
@@ -125,6 +139,7 @@ TEST(DecoderDatabase, ExternalDecoder) { |
info = db.GetDecoderInfo(kPayloadType); |
ASSERT_TRUE(info != NULL); |
EXPECT_EQ(NetEqDecoder::kDecoderPCMu, info->codec_type); |
+ EXPECT_EQ(kCodecName, info->name); |
EXPECT_EQ(&decoder, info->decoder); |
EXPECT_EQ(8000, info->fs_hz); |
EXPECT_TRUE(info->external); |
@@ -146,7 +161,7 @@ TEST(DecoderDatabase, CheckPayloadTypes) { |
for (uint8_t payload_type = 0; payload_type < kNumPayloads; ++payload_type) { |
EXPECT_EQ( |
DecoderDatabase::kOK, |
- db.RegisterPayload(payload_type, NetEqDecoder::kDecoderArbitrary)); |
+ db.RegisterPayload(payload_type, NetEqDecoder::kDecoderArbitrary, "")); |
} |
PacketList packet_list; |
for (int i = 0; i < kNumPayloads + 1; ++i) { |
@@ -185,11 +200,11 @@ TEST(DecoderDatabase, IF_ISAC(ActiveDecoders)) { |
DecoderDatabase db; |
// Load payload types. |
ASSERT_EQ(DecoderDatabase::kOK, |
- db.RegisterPayload(0, NetEqDecoder::kDecoderPCMu)); |
+ db.RegisterPayload(0, NetEqDecoder::kDecoderPCMu, "pcmu")); |
ASSERT_EQ(DecoderDatabase::kOK, |
- db.RegisterPayload(103, NetEqDecoder::kDecoderISAC)); |
+ db.RegisterPayload(103, NetEqDecoder::kDecoderISAC, "isac")); |
ASSERT_EQ(DecoderDatabase::kOK, |
- db.RegisterPayload(13, NetEqDecoder::kDecoderCNGnb)); |
+ db.RegisterPayload(13, NetEqDecoder::kDecoderCNGnb, "cng-nb")); |
// Verify that no decoders are active from the start. |
EXPECT_EQ(NULL, db.GetActiveDecoder()); |
EXPECT_EQ(NULL, db.GetActiveCngDecoder()); |