blob: fdf61208fae4594d8c2d37a6f9eb62078e440d07 [file] [log] [blame]
// 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, params,
AudioDeviceDescription::kDefaultDeviceId, NULL);
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, params,
AudioDeviceDescription::kDefaultDeviceId, NULL);
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, params,
AudioDeviceDescription::kDefaultDeviceId, NULL);
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, params,
AudioDeviceDescription::kDefaultDeviceId, NULL);
ASSERT_TRUE(controller.get());
controller->Record();
controller->Close(base::MessageLoop::QuitWhenIdleClosure());
base::RunLoop().Run();
controller->Close(base::MessageLoop::QuitWhenIdleClosure());
base::RunLoop().Run();
}
} // namespace media