#!/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 sys, os import subprocess import runxpcshelltests as xpcshell import tempfile from automationutils import replaceBackSlashes from mozdevice import devicemanagerADB, devicemanagerSUT, DMError here = os.path.dirname(os.path.abspath(__file__)) # A specialization of XPCShellTests that runs tests on an Android device # via devicemanager. class XPCShellRemote(xpcshell.XPCShellTests, object): def __init__(self, devmgr, options, args): xpcshell.XPCShellTests.__init__(self) self.localLib = options.localLib self.localBin = options.localBin self.options = options self.device = devmgr self.pathMapping = [] self.remoteTestRoot = self.device.getTestRoot("xpcshell") # Terse directory names are used here ("b" for a binaries directory) # to minimize the length of the command line used to execute # xpcshell on the remote device. adb has a limit to the number # of characters used in a shell command, and the xpcshell command # line can be quite complex. self.remoteBinDir = self.remoteJoin(self.remoteTestRoot, "b") self.remoteTmpDir = self.remoteJoin(self.remoteTestRoot, "tmp") self.remoteScriptsDir = self.remoteTestRoot self.remoteComponentsDir = self.remoteJoin(self.remoteTestRoot, "c") self.remoteModulesDir = self.remoteJoin(self.remoteTestRoot, "m") self.profileDir = self.remoteJoin(self.remoteTestRoot, "p") self.remoteDebugger = options.debugger self.remoteDebuggerArgs = options.debuggerArgs self.testingModulesDir = options.testingModulesDir if self.options.objdir: self.xpcDir = os.path.join(self.options.objdir, "_tests/xpcshell") elif os.path.isdir(os.path.join(here, 'tests')): self.xpcDir = os.path.join(here, 'tests') else: print >> sys.stderr, "Couldn't find local xpcshell test directory" sys.exit(1) if options.setup: self.setupUtilities() self.setupModules() self.setupTestDir() if options.localAPK: self.remoteAPK = self.remoteJoin(self.remoteBinDir, os.path.basename(options.localAPK)) self.setAppRoot() def setAppRoot(self): # Determine the application root directory associated with the package # name used by the Fennec APK. self.appRoot = None packageName = None if self.options.localAPK: try: packageName = subprocess.check_output(["unzip", "-p", self.options.localAPK, "package-name.txt"]) if packageName: self.appRoot = self.device.getAppRoot(packageName.strip()) except Exception as detail: print "unable to determine app root: " + detail pass return None def remoteJoin(self, path1, path2): joined = os.path.join(path1, path2) joined = joined.replace('\\', '/') return joined def remoteForLocal(self, local): for mapping in self.pathMapping: if (os.path.abspath(mapping.local) == os.path.abspath(local)): return mapping.remote return local def setupUtilities(self): remotePrefDir = self.remoteJoin(self.remoteBinDir, "defaults/pref") if (self.device.dirExists(self.remoteTmpDir)): self.device.removeDir(self.remoteTmpDir) self.device.mkDir(self.remoteTmpDir) if (not self.device.dirExists(remotePrefDir)): self.device.mkDirs(self.remoteJoin(remotePrefDir, "extra")) if (not self.device.dirExists(self.remoteScriptsDir)): self.device.mkDir(self.remoteScriptsDir) if (not self.device.dirExists(self.remoteComponentsDir)): self.device.mkDir(self.remoteComponentsDir) local = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'head.js') remoteFile = self.remoteJoin(self.remoteScriptsDir, "head.js") self.device.pushFile(local, remoteFile) local = os.path.join(self.localBin, "xpcshell") remoteFile = self.remoteJoin(self.remoteBinDir, "xpcshell") self.device.pushFile(local, remoteFile) local = os.path.join(self.localBin, "components/httpd.js") remoteFile = self.remoteJoin(self.remoteComponentsDir, "httpd.js") self.device.pushFile(local, remoteFile) local = os.path.join(self.localBin, "components/httpd.manifest") remoteFile = self.remoteJoin(self.remoteComponentsDir, "httpd.manifest") self.device.pushFile(local, remoteFile) local = os.path.join(self.localBin, "components/test_necko.xpt") remoteFile = self.remoteJoin(self.remoteComponentsDir, "test_necko.xpt") self.device.pushFile(local, remoteFile) if self.options.localAPK: remoteFile = self.remoteJoin(self.remoteBinDir, os.path.basename(self.options.localAPK)) self.device.pushFile(self.options.localAPK, remoteFile) self.pushLibs() def pushLibs(self): for file in os.listdir(self.localLib): if (file.endswith(".so")): print >> sys.stderr, "Pushing %s.." % file if 'libxul' in file: print >> sys.stderr, "This is a big file, it could take a while." remoteFile = self.remoteJoin(self.remoteBinDir, file) self.device.pushFile(os.path.join(self.localLib, file), remoteFile) # Additional libraries may be found in a sub-directory such as "lib/armeabi-v7a" localArmLib = os.path.join(self.localLib, "lib") if os.path.exists(localArmLib): for root, dirs, files in os.walk(localArmLib): for file in files: if (file.endswith(".so")): remoteFile = self.remoteJoin(self.remoteBinDir, file) self.device.pushFile(os.path.join(root, file), remoteFile) def setupModules(self): if self.testingModulesDir: self.device.pushDir(self.testingModulesDir, self.remoteModulesDir) def setupTestDir(self): push_attempts = 10 for retry in range(1, push_attempts+1): print 'pushing', self.xpcDir, '(attempt %s of %s)' % (retry, push_attempts) try: self.device.pushDir(self.xpcDir, self.remoteScriptsDir) break except DMError: if retry == push_attempts: raise def buildTestList(self): xpcshell.XPCShellTests.buildTestList(self) uniqueTestPaths = set([]) for test in self.alltests: uniqueTestPaths.add(test['here']) for testdir in uniqueTestPaths: abbrevTestDir = os.path.relpath(testdir, self.xpcDir) remoteScriptDir = self.remoteJoin(self.remoteScriptsDir, abbrevTestDir) self.pathMapping.append(PathMapping(testdir, remoteScriptDir)) def buildXpcsCmd(self, testdir): # change base class' paths to remote paths and use base class to build command self.xpcshell = self.remoteJoin(self.remoteBinDir, "xpcw") self.headJSPath = self.remoteJoin(self.remoteScriptsDir, 'head.js') self.httpdJSPath = self.remoteJoin(self.remoteComponentsDir, 'httpd.js') self.httpdManifest = self.remoteJoin(self.remoteComponentsDir, 'httpd.manifest') self.testingModulesDir = self.remoteModulesDir self.testharnessdir = self.remoteScriptsDir xpcshell.XPCShellTests.buildXpcsCmd(self, testdir) # remove "-g -a " and add "--greomni " del(self.xpcsCmd[1:5]) if self.options.localAPK: self.xpcsCmd.insert(3, '--greomni') self.xpcsCmd.insert(4, self.remoteAPK) if self.remoteDebugger: # for example, "/data/local/gdbserver" "localhost:12345" self.xpcsCmd = [ self.remoteDebugger, self.remoteDebuggerArgs, self.xpcsCmd] def getHeadAndTailFiles(self, test): """Override parent method to find files on remote device.""" def sanitize_list(s, kind): for f in s.strip().split(' '): f = f.strip() if len(f) < 1: continue path = self.remoteJoin(self.remoteHere, f) if not self.device.fileExists(path): raise Exception('%s file does not exist: %s' % ( kind, path)) yield path self.remoteHere = self.remoteForLocal(test['here']) return (list(sanitize_list(test['head'], 'head')), list(sanitize_list(test['tail'], 'tail'))) def buildCmdTestFile(self, name): remoteDir = self.remoteForLocal(os.path.dirname(name)) if remoteDir == self.remoteHere: remoteName = os.path.basename(name) else: remoteName = self.remoteJoin(remoteDir, os.path.basename(name)) return ['-e', 'const _TEST_FILE = ["%s"];' % replaceBackSlashes(remoteName)] def setupProfileDir(self): self.device.removeDir(self.profileDir) self.device.mkDir(self.profileDir) if self.interactive or self.singleFile: self.log.info("TEST-INFO | profile dir is %s" % self.profileDir) return self.profileDir def logCommand(self, name, completeCmd, testdir): self.log.info("TEST-INFO | %s | full command: %r" % (name, completeCmd)) self.log.info("TEST-INFO | %s | current directory: %r" % (name, self.remoteHere)) self.log.info("TEST-INFO | %s | environment: %s" % (name, self.env)) def setupLeakLogging(self): self.env["XPCOM_MEM_LEAK_LOG"] = self.remoteJoin(self.profileDir, "leaks.log") return self.env["XPCOM_MEM_LEAK_LOG"] def setLD_LIBRARY_PATH(self, env): env["LD_LIBRARY_PATH"]=self.remoteBinDir def pushWrapper(self): # Rather than executing xpcshell directly, this wrapper script is # used. By setting environment variables and the cwd in the script, # the length of the per-test command line is shortened. This is # often important when using ADB, as there is a limit to the length # of the ADB command line. localWrapper = tempfile.mktemp() f = open(localWrapper, "w") f.write("#!/system/bin/sh\n") for envkey, envval in self.env.iteritems(): f.write("export %s=%s\n" % (envkey, envval)) f.write("cd $1\n") f.write("echo xpcw: cd $1\n") f.write("shift\n") f.write("echo xpcw: xpcshell \"$@\"\n") f.write("%s/xpcshell \"$@\"\n" % self.remoteBinDir) f.close() remoteWrapper = self.remoteJoin(self.remoteBinDir, "xpcw") self.device.pushFile(localWrapper, remoteWrapper) os.remove(localWrapper) self.device.chmodDir(self.remoteBinDir) def buildEnvironment(self): self.env = {} self.buildCoreEnvironment() self.setLD_LIBRARY_PATH(self.env) self.env["MOZ_LINKER_CACHE"]=self.remoteBinDir if self.options.localAPK and self.appRoot: self.env["GRE_HOME"]=self.appRoot self.env["XPCSHELL_TEST_PROFILE_DIR"]=self.profileDir self.env["TMPDIR"]=self.remoteTmpDir self.env["HOME"]=self.profileDir if self.options.setup: self.pushWrapper() def launchProcess(self, cmd, stdout, stderr, env, cwd): cmd.insert(1, self.remoteHere) outputFile = "xpcshelloutput" f = open(outputFile, "w+") self.shellReturnCode = self.device.shell(cmd, f) f.close() # The device manager may have timed out waiting for xpcshell. # Guard against an accumulation of hung processes by killing # them here. Note also that IPC tests may spawn new instances # of xpcshell. self.device.killProcess(cmd[0]) self.device.killProcess("xpcshell") return outputFile def communicate(self, proc): f = open(proc, "r") contents = f.read() f.close() os.remove(proc) return contents, "" def poll(self, proc): if self.device.processExist("xpcshell") is None: return self.getReturnCode(proc) # Process is still running return None def kill(self, proc): return self.device.killProcess("xpcshell", True) def getReturnCode(self, proc): if self.shellReturnCode is not None: return self.shellReturnCode else: return -1 def removeDir(self, dirname): self.device.removeDir(dirname) #TODO: consider creating a separate log dir. We don't have the test file structure, # so we use filename.log. Would rather see ./logs/filename.log def createLogFile(self, test, stdout, leakLogs): try: f = None filename = test.replace('\\', '/').split('/')[-1] + ".log" f = open(filename, "w") f.write(stdout) for leakLog in leakLogs: if os.path.exists(leakLog): leaks = open(leakLog, "r") f.write(leaks.read()) leaks.close() finally: if f <> None: f.close() class RemoteXPCShellOptions(xpcshell.XPCShellOptions): def __init__(self): xpcshell.XPCShellOptions.__init__(self) defaults = {} self.add_option("--deviceIP", action="store", type = "string", dest = "deviceIP", help = "ip address of remote device to test") defaults["deviceIP"] = None self.add_option("--devicePort", action="store", type = "string", dest = "devicePort", help = "port of remote device to test") defaults["devicePort"] = 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("--objdir", action="store", type = "string", dest = "objdir", help = "local objdir, containing xpcshell binaries") defaults["objdir"] = None self.add_option("--apk", action="store", type = "string", dest = "localAPK", help = "local path to Fennec APK") defaults["localAPK"] = None 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("--local-lib-dir", action="store", type = "string", dest = "localLib", help = "local path to library directory") defaults["localLib"] = None self.add_option("--local-bin-dir", action="store", type = "string", dest = "localBin", help = "local path to bin directory") defaults["localBin"] = None self.set_defaults(**defaults) def verifyRemoteOptions(self, options): if options.localLib is None: if options.localAPK and options.objdir: for path in ['dist/fennec', 'fennec/lib']: options.localLib = os.path.join(options.objdir, path) if os.path.isdir(options.localLib): break else: self.error("Couldn't find local library dir, specify --local-lib-dir") elif options.objdir: options.localLib = os.path.join(options.objdir, 'dist/bin') elif os.path.isfile(os.path.join(here, '..', 'bin', 'xpcshell')): # assume tests are being run from a tests.zip options.localLib = os.path.abspath(os.path.join(here, '..', 'bin')) else: self.error("Couldn't find local library dir, specify --local-lib-dir") if options.localBin is None: if options.objdir: for path in ['dist/bin', 'bin']: options.localBin = os.path.join(options.objdir, path) if os.path.isdir(options.localBin): break else: self.error("Couldn't find local binary dir, specify --local-bin-dir") elif os.path.isfile(os.path.join(here, '..', 'bin', 'xpcshell')): # assume tests are being run from a tests.zip options.localBin = os.path.abspath(os.path.join(here, '..', 'bin')) else: self.error("Couldn't find local binary dir, specify --local-bin-dir") return options class PathMapping: def __init__(self, localDir, remoteDir): self.local = localDir self.remote = remoteDir def main(): parser = RemoteXPCShellOptions() options, args = parser.parse_args() if not options.localAPK: for file in os.listdir(os.path.join(options.objdir, "dist")): if (file.endswith(".apk") and file.startswith("fennec")): options.localAPK = os.path.join(options.objdir, "dist") options.localAPK = os.path.join(options.localAPK, file) print >>sys.stderr, "using APK: " + options.localAPK break else: print >>sys.stderr, "Error: please specify an APK" sys.exit(1) options = parser.verifyRemoteOptions(options) if len(args) < 1 and options.manifest is None: print >>sys.stderr, """Usage: %s or: %s --manifest=test.manifest """ % (sys.argv[0], sys.argv[0]) sys.exit(1) if (options.dm_trans == "adb"): if (options.deviceIP): dm = devicemanagerADB.DeviceManagerADB(options.deviceIP, options.devicePort) else: dm = devicemanagerADB.DeviceManagerADB() else: dm = devicemanagerSUT.DeviceManagerSUT(options.deviceIP, options.devicePort) if (options.deviceIP == None): print "Error: you must provide a device IP to connect to via the --device option" sys.exit(1) if options.interactive and not options.testPath: print >>sys.stderr, "Error: You must specify a test filename in interactive mode!" sys.exit(1) if not options.objdir: print >>sys.stderr, "Error: You must specify an objdir" sys.exit(1) xpcsh = XPCShellRemote(dm, options, args) if not xpcsh.runTests(xpcshell='xpcshell', testdirs=args[0:], **options.__dict__): sys.exit(1) if __name__ == '__main__': main()