| // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "base/bind.h" |
| #include "base/macros.h" |
| #include "base/message_loop/message_loop.h" |
| #include "base/run_loop.h" |
| #include "base/synchronization/waitable_event.h" |
| #include "base/test/test_timeouts.h" |
| #include "base/threading/thread.h" |
| #include "media/audio/audio_device_description.h" |
| #include "media/audio/audio_input_controller.h" |
| #include "testing/gmock/include/gmock/gmock.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| using ::testing::_; |
| using ::testing::AnyNumber; |
| using ::testing::AtLeast; |
| using ::testing::Exactly; |
| using ::testing::InvokeWithoutArgs; |
| using ::testing::NotNull; |
| using base::WaitableEvent; |
| |
| namespace media { |
| |
| static const int kSampleRate = AudioParameters::kAudioCDSampleRate; |
| static const int kBitsPerSample = 16; |
| static const ChannelLayout kChannelLayout = CHANNEL_LAYOUT_STEREO; |
| static const int kSamplesPerPacket = kSampleRate / 10; |
| |
| ACTION_P(QuitRunLoop, run_loop) { |
| run_loop->QuitWhenIdle(); |
| } |
| |
| // Posts base::MessageLoop::QuitWhenIdleClosure() on specified message loop |
| // after a certain number of calls given by |limit|. |
| ACTION_P3(CheckCountAndPostQuitTask, count, limit, loop_or_proxy) { |
| if (++*count >= limit) { |
| loop_or_proxy->PostTask(FROM_HERE, |
| base::MessageLoop::QuitWhenIdleClosure()); |
| } |
| } |
| |
| // Closes AudioOutputController synchronously. |
| static void CloseAudioController(AudioInputController* controller) { |
| controller->Close(base::MessageLoop::QuitWhenIdleClosure()); |
| base::RunLoop().Run(); |
| } |
| |
| class MockAudioInputControllerEventHandler |
| : public AudioInputController::EventHandler { |
| public: |
| MockAudioInputControllerEventHandler() {} |
| |
| MOCK_METHOD1(OnCreated, void(AudioInputController* controller)); |
| MOCK_METHOD2(OnError, void(AudioInputController* controller, |
| AudioInputController::ErrorCode error_code)); |
| MOCK_METHOD2(OnLog, |
| void(AudioInputController* controller, |
| const std::string& message)); |
| |
| private: |
| DISALLOW_COPY_AND_ASSIGN(MockAudioInputControllerEventHandler); |
| }; |
| |
| class MockSyncWriter : public AudioInputController::SyncWriter { |
| public: |
| MockSyncWriter() {} |
| |
| MOCK_METHOD4(Write, |
| void(const AudioBus* data, |
| double volume, |
| bool key_pressed, |
| uint32_t hardware_delay_bytes)); |
| MOCK_METHOD0(Close, void()); |
| }; |
| |
| // Test fixture. |
| class AudioInputControllerTest : public testing::Test { |
| public: |
| AudioInputControllerTest() |
| : audio_thread_("AudioThread"), |
| suspend_event_(WaitableEvent::ResetPolicy::AUTOMATIC, |
| WaitableEvent::InitialState::NOT_SIGNALED) { |
| audio_thread_.StartAndWaitForTesting(); |
| audio_manager_ = |
| AudioManager::CreateForTesting(audio_thread_.task_runner()); |
| } |
| ~AudioInputControllerTest() override { |
| audio_task_runner()->PostTask( |
| FROM_HERE, base::Bind(&AudioInputControllerTest::DeleteAudioManager, |
| base::Unretained(this))); |
| audio_thread_.Stop(); |
| } |
| |
| scoped_refptr<base::SingleThreadTaskRunner> audio_task_runner() const { |
| return audio_thread_.task_runner(); |
| } |
| |
| void SuspendAudioThread() { |
| audio_task_runner()->PostTask( |
| FROM_HERE, base::Bind(&AudioInputControllerTest::WaitForResume, |
| base::Unretained(this))); |
| } |
| |
| void ResumeAudioThread() { suspend_event_.Signal(); } |
| |
| private: |
| void DeleteAudioManager() { audio_manager_.reset(); } |
| void WaitForResume() { suspend_event_.Wait(); } |
| |
| protected: |
| base::Thread audio_thread_; |
| base::MessageLoop message_loop_; |
| ScopedAudioManagerPtr audio_manager_; |
| WaitableEvent suspend_event_; |
| |
| private: |
| DISALLOW_COPY_AND_ASSIGN(AudioInputControllerTest); |
| }; |
| |
| // Test AudioInputController for create and close without recording audio. |
| TEST_F(AudioInputControllerTest, CreateAndClose) { |
| base::RunLoop run_loop; |
| |
| MockAudioInputControllerEventHandler event_handler; |
| MockSyncWriter sync_writer; |
| scoped_refptr<AudioInputController> controller; |
| |
| AudioParameters params(AudioParameters::AUDIO_FAKE, kChannelLayout, |
| kSampleRate, kBitsPerSample, kSamplesPerPacket); |
| |
| SuspendAudioThread(); |
| controller = AudioInputController::Create( |
| audio_manager_.get(), &event_handler, &sync_writer, nullptr, nullptr, |
| params, AudioDeviceDescription::kDefaultDeviceId, false); |
| ASSERT_TRUE(controller.get()); |
| EXPECT_CALL(event_handler, OnCreated(controller.get())).Times(Exactly(1)); |
| EXPECT_CALL(event_handler, OnLog(controller.get(), _)); |
| EXPECT_CALL(sync_writer, Close()).Times(Exactly(1)); |
| ResumeAudioThread(); |
| |
| CloseAudioController(controller.get()); |
| |
| audio_thread_.FlushForTesting(); |
| } |
| |
| // Test a normal call sequence of create, record and close. |
| TEST_F(AudioInputControllerTest, RecordAndClose) { |
| MockAudioInputControllerEventHandler event_handler; |
| MockSyncWriter sync_writer; |
| int count = 0; |
| |
| // OnCreated() will be called once. |
| EXPECT_CALL(event_handler, OnCreated(NotNull())) |
| .Times(Exactly(1)); |
| |
| // Write() should be called ten times. |
| EXPECT_CALL(sync_writer, Write(NotNull(), _, _, _)) |
| .Times(AtLeast(10)) |
| .WillRepeatedly( |
| CheckCountAndPostQuitTask(&count, 10, message_loop_.task_runner())); |
| |
| EXPECT_CALL(event_handler, OnLog(_, _)).Times(AnyNumber()); |
| EXPECT_CALL(sync_writer, Close()).Times(Exactly(1)); |
| |
| AudioParameters params(AudioParameters::AUDIO_FAKE, kChannelLayout, |
| kSampleRate, kBitsPerSample, kSamplesPerPacket); |
| |
| // Creating the AudioInputController should render an OnCreated() call. |
| scoped_refptr<AudioInputController> controller = AudioInputController::Create( |
| audio_manager_.get(), &event_handler, &sync_writer, nullptr, nullptr, |
| params, AudioDeviceDescription::kDefaultDeviceId, false); |
| ASSERT_TRUE(controller.get()); |
| |
| controller->Record(); |
| |
| // Record and wait until ten Write() callbacks are received. |
| base::RunLoop().Run(); |
| CloseAudioController(controller.get()); |
| } |
| |
| // Test that AudioInputController rejects insanely large packet sizes. |
| TEST_F(AudioInputControllerTest, SamplesPerPacketTooLarge) { |
| // Create an audio device with a very large packet size. |
| MockAudioInputControllerEventHandler event_handler; |
| MockSyncWriter sync_writer; |
| |
| // OnCreated() shall not be called in this test. |
| EXPECT_CALL(event_handler, OnCreated(NotNull())).Times(Exactly(0)); |
| |
| AudioParameters params(AudioParameters::AUDIO_FAKE, |
| kChannelLayout, |
| kSampleRate, |
| kBitsPerSample, |
| kSamplesPerPacket * 1000); |
| scoped_refptr<AudioInputController> controller = AudioInputController::Create( |
| audio_manager_.get(), &event_handler, &sync_writer, nullptr, nullptr, |
| params, AudioDeviceDescription::kDefaultDeviceId, false); |
| ASSERT_FALSE(controller.get()); |
| } |
| |
| // Test calling AudioInputController::Close multiple times. |
| TEST_F(AudioInputControllerTest, CloseTwice) { |
| MockAudioInputControllerEventHandler event_handler; |
| MockSyncWriter sync_writer; |
| |
| // OnCreated() will be called only once. |
| EXPECT_CALL(event_handler, OnCreated(NotNull())).Times(Exactly(1)); |
| EXPECT_CALL(event_handler, OnLog(_, _)).Times(AnyNumber()); |
| // This callback should still only be called once. |
| EXPECT_CALL(sync_writer, Close()).Times(Exactly(1)); |
| |
| AudioParameters params(AudioParameters::AUDIO_FAKE, |
| kChannelLayout, |
| kSampleRate, |
| kBitsPerSample, |
| kSamplesPerPacket); |
| scoped_refptr<AudioInputController> controller = AudioInputController::Create( |
| audio_manager_.get(), &event_handler, &sync_writer, nullptr, nullptr, |
| params, AudioDeviceDescription::kDefaultDeviceId, false); |
| ASSERT_TRUE(controller.get()); |
| |
| controller->Record(); |
| |
| controller->Close(base::MessageLoop::QuitWhenIdleClosure()); |
| base::RunLoop().Run(); |
| |
| controller->Close(base::MessageLoop::QuitWhenIdleClosure()); |
| base::RunLoop().Run(); |
| } |
| |
| } // namespace media |