gecko-dev/testing/firefox-ui/mach_commands.py

123 строки
4.0 KiB
Python

# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
from __future__ import absolute_import, unicode_literals
import os
import sys
from mozbuild.base import (
MachCommandBase,
MachCommandConditions as conditions,
)
from mach.decorators import (
Command,
CommandProvider,
)
def setup_argument_parser_functional():
from firefox_ui_harness.arguments.base import FirefoxUIArguments
from mozlog.structured import commandline
parser = FirefoxUIArguments()
commandline.add_logging_group(parser)
return parser
def setup_argument_parser_update():
from firefox_ui_harness.arguments.update import UpdateArguments
from mozlog.structured import commandline
parser = UpdateArguments()
commandline.add_logging_group(parser)
return parser
def run_firefox_ui_test(testtype=None, topsrcdir=None, **kwargs):
from mozlog.structured import commandline
from argparse import Namespace
import firefox_ui_harness
if testtype == 'functional':
parser = setup_argument_parser_functional()
else:
parser = setup_argument_parser_update()
test_types = {
'functional': {
'default_tests': [
os.path.join('puppeteer', 'manifest.ini'),
os.path.join('functional', 'manifest.ini'),
],
'cli_module': firefox_ui_harness.cli_functional,
},
'update': {
'default_tests': [
os.path.join('update', 'manifest.ini'),
],
'cli_module': firefox_ui_harness.cli_update,
}
}
fxui_dir = os.path.join(topsrcdir, 'testing', 'firefox-ui')
# Set the resources path which is used to serve test data via wptserve
if not kwargs['server_root']:
kwargs['server_root'] = os.path.join(fxui_dir, 'resources')
# If called via "mach test" a dictionary of tests is passed in
if 'test_objects' in kwargs:
tests = []
for obj in kwargs['test_objects']:
tests.append(obj['file_relpath'])
kwargs['tests'] = tests
elif not kwargs.get('tests'):
# If no tests have been selected, set default ones
kwargs['tests'] = [os.path.join(fxui_dir, 'tests', test)
for test in test_types[testtype]['default_tests']]
kwargs['logger'] = kwargs.pop('log', None)
if not kwargs['logger']:
kwargs['logger'] = commandline.setup_logging('Firefox UI - {} Tests'.format(testtype),
{"mach": sys.stdout})
args = Namespace()
for k, v in kwargs.iteritems():
setattr(args, k, v)
parser.verify_usage(args)
failed = test_types[testtype]['cli_module'].cli(args=vars(args))
if failed > 0:
return 1
else:
return 0
@CommandProvider
class MachCommands(MachCommandBase):
"""Mach command provider for Firefox ui tests."""
@Command('firefox-ui-functional', category='testing',
conditions=[conditions.is_firefox],
description='Run the functional test suite of Firefox UI tests.',
parser=setup_argument_parser_functional,
)
def run_firefox_ui_functional(self, **kwargs):
kwargs['binary'] = kwargs['binary'] or self.get_binary_path('app')
return run_firefox_ui_test(testtype='functional',
topsrcdir=self.topsrcdir, **kwargs)
@Command('firefox-ui-update', category='testing',
conditions=[conditions.is_firefox],
description='Run the update test suite of Firefox UI tests.',
parser=setup_argument_parser_update,
)
def run_firefox_ui_update(self, **kwargs):
kwargs['binary'] = kwargs['binary'] or self.get_binary_path('app')
return run_firefox_ui_test(testtype='update',
topsrcdir=self.topsrcdir, **kwargs)