Fix pipeline integration tests for seek / duration changes.

media/test/pipeline_integration_test_base.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 "media/test/pipeline_integration_test_base.h"
 
 #include <utility>
 
 #include "base/bind.h"
 #include "base/memory/ptr_util.h"
@@ skipping 32 matching lines @@
 const char kNullVideoHash[] = "d41d8cd98f00b204e9800998ecf8427e";
 const char kNullAudioHash[] = "0.00,0.00,0.00,0.00,0.00,0.00,";
 
 PipelineIntegrationTestBase::PipelineIntegrationTestBase()
     : hashing_enabled_(false),
       clockless_playback_(false),
       pipeline_(new PipelineImpl(message_loop_.task_runner(), new MediaLog())),
       ended_(false),
       pipeline_status_(PIPELINE_OK),
       last_video_frame_format_(PIXEL_FORMAT_UNKNOWN),
-      last_video_frame_color_space_(COLOR_SPACE_UNSPECIFIED) {
+      last_video_frame_color_space_(COLOR_SPACE_UNSPECIFIED),
+      current_duration_(kInfiniteDuration) {
   ResetVideoHash();
 }
 
 PipelineIntegrationTestBase::~PipelineIntegrationTestBase() {
   if (pipeline_->IsRunning())
     Stop();
 
   pipeline_.reset();
   base::RunLoop().RunUntilIdle();
 }
@@ skipping 72 matching lines @@
   hashing_enabled_ = test_type & kHashed;
   clockless_playback_ = test_type & kClockless;
 
   EXPECT_CALL(*this, OnMetadata(_))
       .Times(AtMost(1))
       .WillRepeatedly(SaveArg<0>(&metadata_));
   EXPECT_CALL(*this, OnBufferingStateChange(BUFFERING_HAVE_ENOUGH))
       .Times(AnyNumber());
   EXPECT_CALL(*this, OnBufferingStateChange(BUFFERING_HAVE_NOTHING))
       .Times(AnyNumber());
-  EXPECT_CALL(*this, OnDurationChange()).Times(AtMost(1));
+  // Permit at most two calls to OnDurationChange.  CheckDuration will make sure
+  // that no more than one of them is a finite duration.  This allows the
+  // pipeline to call back at the end of the media with the known duration.
+  EXPECT_CALL(*this, OnDurationChange())
+      .Times(AtMost(2))
+      .WillRepeatedly(
+          Invoke(this, &PipelineIntegrationTestBase::CheckDuration));
   EXPECT_CALL(*this, OnVideoNaturalSizeChange(_)).Times(AtMost(1));
   EXPECT_CALL(*this, OnVideoOpacityChange(_)).Times(AtMost(1));
   CreateDemuxer(std::move(data_source));
 
   if (cdm_context) {
     EXPECT_CALL(*this, DecryptorAttached(true));
     pipeline_->SetCdm(
         cdm_context, base::Bind(&PipelineIntegrationTestBase::DecryptorAttached,
                                 base::Unretained(this)));
   }
@@ skipping 228 matching lines @@
   int result;
   if (frame->metadata()->GetInteger(VideoFrameMetadata::COLOR_SPACE, &result))
     last_video_frame_color_space_ = static_cast<ColorSpace>(result);
   if (!hashing_enabled_ || last_frame_ == frame)
     return;
   last_frame_ = frame;
   DVLOG(3) << __FUNCTION__ << " pts=" << frame->timestamp().InSecondsF();
   VideoFrame::HashFrameForTesting(&md5_context_, frame);
 }
 
+void PipelineIntegrationTestBase::CheckDuration() {
+  // Allow the pipeline to specify indefinite duration, then reduce it once
+  // it becomes known.
+  ASSERT_EQ(kInfiniteDuration, current_duration_);
+  base::TimeDelta new_duration = pipeline_->GetMediaDuration();
+  current_duration_ = new_duration;
+}
+
+base::TimeDelta PipelineIntegrationTestBase::GetStartTime() {
+  return demuxer_->GetStartTime();
+}
+
 void PipelineIntegrationTestBase::ResetVideoHash() {
   DVLOG(1) << __FUNCTION__;
   base::MD5Init(&md5_context_);
 }
 
 std::string PipelineIntegrationTestBase::GetVideoHash() {
   DCHECK(hashing_enabled_);
   base::MD5Digest digest;
   base::MD5Final(&digest, &md5_context_);
   return base::MD5DigestToBase16(digest);
@@ skipping 11 matching lines @@
   DCHECK(clockless_playback_);
   return clockless_audio_sink_->render_time();
 }
 
 base::TimeTicks DummyTickClock::NowTicks() {
   now_ += base::TimeDelta::FromSeconds(60);
   return now_;
 }
 
 }  // namespace media