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)
+      : VirtualAudioInputStream(manager, params) {}
+  ~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);
+};
+
+// Test VirtualAudioOutputStream that overrides the CalledOnAudioThread method.
+class TestVirtualAudioOutputStream : public VirtualAudioOutputStream {
+ public:
+  TestVirtualAudioOutputStream(AudioManagerBase* manager,
+                               AudioParameters params,
+                               VirtualAudioInputStream* target)
+      : VirtualAudioOutputStream(manager, params, target) {}
+
+ private:
+  virtual bool CalledOnAudioThread() OVERRIDE { return true; }
+
+  DISALLOW_COPY_AND_ASSIGN(TestVirtualAudioOutputStream);
+};
+
+class MockAudioDeviceListener : public AudioManager::AudioDeviceListener {
+ public:
+  MOCK_METHOD0(OnDeviceChange, void());
+};
+
+class VirtualAudioOutputStreamTest : public testing::Test {
+ public:
+  void AddListenerAndCreateVirtualOnAudioThread(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());
+
+  AudioParameters params(AudioParameters::AUDIO_VIRTUAL, CHANNEL_LAYOUT_MONO,
+      8000, 8, 128);

[DaleCurtis, 2012/12/03 19:27:28]

Bad indent. Fix here and all over.

[justinlin, 2012/12/03 21:18:42]

Done.

+  AudioParameters output_params(AudioParameters::AUDIO_PCM_LINEAR,
+      CHANNEL_LAYOUT_MONO, 8000, 8, 128);
+
+  MockVirtualAudioInputStream input_stream(
+      static_cast<AudioManagerBase*>(audio_manager.get()), params);
+
+  EXPECT_CALL(input_stream, AddOutputStream(_, _)).Times(2);
+  EXPECT_CALL(input_stream, RemoveOutputStream(_, _)).Times(2);
+
+  TestVirtualAudioOutputStream output_stream(
+      static_cast<AudioManagerBase*>(audio_manager.get()), output_params,
+      &input_stream);
+
+  SineWaveAudioSource source(CHANNEL_LAYOUT_STEREO, 200.0, 128);
+  output_stream.Start(&source);
+  output_stream.Stop();
+  output_stream.Start(&source);
+  output_stream.Stop();

[DaleCurtis, 2012/12/03 19:27:28]

No need to Close()?

[justinlin, 2012/12/03 21:18:42]

We can't close here since 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::AddListenerAndCreateVirtualOnAudioThread,
+      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