1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
|
# 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/.
import json
import os
import sys
from argparse import Namespace
from functools import partial
import six
from mach.decorators import Command
here = os.path.abspath(os.path.dirname(__file__))
parser = None
logger = None
def run_test(context, is_junit, **kwargs):
from mochitest_options import ALL_FLAVORS
from mozlog.commandline import setup_logging
if not kwargs.get("log"):
kwargs["log"] = setup_logging("mochitest", kwargs, {"mach": sys.stdout})
global logger
logger = kwargs["log"]
flavor = kwargs.get("flavor") or "mochitest"
if flavor not in ALL_FLAVORS:
for fname, fobj in six.iteritems(ALL_FLAVORS):
if flavor in fobj["aliases"]:
flavor = fname
break
fobj = ALL_FLAVORS[flavor]
kwargs.update(fobj.get("extra_args", {}))
if kwargs.get("allow_software_gl_layers"):
os.environ["MOZ_LAYERS_ALLOW_SOFTWARE_GL"] = "1"
del kwargs["allow_software_gl_layers"]
if kwargs.get("mochitest_suite"):
suite = kwargs["mochitest_suite"]
del kwargs["mochitest_suite"]
elif kwargs.get("test_suite"):
suite = kwargs["test_suite"]
del kwargs["test_suite"]
if kwargs.get("no_run_tests"):
del kwargs["no_run_tests"]
args = Namespace(**kwargs)
args.e10s = context.mozharness_config.get("e10s", args.e10s)
args.certPath = context.certs_dir
if is_junit:
return run_geckoview_junit(context, args)
# subsuite mapping from mozharness configs
subsuites = {
# --mochitest-suite -> --subsuite
"mochitest-chrome-gpu": "gpu",
"mochitest-plain-gpu": "gpu",
"mochitest-media": "media",
"mochitest-browser-chrome-screenshots": "screenshots",
"mochitest-webgl1-core": "webgl1-core",
"mochitest-webgl1-ext": "webgl1-ext",
"mochitest-webgl2-core": "webgl2-core",
"mochitest-webgl2-ext": "webgl2-ext",
"mochitest-webgl2-deqp": "webgl2-deqp",
"mochitest-webgpu": "webgpu",
"mochitest-devtools-chrome": "devtools",
"mochitest-browser-a11y": "a11y",
"mochitest-remote": "remote",
"mochitest-browser-media": "media-bc",
}
args.subsuite = subsuites.get(suite)
if args.subsuite == "devtools":
args.flavor = "browser"
if args.subsuite == "a11y":
args.flavor = "browser"
if args.subsuite == "media-bc":
args.flavor = "browser"
if not args.test_paths:
mh_test_paths = json.loads(os.environ.get("MOZHARNESS_TEST_PATHS", '""'))
if mh_test_paths:
logger.info("Found MOZHARNESS_TEST_PATHS:")
logger.info(str(mh_test_paths))
args.test_paths = []
for k in mh_test_paths:
args.test_paths.extend(mh_test_paths[k])
if args.test_paths:
install_subdir = fobj.get("install_subdir", fobj["suite"])
test_root = os.path.join(context.package_root, "mochitest", install_subdir)
normalize = partial(context.normalize_test_path, test_root)
# pylint --py3k: W1636
args.test_paths = list(map(normalize, args.test_paths))
import mozinfo
if mozinfo.info.get("buildapp") == "mobile/android":
return run_mochitest_android(context, args)
return run_mochitest_desktop(context, args)
def run_mochitest_desktop(context, args):
args.app = args.app or context.firefox_bin
args.utilityPath = context.bin_dir
args.extraProfileFiles.append(os.path.join(context.bin_dir, "plugins"))
args.extraPrefs.append("webgl.force-enabled=true")
args.quiet = True
args.useTestMediaDevices = True
args.screenshotOnFail = True
args.cleanupCrashes = True
args.marionette_startup_timeout = "180"
args.sandboxReadWhitelist.append(context.mozharness_workdir)
if args.flavor == "browser":
args.chunkByRuntime = True
else:
args.chunkByDir = 4
from runtests import run_test_harness
logger.info("mach calling runtests with args: " + str(args))
return run_test_harness(parser, args)
def set_android_args(context, args):
args.app = args.app or "org.mozilla.geckoview.test_runner"
args.utilityPath = context.hostutils
args.xrePath = context.hostutils
config = context.mozharness_config
if config:
host = os.environ.get("HOST_IP", "10.0.2.2")
args.remoteWebServer = config.get("remote_webserver", host)
args.httpPort = config.get("http_port", 8854)
args.sslPort = config.get("ssl_port", 4454)
args.adbPath = config["exes"]["adb"] % {
"abs_work_dir": context.mozharness_workdir
}
args.deviceSerial = os.environ.get("DEVICE_SERIAL", "emulator-5554")
return args
def run_mochitest_android(context, args):
args = set_android_args(context, args)
args.extraProfileFiles.append(
os.path.join(context.package_root, "mochitest", "fonts")
)
from runtestsremote import run_test_harness
logger.info("mach calling runtestsremote with args: " + str(args))
return run_test_harness(parser, args)
def run_geckoview_junit(context, args):
args = set_android_args(context, args)
from runjunit import run_test_harness
# Force fission disabled by default for android
args["disable_fission"] = True
logger.info("mach calling runjunit with args: " + str(args))
return run_test_harness(parser, args)
def add_global_arguments(parser):
parser.add_argument("--test-suite")
parser.add_argument("--mochitest-suite")
parser.add_argument("--download-symbols")
parser.add_argument("--allow-software-gl-layers", action="store_true")
parser.add_argument("--no-run-tests", action="store_true")
def setup_mochitest_argument_parser():
import mozinfo
mozinfo.find_and_update_from_json(here)
app = "generic"
if mozinfo.info.get("buildapp") == "mobile/android":
app = "android"
from mochitest_options import MochitestArgumentParser
global parser
parser = MochitestArgumentParser(app=app)
add_global_arguments(parser)
return parser
def setup_junit_argument_parser():
from runjunit import JunitArgumentParser
global parser
parser = JunitArgumentParser()
add_global_arguments(parser)
return parser
@Command(
"mochitest",
category="testing",
description="Run the mochitest harness.",
parser=setup_mochitest_argument_parser,
)
def mochitest(command_context, **kwargs):
command_context._mach_context.activate_mozharness_venv()
return run_test(command_context._mach_context, False, **kwargs)
@Command(
"geckoview-junit",
category="testing",
description="Run the geckoview-junit harness.",
parser=setup_junit_argument_parser,
)
def geckoview_junit(command_context, **kwargs):
command_context._mach_context.activate_mozharness_venv()
return run_test(command_context._mach_context, True, **kwargs)
|
