media/audio/audio_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 "base/basictypes.h"
   6 #include "base/environment.h"
   7 #include "base/memory/scoped_ptr.h"
   8 #include "base/message_loop.h"
   9 #include "base/threading/platform_thread.h"
  10 #include "media/audio/audio_io.h"
  11 #include "media/audio/audio_manager_base.h"
  12 #include "testing/gtest/include/gtest/gtest.h"
  13
  14 namespace media {
  15
  16 static const int kSamplingRate = 8000;
  17 static const int kSamplesPerPacket = kSamplingRate / 20;
  18
  19 // This class allows to find out if the callbacks are occurring as
  20 // expected and if any error has been reported.
  21 class TestInputCallback : public AudioInputStream::AudioInputCallback {
  22  public:
  23   explicit TestInputCallback(int max_data_bytes)
  24       : callback_count_(0),
  25         had_error_(0),
  26         max_data_bytes_(max_data_bytes) {
  27   }
  28   virtual void OnData(AudioInputStream* stream,
  29                       const uint8* data,
  30                       uint32 size,
  31                       uint32 hardware_delay_bytes,
  32                       double volume) OVERRIDE {
  33     ++callback_count_;
  34     // Read the first byte to make sure memory is good.
  35     if (size) {
  36       ASSERT_LE(static_cast<int>(size), max_data_bytes_);
  37       int value = data[0];
  38       EXPECT_GE(value, 0);
  39     }
  40   }
  41   virtual void OnClose(AudioInputStream* stream) OVERRIDE {}
  42   virtual void OnError(AudioInputStream* stream) OVERRIDE {
  43     ++had_error_;
  44   }
  45   // Returns how many times OnData() has been called.
  46   int callback_count() const {
  47     return callback_count_;
  48   }
  49   // Returns how many times the OnError callback was called.
  50   int had_error() const {
  51     return had_error_;
  52   }
  53
  54  private:
  55   int callback_count_;
  56   int had_error_;
  57   int max_data_bytes_;
  58 };
  59
  60 static bool CanRunAudioTests(AudioManager* audio_man) {
  61   bool has_input = audio_man->HasAudioInputDevices();
  62
  63   if (!has_input)
  64     LOG(WARNING) << "No input devices detected";
  65
  66   return has_input;
  67 }
  68
  69 static AudioInputStream* CreateTestAudioInputStream(AudioManager* audio_man) {
  70   AudioInputStream* ais = audio_man->MakeAudioInputStream(
  71       AudioParameters(AudioParameters::AUDIO_PCM_LINEAR, CHANNEL_LAYOUT_STEREO,
  72                       kSamplingRate, 16, kSamplesPerPacket),
  73                       AudioManagerBase::kDefaultDeviceId);
  74   EXPECT_TRUE(NULL != ais);
  75   return ais;
  76 }
  77
  78 // Test that AudioInputStream rejects out of range parameters.
  79 TEST(AudioInputTest, SanityOnMakeParams) {
  80   scoped_ptr<AudioManager> audio_man(AudioManager::Create());
  81   if (!CanRunAudioTests(audio_man.get()))
  82     return;
  83
  84   AudioParameters::Format fmt = AudioParameters::AUDIO_PCM_LINEAR;
  85   EXPECT_TRUE(NULL == audio_man->MakeAudioInputStream(
  86       AudioParameters(fmt, CHANNEL_LAYOUT_7_1, 8000, 16,
  87                       kSamplesPerPacket), AudioManagerBase::kDefaultDeviceId));
  88   EXPECT_TRUE(NULL == audio_man->MakeAudioInputStream(
  89       AudioParameters(fmt, CHANNEL_LAYOUT_MONO, 1024 * 1024, 16,
  90                       kSamplesPerPacket), AudioManagerBase::kDefaultDeviceId));
  91   EXPECT_TRUE(NULL == audio_man->MakeAudioInputStream(
  92       AudioParameters(fmt, CHANNEL_LAYOUT_STEREO, 8000, 80,
  93                       kSamplesPerPacket), AudioManagerBase::kDefaultDeviceId));
  94   EXPECT_TRUE(NULL == audio_man->MakeAudioInputStream(
  95       AudioParameters(fmt, CHANNEL_LAYOUT_STEREO, 8000, 80,
  96                       1000 * kSamplesPerPacket),
  97                       AudioManagerBase::kDefaultDeviceId));
  98   EXPECT_TRUE(NULL == audio_man->MakeAudioInputStream(
  99       AudioParameters(fmt, CHANNEL_LAYOUT_UNSUPPORTED, 8000, 16,
 100                       kSamplesPerPacket), AudioManagerBase::kDefaultDeviceId));
 101   EXPECT_TRUE(NULL == audio_man->MakeAudioInputStream(
 102       AudioParameters(fmt, CHANNEL_LAYOUT_STEREO, -8000, 16,
 103                       kSamplesPerPacket), AudioManagerBase::kDefaultDeviceId));
 104   EXPECT_TRUE(NULL == audio_man->MakeAudioInputStream(
 105       AudioParameters(fmt, CHANNEL_LAYOUT_STEREO, 8000, -16,
 106                       kSamplesPerPacket), AudioManagerBase::kDefaultDeviceId));
 107   EXPECT_TRUE(NULL == audio_man->MakeAudioInputStream(
 108       AudioParameters(fmt, CHANNEL_LAYOUT_STEREO, 8000, 16, -1024),
 109       AudioManagerBase::kDefaultDeviceId));
 110 }
 111
 112 // Test create and close of an AudioInputStream without recording audio.
 113 TEST(AudioInputTest, CreateAndClose) {
 114   scoped_ptr<AudioManager> audio_man(AudioManager::Create());
 115   if (!CanRunAudioTests(audio_man.get()))
 116     return;
 117   AudioInputStream* ais = CreateTestAudioInputStream(audio_man.get());
 118   ais->Close();
 119 }
 120
 121 // Test create, open and close of an AudioInputStream without recording audio.
 122 TEST(AudioInputTest, OpenAndClose) {
 123   scoped_ptr<AudioManager> audio_man(AudioManager::Create());
 124   if (!CanRunAudioTests(audio_man.get()))
 125     return;
 126   AudioInputStream* ais = CreateTestAudioInputStream(audio_man.get());
 127   EXPECT_TRUE(ais->Open());
 128   ais->Close();
 129 }
 130
 131 // Test create, open, stop and close of an AudioInputStream without recording.
 132 TEST(AudioInputTest, OpenStopAndClose) {
 133   scoped_ptr<AudioManager> audio_man(AudioManager::Create());
 134   if (!CanRunAudioTests(audio_man.get()))
 135     return;
 136   AudioInputStream* ais = CreateTestAudioInputStream(audio_man.get());
 137   EXPECT_TRUE(ais->Open());
 138   ais->Stop();
 139   ais->Close();
 140 }
 141
 142 // Test a normal recording sequence using an AudioInputStream.
 143 TEST(AudioInputTest, Record) {
 144   scoped_ptr<AudioManager> audio_man(AudioManager::Create());
 145   if (!CanRunAudioTests(audio_man.get()))
 146     return;
 147   MessageLoop message_loop(MessageLoop::TYPE_DEFAULT);
 148   AudioInputStream* ais = CreateTestAudioInputStream(audio_man.get());
 149   EXPECT_TRUE(ais->Open());
 150
 151   TestInputCallback test_callback(kSamplesPerPacket * 4);
 152   ais->Start(&test_callback);
 153   // Verify at least 500ms worth of audio was recorded, after giving sufficient
 154   // extra time.
 155   message_loop.PostDelayedTask(
 156       FROM_HERE,
 157       MessageLoop::QuitClosure(),
 158       base::TimeDelta::FromMilliseconds(690));
 159   message_loop.Run();
 160   EXPECT_GE(test_callback.callback_count(), 1);
 161   EXPECT_FALSE(test_callback.had_error());
 162
 163   ais->Stop();
 164   ais->Close();
 165 }
 166
 167 }  // namespace media