Support async tests which use Hypothesis

.gitignore

@@ -6,3 +6,4 @@ dist
 doc/_build
 virtualenv/
 venv*/
+.hypothesis/

.travis.yml

@@ -130,7 +130,7 @@ matrix:
 
 install:
 - curl https://bootstrap.pypa.io/get-pip.py | python
-- python -m pip install --upgrade wheel>=0.30.0 setuptools>=36.6.0
+- python -m pip install --upgrade wheel>=0.30.0 setuptools>=36.6.0 hypothesis>=3.64
 
 script:
 - python -m unittest --verbose test

AUTHORS

@@ -8,3 +8,4 @@ Pyotr Ermishkin <[email protected]>
 Random User <[email protected]>
 Rémi Cardona <[email protected]>
 Sviatoslav Sydorenko <[email protected]>
+Stefan Tjarks <[email protected]>

asynctest/case.py

@@ -347,6 +347,13 @@ def debug(self):
             self._unset_loop()
 
     def _run_test_method(self, method):
+        # If the coroutine is decorated by hypothesis wrap
+        # it before running.
+        if getattr(method, "is_hypothesis_test", False) and asyncio.iscoroutinefunction(
+            method.hypothesis.inner_test
+        ):
+            method.hypothesis.inner_test = wrap_in_sync(method.hypothesis.inner_test)
+
         # If the method is a coroutine or returns a coroutine, run it on the
         # loop
         result = method()
@@ -508,3 +515,25 @@ def ignore_loop(func=None):
                   "fail_on(unused_loop=False)", DeprecationWarning)
     checker = asynctest._fail_on._fail_on({"unused_loop": False})
     return checker if func is None else checker(func)
+
+
+def wrap_in_sync(func):
+    """Return a sync wrapper around an async function executing it in the
+    current event loop.
+    """
+
+    @functools.wraps(func)
+    def inner(**kwargs):
+        coro = func(**kwargs)
+        if coro is not None:
+            task = asyncio.ensure_future(coro)
+            try:
+                asyncio.get_event_loop().run_until_complete(task)
+            except BaseException:
+                # run_until_complete doesn't get the result from exceptions
+                # that are not subclasses of `Exception`. Consume all
+                # exceptions to prevent asyncio's warning from logging.
+                if task.done() and not task.cancelled():
+                    task.exception()
+                raise
+    return inner

test/test_hypothesis_integration.py

@@ -0,0 +1,15 @@
+"""Tests for the Hypothesis integration, which wraps async functions in a
+sync shim for Hypothesis.
+"""
+
+import asynctest
+
+from hypothesis import given, strategies as st
+
+
+class TestHypothesisIntegration(asynctest.TestCase):
+
+    @given(st.integers())
+    async def test_mark_inner(self, n):
+        assert isinstance(n, int)
+