Decouple 'give me more data' and 'rendered end of stream' audio callbacks.

content/renderer/media/audio_renderer_impl_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/bind.h"
 #include "base/bind_helpers.h"
+#include "base/gtest_prod_util.h"
 #include "base/message_loop.h"
 #include "base/process_util.h"
 #include "base/synchronization/waitable_event.h"
 #include "base/test/test_timeouts.h"
 #include "base/time.h"
 #include "content/common/child_process.h"
 #include "content/common/child_thread.h"
 #include "content/renderer/media/audio_renderer_impl.h"
 #include "content/renderer/mock_content_renderer_client.h"
 #include "content/renderer/render_process.h"
@@ skipping 33 matching lines @@
 void WaitCallback(base::WaitableEvent* event) {
   event->Signal();
 }
 
 // Class we would be testing.
 class TestAudioRendererImpl : public AudioRendererImpl {
  public:
   explicit TestAudioRendererImpl(media::AudioRendererSink* sink)
       : AudioRendererImpl(sink) {
   }
+
+ protected:
+  FRIEND_TEST_ALL_PREFIXES(AudioRendererImplTest, OnRenderEndOfStream);
+  MOCK_METHOD0(SignalEndOfStream, void());
+
+ private:
+  FRIEND_TEST_ALL_PREFIXES(AudioRendererImplTest, OnRenderEndOfStream);
+  MOCK_METHOD0(OnRenderEndOfStreamDelay, int64());
 };
 
 class AudioRendererImplTest
     : public ::testing::Test,
       public IPC::Channel::Listener {
  public:
   // IPC::Channel::Listener implementation.
   virtual bool OnMessageReceived(const IPC::Message& message) {
       NOTIMPLEMENTED();
       return true;
@@ skipping 73 matching lines @@
     EXPECT_TRUE(event_->TimedWait(
         base::TimeDelta::FromMilliseconds(TestTimeouts::action_timeout_ms())));
   }
 
   MessageLoopForIO message_loop_;
   content::MockContentRendererClient mock_content_renderer_client_;
   scoped_ptr<IPC::Channel> channel_;
   RenderThreadImpl* render_thread_;  // owned by mock_process_
   scoped_ptr<MockRenderProcess> mock_process_;
   scoped_refptr<media::MockAudioDecoder> decoder_;
-  scoped_refptr<AudioRendererImpl> renderer_;
+  scoped_refptr<TestAudioRendererImpl> renderer_;
   scoped_ptr<base::WaitableEvent> event_;
 
  private:
   DISALLOW_COPY_AND_ASSIGN(AudioRendererImplTest);
 };
 
 const int AudioRendererImplTest::kSize = 1024;
 
 TEST_F(AudioRendererImplTest, SetPlaybackRate) {
   // Execute SetPlaybackRate() codepath by toggling play/pause.
@@ skipping 24 matching lines @@
   base::Time time_now = base::Time();  // Null time by default.
   renderer_->set_earliest_end_time(time_now);
   renderer_->UpdateEarliestEndTime(renderer_->bytes_per_second(),
                                    base::TimeDelta::FromMilliseconds(100),
                                    time_now);
   int time_delta = (renderer_->earliest_end_time() - time_now).InMilliseconds();
   EXPECT_EQ(1100, time_delta);
   renderer_->Stop(media::NewExpectedClosure());
   WaitForIOThreadCompletion();
 }
+
+TEST_F(AudioRendererImplTest, OnRenderEndOfStream) {
+  renderer_->SetPlaybackRate(1.0f);
+  WaitForIOThreadCompletion();
+  renderer_->set_recieved_end_of_stream(true);
+  EXPECT_CALL(*renderer_, OnRenderEndOfStreamDelay())
+      .WillOnce(Return(0));
+  EXPECT_CALL(*renderer_, SignalEndOfStream());
+  renderer_->OnRenderEndOfStream();
+  renderer_->Stop(media::NewExpectedClosure());
+  WaitForIOThreadCompletion();
+}