Commit 8f8b2139 authored by Denis Bilenko's avatar Denis Bilenko

greentest/helper.py: allow customizing monkey.patch_all

parent a130025d
...@@ -19,9 +19,11 @@ class ContainsAll(object): ...@@ -19,9 +19,11 @@ class ContainsAll(object):
return True return True
def patch_all(timeout=None): def patch_all(**kwargs):
timeout = kwargs.pop('timeout', None)
kwargs.setdefault('aggressive', True)
from gevent import monkey from gevent import monkey
monkey.patch_all(aggressive=True) monkey.patch_all(**kwargs)
import unittest import unittest
import greentest import greentest
unittest.TestCase = greentest.TestCase0 unittest.TestCase = greentest.TestCase0
...@@ -38,8 +40,9 @@ def imp_find_dotted_module(name): ...@@ -38,8 +40,9 @@ def imp_find_dotted_module(name):
return result return result
def prepare_stdlib_test(filename): def prepare_stdlib_test(filename, **kwargs):
patch_all(timeout=20) kwargs.setdefault('timeout', 20)
patch_all(**kwargs)
import test import test
try: try:
from test import test_support from test import test_support
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment