From 94700de7a3d9c499dc8ee13d969c3b36126bdc5b Mon Sep 17 00:00:00 2001 From: mathieui Date: Sun, 14 Feb 2021 11:33:47 +0100 Subject: [PATCH 1/2] run_tests: add a command-line option to enable debug logs --- run_tests.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/run_tests.py b/run_tests.py index 3b9df045..f5103b15 100755 --- a/run_tests.py +++ b/run_tests.py @@ -10,7 +10,7 @@ from importlib import import_module from pathlib import Path -def run_tests(filenames=None): +def run_tests(filenames=None, debug=False): """ Find and run all tests in the tests/ directory. @@ -31,9 +31,12 @@ def run_tests(filenames=None): tests = unittest.TestSuite(suites) runner = unittest.TextTestRunner(verbosity=2) - # Disable logging output - logging.basicConfig(level=100) - logging.disable(100) + if debug: + logging.basicConfig(level='DEBUG') + else: + # Disable logging output + logging.basicConfig(level=100) + logging.disable(100) result = runner.run(tests) return result @@ -58,9 +61,10 @@ class TestCommand(Command): if __name__ == '__main__': parser = ArgumentParser(description='Run unit tests.') parser.add_argument('tests', metavar='TEST', nargs='*', help='list of tests to run, or nothing to run them all') + parser.add_argument('-d', '--debug', action='store_true', dest='debug', default=False, help='enable debug output') args = parser.parse_args() - result = run_tests(args.tests) + result = run_tests(args.tests, args.debug) print("" % ( "xmlns='http//andyet.net/protocol/tests'", result.testsRun, len(result.errors), From 7e3e056e6b630411a1d05ffbfbc11b9fe4d45f5c Mon Sep 17 00:00:00 2001 From: mathieui Date: Sun, 14 Feb 2021 11:35:03 +0100 Subject: [PATCH 2/2] slixtest: add more shortcuts for asyncio workflow --- slixmpp/test/slixtest.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/slixmpp/test/slixtest.py b/slixmpp/test/slixtest.py index 668b02c0..d0e380f7 100644 --- a/slixmpp/test/slixtest.py +++ b/slixmpp/test/slixtest.py @@ -434,7 +434,9 @@ class SlixTest(unittest.TestCase): timeout -- Time to wait in seconds for data to be received by a live connection. """ + self.wait_() self.xmpp.data_received(data) + self.wait_() def recv_header(self, sto='', sfrom='', @@ -623,10 +625,20 @@ class SlixTest(unittest.TestCase): loop = asyncio.get_event_loop() future = asyncio.ensure_future(self.xmpp.run_filters(), loop=loop) queue = self.xmpp.waiting_queue - print(queue) loop.run_until_complete(queue.join()) future.cancel() + def wait_(self): + async def yield_some(): + for i in range(100): + await asyncio.sleep(0) + loop = asyncio.get_event_loop() + loop.run_until_complete(yield_some()) + + def run_coro(self, coro): + loop = asyncio.get_event_loop() + loop.run_until_complete(coro) + def stream_close(self): """ Disconnect the dummy XMPP client.