author Henrik Skupin <mail@hskupin.info>
Tue, 14 May 2019 19:29:53 +0000
changeset 535711 d12eb6d840fc2dbbf0746b3dbbbd0ca34352ca75
parent 481719 16cad8d7cb82fd2e8cd71e93a285a9e55120d386
child 551640 2cee544a6859c721ea6abe2f47dd492e80658489
permissions -rw-r--r--
Bug 1528236 - [raptor] Fix root permission failures for device tuning. r=perftest-reviewers,stephendonner Differential Revision: https://phabricator.services.mozilla.com/D30922

# 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 (
    MachCommandConditions as conditions,

from mach.decorators import (

def setup_argument_parser_functional():
    from firefox_ui_harness.arguments.base import FirefoxUIArguments
    from mozlog.structured import commandline
    parser = FirefoxUIArguments()
    return parser

def setup_argument_parser_update():
    from firefox_ui_harness.arguments.update import UpdateArguments
    from mozlog.structured import commandline
    parser = UpdateArguments()
    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()
        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']:
        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)


    failed = test_types[testtype]['cli_module'].cli(args=vars(args))

    if failed > 0:
        return 1
        return 0

class MachCommands(MachCommandBase):
    """Mach command provider for Firefox ui tests."""

    @Command('firefox-ui-functional', category='testing',
             description='Run the functional test suite of Firefox UI tests.',
    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',
             description='Run the update test suite of Firefox UI tests.',
    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)