mmgen/mmgen-wallet
1
Fork 0
Code Releases 28 Wiki Activity

unit_tests.py.process_bad_data(): support coroutines

Browse source
This commit is contained in:
The MMGen Project 2024-03-04 10:30:36 +00:00
commit 0d3086fc61
Signed by: mmgen
GPG key ID: 3F8B1861E32B7DA2
1 changed files with 6 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

10
test/unit_tests.py
View file

@ -20,7 +20,7 @@
test/unit_tests.py: Unit tests for the MMGen suite
"""
import sys,os,time,importlib,platform
import sys,os,time,importlib,platform,asyncio
try:
from include.test_init import repo_root
@ -35,7 +35,7 @@ if not os.getenv('MMGEN_DEVTOOLS'):
from mmgen.cfg import Config,gc
from mmgen.color import green,gray,brown,orange
from mmgen.util import msg,gmsg,ymsg,Msg,async_run
from mmgen.util import msg,gmsg,ymsg,Msg
from test.include.common import set_globals,end_msg
@ -130,7 +130,9 @@ class UnitTestHelpers:
for (desc,exc_chk,emsg_chk,func) in data:
try:
cfg._util.vmsg_r(' {}{:{w}}'.format(pfx, desc+':', w=desc_w+1))
func()
ret = func()
if type(ret).__name__ == 'coroutine':
asyncio.run(ret)
except Exception as e:
exc = type(e).__name__
emsg = e.args[0]
@ -158,7 +160,7 @@ def run_test(test,subtest=None):
try:
ret = getattr(t,subtest.replace('-','_'))(test,UnitTestHelpers(subtest))
if type(ret).__name__ == 'coroutine':
ret = async_run(ret)
ret = asyncio.run(ret)
except:
if getattr(t,'silence_output',False):
t._end_silence()