diff options
author | Matt A. Tobin <mattatobin@localhost.localdomain> | 2018-02-02 04:16:08 -0500 |
---|---|---|
committer | Matt A. Tobin <mattatobin@localhost.localdomain> | 2018-02-02 04:16:08 -0500 |
commit | 5f8de423f190bbb79a62f804151bc24824fa32d8 (patch) | |
tree | 10027f336435511475e392454359edea8e25895d /testing/mozbase/mozdevice/sut_tests/runtests.py | |
parent | 49ee0794b5d912db1f95dce6eb52d781dc210db5 (diff) | |
download | UXP-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/mozbase/mozdevice/sut_tests/runtests.py')
-rw-r--r-- | testing/mozbase/mozdevice/sut_tests/runtests.py | 96 |
1 files changed, 96 insertions, 0 deletions
diff --git a/testing/mozbase/mozdevice/sut_tests/runtests.py b/testing/mozbase/mozdevice/sut_tests/runtests.py new file mode 100644 index 000000000..fffc306e3 --- /dev/null +++ b/testing/mozbase/mozdevice/sut_tests/runtests.py @@ -0,0 +1,96 @@ +# 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 optparse import OptionParser +import logging +import os +import re +import sys +import unittest + +import dmunit +import genfiles + + +def main(ip, port, heartbeat_port, scripts, directory, isTestDevice, verbose): + dmunit.ip = ip + dmunit.port = port + dmunit.heartbeat_port = heartbeat_port + if verbose: + dmunit.log_level = logging.DEBUG + + suite = unittest.TestSuite() + + genfiles.gen_test_files() + + if scripts: + # Ensure the user didn't include the .py on the name of the test file + # (and get rid of it if they did) + scripts = map(lambda x: x.split('.')[0], scripts) + else: + # Go through the directory and pick up everything + # named test_*.py and run it + testfile = re.compile('^test_.*\.py$') + files = os.listdir(directory) + + for f in files: + if testfile.match(f): + scripts.append(f.split('.')[0]) + + testLoader = dmunit.DeviceManagerTestLoader(isTestDevice) + for s in scripts: + suite.addTest(testLoader.loadTestsFromModuleName(s)) + unittest.TextTestRunner(verbosity=2).run(suite) + + genfiles.clean_test_files() + + +if __name__ == "__main__": + + default_ip = '127.0.0.1' + default_port = 20701 + + env_ip, _, env_port = os.getenv('TEST_DEVICE', '').partition(':') + if env_port: + try: + env_port = int(env_port) + except ValueError: + print >> sys.stderr, "Port in TEST_DEVICE should be an integer." + sys.exit(1) + + # Deal with the options + parser = OptionParser() + parser.add_option("--ip", action="store", type="string", dest="ip", + help="IP address for device running SUTAgent, defaults " + "to what's provided in $TEST_DEVICE or 127.0.0.1", + default=(env_ip or default_ip)) + + parser.add_option("--port", action="store", type="int", dest="port", + help="Port of SUTAgent on device, defaults to " + "what's provided in $TEST_DEVICE or 20701", + default=(env_port or default_port)) + + parser.add_option("--heartbeat", action="store", type="int", + dest="heartbeat_port", help="Port for heartbeat/data " + "channel, defaults to 20700", default=20700) + + parser.add_option("--script", action="append", type="string", + dest="scripts", help="Name of test script to run, " + "can be specified multiple times", default=[]) + + parser.add_option("--directory", action="store", type="string", dest="dir", + help="Directory to look for tests in, defaults to " + "current directory", default=os.getcwd()) + + parser.add_option("--testDevice", action="store_true", dest="isTestDevice", + help="Specifies that the device is a local test agent", + default=False) + + parser.add_option("-v", "--verbose", action="store_true", dest="verbose", + help="Verbose DeviceManager output", default=False) + + (options, args) = parser.parse_args() + + main(options.ip, options.port, options.heartbeat_port, options.scripts, + options.dir, options.isTestDevice, options.verbose) |