content/test/gpu/gpu_tests/webcodecs_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 gpu_integration_test
from gpu_tests import path_util

html_path = os.path.join(path_util.GetChromiumSrcDir(), 'content', 'test',
                         'data', 'gpu', 'webcodecs')
data_path = os.path.join(path_util.GetChromiumSrcDir(), 'media', 'test', 'data')

frame_sources = ["camera", "capture", "offscreen", "hw_decoder", "sw_decoder"]
codecs = ["avc1.42001E", "vp8", "vp09.00.10.08"]
accelerations = ["prefer-hardware", "prefer-software"]


class WebCodecsIntegrationTest(gpu_integration_test.GpuIntegrationTest):
  @classmethod
  def Name(cls):
    return 'webcodecs'

  @classmethod
  def GenerateGpuTests(cls, options):
    for source_type in frame_sources:
      yield ('WebCodecs_DrawImage_' + source_type, 'draw-image.html',
             ('{ source_type : "%s" }' % (source_type)))
      yield ('WebCodecs_TexImage2d_' + source_type, 'tex-image-2d.html',
             ('{ source_type : "%s" }' % (source_type)))

    for codec in codecs:
      yield ('WebCodecs_EncodeDecode_' + codec, 'encode-decode.html',
             ('{ codec : "%s" }' % codec))

    for source_type in frame_sources:
      for codec in codecs:
        for acc in accelerations:
          args = (source_type, codec, acc)
          yield ('WebCodecs_Encode_%s_%s_%s' % args, 'encode.html',
                 ('{ source_type : "%s", codec : "%s", acceleration : "%s" }' %
                  args))

    for codec in codecs:
      for acc in accelerations:
        args = ("camera", codec, acc)
        yield ('WebCodecs_Realtime_%s_%s_%s' % args, 'realtime.html',
               ('{ source_type : "%s", codec : "%s", acceleration : "%s" }' %
                args))

  def RunActualGpuTest(self, test_path, *args):
    url = self.UrlOfStaticFilePath(html_path + '/' + test_path)
    tab = self.tab
    arg_obj = args[0]
    tab.Navigate(url)
    tab.action_runner.WaitForJavaScriptCondition(
        'document.readyState == "complete"')
    tab.EvaluateJavaScript('TEST.run(' + str(arg_obj) + ')')
    tab.action_runner.WaitForJavaScriptCondition('TEST.finished', timeout=60)
    if not tab.EvaluateJavaScript('TEST.success'):
      self.fail('Test failure:' + tab.EvaluateJavaScript('TEST.summary()'))

  @classmethod
  def SetUpProcess(cls):
    super(WebCodecsIntegrationTest, cls).SetUpProcess()
    cls.CustomizeBrowserArgs([
        "--enable-blink-features=MediaStreamInsertableStreams",
        '--enable-blink-features=WebCodecs',
        '--use-fake-device-for-media-stream',
        '--use-fake-ui-for-media-stream',
    ])
    cls.StartBrowser()
    cls.SetStaticServerDirs([html_path, data_path])

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


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