forked from goodboy/tractor
1
0
Fork 0

Handle windows in `@tractor_test` as well

try_trip^2
Tyler Goodlet 2020-01-26 23:44:47 -05:00
parent a6b249cd52
commit 43cca122f5
1 changed files with 9 additions and 1 deletions

View File

@ -1,4 +1,5 @@
import inspect import inspect
import platform
from functools import partial, wraps from functools import partial, wraps
from .. import run from .. import run
@ -29,7 +30,7 @@ def tractor_test(fn):
*args, *args,
loglevel=None, loglevel=None,
arb_addr=None, arb_addr=None,
start_method='trio_run_in_process', start_method=None,
**kwargs **kwargs
): ):
# __tracebackhide__ = True # __tracebackhide__ = True
@ -41,6 +42,13 @@ def tractor_test(fn):
# allows test suites to define a 'loglevel' fixture # allows test suites to define a 'loglevel' fixture
# that activates the internal logging # that activates the internal logging
kwargs['loglevel'] = loglevel kwargs['loglevel'] = loglevel
if start_method is None:
if platform.system() == "Windows":
start_method = 'spawn'
else:
start_method = 'trio_run_in_process'
if 'start_method' in inspect.signature(fn).parameters: if 'start_method' in inspect.signature(fn).parameters:
# set of subprocess spawning backends # set of subprocess spawning backends
kwargs['start_method'] = start_method kwargs['start_method'] = start_method