src/components/copresence/mediums/audio/audio_recorder_unittest.cc

   1 // Copyright 2014 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 "components/copresence/mediums/audio/audio_recorder.h"
   6
   7 #include "base/bind.h"
   8 #include "base/memory/aligned_memory.h"
   9 #include "base/run_loop.h"
  10 #include "components/copresence/public/copresence_constants.h"
  11 #include "components/copresence/test/audio_test_support.h"
  12 #include "content/public/test/test_browser_thread_bundle.h"
  13 #include "media/audio/audio_manager.h"
  14 #include "media/audio/audio_manager_base.h"
  15 #include "media/base/audio_bus.h"
  16 #include "testing/gtest/include/gtest/gtest.h"
  17
  18 namespace {
  19
  20 class TestAudioInputStream : public media::AudioInputStream {
  21  public:
  22   TestAudioInputStream(const media::AudioParameters& params,
  23                        const std::vector<float*> channel_data,
  24                        size_t samples)
  25       : callback_(NULL), params_(params) {
  26     buffer_ = media::AudioBus::CreateWrapper(2);
  27     for (size_t i = 0; i < channel_data.size(); ++i)
  28       buffer_->SetChannelData(i, channel_data[i]);
  29     buffer_->set_frames(samples);
  30   }
  31
  32   virtual ~TestAudioInputStream() {}
  33
  34   virtual bool Open() OVERRIDE { return true; }
  35   virtual void Start(AudioInputCallback* callback) OVERRIDE {
  36     DCHECK(callback);
  37     callback_ = callback;
  38     media::AudioManager::Get()->GetTaskRunner()->PostTask(
  39         FROM_HERE,
  40         base::Bind(&TestAudioInputStream::SimulateRecording,
  41                    base::Unretained(this)));
  42   }
  43   virtual void Stop() OVERRIDE {}
  44   virtual void Close() OVERRIDE {}
  45   virtual double GetMaxVolume() OVERRIDE { return 1.0; }
  46   virtual void SetVolume(double volume) OVERRIDE {}
  47   virtual double GetVolume() OVERRIDE { return 1.0; }
  48   virtual void SetAutomaticGainControl(bool enabled) OVERRIDE {}
  49   virtual bool GetAutomaticGainControl() OVERRIDE { return true; }
  50
  51  private:
  52   void SimulateRecording() {
  53     const int fpb = params_.frames_per_buffer();
  54     for (int i = 0; i < buffer_->frames() / fpb; ++i) {
  55       scoped_ptr<media::AudioBus> source = media::AudioBus::Create(2, fpb);
  56       buffer_->CopyPartialFramesTo(i * fpb, fpb, 0, source.get());
  57       callback_->OnData(this, source.get(), fpb, 1.0);
  58     }
  59   }
  60
  61   AudioInputCallback* callback_;
  62   media::AudioParameters params_;
  63   scoped_ptr<media::AudioBus> buffer_;
  64
  65   DISALLOW_COPY_AND_ASSIGN(TestAudioInputStream);
  66 };
  67
  68 }  // namespace
  69
  70 namespace copresence {
  71
  72 class AudioRecorderTest : public testing::Test {
  73  public:
  74   AudioRecorderTest() : total_samples_(0), recorder_(NULL) {
  75     if (!media::AudioManager::Get())
  76       media::AudioManager::CreateForTesting();
  77   }
  78
  79   virtual ~AudioRecorderTest() {
  80     DeleteRecorder();
  81     for (size_t i = 0; i < channel_data_.size(); ++i)
  82       base::AlignedFree(channel_data_[i]);
  83   }
  84
  85   void CreateSimpleRecorder() {
  86     DeleteRecorder();
  87     recorder_ = new AudioRecorder(
  88         base::Bind(&AudioRecorderTest::DecodeSamples, base::Unretained(this)));
  89     recorder_->Initialize();
  90   }
  91
  92   void CreateRecorder(size_t channels,
  93                       size_t sample_rate,
  94                       size_t bits_per_sample,
  95                       size_t samples) {
  96     DeleteRecorder();
  97     params_.Reset(media::AudioParameters::AUDIO_PCM_LOW_LATENCY,
  98                   kDefaultChannelLayout,
  99                   channels,
 100                   sample_rate,
 101                   bits_per_sample,
 102                   4096);
 103
 104     channel_data_.clear();
 105     channel_data_.push_back(GenerateSamples(0x1337, samples));
 106     channel_data_.push_back(GenerateSamples(0x7331, samples));
 107
 108     total_samples_ = samples;
 109
 110     recorder_ = new AudioRecorder(
 111         base::Bind(&AudioRecorderTest::DecodeSamples, base::Unretained(this)));
 112     recorder_->set_input_stream_for_testing(
 113         new TestAudioInputStream(params_, channel_data_, samples));
 114     recorder_->set_params_for_testing(new media::AudioParameters(params_));
 115     recorder_->Initialize();
 116   }
 117
 118   void DeleteRecorder() {
 119     if (!recorder_)
 120       return;
 121     recorder_->Finalize();
 122     recorder_ = NULL;
 123   }
 124
 125   void RecordAndVerifySamples() {
 126     received_samples_.clear();
 127     run_loop_.reset(new base::RunLoop());
 128     recorder_->Record();
 129     run_loop_->Run();
 130   }
 131
 132   void DecodeSamples(const std::string& samples) {
 133     received_samples_ += samples;
 134     // We expect one less decode than our total samples would ideally have
 135     // triggered since we process data in 4k chunks. So our sample processing
 136     // will never rarely be perfectly aligned with 0.5s worth of samples, hence
 137     // we will almost always run with a buffer of leftover samples that will
 138     // not get sent to this callback since the recorder will be waiting for
 139     // more data.
 140     const size_t decode_buffer = params_.sample_rate() / 2;  // 0.5s
 141     const size_t expected_samples =
 142         (total_samples_ / decode_buffer - 1) * decode_buffer;
 143     const size_t expected_samples_size =
 144         expected_samples * sizeof(float) * params_.channels();
 145     if (received_samples_.size() == expected_samples_size) {
 146       VerifySamples();
 147       run_loop_->Quit();
 148     }
 149   }
 150
 151   void VerifySamples() {
 152     int differences = 0;
 153
 154     float* buffer_view =
 155         reinterpret_cast<float*>(string_as_array(&received_samples_));
 156     const int channels = params_.channels();
 157     const int frames =
 158         received_samples_.size() / sizeof(float) / params_.channels();
 159     for (int ch = 0; ch < channels; ++ch) {
 160       for (int si = 0, di = ch; si < frames; ++si, di += channels)
 161         differences += (buffer_view[di] != channel_data_[ch][si]);
 162     }
 163
 164     ASSERT_EQ(0, differences);
 165   }
 166
 167  protected:
 168   float* GenerateSamples(int random_seed, size_t size) {
 169     float* samples = static_cast<float*>(base::AlignedAlloc(
 170         size * sizeof(float), media::AudioBus::kChannelAlignment));
 171     PopulateSamples(0x1337, size, samples);
 172     return samples;
 173   }
 174   bool IsRecording() {
 175     recorder_->FlushAudioLoopForTesting();
 176     return recorder_->is_recording_;
 177   }
 178
 179   std::vector<float*> channel_data_;
 180   media::AudioParameters params_;
 181   size_t total_samples_;
 182
 183   AudioRecorder* recorder_;
 184
 185   std::string received_samples_;
 186
 187   scoped_ptr<base::RunLoop> run_loop_;
 188   content::TestBrowserThreadBundle thread_bundle_;
 189 };
 190
 191 // TODO(rkc): These tests are broken on all platforms.
 192 // On Windows and Mac, we cannot use non-OS params. The tests need to be
 193 // rewritten to use the params provided to us by the audio manager
 194 // rather than setting our own params.
 195 // On Linux, there is a memory leak in the audio code during initialization.
 196 #define MAYBE_BasicRecordAndStop DISABLED_BasicRecordAndStop
 197 #define MAYBE_OutOfOrderRecordAndStopMultiple DISABLED_OutOfOrderRecordAndStopMultiple
 198 #define MAYBE_RecordingEndToEnd DISABLED_RecordingEndToEnd
 199
 200 TEST_F(AudioRecorderTest, MAYBE_BasicRecordAndStop) {
 201   CreateSimpleRecorder();
 202
 203   recorder_->Record();
 204   EXPECT_TRUE(IsRecording());
 205   recorder_->Stop();
 206   EXPECT_FALSE(IsRecording());
 207   recorder_->Record();
 208
 209   EXPECT_TRUE(IsRecording());
 210   recorder_->Stop();
 211   EXPECT_FALSE(IsRecording());
 212   recorder_->Record();
 213
 214   EXPECT_TRUE(IsRecording());
 215   recorder_->Stop();
 216   EXPECT_FALSE(IsRecording());
 217
 218   DeleteRecorder();
 219 }
 220
 221 TEST_F(AudioRecorderTest, MAYBE_OutOfOrderRecordAndStopMultiple) {
 222   CreateSimpleRecorder();
 223
 224   recorder_->Stop();
 225   recorder_->Stop();
 226   recorder_->Stop();
 227   EXPECT_FALSE(IsRecording());
 228
 229   recorder_->Record();
 230   recorder_->Record();
 231   EXPECT_TRUE(IsRecording());
 232
 233   recorder_->Stop();
 234   recorder_->Stop();
 235   EXPECT_FALSE(IsRecording());
 236
 237   DeleteRecorder();
 238 }
 239
 240 TEST_F(AudioRecorderTest, MAYBE_RecordingEndToEnd) {
 241   const int kNumSamples = 48000 * 3;
 242   CreateRecorder(
 243       kDefaultChannels, kDefaultSampleRate, kDefaultBitsPerSample, kNumSamples);
 244
 245   RecordAndVerifySamples();
 246
 247   DeleteRecorder();
 248 }
 249
 250 // TODO(rkc): Add tests with recording different sample rates.
 251
 252 }  // namespace copresence