content/test/gpu/gpu_tests/mediapipe_integration_test.py

# Copyright 2021 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.

from __future__ import print_function

import os
import sys

from gpu_tests import common_browser_args as cba
from gpu_tests import gpu_integration_test
from gpu_tests import path_util

# Tests will be stored individually in subdirectories underneath this base
# directory.
_DATA_PATH = os.path.join(path_util.GetChromiumSrcDir(), 'content', 'test',
                          'data', 'gpu', 'mediapipe')


class MediaPipeIntegrationTest(gpu_integration_test.GpuIntegrationTest):
  """Google MediaPipe Tests

  The tests are generated by MediaPipe's test_harness and placed into a
  zip file that is stored on the cloud.  See mediapipe_update.py.

  Iterate through the subdirectories under _DATA_PATH and runs each test
  individually.
  """

  @classmethod
  def Name(cls):
    return 'mediapipe'

  @classmethod
  def SetUpProcess(cls):
    super(MediaPipeIntegrationTest, cls).SetUpProcess()
    cls.CustomizeBrowserArgs([
        cba.FORCE_BROWSER_CRASH_ON_GPU_CRASH,
    ])

    cls.SetStaticServerDirs([_DATA_PATH])
    cls.StartBrowser()

  @classmethod
  def GenerateGpuTests(cls, options):
    for entry in next(os.walk(_DATA_PATH))[1]:
      yield ('MediaPipe_mediapipe_%s' % entry, _get_test_html(entry), ())

  def RunActualGpuTest(self, url, *_):
    action_runner = self.tab.action_runner
    action_runner.Navigate(self.UrlOfStaticFilePath(url))
    action_runner.WaitForJavaScriptCondition('window.runTest !== undefined')
    action_runner.EvaluateJavaScript('window.runTest()')
    # 120s timeout: some of these tests time out on bots pretty regularly, even
    # though they are fast enough locally. Set a large timeout value to deflake
    # the bots. See crbug.com/1219013 for details.
    action_runner.WaitForJavaScriptCondition('window.isTestComplete()',
                                             timeout=120)
    errors = action_runner.EvaluateJavaScript('window.getErrors()')

    if errors:
      self.fail(errors)

  @classmethod
  def ExpectationsFiles(cls):
    return [
        os.path.join(os.path.dirname(os.path.abspath(__file__)),
                     'test_expectations', 'mediapipe_expectations.txt'),
    ]


def _get_test_html(entry):
  return '%s/_CLICK_ME_TO_RUN_%s_LOCALLY.html' % (entry, entry)


def load_tests(loader, tests, pattern):
  del loader, tests, pattern  # Unused.
  return gpu_integration_test.LoadAllTestsInModule(sys.modules[__name__])