Add support for test methods

- Inspired on the default implementation for pytest_pyfunc_call for normal test functions
- Refactor pytest_pyfunc_call to remove duplication

fixes #5

Author: nicoddemus

pytest_asyncio/plugin.py

@@ -27,30 +27,21 @@ def pytest_pycollect_makeitem(collector, name, obj):
             return list(collector._genfunctions(name, obj))
 
 
-def _argnames(func):
-    spec = inspect.getfullargspec(func)
-    if spec.defaults:
-        return spec.args[:-len(spec.defaults)]
-    return spec.args
-
-
 @pytest.mark.tryfirst
 def pytest_pyfunc_call(pyfuncitem):
-    """A hook wrapper."""
-    if 'asyncio_process_pool' in pyfuncitem.keywords:
-        event_loop = pyfuncitem.funcargs.get('event_loop_process_pool')
-        funcargs = dict((arg, pyfuncitem.funcargs[arg])
-                        for arg in _argnames(pyfuncitem.obj))
-        event_loop.run_until_complete(asyncio.async(pyfuncitem.obj(**funcargs)))
-        # prevent other pyfunc calls from executing
-        return True
-    elif 'asyncio' in pyfuncitem.keywords:
-        event_loop = pyfuncitem.funcargs.get('event_loop')
-        funcargs = dict((arg, pyfuncitem.funcargs[arg])
-                        for arg in _argnames(pyfuncitem.obj))
-        event_loop.run_until_complete(asyncio.async(pyfuncitem.obj(**funcargs)))
-        # prevent other pyfunc calls from executing
-        return True
+    """
+    Run asyncio marked test functions in an event loop instead of a normal
+    function call.
+    """
+    for fixture_name in ('asyncio_process_pool', 'asyncio'):
+        if fixture_name in pyfuncitem.funcargs:
+            event_loop = pyfuncitem.funcargs[fixture_name]
+            funcargs = pyfuncitem.funcargs
+            testargs = {(arg, funcargs[arg])
+                        for arg in pyfuncitem._fixtureinfo.argnames}
+            event_loop.run_until_complete(
+                asyncio.async(pyfuncitem.obj(**testargs)))
+            return True
 
 
 def pytest_runtest_setup(item):

tests/test_simple.py

@@ -87,4 +87,15 @@ def closer(_, writer):
                                         port=unused_tcp_port)
 
     server1.close()
-    yield from server1.wait_closed()
+    yield from server1.wait_closed()
+
+
+class Test:
+    """Test that asyncio marked functions work in test methods."""
+
+    @pytest.mark.asyncio
+    def test_asyncio_marker_method(self):
+        """Test the asyncio pytest marker in a Test class."""
+        url = 'http://httpbin.org/get'
+        resp = yield from simple_http_client(url)
+        assert b'HTTP/1.1 200 OK' in resp