media::WebAudioSourceProviderImpl, add support for getting a copy of passing AudioBuses

media/blink/webaudiosourceprovider_impl_unittest.cc

 // Copyright 2013 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 <stddef.h>
 
+#include "base/bind.h"
 #include "base/macros.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
 #include "media/audio/audio_parameters.h"
 #include "media/base/fake_audio_render_callback.h"
 #include "media/base/mock_audio_renderer_sink.h"
 #include "media/blink/webaudiosourceprovider_impl.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "third_party/WebKit/public/platform/WebAudioSourceProviderClient.h"
 
+using ::testing::_;
+
 namespace media {
 
 namespace {
 const float kTestVolume = 0.25;
 }  // namespace
 
 class WebAudioSourceProviderImplTest
     : public testing::Test,
       public blink::WebAudioSourceProviderClient {
  public:
@@ skipping 50 matching lines @@
                  sizeof(*bus1->channel(ch)) * bus1->frames()) != 0) {
         return false;
       }
     }
     return true;
   }
 
   // blink::WebAudioSourceProviderClient implementation.
   MOCK_METHOD2(setFormat, void(size_t numberOfChannels, float sampleRate));
 
+  // CopyAudioCB. Added forwarder method due to GMock troubles with scoped_ptr.
+  MOCK_METHOD4(DoCopyAudioCB,
+               void(AudioBus*,
+                    uint32_t delay_milliseconds,
+                    int channels,
+                    int sample_rate));
+  void OnAudioBus(scoped_ptr<AudioBus> bus,
+                  uint32_t delay_milliseconds,
+                  int channels,
+                  int sample_rate) {
+    DoCopyAudioCB(bus.get(), delay_milliseconds, channels, sample_rate);
+  }
+
+  int Render(AudioBus* audio_bus) {
+    return wasp_impl_->RenderForTesting(audio_bus);
+  }
+
  protected:
   AudioParameters params_;
   FakeAudioRenderCallback fake_callback_;
   scoped_refptr<MockAudioRendererSink> mock_sink_;
   scoped_refptr<WebAudioSourceProviderImpl> wasp_impl_;
   base::MessageLoop message_loop_;
 
   DISALLOW_COPY_AND_ASSIGN(WebAudioSourceProviderImplTest);
 };
 
@@ skipping 12 matching lines @@
   base::RunLoop().RunUntilIdle();
 
   // setClient() with the same client should do nothing.
   wasp_impl_->setClient(this);
   base::RunLoop().RunUntilIdle();
 }
 
 // Verify AudioRendererSink functionality w/ and w/o a client.
 TEST_F(WebAudioSourceProviderImplTest, SinkMethods) {
   wasp_impl_->Initialize(params_, &fake_callback_);
-  ASSERT_EQ(mock_sink_->callback(), &fake_callback_);
 
   // Without a client all WASP calls should fall through to the underlying sink.
   CallAllSinkMethodsAndVerify(true);
 
   // With a client no calls should reach the Stop()'d sink.  Also, setClient()
   // should propagate the params provided during Initialize() at call time.
   SetClient(this);
   CallAllSinkMethodsAndVerify(false);
 
   // Removing the client should cause WASP to revert to the underlying sink.
@@ skipping 102 matching lines @@
   ASSERT_TRUE(CompareBusses(bus1.get(), bus2.get()));
 
   // Stop() should return silence.
   wasp_impl_->Stop();
   bus1->channel(0)[0] = 1;
   bus2->Zero();
   wasp_impl_->provideInput(audio_data, params_.frames_per_buffer());
   ASSERT_TRUE(CompareBusses(bus1.get(), bus2.get()));
 }
 
+// Verify CopyAudioCB is called if registered.
+TEST_F(WebAudioSourceProviderImplTest, CopyAudioCB) {
+  testing::InSequence s;
+  wasp_impl_->Initialize(params_, &fake_callback_);
+  wasp_impl_->RegisterCopyAudioCallback(base::Bind(
+      &WebAudioSourceProviderImplTest::OnAudioBus, base::Unretained(this)));
+
+  const scoped_ptr<AudioBus> bus1 = AudioBus::Create(params_);
+  EXPECT_CALL(*this,
+              DoCopyAudioCB(_, 0, params_.channels(), params_.sample_rate()))
+      .Times(1);
+  Render(bus1.get());
+
+  wasp_impl_->ResetCopyAudioCallback();
+  EXPECT_CALL(*this, DoCopyAudioCB(_, _, _, _)).Times(0);
+  Render(bus1.get());
+
+  testing::Mock::VerifyAndClear(mock_sink_.get());
+}
+
 }  // namespace media