Interpolate media time for mojo rendering pipeline.

media/mojo/clients/mojo_renderer_unittest.cc

 // Copyright 2016 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 <stdint.h>
 
 #include "base/bind.h"
 #include "base/macros.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
 #include "base/threading/platform_thread.h"
+#include "base/timer/elapsed_timer.h"
 #include "media/base/audio_renderer_sink.h"
 #include "media/base/cdm_config.h"
 #include "media/base/cdm_context.h"
 #include "media/base/gmock_callback_support.h"
 #include "media/base/mock_filters.h"
 #include "media/base/test_helpers.h"
 #include "media/base/video_renderer_sink.h"
 #include "media/cdm/default_cdm_factory.h"
 #include "media/mojo/clients/mojo_renderer.h"
 #include "media/mojo/common/media_type_converters.h"
 #include "media/mojo/interfaces/content_decryption_module.mojom.h"
 #include "media/mojo/interfaces/renderer.mojom.h"
 #include "media/mojo/services/mojo_cdm_service.h"
 #include "media/mojo/services/mojo_cdm_service_context.h"
 #include "media/mojo/services/mojo_renderer_service.h"
 #include "media/renderers/video_overlay_factory.h"
 #include "mojo/public/cpp/bindings/interface_request.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using ::testing::_;
 using ::testing::DoAll;
 using ::testing::Return;
 using ::testing::SaveArg;
 using ::testing::StrictMock;
 
 namespace media {
 
+namespace {
 const int64_t kStartPlayingTimeInMs = 100;
 const char kClearKeyKeySystem[] = "org.w3.clearkey";
 
-ACTION_P2(SetError, renderer_client, error) {
-  renderer_client->OnError(error);
+ACTION_P2(GetMediaTime, start_time, elapsed_timer) {
+  return start_time + elapsed_timer->Elapsed();
 }
 
+void WaitFor(base::TimeDelta duration) {
+  base::RunLoop run_loop;
+  base::ThreadTaskRunnerHandle::Get()->PostDelayedTask(
+      FROM_HERE, run_loop.QuitClosure(), duration);
+  run_loop.Run();
+}
+}  // namespace
+
 class MojoRendererTest : public ::testing::Test {
  public:
   MojoRendererTest() {
     std::unique_ptr<StrictMock<MockRenderer>> mock_renderer(
         new StrictMock<MockRenderer>());
     mock_renderer_ = mock_renderer.get();
 
     mojom::RendererPtr remote_renderer;
 
     mojo_renderer_service_ = new MojoRendererService(
@@ skipping 263 matching lines @@
 
 TEST_F(MojoRendererTest, StartPlayingFrom) {
   Initialize();
   Play();
 }
 
 TEST_F(MojoRendererTest, GetMediaTime) {
   Initialize();
   EXPECT_EQ(base::TimeDelta(), mojo_renderer_->GetMediaTime());
 
-  Play();
+  const base::TimeDelta kSleepTime = base::TimeDelta::FromMilliseconds(500);
+  const base::TimeDelta kStartTime =
+      base::TimeDelta::FromMilliseconds(kStartPlayingTimeInMs);
 
-  // Time is updated periodically with a short delay.
-  const base::TimeDelta kUpdatedTime = base::TimeDelta::FromMilliseconds(500);
+  // Media time should not advance since playback rate is 0.
+  EXPECT_CALL(*mock_renderer_, SetPlaybackRate(0));
+  EXPECT_CALL(*mock_renderer_, StartPlayingFrom(kStartTime));
   EXPECT_CALL(*mock_renderer_, GetMediaTime())
-      .WillRepeatedly(Return(kUpdatedTime));
-  base::PlatformThread::Sleep(base::TimeDelta::FromMilliseconds(100));
-  base::RunLoop().RunUntilIdle();
-  EXPECT_EQ(kUpdatedTime, mojo_renderer_->GetMediaTime());
+      .WillRepeatedly(Return(kStartTime));
+  mojo_renderer_->SetPlaybackRate(0);
+  mojo_renderer_->StartPlayingFrom(kStartTime);
+  WaitFor(kSleepTime);
+  EXPECT_EQ(kStartTime, mojo_renderer_->GetMediaTime());
+
+  // Media time should now advance since playback rate is > 0.
+  std::unique_ptr<base::ElapsedTimer> elapsed_timer(new base::ElapsedTimer);
+  EXPECT_CALL(*mock_renderer_, SetPlaybackRate(1.0));
+  EXPECT_CALL(*mock_renderer_, GetMediaTime())
+      .WillRepeatedly(GetMediaTime(kStartTime, elapsed_timer.get()));
+  mojo_renderer_->SetPlaybackRate(1.0);
+  WaitFor(kSleepTime);
+  EXPECT_GT(mojo_renderer_->GetMediaTime(), kStartTime);
+
+  // Flushing should pause media-time updates.
+  EXPECT_CALL(*mock_renderer_, Flush(_)).WillOnce(RunClosure<0>());
+  Flush();
+  base::TimeDelta pause_time = mojo_renderer_->GetMediaTime();
+  EXPECT_GT(pause_time, kStartTime);
+  WaitFor(kSleepTime);
+  EXPECT_EQ(pause_time, mojo_renderer_->GetMediaTime());
+  Destroy();
 }
 
 TEST_F(MojoRendererTest, OnEnded) {
   Initialize();
   Play();
 
   EXPECT_CALL(renderer_client_, OnEnded()).Times(1);
   remote_renderer_client_->OnEnded();
   base::RunLoop().RunUntilIdle();
 }
@@ skipping 16 matching lines @@
       .WillRepeatedly(RunCallback<1>(true));
   EXPECT_CALL(*this, OnCdmAttached(false));
   mojo_renderer_->SetCdm(
       &cdm_context_,
       base::Bind(&MojoRendererTest::OnCdmAttached, base::Unretained(this)));
   Destroy();
 }
 
 TEST_F(MojoRendererTest, Destroy_PendingSetCdm) {
   Initialize();
+
   EXPECT_CALL(*mock_renderer_, Flush(_)).WillRepeatedly(RunClosure<0>());
   EXPECT_CALL(*this, OnFlushed());
   mojo_renderer_->Flush(
       base::Bind(&MojoRendererTest::OnFlushed, base::Unretained(this)));
   Destroy();
 }
 
 // TODO(xhwang): Add more tests on OnError. For example, ErrorDuringFlush,
 // ErrorAfterFlush etc.
 
 TEST_F(MojoRendererTest, ErrorDuringPlayback) {
   Initialize();
 
   EXPECT_CALL(renderer_client_, OnError(PIPELINE_ERROR_DECODE)).Times(1);
 
   Play();
   remote_renderer_client_->OnError(PIPELINE_ERROR_DECODE);
   base::RunLoop().RunUntilIdle();
 
   EXPECT_CALL(*mock_renderer_, SetPlaybackRate(0.0)).Times(1);
   mojo_renderer_->SetPlaybackRate(0.0);
   Flush();
 }
 
 }  // namespace media