summaryrefslogtreecommitdiffstats
path: root/testing/remotecppunittests.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/remotecppunittests.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/remotecppunittests.py')
-rw-r--r--testing/remotecppunittests.py283
1 files changed, 283 insertions, 0 deletions
diff --git a/testing/remotecppunittests.py b/testing/remotecppunittests.py
new file mode 100644
index 000000000..197cae8b4
--- /dev/null
+++ b/testing/remotecppunittests.py
@@ -0,0 +1,283 @@
+#!/usr/bin/env 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/.
+
+import os, sys
+import subprocess
+import tempfile
+from zipfile import ZipFile
+import runcppunittests as cppunittests
+import mozcrash
+import mozfile
+import mozinfo
+import mozlog
+import StringIO
+import posixpath
+from mozdevice import devicemanager, devicemanagerADB, devicemanagerSUT
+
+try:
+ from mozbuild.base import MozbuildObject
+ build_obj = MozbuildObject.from_environment()
+except ImportError:
+ build_obj = None
+
+class RemoteCPPUnitTests(cppunittests.CPPUnitTests):
+ def __init__(self, devmgr, options, progs):
+ cppunittests.CPPUnitTests.__init__(self)
+ self.options = options
+ self.device = devmgr
+ self.remote_test_root = self.device.deviceRoot + "/cppunittests"
+ self.remote_bin_dir = posixpath.join(self.remote_test_root, "b")
+ self.remote_tmp_dir = posixpath.join(self.remote_test_root, "tmp")
+ self.remote_home_dir = posixpath.join(self.remote_test_root, "h")
+ if options.setup:
+ self.setup_bin(progs)
+
+ def setup_bin(self, progs):
+ if not self.device.dirExists(self.remote_test_root):
+ self.device.mkDir(self.remote_test_root)
+ if self.device.dirExists(self.remote_tmp_dir):
+ self.device.removeDir(self.remote_tmp_dir)
+ self.device.mkDir(self.remote_tmp_dir)
+ if self.device.dirExists(self.remote_bin_dir):
+ self.device.removeDir(self.remote_bin_dir)
+ self.device.mkDir(self.remote_bin_dir)
+ if self.device.dirExists(self.remote_home_dir):
+ self.device.removeDir(self.remote_home_dir)
+ self.device.mkDir(self.remote_home_dir)
+ self.push_libs()
+ self.push_progs(progs)
+ self.device.chmodDir(self.remote_bin_dir)
+
+ def push_libs(self):
+ if self.options.local_apk:
+ with mozfile.TemporaryDirectory() as tmpdir:
+ apk_contents = ZipFile(self.options.local_apk)
+
+ for info in apk_contents.infolist():
+ if info.filename.endswith(".so"):
+ print >> sys.stderr, "Pushing %s.." % info.filename
+ remote_file = posixpath.join(self.remote_bin_dir, os.path.basename(info.filename))
+ apk_contents.extract(info, tmpdir)
+ local_file = os.path.join(tmpdir, info.filename)
+ with open(local_file) as f:
+ # Decompress xz-compressed file.
+ if f.read(5)[1:] == '7zXZ':
+ cmd = ['xz', '-df', '--suffix', '.so', local_file]
+ subprocess.check_output(cmd)
+ # xz strips the ".so" file suffix.
+ os.rename(local_file[:-3], local_file)
+ self.device.pushFile(local_file, remote_file)
+
+ elif self.options.local_lib:
+ for file in os.listdir(self.options.local_lib):
+ if file.endswith(".so"):
+ print >> sys.stderr, "Pushing %s.." % file
+ remote_file = posixpath.join(self.remote_bin_dir, file)
+ local_file = os.path.join(self.options.local_lib, file)
+ self.device.pushFile(local_file, remote_file)
+ # Additional libraries may be found in a sub-directory such as "lib/armeabi-v7a"
+ for subdir in ["assets", "lib"]:
+ local_arm_lib = os.path.join(self.options.local_lib, subdir)
+ if os.path.isdir(local_arm_lib):
+ for root, dirs, files in os.walk(local_arm_lib):
+ for file in files:
+ if (file.endswith(".so")):
+ print >> sys.stderr, "Pushing %s.." % file
+ remote_file = posixpath.join(self.remote_bin_dir, file)
+ local_file = os.path.join(root, file)
+ self.device.pushFile(local_file, remote_file)
+
+ def push_progs(self, progs):
+ for local_file in progs:
+ remote_file = posixpath.join(self.remote_bin_dir, os.path.basename(local_file))
+ self.device.pushFile(local_file, remote_file)
+
+ def build_environment(self):
+ env = self.build_core_environment()
+ env['LD_LIBRARY_PATH'] = self.remote_bin_dir
+ env["TMPDIR"]=self.remote_tmp_dir
+ env["HOME"]=self.remote_home_dir
+ env["MOZILLA_FIVE_HOME"] = self.remote_home_dir
+ env["MOZ_XRE_DIR"] = self.remote_bin_dir
+ if self.options.add_env:
+ for envdef in self.options.add_env:
+ envdef_parts = envdef.split("=", 1)
+ if len(envdef_parts) == 2:
+ env[envdef_parts[0]] = envdef_parts[1]
+ elif len(envdef_parts) == 1:
+ env[envdef_parts[0]] = ""
+ else:
+ self.log.warning("invalid --addEnv option skipped: %s" % envdef)
+
+ return env
+
+ def run_one_test(self, prog, env, symbols_path=None, interactive=False,
+ timeout_factor=1):
+ """
+ Run a single C++ unit test program remotely.
+
+ Arguments:
+ * prog: The path to the test program to run.
+ * env: The environment to use for running the program.
+ * symbols_path: A path to a directory containing Breakpad-formatted
+ symbol files for producing stack traces on crash.
+ * timeout_factor: An optional test-specific timeout multiplier.
+
+ Return True if the program exits with a zero status, False otherwise.
+ """
+ basename = os.path.basename(prog)
+ remote_bin = posixpath.join(self.remote_bin_dir, basename)
+ self.log.test_start(basename)
+ buf = StringIO.StringIO()
+ test_timeout = cppunittests.CPPUnitTests.TEST_PROC_TIMEOUT * timeout_factor
+ returncode = self.device.shell([remote_bin], buf, env=env, cwd=self.remote_home_dir,
+ timeout=test_timeout)
+ self.log.process_output(basename, "\n%s" % buf.getvalue(),
+ command=[remote_bin])
+ with mozfile.TemporaryDirectory() as tempdir:
+ self.device.getDirectory(self.remote_home_dir, tempdir)
+ if mozcrash.check_for_crashes(tempdir, symbols_path,
+ test_name=basename):
+ self.log.test_end(basename, status='CRASH', expected='PASS')
+ return False
+ result = returncode == 0
+ if not result:
+ self.log.test_end(basename, status='FAIL', expected='PASS',
+ message=("test failed with return code %s" %
+ returncode))
+ else:
+ self.log.test_end(basename, status='PASS', expected='PASS')
+ return result
+
+class RemoteCPPUnittestOptions(cppunittests.CPPUnittestOptions):
+ def __init__(self):
+ cppunittests.CPPUnittestOptions.__init__(self)
+ defaults = {}
+
+ self.add_option("--deviceIP", action="store",
+ type = "string", dest = "device_ip",
+ help = "ip address of remote device to test")
+ defaults["device_ip"] = None
+
+ self.add_option("--devicePort", action="store",
+ type = "string", dest = "device_port",
+ help = "port of remote device to test")
+ defaults["device_port"] = 20701
+
+ self.add_option("--dm_trans", action="store",
+ type = "string", dest = "dm_trans",
+ help = "the transport to use to communicate with device: [adb|sut]; default=sut")
+ defaults["dm_trans"] = "sut"
+
+ self.add_option("--noSetup", action="store_false",
+ dest = "setup",
+ help = "do not copy any files to device (to be used only if device is already setup)")
+ defaults["setup"] = True
+
+ self.add_option("--localLib", action="store",
+ type = "string", dest = "local_lib",
+ help = "location of libraries to push -- preferably stripped")
+ defaults["local_lib"] = None
+
+ self.add_option("--apk", action="store",
+ type = "string", dest = "local_apk",
+ help = "local path to Fennec APK")
+ defaults["local_apk"] = None
+
+ self.add_option("--localBinDir", action="store",
+ type = "string", dest = "local_bin",
+ help = "local path to bin directory")
+ defaults["local_bin"] = build_obj.bindir if build_obj is not None else None
+
+ self.add_option("--remoteTestRoot", action = "store",
+ type = "string", dest = "remote_test_root",
+ help = "remote directory to use as test root (eg. /data/local/tests)")
+ # /data/local/tests is used because it is usually not possible to set +x permissions
+ # on binaries on /mnt/sdcard
+ defaults["remote_test_root"] = "/data/local/tests"
+
+ self.add_option("--with-b2g-emulator", action = "store",
+ type = "string", dest = "with_b2g_emulator",
+ help = "Start B2G Emulator (specify path to b2g home)")
+ self.add_option("--emulator", default="arm", choices=["x86", "arm"],
+ help = "Architecture of emulator to use: x86 or arm")
+ self.add_option("--addEnv", action = "append",
+ type = "string", dest = "add_env",
+ help = "additional remote environment variable definitions (eg. --addEnv \"somevar=something\")")
+ defaults["add_env"] = None
+
+ self.set_defaults(**defaults)
+
+def run_test_harness(options, args):
+ if options.with_b2g_emulator:
+ from mozrunner import B2GEmulatorRunner
+ runner = B2GEmulatorRunner(arch=options.emulator, b2g_home=options.with_b2g_emulator)
+ runner.start()
+ if options.dm_trans == "adb":
+ if options.with_b2g_emulator:
+ # because we just started the emulator, we need more than the
+ # default number of retries here.
+ retryLimit = 50
+ else:
+ retryLimit = 5
+ try:
+ if options.device_ip:
+ dm = devicemanagerADB.DeviceManagerADB(options.device_ip, options.device_port, packageName=None, deviceRoot=options.remote_test_root, retryLimit=retryLimit)
+ else:
+ dm = devicemanagerADB.DeviceManagerADB(packageName=None, deviceRoot=options.remote_test_root, retryLimit=retryLimit)
+ except:
+ if options.with_b2g_emulator:
+ runner.cleanup()
+ runner.wait()
+ raise
+ else:
+ dm = devicemanagerSUT.DeviceManagerSUT(options.device_ip, options.device_port, deviceRoot=options.remote_test_root)
+ if not options.device_ip:
+ print "Error: you must provide a device IP to connect to via the --deviceIP option"
+ sys.exit(1)
+
+ options.xre_path = os.path.abspath(options.xre_path)
+ cppunittests.update_mozinfo()
+ progs = cppunittests.extract_unittests_from_args(args,
+ mozinfo.info,
+ options.manifest_path)
+ tester = RemoteCPPUnitTests(dm, options, [item[0] for item in progs])
+ try:
+ result = tester.run_tests(progs, options.xre_path, options.symbols_path)
+ finally:
+ if options.with_b2g_emulator:
+ runner.cleanup()
+ runner.wait()
+ return result
+
+def main():
+ parser = RemoteCPPUnittestOptions()
+ mozlog.commandline.add_logging_group(parser)
+ options, args = parser.parse_args()
+ if not args:
+ print >>sys.stderr, """Usage: %s <test binary> [<test binary>...]""" % sys.argv[0]
+ sys.exit(1)
+ if options.local_lib is not None and not os.path.isdir(options.local_lib):
+ print >>sys.stderr, """Error: --localLib directory %s not found""" % options.local_lib
+ sys.exit(1)
+ if options.local_apk is not None and not os.path.isfile(options.local_apk):
+ print >>sys.stderr, """Error: --apk file %s not found""" % options.local_apk
+ sys.exit(1)
+ if not options.xre_path:
+ print >>sys.stderr, """Error: --xre-path is required"""
+ sys.exit(1)
+
+ log = mozlog.commandline.setup_logging("remotecppunittests", options,
+ {"tbpl": sys.stdout})
+ try:
+ result = run_test_harness(options, args)
+ except Exception, e:
+ log.error(str(e))
+ result = False
+ sys.exit(0 if result else 1)
+
+if __name__ == '__main__':
+ main()