MediaCodecLoop unit tests

media/base/android/media_codec_loop.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 "media/base/android/media_codec_loop.h"
 
-#include "base/android/build_info.h"
 #include "base/bind.h"
 #include "base/callback_helpers.h"
 #include "base/logging.h"
-#include "base/threading/thread_task_runner_handle.h"
-#include "media/base/android/sdk_media_codec_bridge.h"
-#include "media/base/audio_buffer.h"
-#include "media/base/audio_timestamp_helper.h"
 #include "media/base/bind_to_current_loop.h"
 #include "media/base/timestamp_constants.h"
 
 namespace media {
 
-constexpr base::TimeDelta kDecodePollDelay =
-    base::TimeDelta::FromMilliseconds(10);
-constexpr base::TimeDelta kNoWaitTimeout = base::TimeDelta::FromMicroseconds(0);
-constexpr base::TimeDelta kIdleTimerTimeout = base::TimeDelta::FromSeconds(1);
+// Declaring these as constexpr variables doesn't work in windows -- they
+// always are 0.  The exception is FromMicroseconds, which doesn't do any
+// conversion.  However, declaring these as constexpr functions seesm to work
+// fine everywhere.  We care that this works in windows because our unit tests
+// run on non-android platforms.
+constexpr base::TimeDelta DecodePollDelay() {
+  return base::TimeDelta::FromMilliseconds(10);
+}
 
-static inline bool codec_flush_requires_destruction() {
-  // Return true if and only if Flush() isn't supported / doesn't work.
-  // Prior to JellyBean-MR2, flush() had several bugs (b/8125974, b/8347958) so
-  // we have to completely destroy and recreate the codec there.
-  return base::android::BuildInfo::GetInstance()->sdk_int() < 18;
+constexpr base::TimeDelta NoWaitTimeout() {
+  return base::TimeDelta::FromMicroseconds(0);
+}
+
+constexpr base::TimeDelta IdleTimerTimeout() {
+  return base::TimeDelta::FromSeconds(1);
 }
 
 MediaCodecLoop::InputData::InputData() {}
 
 MediaCodecLoop::InputData::InputData(const InputData& other)
     : memory(other.memory),
       length(other.length),
       key_id(other.key_id),
       iv(other.iv),
       subsamples(other.subsamples),
       presentation_time(other.presentation_time),
       is_eos(other.is_eos),
       is_encrypted(other.is_encrypted) {}
 
 MediaCodecLoop::InputData::~InputData() {}
 
-MediaCodecLoop::MediaCodecLoop(Client* client,
-                               std::unique_ptr<MediaCodecBridge> media_codec)
+MediaCodecLoop::MediaCodecLoop(
+    int sdk_int,
+    Client* client,
+    std::unique_ptr<MediaCodecBridge> media_codec,
+    scoped_refptr<base::SingleThreadTaskRunner> task_runner)
     : state_(STATE_READY),
       client_(client),
       media_codec_(std::move(media_codec)),
       pending_input_buf_index_(kInvalidBufferIndex),
+      sdk_int_(sdk_int),
       weak_factory_(this) {
+  if (task_runner)
+    io_timer_.SetTaskRunner(task_runner);
   // TODO(liberato): should this DCHECK?
   if (media_codec_ == nullptr)
     SetState(STATE_ERROR);
 }
 
 MediaCodecLoop::~MediaCodecLoop() {
   io_timer_.Stop();
 }
 
+void MediaCodecLoop::SetTestTickClock(base::TickClock* test_tick_clock) {
+  test_tick_clock_ = test_tick_clock;
+}
+
 void MediaCodecLoop::OnKeyAdded() {
   if (state_ == STATE_WAITING_FOR_KEY)
     SetState(STATE_READY);
 
   DoPendingWork();
 }
 
 bool MediaCodecLoop::TryFlush() {
   // We do not clear the input queue here.  It depends on the caller.
   // For decoder reset, then it is appropriate.  Otherwise, the requests might
   // simply be sent to us later, such as on a format change.
 
   // STATE_DRAINED seems like it allows flush, but it causes test failures.
   // crbug.com/624878
   if (state_ == STATE_ERROR || state_ == STATE_DRAINED)
     return false;
 
-  if (codec_flush_requires_destruction())
+  if (CodecNeedsFlushWorkaround())
     return false;
 
   // Actually try to flush!
   io_timer_.Stop();
 
   if (media_codec_->Flush() != MEDIA_CODEC_OK) {
     // TODO(liberato): we might not want to notify the client about this.
     SetState(STATE_ERROR);
     return false;
   }
@@ skipping 50 matching lines @@
   // MediaCodec's QueueSecureInputBuffer().
   if (pending_input_buf_index_ != kInvalidBufferIndex) {
     InputBuffer result(pending_input_buf_index_, true);
     pending_input_buf_index_ = kInvalidBufferIndex;
     return result;
   }
 
   int input_buf_index = kInvalidBufferIndex;
 
   media::MediaCodecStatus status =
-      media_codec_->DequeueInputBuffer(kNoWaitTimeout, &input_buf_index);
+      media_codec_->DequeueInputBuffer(NoWaitTimeout(), &input_buf_index);
   switch (status) {
     case media::MEDIA_CODEC_DEQUEUE_INPUT_AGAIN_LATER:
       break;
 
     case media::MEDIA_CODEC_ERROR:
       DLOG(ERROR) << __FUNCTION__
                   << ": MEDIA_CODEC_ERROR from DequeInputBuffer";
       SetState(STATE_ERROR);
       break;
 
@@ skipping 81 matching lines @@
 
 bool MediaCodecLoop::ProcessOneOutputBuffer() {
   // TODO(liberato): When merging AVDA, we will also have to ask the client if
   // it can accept another output buffer.
 
   if (state_ == STATE_ERROR)
     return false;
 
   OutputBuffer out;
   MediaCodecStatus status = media_codec_->DequeueOutputBuffer(
-      kNoWaitTimeout, &out.index, &out.offset, &out.size, &out.pts, &out.is_eos,
-      &out.is_key_frame);
+      NoWaitTimeout(), &out.index, &out.offset, &out.size, &out.pts,
+      &out.is_eos, &out.is_key_frame);
 
   bool did_work = false;
   switch (status) {
     case MEDIA_CODEC_OUTPUT_BUFFERS_CHANGED:
       // Output buffers are replaced in MediaCodecBridge, nothing to do.
       did_work = true;
       break;
 
     case MEDIA_CODEC_OUTPUT_FORMAT_CHANGED:
       if (!client_->OnOutputFormatChanged())
@@ skipping 39 matching lines @@
       SetState(STATE_ERROR);
       break;
   }
 
   return did_work;
 }
 
 void MediaCodecLoop::ManageTimer(bool did_work) {
   bool should_be_running = true;
 
-  base::TimeTicks now = base::TimeTicks::Now();
+  // One might also use DefaultTickClock, but then ownership becomes harder.
+  base::TimeTicks now = (test_tick_clock_ ? test_tick_clock_->NowTicks()
+                                          : base::TimeTicks::Now());
   if (did_work || idle_time_begin_ == base::TimeTicks()) {
     idle_time_begin_ = now;
   } else {
     // Make sure that we have done work recently enough, else stop the timer.
-    if (now - idle_time_begin_ > kIdleTimerTimeout)
+    if (now - idle_time_begin_ > IdleTimerTimeout())
       should_be_running = false;
   }
 
   if (should_be_running && !io_timer_.IsRunning()) {
-    io_timer_.Start(FROM_HERE, kDecodePollDelay, this,
+    io_timer_.Start(FROM_HERE, DecodePollDelay(), this,
                     &MediaCodecLoop::DoPendingWork);
   } else if (!should_be_running && io_timer_.IsRunning()) {
     io_timer_.Stop();
   }
 }
 
 void MediaCodecLoop::SetState(State new_state) {
   const State old_state = state_;
   state_ = new_state;
   if (old_state != new_state && new_state == STATE_ERROR)
     client_->OnCodecLoopError();
 }
 
 MediaCodecBridge* MediaCodecLoop::GetCodec() const {
   return media_codec_.get();
 }
 
+bool MediaCodecLoop::CodecNeedsFlushWorkaround() const {
+  // Return true if and only if Flush() isn't supported / doesn't work.
+  // Prior to JellyBean-MR2, flush() had several bugs (b/8125974, b/8347958) so
+  // we have to completely destroy and recreate the codec there.
+  // TODO(liberato): MediaCodecUtil implements the same function.  We should
+  // call that one, except that it doesn't compile outside of android right now.
+  return sdk_int_ < 18;
+}
+
 // static
 const char* MediaCodecLoop::AsString(State state) {
 #define RETURN_STRING(x) \
   case x:                \
     return #x;
 
   switch (state) {
     RETURN_STRING(STATE_READY);
     RETURN_STRING(STATE_WAITING_FOR_KEY);
     RETURN_STRING(STATE_DRAINING);
     RETURN_STRING(STATE_DRAINED);
     RETURN_STRING(STATE_ERROR);
   }
 #undef RETURN_STRING
 
   NOTREACHED() << "Unknown state " << state;
   return nullptr;
 }
 
 }  // namespace media