OLD | NEW |
| (Empty) |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include <stddef.h> | |
6 | |
7 #include "content/renderer/media/media_stream_audio_level_calculator.h" | |
8 #include "content/renderer/media/webrtc/webrtc_local_audio_track_adapter.h" | |
9 #include "content/renderer/media/webrtc_audio_capturer.h" | |
10 #include "content/renderer/media/webrtc_local_audio_track.h" | |
11 #include "testing/gmock/include/gmock/gmock.h" | |
12 #include "testing/gtest/include/gtest/gtest.h" | |
13 #include "third_party/webrtc/api/mediastreaminterface.h" | |
14 | |
15 using ::testing::_; | |
16 using ::testing::AnyNumber; | |
17 | |
18 namespace content { | |
19 | |
20 namespace { | |
21 | |
22 class MockWebRtcAudioSink : public webrtc::AudioTrackSinkInterface { | |
23 public: | |
24 MockWebRtcAudioSink() {} | |
25 ~MockWebRtcAudioSink() {} | |
26 MOCK_METHOD5(OnData, void(const void* audio_data, | |
27 int bits_per_sample, | |
28 int sample_rate, | |
29 size_t number_of_channels, | |
30 size_t number_of_frames)); | |
31 }; | |
32 | |
33 } // namespace | |
34 | |
35 class WebRtcLocalAudioTrackAdapterTest : public ::testing::Test { | |
36 public: | |
37 WebRtcLocalAudioTrackAdapterTest() | |
38 : params_(media::AudioParameters::AUDIO_PCM_LOW_LATENCY, | |
39 media::CHANNEL_LAYOUT_STEREO, 48000, 16, 480), | |
40 adapter_(WebRtcLocalAudioTrackAdapter::Create(std::string(), NULL)) { | |
41 track_.reset(new WebRtcLocalAudioTrack(adapter_.get())); | |
42 } | |
43 | |
44 protected: | |
45 void SetUp() override { | |
46 track_->OnSetFormat(params_); | |
47 EXPECT_TRUE(track_->GetAudioAdapter()->enabled()); | |
48 } | |
49 | |
50 media::AudioParameters params_; | |
51 scoped_refptr<WebRtcLocalAudioTrackAdapter> adapter_; | |
52 std::unique_ptr<WebRtcLocalAudioTrack> track_; | |
53 }; | |
54 | |
55 // Adds and Removes a WebRtcAudioSink to a local audio track. | |
56 TEST_F(WebRtcLocalAudioTrackAdapterTest, AddAndRemoveSink) { | |
57 // Add a sink to the webrtc track. | |
58 std::unique_ptr<MockWebRtcAudioSink> sink(new MockWebRtcAudioSink()); | |
59 webrtc::AudioTrackInterface* webrtc_track = | |
60 static_cast<webrtc::AudioTrackInterface*>(adapter_.get()); | |
61 webrtc_track->AddSink(sink.get()); | |
62 | |
63 // Send a packet via |track_| and the data should reach the sink of the | |
64 // |adapter_|. | |
65 const std::unique_ptr<media::AudioBus> audio_bus = | |
66 media::AudioBus::Create(params_); | |
67 // While this test is not checking the signal data being passed around, the | |
68 // implementation in WebRtcLocalAudioTrack reads the data for its signal level | |
69 // computation. Initialize all samples to zero to make the memory sanitizer | |
70 // happy. | |
71 audio_bus->Zero(); | |
72 | |
73 base::TimeTicks estimated_capture_time = base::TimeTicks::Now(); | |
74 EXPECT_CALL(*sink, | |
75 OnData(_, 16, params_.sample_rate(), params_.channels(), | |
76 params_.frames_per_buffer())); | |
77 track_->Capture(*audio_bus, estimated_capture_time); | |
78 | |
79 // Remove the sink from the webrtc track. | |
80 webrtc_track->RemoveSink(sink.get()); | |
81 sink.reset(); | |
82 | |
83 // Verify that no more callback gets into the sink. | |
84 estimated_capture_time += | |
85 params_.frames_per_buffer() * base::TimeDelta::FromSeconds(1) / | |
86 params_.sample_rate(); | |
87 track_->Capture(*audio_bus, estimated_capture_time); | |
88 } | |
89 | |
90 TEST_F(WebRtcLocalAudioTrackAdapterTest, GetSignalLevel) { | |
91 webrtc::AudioTrackInterface* webrtc_track = | |
92 static_cast<webrtc::AudioTrackInterface*>(adapter_.get()); | |
93 int signal_level = -1; | |
94 EXPECT_FALSE(webrtc_track->GetSignalLevel(&signal_level)); | |
95 MediaStreamAudioLevelCalculator calculator; | |
96 adapter_->SetLevel(calculator.level()); | |
97 signal_level = -1; | |
98 EXPECT_TRUE(webrtc_track->GetSignalLevel(&signal_level)); | |
99 EXPECT_EQ(0, signal_level); | |
100 } | |
101 | |
102 } // namespace content | |
OLD | NEW |