This commit is contained in:
Jack Robison 2018-07-29 17:48:20 -04:00
parent d6695c7925
commit 8e004a8744
No known key found for this signature in database
GPG key ID: DF25C68FE0239BB2

View file

@ -48,22 +48,22 @@ def test(ext_port=4446, int_port=4446, proto='UDP', timeout=1):
@defer.inlineCallbacks @defer.inlineCallbacks
def run_tests(timeout=1): def run_tests():
if len(sys.argv) > 1:
log.setLevel(logging.DEBUG)
timeout = int(sys.argv[1])
else:
timeout = 1
for p in ['UDP']: for p in ['UDP']:
yield test(proto=p, timeout=timeout) yield test(proto=p, timeout=timeout)
def main(timeout): def main():
d = run_tests(timeout) d = run_tests()
d.addErrback(log.exception) d.addErrback(log.exception)
d.addBoth(lambda _: reactor.callLater(0, reactor.stop)) d.addBoth(lambda _: reactor.callLater(0, reactor.stop))
reactor.run() reactor.run()
if __name__ == "__main__": if __name__ == "__main__":
if len(sys.argv) > 1: main()
log.setLevel(logging.DEBUG)
timeout = int(sys.argv[1])
else:
timeout = 1
main(timeout)