audio_input_stream_data_interceptor_unittest.cc 10 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314
  1. // Copyright 2017 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. #include "media/audio/audio_input_stream_data_interceptor.h"
  5. #include <utility>
  6. #include "base/bind.h"
  7. #include "base/callback.h"
  8. #include "base/memory/ptr_util.h"
  9. #include "media/audio/audio_debug_recording_helper.h"
  10. #include "media/audio/audio_io.h"
  11. #include "testing/gmock/include/gmock/gmock.h"
  12. #include "testing/gtest/include/gtest/gtest.h"
  13. namespace media {
  14. namespace {
  15. using testing::Return;
  16. using testing::StrictMock;
  17. using testing::Mock;
  18. const double kMaxVolume = 0.1234;
  19. const double kNewVolume = 0.2345;
  20. const double kVolume = 0.3456;
  21. class MockStream : public AudioInputStream {
  22. public:
  23. MockStream() = default;
  24. ~MockStream() override = default;
  25. MOCK_METHOD0(Open, AudioInputStream::OpenOutcome());
  26. MOCK_METHOD1(Start, void(AudioInputStream::AudioInputCallback*));
  27. MOCK_METHOD0(Stop, void());
  28. MOCK_METHOD0(Close, void());
  29. MOCK_METHOD0(GetMaxVolume, double());
  30. MOCK_METHOD1(SetVolume, void(double));
  31. MOCK_METHOD0(GetVolume, double());
  32. MOCK_METHOD1(SetAutomaticGainControl, bool(bool));
  33. MOCK_METHOD0(GetAutomaticGainControl, bool());
  34. MOCK_METHOD0(IsMuted, bool());
  35. MOCK_METHOD1(SetOutputDeviceForAec, void(const std::string&));
  36. };
  37. class MockDebugRecorder : public AudioDebugRecorder {
  38. public:
  39. MockDebugRecorder() = default;
  40. ~MockDebugRecorder() override = default;
  41. MOCK_METHOD1(OnData, void(const AudioBus* source));
  42. };
  43. class MockCallback : public AudioInputStream::AudioInputCallback {
  44. public:
  45. MockCallback() = default;
  46. ~MockCallback() override = default;
  47. MOCK_METHOD3(OnData, void(const AudioBus*, base::TimeTicks, double));
  48. MOCK_METHOD0(OnError, void());
  49. };
  50. class MockDebugRecorderFactory {
  51. public:
  52. MockDebugRecorderFactory() = default;
  53. ~MockDebugRecorderFactory() { DCHECK(!prepared_recorder_); }
  54. std::unique_ptr<AudioDebugRecorder> CreateDebugRecorder() {
  55. DCHECK(prepared_recorder_);
  56. return std::move(prepared_recorder_);
  57. }
  58. void ExpectRecorderCreation(
  59. std::unique_ptr<AudioDebugRecorder> recorder_ptr) {
  60. DCHECK(!prepared_recorder_);
  61. prepared_recorder_ = std::move(recorder_ptr);
  62. DCHECK(prepared_recorder_);
  63. }
  64. private:
  65. std::unique_ptr<AudioDebugRecorder> prepared_recorder_;
  66. };
  67. void TestSetAutomaticGainControl(bool enable, bool agc_is_supported) {
  68. MockDebugRecorderFactory factory;
  69. StrictMock<MockStream> stream;
  70. AudioInputStream* interceptor = new AudioInputStreamDataInterceptor(
  71. base::BindRepeating(&MockDebugRecorderFactory::CreateDebugRecorder,
  72. base::Unretained(&factory)),
  73. &stream);
  74. EXPECT_CALL(stream, SetAutomaticGainControl(enable))
  75. .WillOnce(Return(agc_is_supported));
  76. EXPECT_EQ(interceptor->SetAutomaticGainControl(enable), agc_is_supported);
  77. Mock::VerifyAndClearExpectations(&stream);
  78. EXPECT_CALL(stream, Close());
  79. interceptor->Close();
  80. }
  81. } // namespace
  82. TEST(AudioInputStreamDataInterceptorTest, Open) {
  83. MockDebugRecorderFactory factory;
  84. StrictMock<MockStream> stream;
  85. AudioInputStream* interceptor = new AudioInputStreamDataInterceptor(
  86. base::BindRepeating(&MockDebugRecorderFactory::CreateDebugRecorder,
  87. base::Unretained(&factory)),
  88. &stream);
  89. EXPECT_CALL(stream, Open());
  90. interceptor->Open();
  91. Mock::VerifyAndClearExpectations(&stream);
  92. EXPECT_CALL(stream, Close());
  93. interceptor->Close();
  94. }
  95. TEST(AudioInputStreamDataInterceptorTest, Start) {
  96. MockDebugRecorderFactory factory;
  97. StrictMock<MockStream> stream;
  98. StrictMock<MockCallback> callback;
  99. auto* recorder = new StrictMock<MockDebugRecorder>();
  100. std::unique_ptr<AudioBus> audio_bus = AudioBus::Create(1, 1);
  101. AudioInputStreamDataInterceptor* interceptor =
  102. new AudioInputStreamDataInterceptor(
  103. base::BindRepeating(&MockDebugRecorderFactory::CreateDebugRecorder,
  104. base::Unretained(&factory)),
  105. &stream);
  106. EXPECT_CALL(stream, Start(interceptor));
  107. factory.ExpectRecorderCreation(base::WrapUnique(recorder));
  108. interceptor->Start(&callback);
  109. Mock::VerifyAndClearExpectations(&stream);
  110. base::TimeTicks time = base::TimeTicks::Now();
  111. // Audio data should be passed to both callback and recorder.
  112. EXPECT_CALL(callback, OnData(audio_bus.get(), time, kVolume));
  113. EXPECT_CALL(*recorder, OnData(audio_bus.get()));
  114. interceptor->OnData(audio_bus.get(), time, kVolume);
  115. Mock::VerifyAndClearExpectations(&callback);
  116. Mock::VerifyAndClearExpectations(recorder);
  117. // Errors should be propagated to the renderer
  118. EXPECT_CALL(callback, OnError());
  119. interceptor->OnError();
  120. Mock::VerifyAndClearExpectations(&callback);
  121. EXPECT_CALL(stream, Close());
  122. interceptor->Close();
  123. }
  124. TEST(AudioInputStreamDataInterceptorTest, Stop) {
  125. MockDebugRecorderFactory factory;
  126. StrictMock<MockStream> stream;
  127. AudioInputStream* interceptor = new AudioInputStreamDataInterceptor(
  128. base::BindRepeating(&MockDebugRecorderFactory::CreateDebugRecorder,
  129. base::Unretained(&factory)),
  130. &stream);
  131. EXPECT_CALL(stream, Stop());
  132. interceptor->Stop();
  133. Mock::VerifyAndClearExpectations(&stream);
  134. EXPECT_CALL(stream, Close());
  135. interceptor->Close();
  136. }
  137. TEST(AudioInputStreamDataInterceptorTest, Close) {
  138. MockDebugRecorderFactory factory;
  139. StrictMock<MockStream> stream;
  140. AudioInputStream* interceptor = new AudioInputStreamDataInterceptor(
  141. base::BindRepeating(&MockDebugRecorderFactory::CreateDebugRecorder,
  142. base::Unretained(&factory)),
  143. &stream);
  144. EXPECT_CALL(stream, Close());
  145. interceptor->Close();
  146. }
  147. TEST(AudioInputStreamDataInterceptorTest, GetMaxVolume) {
  148. MockDebugRecorderFactory factory;
  149. StrictMock<MockStream> stream;
  150. AudioInputStream* interceptor = new AudioInputStreamDataInterceptor(
  151. base::BindRepeating(&MockDebugRecorderFactory::CreateDebugRecorder,
  152. base::Unretained(&factory)),
  153. &stream);
  154. EXPECT_CALL(stream, GetMaxVolume()).WillOnce(Return(kMaxVolume));
  155. EXPECT_EQ(interceptor->GetMaxVolume(), kMaxVolume);
  156. Mock::VerifyAndClearExpectations(&stream);
  157. EXPECT_CALL(stream, Close());
  158. interceptor->Close();
  159. }
  160. TEST(AudioInputStreamDataInterceptorTest, SetVolume) {
  161. MockDebugRecorderFactory factory;
  162. StrictMock<MockStream> stream;
  163. AudioInputStream* interceptor = new AudioInputStreamDataInterceptor(
  164. base::BindRepeating(&MockDebugRecorderFactory::CreateDebugRecorder,
  165. base::Unretained(&factory)),
  166. &stream);
  167. EXPECT_CALL(stream, SetVolume(kNewVolume));
  168. interceptor->SetVolume(kNewVolume);
  169. Mock::VerifyAndClearExpectations(&stream);
  170. EXPECT_CALL(stream, Close());
  171. interceptor->Close();
  172. }
  173. TEST(AudioInputStreamDataInterceptorTest, GetVolume) {
  174. MockDebugRecorderFactory factory;
  175. StrictMock<MockStream> stream;
  176. AudioInputStream* interceptor = new AudioInputStreamDataInterceptor(
  177. base::BindRepeating(&MockDebugRecorderFactory::CreateDebugRecorder,
  178. base::Unretained(&factory)),
  179. &stream);
  180. EXPECT_CALL(stream, GetVolume()).WillOnce(Return(kVolume));
  181. EXPECT_EQ(interceptor->GetVolume(), kVolume);
  182. Mock::VerifyAndClearExpectations(&stream);
  183. EXPECT_CALL(stream, Close());
  184. interceptor->Close();
  185. }
  186. TEST(AudioInputStreamDataInterceptorTest,
  187. SetAutomaticGainControlTrueWhenSupported) {
  188. TestSetAutomaticGainControl(true, true);
  189. }
  190. TEST(AudioInputStreamDataInterceptorTest,
  191. SetAutomaticGainControlFalseWhenSupported) {
  192. TestSetAutomaticGainControl(false, true);
  193. }
  194. TEST(AudioInputStreamDataInterceptorTest,
  195. SetAutomaticGainControlTrueWhenNotSupported) {
  196. TestSetAutomaticGainControl(true, false);
  197. }
  198. TEST(AudioInputStreamDataInterceptorTest,
  199. SetAutomaticGainControlFalseWhenNotSupported) {
  200. TestSetAutomaticGainControl(false, false);
  201. }
  202. TEST(AudioInputStreamDataInterceptorTest, GetAutomaticGainControl_True) {
  203. MockDebugRecorderFactory factory;
  204. StrictMock<MockStream> stream;
  205. AudioInputStream* interceptor = new AudioInputStreamDataInterceptor(
  206. base::BindRepeating(&MockDebugRecorderFactory::CreateDebugRecorder,
  207. base::Unretained(&factory)),
  208. &stream);
  209. EXPECT_CALL(stream, GetAutomaticGainControl()).WillOnce(Return(true));
  210. EXPECT_EQ(interceptor->GetAutomaticGainControl(), true);
  211. Mock::VerifyAndClearExpectations(&stream);
  212. EXPECT_CALL(stream, Close());
  213. interceptor->Close();
  214. }
  215. TEST(AudioInputStreamDataInterceptorTest, GetAutomaticGainControl_False) {
  216. MockDebugRecorderFactory factory;
  217. StrictMock<MockStream> stream;
  218. AudioInputStream* interceptor = new AudioInputStreamDataInterceptor(
  219. base::BindRepeating(&MockDebugRecorderFactory::CreateDebugRecorder,
  220. base::Unretained(&factory)),
  221. &stream);
  222. EXPECT_CALL(stream, GetAutomaticGainControl()).WillOnce(Return(false));
  223. EXPECT_EQ(interceptor->GetAutomaticGainControl(), false);
  224. Mock::VerifyAndClearExpectations(&stream);
  225. EXPECT_CALL(stream, Close());
  226. interceptor->Close();
  227. }
  228. TEST(AudioInputStreamDataInterceptorTest, IsMuted_True) {
  229. MockDebugRecorderFactory factory;
  230. StrictMock<MockStream> stream;
  231. AudioInputStream* interceptor = new AudioInputStreamDataInterceptor(
  232. base::BindRepeating(&MockDebugRecorderFactory::CreateDebugRecorder,
  233. base::Unretained(&factory)),
  234. &stream);
  235. EXPECT_CALL(stream, IsMuted()).WillOnce(Return(true));
  236. EXPECT_EQ(interceptor->IsMuted(), true);
  237. Mock::VerifyAndClearExpectations(&stream);
  238. EXPECT_CALL(stream, Close());
  239. interceptor->Close();
  240. }
  241. TEST(AudioInputStreamDataInterceptorTest, IsMuted_False) {
  242. MockDebugRecorderFactory factory;
  243. StrictMock<MockStream> stream;
  244. AudioInputStream* interceptor = new AudioInputStreamDataInterceptor(
  245. base::BindRepeating(&MockDebugRecorderFactory::CreateDebugRecorder,
  246. base::Unretained(&factory)),
  247. &stream);
  248. EXPECT_CALL(stream, IsMuted()).WillOnce(Return(false));
  249. EXPECT_EQ(interceptor->IsMuted(), false);
  250. Mock::VerifyAndClearExpectations(&stream);
  251. EXPECT_CALL(stream, Close());
  252. interceptor->Close();
  253. }
  254. } // namespace media