зеркало из https://github.com/mozilla/gecko-dev.git
80 строки
2.4 KiB
Python
80 строки
2.4 KiB
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 glob
|
|
import optparse
|
|
import traceback
|
|
import WebIDL
|
|
|
|
class TestHarness(object):
|
|
def __init__(self, test, verbose):
|
|
self.test = test
|
|
self.verbose = verbose
|
|
self.printed_intro = False
|
|
|
|
def start(self):
|
|
if self.verbose:
|
|
self.maybe_print_intro()
|
|
|
|
def finish(self):
|
|
if self.verbose or self.printed_intro:
|
|
print "Finished test %s" % self.test
|
|
|
|
def maybe_print_intro(self):
|
|
if not self.printed_intro:
|
|
print "Starting test %s" % self.test
|
|
self.printed_intro = True
|
|
|
|
def test_pass(self, msg):
|
|
if self.verbose:
|
|
print "TEST-PASS | %s" % msg
|
|
|
|
def test_fail(self, msg):
|
|
self.maybe_print_intro()
|
|
print "TEST-UNEXPECTED-FAIL | %s" % msg
|
|
|
|
def ok(self, condition, msg):
|
|
if condition:
|
|
self.test_pass(msg)
|
|
else:
|
|
self.test_fail(msg)
|
|
|
|
def check(self, a, b, msg):
|
|
if a == b:
|
|
self.test_pass(msg)
|
|
else:
|
|
self.test_fail(msg)
|
|
print "\tGot %s expected %s" % (a, b)
|
|
|
|
def run_tests(tests, verbose):
|
|
testdir = os.path.join(os.path.dirname(__file__), 'tests')
|
|
if not tests:
|
|
tests = glob.iglob(os.path.join(testdir, "*.py"))
|
|
sys.path.append(testdir)
|
|
|
|
for test in tests:
|
|
(testpath, ext) = os.path.splitext(os.path.basename(test))
|
|
_test = __import__(testpath, globals(), locals(), ['WebIDLTest'])
|
|
|
|
harness = TestHarness(test, verbose)
|
|
harness.start()
|
|
try:
|
|
_test.WebIDLTest.__call__(WebIDL.Parser(), harness)
|
|
except Exception, ex:
|
|
print "TEST-UNEXPECTED-FAIL | Unhandled exception in test %s: %s" % (testpath, ex)
|
|
traceback.print_exc()
|
|
finally:
|
|
harness.finish()
|
|
|
|
if __name__ == '__main__':
|
|
usage = """%prog [OPTIONS] [TESTS]
|
|
Where TESTS are relative to the tests directory."""
|
|
parser = optparse.OptionParser(usage=usage)
|
|
parser.add_option('-q', '--quiet', action='store_false', dest='verbose', default=True,
|
|
help="Don't print passing tests.")
|
|
options, tests = parser.parse_args()
|
|
|
|
run_tests(tests, verbose=options.verbose)
|