From 61f9345037a7bb8ad70bd277c9e4e2bff2540780 Mon Sep 17 00:00:00 2001 From: Denis Bilenko <denis.bilenko@gmail.com> Date: Fri, 29 Jan 2010 17:05:42 +0600 Subject: [PATCH] add patch_all helper to greentest --- greentest/greentest.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/greentest/greentest.py b/greentest/greentest.py index 922f2376..86faad46 100644 --- a/greentest/greentest.py +++ b/greentest/greentest.py @@ -171,3 +171,12 @@ class GenericGetTestCase(TestCase): delay = time.time() - start assert 0.01 - 0.001 <= delay < 0.01 + 0.01 + 0.1, delay + +def patch_all(timeout=None): + from gevent import monkey + monkey.patch_all(aggressive=True) + import unittest + unittest.TestCase = TestCase + if timeout is not None: + TestCase.__timeout__ = timeout + -- 2.30.9