Browser-wide audio mirroring for TabCapture API

media/audio/virtual_audio_output_stream_unittest.cc

+// Copyright (c) 2012 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 "base/message_loop.h"
+#include "base/synchronization/waitable_event.h"
+#include "media/audio/audio_manager.h"
+#include "media/audio/simple_sources.h"
+#include "media/audio/virtual_audio_input_stream.h"
+#include "media/audio/virtual_audio_output_stream.h"
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+using ::testing::_;
+
+namespace media {
+
+class MockVirtualAudioInputStream : public VirtualAudioInputStream {
+ public:
+  MockVirtualAudioInputStream(AudioManagerBase* manager,
+                              AudioParameters params,
+                              base::MessageLoopProxy* message_loop)
+      : VirtualAudioInputStream(manager, params, message_loop) {}
+  ~MockVirtualAudioInputStream() {}
+
+  MOCK_METHOD2(AddOutputStream, void(VirtualAudioOutputStream* stream,
+                                     const AudioParameters& output_params));
+  MOCK_METHOD2(RemoveOutputStream, void(VirtualAudioOutputStream* stream,
+                                        const AudioParameters& output_params));
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(MockVirtualAudioInputStream);
+};
+
+class MockAudioDeviceListener : public AudioManager::AudioDeviceListener {
+ public:
+  MOCK_METHOD0(OnDeviceChange, void());
+};
+
+class VirtualAudioOutputStreamTest : public testing::Test {
+ public:
+  void ListenAndCreateVirtualOnAudioThread(
+      AudioManager* manager, AudioManager::AudioDeviceListener* listener) {
+    manager->AddOutputDeviceChangeListener(listener);
+
+    AudioParameters params(
+        AudioParameters::AUDIO_VIRTUAL, CHANNEL_LAYOUT_MONO, 8000, 8, 128);
+    AudioInputStream* stream = manager->MakeAudioInputStream(params, "1");
+    stream->Close();
+    signal_.Signal();
+  }
+
+  void RemoveListenerOnAudioThread(
+      AudioManager* manager, AudioManager::AudioDeviceListener* listener) {
+    manager->RemoveOutputDeviceChangeListener(listener);
+    signal_.Signal();
+  }
+
+ protected:
+  VirtualAudioOutputStreamTest() : signal_(false, false) {}
+
+  base::WaitableEvent signal_;
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(VirtualAudioOutputStreamTest);
+};
+
+TEST_F(VirtualAudioOutputStreamTest, StartStopStartStop) {
+  scoped_ptr<AudioManager> audio_manager(AudioManager::Create());
+
+  MessageLoop message_loop;
+
+  AudioParameters params(
+      AudioParameters::AUDIO_VIRTUAL, CHANNEL_LAYOUT_MONO, 8000, 8, 128);
+  AudioParameters output_params(
+      AudioParameters::AUDIO_PCM_LINEAR, CHANNEL_LAYOUT_MONO, 8000, 8, 128);
+
+  MockVirtualAudioInputStream input_stream(
+      static_cast<AudioManagerBase*>(audio_manager.get()),
+      params,
+      message_loop.message_loop_proxy());
+
+  EXPECT_CALL(input_stream, AddOutputStream(_, _)).Times(2);
+  EXPECT_CALL(input_stream, RemoveOutputStream(_, _)).Times(2);
+
+  scoped_ptr<VirtualAudioOutputStream> output_stream(
+      VirtualAudioOutputStream::MakeStream(
+          static_cast<AudioManagerBase*>(audio_manager.get()),
+          output_params,
+          message_loop.message_loop_proxy(),
+          &input_stream));
+
+  SineWaveAudioSource source(CHANNEL_LAYOUT_STEREO, 200.0, 128);
+  output_stream->Start(&source);
+  output_stream->Stop();
+  output_stream->Start(&source);
+  output_stream->Stop();
+  // Can't Close() here because we didn't create this output stream is not owned
+  // by the audio manager.
+}
+
+// Tests that we get notifications to reattach output streams when we create a
+// VirtualAudioInputStream.
+TEST_F(VirtualAudioOutputStreamTest, OutputStreamsNotified) {
+  scoped_ptr<AudioManager> audio_manager(AudioManager::Create());
+
+  MockAudioDeviceListener mock_listener;
+  EXPECT_CALL(mock_listener, OnDeviceChange()).Times(2);
+
+  audio_manager->GetMessageLoop()->PostTask(
+      FROM_HERE, base::Bind(
+          &VirtualAudioOutputStreamTest::ListenAndCreateVirtualOnAudioThread,
+          base::Unretained(this),
+          audio_manager.get(),
+          &mock_listener));
+
+  signal_.Wait();
+
+  audio_manager->GetMessageLoop()->PostTask(
+      FROM_HERE, base::Bind(
+          &VirtualAudioOutputStreamTest::RemoveListenerOnAudioThread,
+          base::Unretained(this),
+          audio_manager.get(),
+          &mock_listener));
+
+  signal_.Wait();
+}
+
+}  // namespace media