solenberg | c7a8b08 | 2015-10-16 14:35:07 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (c) 2015 The WebRTC project authors. All Rights Reserved. |
| 3 | * |
| 4 | * Use of this source code is governed by a BSD-style license |
| 5 | * that can be found in the LICENSE file in the root of the source |
| 6 | * tree. An additional intellectual property rights grant can be found |
| 7 | * in the file PATENTS. All contributing project authors may |
| 8 | * be found in the AUTHORS file in the root of the source tree. |
| 9 | */ |
| 10 | |
| 11 | #include "testing/gtest/include/gtest/gtest.h" |
| 12 | |
| 13 | #include "webrtc/audio/audio_send_stream.h" |
solenberg | 566ef24 | 2015-11-06 15:34:49 -0800 | [diff] [blame] | 14 | #include "webrtc/audio/audio_state.h" |
solenberg | 85a0496 | 2015-10-27 03:35:21 -0700 | [diff] [blame] | 15 | #include "webrtc/audio/conversion.h" |
Fredrik Solenberg | 0ccae13 | 2015-11-03 10:15:49 +0100 | [diff] [blame] | 16 | #include "webrtc/test/mock_voice_engine.h" |
solenberg | c7a8b08 | 2015-10-16 14:35:07 -0700 | [diff] [blame] | 17 | |
| 18 | namespace webrtc { |
solenberg | 85a0496 | 2015-10-27 03:35:21 -0700 | [diff] [blame] | 19 | namespace test { |
Fredrik Solenberg | 0ccae13 | 2015-11-03 10:15:49 +0100 | [diff] [blame] | 20 | namespace { |
| 21 | |
solenberg | 3a94154 | 2015-11-16 07:34:50 -0800 | [diff] [blame^] | 22 | using testing::_; |
| 23 | using testing::Return; |
| 24 | |
Fredrik Solenberg | 0ccae13 | 2015-11-03 10:15:49 +0100 | [diff] [blame] | 25 | const int kChannelId = 1; |
| 26 | const uint32_t kSsrc = 1234; |
solenberg | 3a94154 | 2015-11-16 07:34:50 -0800 | [diff] [blame^] | 27 | const char* kCName = "foo_name"; |
| 28 | const int kAudioLevelId = 2; |
| 29 | const int kAbsSendTimeId = 3; |
solenberg | 566ef24 | 2015-11-06 15:34:49 -0800 | [diff] [blame] | 30 | const int kEchoDelayMedian = 254; |
| 31 | const int kEchoDelayStdDev = -3; |
| 32 | const int kEchoReturnLoss = -65; |
| 33 | const int kEchoReturnLossEnhancement = 101; |
| 34 | const unsigned int kSpeechInputLevel = 96; |
| 35 | const CallStatistics kCallStats = { |
| 36 | 1345, 1678, 1901, 1234, 112, 13456, 17890, 1567, -1890, -1123}; |
| 37 | const CodecInst kCodecInst = {-121, "codec_name_send", 48000, -231, -451, -671}; |
| 38 | const ReportBlock kReportBlock = {456, 780, 123, 567, 890, 132, 143, 13354}; |
| 39 | |
| 40 | struct ConfigHelper { |
| 41 | ConfigHelper() : stream_config_(nullptr) { |
solenberg | 3a94154 | 2015-11-16 07:34:50 -0800 | [diff] [blame^] | 42 | using testing::StrEq; |
| 43 | |
solenberg | 566ef24 | 2015-11-06 15:34:49 -0800 | [diff] [blame] | 44 | EXPECT_CALL(voice_engine_, |
solenberg | 3a94154 | 2015-11-16 07:34:50 -0800 | [diff] [blame^] | 45 | RegisterVoiceEngineObserver(_)).WillOnce(Return(0)); |
solenberg | 566ef24 | 2015-11-06 15:34:49 -0800 | [diff] [blame] | 46 | EXPECT_CALL(voice_engine_, |
solenberg | 3a94154 | 2015-11-16 07:34:50 -0800 | [diff] [blame^] | 47 | DeRegisterVoiceEngineObserver()).WillOnce(Return(0)); |
solenberg | 566ef24 | 2015-11-06 15:34:49 -0800 | [diff] [blame] | 48 | AudioState::Config config; |
| 49 | config.voice_engine = &voice_engine_; |
| 50 | audio_state_ = AudioState::Create(config); |
solenberg | 3a94154 | 2015-11-16 07:34:50 -0800 | [diff] [blame^] | 51 | |
| 52 | EXPECT_CALL(voice_engine_, SetRTCPStatus(kChannelId, true)) |
| 53 | .WillOnce(Return(0)); |
| 54 | EXPECT_CALL(voice_engine_, SetLocalSSRC(kChannelId, kSsrc)) |
| 55 | .WillOnce(Return(0)); |
| 56 | EXPECT_CALL(voice_engine_, SetRTCP_CNAME(kChannelId, StrEq(kCName))) |
| 57 | .WillOnce(Return(0)); |
| 58 | EXPECT_CALL(voice_engine_, |
| 59 | SetSendAbsoluteSenderTimeStatus(kChannelId, true, kAbsSendTimeId)) |
| 60 | .WillOnce(Return(0)); |
| 61 | EXPECT_CALL(voice_engine_, |
| 62 | SetSendAudioLevelIndicationStatus(kChannelId, true, kAudioLevelId)) |
| 63 | .WillOnce(Return(0)); |
solenberg | 566ef24 | 2015-11-06 15:34:49 -0800 | [diff] [blame] | 64 | stream_config_.voe_channel_id = kChannelId; |
| 65 | stream_config_.rtp.ssrc = kSsrc; |
solenberg | 3a94154 | 2015-11-16 07:34:50 -0800 | [diff] [blame^] | 66 | stream_config_.rtp.c_name = kCName; |
| 67 | stream_config_.rtp.extensions.push_back( |
| 68 | RtpExtension(RtpExtension::kAudioLevel, kAudioLevelId)); |
| 69 | stream_config_.rtp.extensions.push_back( |
| 70 | RtpExtension(RtpExtension::kAbsSendTime, kAbsSendTimeId)); |
solenberg | 566ef24 | 2015-11-06 15:34:49 -0800 | [diff] [blame] | 71 | } |
| 72 | |
| 73 | AudioSendStream::Config& config() { return stream_config_; } |
| 74 | rtc::scoped_refptr<AudioState> audio_state() { return audio_state_; } |
| 75 | |
| 76 | void SetupMockForGetStats() { |
solenberg | 3a94154 | 2015-11-16 07:34:50 -0800 | [diff] [blame^] | 77 | using testing::DoAll; |
| 78 | using testing::SetArgPointee; |
| 79 | using testing::SetArgReferee; |
| 80 | |
solenberg | 566ef24 | 2015-11-06 15:34:49 -0800 | [diff] [blame] | 81 | std::vector<ReportBlock> report_blocks; |
| 82 | webrtc::ReportBlock block = kReportBlock; |
| 83 | report_blocks.push_back(block); // Has wrong SSRC. |
| 84 | block.source_SSRC = kSsrc; |
| 85 | report_blocks.push_back(block); // Correct block. |
| 86 | block.fraction_lost = 0; |
| 87 | report_blocks.push_back(block); // Duplicate SSRC, bad fraction_lost. |
| 88 | |
solenberg | 566ef24 | 2015-11-06 15:34:49 -0800 | [diff] [blame] | 89 | EXPECT_CALL(voice_engine_, GetLocalSSRC(kChannelId, _)) |
| 90 | .WillRepeatedly(DoAll(SetArgReferee<1>(0), Return(0))); |
| 91 | EXPECT_CALL(voice_engine_, GetRTCPStatistics(kChannelId, _)) |
| 92 | .WillRepeatedly(DoAll(SetArgReferee<1>(kCallStats), Return(0))); |
| 93 | EXPECT_CALL(voice_engine_, GetSendCodec(kChannelId, _)) |
| 94 | .WillRepeatedly(DoAll(SetArgReferee<1>(kCodecInst), Return(0))); |
| 95 | EXPECT_CALL(voice_engine_, GetRemoteRTCPReportBlocks(kChannelId, _)) |
| 96 | .WillRepeatedly(DoAll(SetArgPointee<1>(report_blocks), Return(0))); |
| 97 | EXPECT_CALL(voice_engine_, GetSpeechInputLevelFullRange(_)) |
| 98 | .WillRepeatedly(DoAll(SetArgReferee<0>(kSpeechInputLevel), Return(0))); |
| 99 | EXPECT_CALL(voice_engine_, GetEcMetricsStatus(_)) |
| 100 | .WillRepeatedly(DoAll(SetArgReferee<0>(true), Return(0))); |
| 101 | EXPECT_CALL(voice_engine_, GetEchoMetrics(_, _, _, _)) |
| 102 | .WillRepeatedly(DoAll(SetArgReferee<0>(kEchoReturnLoss), |
| 103 | SetArgReferee<1>(kEchoReturnLossEnhancement), |
| 104 | Return(0))); |
| 105 | EXPECT_CALL(voice_engine_, GetEcDelayMetrics(_, _, _)) |
| 106 | .WillRepeatedly(DoAll(SetArgReferee<0>(kEchoDelayMedian), |
| 107 | SetArgReferee<1>(kEchoDelayStdDev), Return(0))); |
| 108 | } |
| 109 | |
| 110 | private: |
solenberg | 3a94154 | 2015-11-16 07:34:50 -0800 | [diff] [blame^] | 111 | testing::StrictMock<MockVoiceEngine> voice_engine_; |
solenberg | 566ef24 | 2015-11-06 15:34:49 -0800 | [diff] [blame] | 112 | rtc::scoped_refptr<AudioState> audio_state_; |
| 113 | AudioSendStream::Config stream_config_; |
| 114 | }; |
Fredrik Solenberg | 0ccae13 | 2015-11-03 10:15:49 +0100 | [diff] [blame] | 115 | } // namespace |
solenberg | c7a8b08 | 2015-10-16 14:35:07 -0700 | [diff] [blame] | 116 | |
| 117 | TEST(AudioSendStreamTest, ConfigToString) { |
solenberg | c7a8b08 | 2015-10-16 14:35:07 -0700 | [diff] [blame] | 118 | AudioSendStream::Config config(nullptr); |
Fredrik Solenberg | 0ccae13 | 2015-11-03 10:15:49 +0100 | [diff] [blame] | 119 | config.rtp.ssrc = kSsrc; |
solenberg | c7a8b08 | 2015-10-16 14:35:07 -0700 | [diff] [blame] | 120 | config.rtp.extensions.push_back( |
| 121 | RtpExtension(RtpExtension::kAbsSendTime, kAbsSendTimeId)); |
solenberg | 3a94154 | 2015-11-16 07:34:50 -0800 | [diff] [blame^] | 122 | config.rtp.c_name = kCName; |
Fredrik Solenberg | 0ccae13 | 2015-11-03 10:15:49 +0100 | [diff] [blame] | 123 | config.voe_channel_id = kChannelId; |
solenberg | c7a8b08 | 2015-10-16 14:35:07 -0700 | [diff] [blame] | 124 | config.cng_payload_type = 42; |
| 125 | config.red_payload_type = 17; |
Fredrik Solenberg | 0ccae13 | 2015-11-03 10:15:49 +0100 | [diff] [blame] | 126 | EXPECT_EQ( |
| 127 | "{rtp: {ssrc: 1234, extensions: [{name: " |
solenberg | 3a94154 | 2015-11-16 07:34:50 -0800 | [diff] [blame^] | 128 | "http://www.webrtc.org/experiments/rtp-hdrext/abs-send-time, id: 3}], " |
| 129 | "c_name: foo_name}, voe_channel_id: 1, cng_payload_type: 42, " |
| 130 | "red_payload_type: 17}", |
solenberg | 85a0496 | 2015-10-27 03:35:21 -0700 | [diff] [blame] | 131 | config.ToString()); |
solenberg | c7a8b08 | 2015-10-16 14:35:07 -0700 | [diff] [blame] | 132 | } |
| 133 | |
| 134 | TEST(AudioSendStreamTest, ConstructDestruct) { |
solenberg | 566ef24 | 2015-11-06 15:34:49 -0800 | [diff] [blame] | 135 | ConfigHelper helper; |
| 136 | internal::AudioSendStream send_stream(helper.config(), helper.audio_state()); |
solenberg | c7a8b08 | 2015-10-16 14:35:07 -0700 | [diff] [blame] | 137 | } |
solenberg | 85a0496 | 2015-10-27 03:35:21 -0700 | [diff] [blame] | 138 | |
| 139 | TEST(AudioSendStreamTest, GetStats) { |
solenberg | 566ef24 | 2015-11-06 15:34:49 -0800 | [diff] [blame] | 140 | ConfigHelper helper; |
| 141 | internal::AudioSendStream send_stream(helper.config(), helper.audio_state()); |
| 142 | helper.SetupMockForGetStats(); |
solenberg | 85a0496 | 2015-10-27 03:35:21 -0700 | [diff] [blame] | 143 | AudioSendStream::Stats stats = send_stream.GetStats(); |
Fredrik Solenberg | 0ccae13 | 2015-11-03 10:15:49 +0100 | [diff] [blame] | 144 | EXPECT_EQ(kSsrc, stats.local_ssrc); |
| 145 | EXPECT_EQ(static_cast<int64_t>(kCallStats.bytesSent), stats.bytes_sent); |
| 146 | EXPECT_EQ(kCallStats.packetsSent, stats.packets_sent); |
| 147 | EXPECT_EQ(static_cast<int32_t>(kReportBlock.cumulative_num_packets_lost), |
solenberg | 85a0496 | 2015-10-27 03:35:21 -0700 | [diff] [blame] | 148 | stats.packets_lost); |
Fredrik Solenberg | 0ccae13 | 2015-11-03 10:15:49 +0100 | [diff] [blame] | 149 | EXPECT_EQ(Q8ToFloat(kReportBlock.fraction_lost), stats.fraction_lost); |
| 150 | EXPECT_EQ(std::string(kCodecInst.plname), stats.codec_name); |
| 151 | EXPECT_EQ(static_cast<int32_t>(kReportBlock.extended_highest_sequence_number), |
solenberg | 85a0496 | 2015-10-27 03:35:21 -0700 | [diff] [blame] | 152 | stats.ext_seqnum); |
Fredrik Solenberg | 0ccae13 | 2015-11-03 10:15:49 +0100 | [diff] [blame] | 153 | EXPECT_EQ(static_cast<int32_t>(kReportBlock.interarrival_jitter / |
| 154 | (kCodecInst.plfreq / 1000)), |
| 155 | stats.jitter_ms); |
| 156 | EXPECT_EQ(kCallStats.rttMs, stats.rtt_ms); |
| 157 | EXPECT_EQ(static_cast<int32_t>(kSpeechInputLevel), stats.audio_level); |
solenberg | 85a0496 | 2015-10-27 03:35:21 -0700 | [diff] [blame] | 158 | EXPECT_EQ(-1, stats.aec_quality_min); |
Fredrik Solenberg | 0ccae13 | 2015-11-03 10:15:49 +0100 | [diff] [blame] | 159 | EXPECT_EQ(kEchoDelayMedian, stats.echo_delay_median_ms); |
| 160 | EXPECT_EQ(kEchoDelayStdDev, stats.echo_delay_std_ms); |
| 161 | EXPECT_EQ(kEchoReturnLoss, stats.echo_return_loss); |
| 162 | EXPECT_EQ(kEchoReturnLossEnhancement, stats.echo_return_loss_enhancement); |
solenberg | 85a0496 | 2015-10-27 03:35:21 -0700 | [diff] [blame] | 163 | EXPECT_FALSE(stats.typing_noise_detected); |
| 164 | } |
solenberg | 566ef24 | 2015-11-06 15:34:49 -0800 | [diff] [blame] | 165 | |
| 166 | TEST(AudioSendStreamTest, GetStatsTypingNoiseDetected) { |
| 167 | ConfigHelper helper; |
| 168 | internal::AudioSendStream send_stream(helper.config(), helper.audio_state()); |
| 169 | helper.SetupMockForGetStats(); |
| 170 | EXPECT_FALSE(send_stream.GetStats().typing_noise_detected); |
| 171 | |
| 172 | internal::AudioState* internal_audio_state = |
| 173 | static_cast<internal::AudioState*>(helper.audio_state().get()); |
| 174 | VoiceEngineObserver* voe_observer = |
| 175 | static_cast<VoiceEngineObserver*>(internal_audio_state); |
| 176 | voe_observer->CallbackOnError(-1, VE_TYPING_NOISE_WARNING); |
| 177 | EXPECT_TRUE(send_stream.GetStats().typing_noise_detected); |
| 178 | voe_observer->CallbackOnError(-1, VE_TYPING_NOISE_OFF_WARNING); |
| 179 | EXPECT_FALSE(send_stream.GetStats().typing_noise_detected); |
| 180 | } |
solenberg | 85a0496 | 2015-10-27 03:35:21 -0700 | [diff] [blame] | 181 | } // namespace test |
solenberg | c7a8b08 | 2015-10-16 14:35:07 -0700 | [diff] [blame] | 182 | } // namespace webrtc |