src/media/audio/cras/cras_input_unittest.cc

   1 // Copyright (c) 2012 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 <unistd.h>
   6
   7 #include <string>
   8
   9 #include "base/synchronization/waitable_event.h"
  10 #include "base/test/test_timeouts.h"
  11 #include "base/time/time.h"
  12 #include "media/audio/cras/audio_manager_cras.h"
  13 #include "media/audio/cras/cras_input.h"
  14 #include "testing/gmock/include/gmock/gmock.h"
  15 #include "testing/gtest/include/gtest/gtest.h"
  16
  17 using testing::_;
  18 using testing::AtLeast;
  19 using testing::Ge;
  20 using testing::InvokeWithoutArgs;
  21 using testing::StrictMock;
  22
  23 namespace media {
  24
  25 class MockAudioInputCallback : public AudioInputStream::AudioInputCallback {
  26  public:
  27   MOCK_METHOD5(OnData, void(
  28       AudioInputStream*, const uint8*, uint32, uint32, double));
  29   MOCK_METHOD1(OnError, void(AudioInputStream*));
  30   MOCK_METHOD1(OnClose, void(AudioInputStream*));
  31 };
  32
  33 class MockAudioManagerCrasInput : public AudioManagerCras {
  34  public:
  35   // We need to override this function in order to skip checking the number
  36   // of active output streams. It is because the number of active streams
  37   // is managed inside MakeAudioInputStream, and we don't use
  38   // MakeAudioInputStream to create the stream in the tests.
  39   virtual void ReleaseInputStream(AudioInputStream* stream) OVERRIDE {
  40     DCHECK(stream);
  41     delete stream;
  42   }
  43 };
  44
  45 class CrasInputStreamTest : public testing::Test {
  46  protected:
  47   CrasInputStreamTest() {
  48     mock_manager_.reset(new StrictMock<MockAudioManagerCrasInput>());
  49   }
  50
  51   virtual ~CrasInputStreamTest() {
  52   }
  53
  54   CrasInputStream* CreateStream(ChannelLayout layout) {
  55     return CreateStream(layout, kTestFramesPerPacket);
  56   }
  57
  58   CrasInputStream* CreateStream(ChannelLayout layout,
  59                                 int32 samples_per_packet) {
  60     AudioParameters params(kTestFormat,
  61                            layout,
  62                            kTestSampleRate,
  63                            kTestBitsPerSample,
  64                            samples_per_packet);
  65     return new CrasInputStream(params, mock_manager_.get(),
  66                                AudioManagerBase::kDefaultDeviceId);
  67   }
  68
  69   void CaptureSomeFrames(const AudioParameters &params,
  70                          unsigned int duration_ms) {
  71     CrasInputStream* test_stream = new CrasInputStream(
  72         params, mock_manager_.get(), AudioManagerBase::kDefaultDeviceId);
  73
  74     ASSERT_TRUE(test_stream->Open());
  75
  76     // Allow 8 frames variance for SRC in the callback.  Different numbers of
  77     // samples can be provided when doing non-integer SRC.  For example
  78     // converting from 192k to 44.1k is a ratio of 4.35 to 1.
  79     MockAudioInputCallback mock_callback;
  80     unsigned int expected_size = (kTestFramesPerPacket - 8) *
  81                                  params.channels() *
  82                                  params.bits_per_sample() / 8;
  83
  84     base::WaitableEvent event(false, false);
  85
  86     EXPECT_CALL(mock_callback,
  87                 OnData(test_stream, _, Ge(expected_size), _, _))
  88         .WillOnce(InvokeWithoutArgs(&event, &base::WaitableEvent::Signal));
  89
  90     test_stream->Start(&mock_callback);
  91
  92     // Wait for samples to be captured.
  93     EXPECT_TRUE(event.TimedWait(TestTimeouts::action_timeout()));
  94
  95     test_stream->Stop();
  96
  97     EXPECT_CALL(mock_callback, OnClose(test_stream)).Times(1);
  98     test_stream->Close();
  99   }
 100
 101   static const unsigned int kTestBitsPerSample;
 102   static const unsigned int kTestCaptureDurationMs;
 103   static const ChannelLayout kTestChannelLayout;
 104   static const AudioParameters::Format kTestFormat;
 105   static const uint32 kTestFramesPerPacket;
 106   static const int kTestSampleRate;
 107
 108   scoped_ptr<StrictMock<MockAudioManagerCrasInput> > mock_manager_;
 109
 110  private:
 111   DISALLOW_COPY_AND_ASSIGN(CrasInputStreamTest);
 112 };
 113
 114 const unsigned int CrasInputStreamTest::kTestBitsPerSample = 16;
 115 const unsigned int CrasInputStreamTest::kTestCaptureDurationMs = 250;
 116 const ChannelLayout CrasInputStreamTest::kTestChannelLayout =
 117     CHANNEL_LAYOUT_STEREO;
 118 const AudioParameters::Format CrasInputStreamTest::kTestFormat =
 119     AudioParameters::AUDIO_PCM_LINEAR;
 120 const uint32 CrasInputStreamTest::kTestFramesPerPacket = 1000;
 121 const int CrasInputStreamTest::kTestSampleRate = 44100;
 122
 123 TEST_F(CrasInputStreamTest, OpenMono) {
 124   CrasInputStream* test_stream = CreateStream(CHANNEL_LAYOUT_MONO);
 125   EXPECT_TRUE(test_stream->Open());
 126   test_stream->Close();
 127 }
 128
 129 TEST_F(CrasInputStreamTest, OpenStereo) {
 130   CrasInputStream* test_stream = CreateStream(CHANNEL_LAYOUT_STEREO);
 131   EXPECT_TRUE(test_stream->Open());
 132   test_stream->Close();
 133 }
 134
 135 TEST_F(CrasInputStreamTest, BadBitsPerSample) {
 136   AudioParameters bad_bps_params(kTestFormat,
 137                                  kTestChannelLayout,
 138                                  kTestSampleRate,
 139                                  kTestBitsPerSample - 1,
 140                                  kTestFramesPerPacket);
 141   CrasInputStream* test_stream = new CrasInputStream(
 142       bad_bps_params, mock_manager_.get(), AudioManagerBase::kDefaultDeviceId);
 143   EXPECT_FALSE(test_stream->Open());
 144   test_stream->Close();
 145 }
 146
 147 TEST_F(CrasInputStreamTest, BadFormat) {
 148   AudioParameters bad_format_params(AudioParameters::AUDIO_LAST_FORMAT,
 149                                     kTestChannelLayout,
 150                                     kTestSampleRate,
 151                                     kTestBitsPerSample,
 152                                     kTestFramesPerPacket);
 153   CrasInputStream* test_stream = new CrasInputStream(
 154       bad_format_params, mock_manager_.get(),
 155       AudioManagerBase::kDefaultDeviceId);
 156   EXPECT_FALSE(test_stream->Open());
 157   test_stream->Close();
 158 }
 159
 160 TEST_F(CrasInputStreamTest, BadSampleRate) {
 161   AudioParameters bad_rate_params(kTestFormat,
 162                                   kTestChannelLayout,
 163                                   0,
 164                                   kTestBitsPerSample,
 165                                   kTestFramesPerPacket);
 166   CrasInputStream* test_stream = new CrasInputStream(
 167       bad_rate_params, mock_manager_.get(), AudioManagerBase::kDefaultDeviceId);
 168   EXPECT_FALSE(test_stream->Open());
 169   test_stream->Close();
 170 }
 171
 172 TEST_F(CrasInputStreamTest, SetGetVolume) {
 173   CrasInputStream* test_stream = CreateStream(CHANNEL_LAYOUT_MONO);
 174   EXPECT_TRUE(test_stream->Open());
 175
 176   double max_volume = test_stream->GetMaxVolume();
 177   EXPECT_GE(max_volume, 1.0);
 178
 179   test_stream->SetVolume(max_volume / 2);
 180
 181   double new_volume = test_stream->GetVolume();
 182
 183   EXPECT_GE(new_volume, 0.0);
 184   EXPECT_LE(new_volume, max_volume);
 185
 186   test_stream->Close();
 187 }
 188
 189 TEST_F(CrasInputStreamTest, CaptureFrames) {
 190   const unsigned int rates[] =
 191       {8000, 16000, 22050, 32000, 44100, 48000, 96000, 192000};
 192
 193   for (unsigned int i = 0; i < ARRAY_SIZE(rates); i++) {
 194     SCOPED_TRACE(testing::Message() << "Mono " << rates[i] << "Hz");
 195     AudioParameters params_mono(kTestFormat,
 196                                 CHANNEL_LAYOUT_MONO,
 197                                 rates[i],
 198                                 kTestBitsPerSample,
 199                                 kTestFramesPerPacket);
 200     CaptureSomeFrames(params_mono, kTestCaptureDurationMs);
 201   }
 202
 203   for (unsigned int i = 0; i < ARRAY_SIZE(rates); i++) {
 204     SCOPED_TRACE(testing::Message() << "Stereo " << rates[i] << "Hz");
 205     AudioParameters params_stereo(kTestFormat,
 206                                   CHANNEL_LAYOUT_STEREO,
 207                                   rates[i],
 208                                   kTestBitsPerSample,
 209                                   kTestFramesPerPacket);
 210     CaptureSomeFrames(params_stereo, kTestCaptureDurationMs);
 211   }
 212 }
 213
 214 }  // namespace media