arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-19 12:58:13 +00:00
Code Issues Releases Wiki Activity

Merge pull request #319 from ph4r05/tests_async

Browse Source 
tests: support async test cases added
This commit is contained in:
Tomas Susanka 2018-08-20 16:09:44 +02:00 committed by GitHub
commit aab3a5eed6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/unittest.py
View File

@ -1,4 +1,5 @@
from trezor.utils import ensure from trezor.utils import ensure
from utest import assert_async
class SkipTest(Exception): class SkipTest(Exception):
@ -178,12 +179,16 @@ def run_class(c, test_result):
print('class', c.__qualname__) print('class', c.__qualname__)
for name in dir(o): for name in dir(o):
if name.startswith("test"): if name.startswith("test"):
is_async = name.startswith("test_async")
print(' ', name, end=' ...') print(' ', name, end=' ...')
m = getattr(o, name) m = getattr(o, name)
try: try:
set_up() set_up()
test_result.testsRun += 1 test_result.testsRun += 1
m() if is_async:
assert_async(m(), [(None, StopIteration()), ])
else:
m()
tear_down() tear_down()
print(" ok") print(" ok")
except SkipTest as e: except SkipTest as e: