Convert audio samples in Whispernet.

components/audio_modem/audio_recorder_unittest.cc

 // Copyright 2015 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 "components/audio_modem/audio_recorder.h"
 
 #include <vector>
 
 #include "base/bind.h"
 #include "base/memory/aligned_memory.h"
 #include "base/run_loop.h"
 #include "components/audio_modem/audio_recorder_impl.h"
 #include "components/audio_modem/public/audio_modem_types.h"
 #include "components/audio_modem/test/random_samples.h"
 #include "content/public/test/test_browser_thread_bundle.h"
 #include "media/audio/audio_manager.h"
 #include "media/audio/audio_manager_base.h"
 #include "media/base/audio_bus.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace {
 
+const size_t kSomeNumber = 0x9999;
+
 class TestAudioInputStream : public media::AudioInputStream {
  public:
   TestAudioInputStream(const media::AudioParameters& params,
                        const std::vector<float*> channel_data,
                        size_t samples)
       : callback_(nullptr), params_(params) {
     buffer_ = media::AudioBus::CreateWrapper(2);
     for (size_t i = 0; i < channel_data.size(); ++i)
       buffer_->SetChannelData(i, channel_data[i]);
     buffer_->set_frames(samples);
@@ skipping 47 matching lines @@
       media::AudioManager::CreateForTesting();
   }
 
   ~AudioRecorderTest() override {
     DeleteRecorder();
     for (size_t i = 0; i < channel_data_.size(); ++i)
       base::AlignedFree(channel_data_[i]);
   }
 
   void CreateSimpleRecorder() {
-    DeleteRecorder();
-    recorder_ = new AudioRecorderImpl();
-    recorder_->Initialize(
-        base::Bind(&AudioRecorderTest::DecodeSamples, base::Unretained(this)));
+    // If we have input devices, we'll create a recorder which uses a real
+    // input stream, if not, we'll create a recorder which uses our mock input
+    // stream.
+    if (media::AudioManager::Get()->HasAudioInputDevices()) {
+      DeleteRecorder();
+      recorder_ = new AudioRecorderImpl();
+      recorder_->Initialize(base::Bind(&AudioRecorderTest::DecodeSamples,
+                                       base::Unretained(this)));
+    } else {
+      CreateRecorder(kSomeNumber);
+    }
   }
 
-  void CreateRecorder(size_t channels,
-                      size_t sample_rate,
-                      size_t bits_per_sample,
-                      size_t samples) {
+  void CreateRecorder(size_t samples) {
     DeleteRecorder();
-    params_.Reset(media::AudioParameters::AUDIO_PCM_LOW_LATENCY,
-                  kDefaultChannelLayout,
-                  channels,
-                  sample_rate,
-                  bits_per_sample,
-                  4096);
+
+    params_ = media::AudioManager::Get()->GetInputStreamParameters(
+        media::AudioManagerBase::kDefaultDeviceId);
 
     channel_data_.clear();
     channel_data_.push_back(GenerateSamples(0x1337, samples));
     channel_data_.push_back(GenerateSamples(0x7331, samples));
 
     total_samples_ = samples;
 
     recorder_ = new AudioRecorderImpl();
     recorder_->set_input_stream_for_testing(
         new TestAudioInputStream(params_, channel_data_, samples));
@@ skipping 69 matching lines @@
 
   // Deleted by calling Finalize() on the object.
   AudioRecorderImpl* recorder_;
 
   std::string received_samples_;
 
   scoped_ptr<base::RunLoop> run_loop_;
   content::TestBrowserThreadBundle thread_bundle_;
 };
 
-// TODO(rkc): These tests are broken on all platforms.
-// On Windows and Mac, we cannot use non-OS params. The tests need to be
-// rewritten to use the params provided to us by the audio manager
-// rather than setting our own params.
-// On Linux, there is a memory leak in the audio code during initialization.
-#define MAYBE_BasicRecordAndStop DISABLED_BasicRecordAndStop
-#define MAYBE_OutOfOrderRecordAndStopMultiple \
-    DISABLED_OutOfOrderRecordAndStopMultiple
-#define MAYBE_RecordingEndToEnd DISABLED_RecordingEndToEnd
-
-TEST_F(AudioRecorderTest, MAYBE_BasicRecordAndStop) {
+TEST_F(AudioRecorderTest, BasicRecordAndStop) {
   CreateSimpleRecorder();
 
   recorder_->Record();
   EXPECT_TRUE(IsRecording());
   recorder_->Stop();
   EXPECT_FALSE(IsRecording());
   recorder_->Record();
 
   EXPECT_TRUE(IsRecording());
   recorder_->Stop();
   EXPECT_FALSE(IsRecording());
   recorder_->Record();
 
   EXPECT_TRUE(IsRecording());
   recorder_->Stop();
   EXPECT_FALSE(IsRecording());
 
   DeleteRecorder();
 }
 
-TEST_F(AudioRecorderTest, MAYBE_OutOfOrderRecordAndStopMultiple) {
+TEST_F(AudioRecorderTest, OutOfOrderRecordAndStopMultiple) {
   CreateSimpleRecorder();
 
   recorder_->Stop();
   recorder_->Stop();
   recorder_->Stop();
   EXPECT_FALSE(IsRecording());
 
   recorder_->Record();
   recorder_->Record();
   EXPECT_TRUE(IsRecording());
 
   recorder_->Stop();
   recorder_->Stop();
   EXPECT_FALSE(IsRecording());
 
   DeleteRecorder();
 }
 
-TEST_F(AudioRecorderTest, MAYBE_RecordingEndToEnd) {
+TEST_F(AudioRecorderTest, RecordingEndToEnd) {
   const int kNumSamples = 48000 * 3;
-  CreateRecorder(
-      kDefaultChannels, kDefaultSampleRate, kDefaultBitsPerSample, kNumSamples);
+  CreateRecorder(kNumSamples);
 
   RecordAndVerifySamples();
 
   DeleteRecorder();
 }
 
 // TODO(rkc): Add tests with recording different sample rates.
 
 }  // namespace audio_modem