vea_unittest: Add "--disable_flush" argument.

We added Flush() function in VEA interface, but not all devices and
drivers support this. This CL adds a new argument "--disable_flush"
into VEA unittest to skip checking the flush function.

Bug: chromium:829276
Test: Pass video_encode_accelerator_unittest --disable_flush at
      peach-pit device

Cq-Include-Trybots: luci.chromium.try:android_optional_gpu_tests_rel;luci.chromium.try:linux_optional_gpu_tests_rel;luci.chromium.try:mac_optional_gpu_tests_rel;master.tryserver.chromium.win:win_optional_gpu_tests_rel
Change-Id: I82477a75178ec6aeb3ebf382b601005455c957a4
Reviewed-on: https://chromium-review.googlesource.com/999233Reviewed-by: Ricky Liang <jcliang@chromium.org>
Commit-Queue: Chih-Yu Huang <akahuang@chromium.org>
Cr-Commit-Position: refs/heads/master@{#549508}

media/gpu/video_encode_accelerator_unittest.cc

@@ -147,6 +147,11 @@ const base::FilePath::CharType* g_default_in_parameters =
 // test.
 bool g_fake_encoder = false;
 
+// Skip checking the flush functionality. Currently only Chrome OS devices
+// support flush, so we need this to skip the check. This is set by the
+// command line switch "--disable_flush".
+bool g_disable_flush = false;
+
 // Environment to store test stream data for all test cases.
 class VideoEncodeAcceleratorTestEnvironment;
 VideoEncodeAcceleratorTestEnvironment* g_env;
@@ -1193,6 +1198,7 @@ class VEAClient : public VEAClientBase {
   // frames at BitstreamBufferReady() and verify the number after flush is
   // completed.
   void FlushEncoderDone(bool success);
+  void FlushEncoderSuccessfully();
 
   // Timeout function to check the flush callback function is called in the
   // short period.
@@ -1649,6 +1655,10 @@ void VEAClient::BitstreamBufferReady(int32_t bitstream_buffer_id,
           static_cast<int>(payload_size)));
       quality_validator_->AddDecodeBuffer(buffer);
     }
+    // If flush is disabled, pretend flush is done when all frames are received.
+    if (g_disable_flush && num_encoded_frames_ == num_frames_to_encode_) {
+      FlushEncoderSuccessfully();
+    }
 
     if (save_to_file_) {
       if (IsVP8(test_stream_->requested_profile))
@@ -1918,6 +1928,9 @@ void VEAClient::FlushEncoder() {
   DCHECK(thread_checker_.CalledOnValidThread());
   LOG_ASSERT(num_frames_submitted_to_encoder_ == num_frames_to_encode_);
 
+  if (g_disable_flush)
+    return;
+
   encoder_->Flush(
       base::BindOnce(&VEAClient::FlushEncoderDone, base::Unretained(this)));
   // We might receive the last frame before calling Flush(). In this case we set
@@ -1941,6 +1954,11 @@ void VEAClient::FlushEncoderDone(bool success) {
     SetState(CS_ERROR);
     return;
   }
+  FlushEncoderSuccessfully();
+}
+
+void VEAClient::FlushEncoderSuccessfully() {
+  DCHECK(thread_checker_.CalledOnValidThread());
 
   SetState(CS_FLUSHED);
   if (!quality_validator_) {
@@ -2530,6 +2548,12 @@ int main(int argc, char** argv) {
   DCHECK(cmd_line);
 
   bool run_at_fps = false;
+#if defined(OS_CHROMEOS)
+  // Currently only ARC++ uses flush function, we only verify it on Chrome OS.
+  media::g_disable_flush = false;
+#else
+  media::g_disable_flush = true;
+#endif
   bool needs_encode_latency = false;
   bool verify_all_output = false;
   base::FilePath log_path;
@@ -2565,6 +2589,10 @@ int main(int argc, char** argv) {
       run_at_fps = true;
       continue;
     }
+    if (it->first == "disable_flush") {
+      media::g_disable_flush = true;
+      continue;
+    }
     if (it->first == "verify_all_output") {
       verify_all_output = true;
       continue;