Allow unit tests to use a mock audio input controller.

media/audio/audio_input_controller.h

 // Copyright (c) 2010 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.
 
 #ifndef MEDIA_AUDIO_AUDIO_INPUT_CONTROLLER_H_
 #define MEDIA_AUDIO_AUDIO_INPUT_CONTROLLER_H_
 
 #include "base/lock.h"
 #include "base/ref_counted.h"
 #include "base/scoped_ptr.h"
@@ skipping 31 matching lines @@
   class EventHandler {
    public:
     virtual ~EventHandler() {}
     virtual void OnCreated(AudioInputController* controller) = 0;
     virtual void OnRecording(AudioInputController* controller) = 0;
     virtual void OnError(AudioInputController* controller, int error_code) = 0;
     virtual void OnData(AudioInputController* controller, const uint8* data,
                         uint32 size) = 0;
   };
 
+  // AudioInputController::Create uses the currently registered Factory to
+  // create the AudioInputController. Factory is intended for testing.
+  class Factory {
+   public:
+    virtual AudioInputController* Create(EventHandler* event_handler,
+                                         AudioManager::Format format,
+                                         int channels,
+                                         int sample_rate,
+                                         int bits_per_sample,
+                                         int samples_per_packet) = 0;
+
+   protected:
+    virtual ~Factory() {}
+  };
+
   virtual ~AudioInputController();
 
   // Factory method for creating an AudioInputController.
   // If successful, an audio input controller thread is created. The audio
   // device will be created on the new thread and when that is done event
   // handler will receive a OnCreated() call.
   static scoped_refptr<AudioInputController> Create(
       EventHandler* event_handler,
       AudioManager::Format format,    // Format of the stream.
       int channels,                   // Number of channels.
       int sample_rate,                // Sampling frequency/rate.
       int bits_per_sample,            // Number of bits per sample.
       int samples_per_packet);        // Size of the hardware buffer.
 
+  // Sets the factory used by the static method Create. AudioInputController
+  // does not take ownership of |factory|. A value of NULL results in an
+  // AudioInputController being created directly.
+#if defined(UNIT_TEST)
+  static void set_factory(Factory* factory) { factory_ = factory; }
+#endif
+
   // Starts recording in this audio input stream.
-  void Record();
+  virtual void Record();
 
   // Closes the audio input stream and shutdown the audio input controller
   // thread. This method returns only after all operations are completed. This
   // input controller cannot be used after this method is called.
   //
   // It is safe to call this method more than once. Calls after the first one
   // will have no effect.
-  void Close();
+  virtual void Close();
 
   ///////////////////////////////////////////////////////////////////////////
   // AudioInputCallback methods.
   virtual void OnData(AudioInputStream* stream, const uint8* src, uint32 size);
   virtual void OnClose(AudioInputStream* stream);
   virtual void OnError(AudioInputStream* stream, int code);
 
- private:
+ protected:
   // Internal state of the source.
   enum State {
     kEmpty,
     kCreated,
     kRecording,
     kClosed,
     kError
   };
 
   AudioInputController(EventHandler* handler);
@@ skipping 12 matching lines @@
   // |state_| is written on the audio input controller thread and is read on
   // the hardware audio thread. These operations need to be locked. But lock
   // is not required for reading on the audio input controller thread.
   State state_;
 
   Lock lock_;
 
   // The audio input controller thread that this object runs on.
   base::Thread thread_;
 
+  static Factory* factory_;
+
   DISALLOW_COPY_AND_ASSIGN(AudioInputController);
 };
 
 }  // namespace media
 
 #endif  //  MEDIA_AUDIO_AUDIO_INPUT_CONTROLLER_H_