summaryrefslogtreecommitdiffstats
path: root/testing/firefox-ui/mach_commands.py
diff options
context:
space:
mode:
authorMatt A. Tobin <mattatobin@localhost.localdomain>2018-02-02 04:16:08 -0500
committerMatt A. Tobin <mattatobin@localhost.localdomain>2018-02-02 04:16:08 -0500
commit5f8de423f190bbb79a62f804151bc24824fa32d8 (patch)
tree10027f336435511475e392454359edea8e25895d /testing/firefox-ui/mach_commands.py
parent49ee0794b5d912db1f95dce6eb52d781dc210db5 (diff)
downloadUXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.gz
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.lz
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.tar.xz
UXP-5f8de423f190bbb79a62f804151bc24824fa32d8.zip
Add m-esr52 at 52.6.0
Diffstat (limited to 'testing/firefox-ui/mach_commands.py')
-rw-r--r--testing/firefox-ui/mach_commands.py120
1 files changed, 120 insertions, 0 deletions
diff --git a/testing/firefox-ui/mach_commands.py b/testing/firefox-ui/mach_commands.py
new file mode 100644
index 000000000..368b673a2
--- /dev/null
+++ b/testing/firefox-ui/mach_commands.py
@@ -0,0 +1,120 @@
+# 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'] = 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)