figured out loop_factory :)

Author: Vizonex

pytest_asyncio/plugin.py

@@ -49,6 +49,7 @@
     PytestPluginManager,
 )
 
+from typing import Callable
 if sys.version_info >= (3, 10):
     from typing import ParamSpec
 else:
@@ -116,6 +117,7 @@ def fixture(
     *,
     scope: _ScopeName | Callable[[str, Config], _ScopeName] = ...,
     loop_scope: _ScopeName | None = ...,
+    loop_factory: _ScopeName | Callable[[], AbstractEventLoop] = ...,
     params: Iterable[object] | None = ...,
     autouse: bool = ...,
     ids: (
@@ -133,6 +135,7 @@ def fixture(
     *,
     scope: _ScopeName | Callable[[str, Config], _ScopeName] = ...,
     loop_scope: _ScopeName | None = ...,
+    loop_factory: _ScopeName | Callable[[], AbstractEventLoop] = ...,
     params: Iterable[object] | None = ...,
     autouse: bool = ...,
     ids: (
@@ -147,20 +150,21 @@ def fixture(
 def fixture(
     fixture_function: FixtureFunction[_P, _R] | None = None,
     loop_scope: _ScopeName | None = None,
+    loop_factory: _ScopeName | Callable[[], AbstractEventLoop] = ...,
     **kwargs: Any,
 ) -> (
     FixtureFunction[_P, _R]
     | Callable[[FixtureFunction[_P, _R]], FixtureFunction[_P, _R]]
 ):
     if fixture_function is not None:
-        _make_asyncio_fixture_function(fixture_function, loop_scope)
+        _make_asyncio_fixture_function(fixture_function, loop_scope, loop_factory)
         return pytest.fixture(fixture_function, **kwargs)
 
     else:
 
         @functools.wraps(fixture)
         def inner(fixture_function: FixtureFunction[_P, _R]) -> FixtureFunction[_P, _R]:
-            return fixture(fixture_function, loop_scope=loop_scope, **kwargs)
+            return fixture(fixture_function, loop_factory=loop_factory, loop_scope=loop_scope, **kwargs)
 
         return inner
 
@@ -170,12 +174,13 @@ def _is_asyncio_fixture_function(obj: Any) -> bool:
     return getattr(obj, "_force_asyncio_fixture", False)
 
 
-def _make_asyncio_fixture_function(obj: Any, loop_scope: _ScopeName | None) -> None:
+def _make_asyncio_fixture_function(obj: Any, loop_scope: _ScopeName | None, loop_factory: _ScopeName | None) -> None:
     if hasattr(obj, "__func__"):
         # instance method, check the function object
         obj = obj.__func__
     obj._force_asyncio_fixture = True
     obj._loop_scope = loop_scope
+    obj._loop_factory = loop_factory
 
 
 def _is_coroutine_or_asyncgen(obj: Any) -> bool:
@@ -234,14 +239,14 @@ def pytest_report_header(config: Config) -> list[str]:
 
 
 def _fixture_synchronizer(
-    fixturedef: FixtureDef, runner: Runner, request: FixtureRequest
+    fixturedef: FixtureDef, runner: Runner, request: FixtureRequest, loop_factory: Callable[[], AbstractEventLoop]
 ) -> Callable:
     """Returns a synchronous function evaluating the specified fixture."""
     fixture_function = resolve_fixture_function(fixturedef, request)
     if inspect.isasyncgenfunction(fixturedef.func):
-        return _wrap_asyncgen_fixture(fixture_function, runner, request)  # type: ignore[arg-type]
+        return _wrap_asyncgen_fixture(fixture_function, runner, request, loop_factory)  # type: ignore[arg-type]
     elif inspect.iscoroutinefunction(fixturedef.func):
-        return _wrap_async_fixture(fixture_function, runner, request)  # type: ignore[arg-type]
+        return _wrap_async_fixture(fixture_function, runner, request, loop_factory)  # type: ignore[arg-type]
     else:
         return fixturedef.func
 
@@ -256,6 +261,7 @@ def _wrap_asyncgen_fixture(
     ],
     runner: Runner,
     request: FixtureRequest,
+    loop_factory:Callable[[], AbstractEventLoop]
 ) -> Callable[AsyncGenFixtureParams, AsyncGenFixtureYieldType]:
     @functools.wraps(fixture_function)
     def _asyncgen_fixture_wrapper(
@@ -285,6 +291,9 @@ async def async_finalizer() -> None:
                     msg = "Async generator fixture didn't stop."
                     msg += "Yield only once."
                     raise ValueError(msg)
+            if loop_factory:
+                _loop = loop_factory()
+                asyncio.set_event_loop(_loop)
 
             runner.run(async_finalizer(), context=context)
             if reset_contextvars is not None:
@@ -306,6 +315,7 @@ def _wrap_async_fixture(
     ],
     runner: Runner,
     request: FixtureRequest,
+    loop_factory: Callable[[], AbstractEventLoop] | None = None
 ) -> Callable[AsyncFixtureParams, AsyncFixtureReturnType]:
 
     @functools.wraps(fixture_function)  # type: ignore[arg-type]
@@ -318,8 +328,12 @@ async def setup():
             return res
 
         context = contextvars.copy_context()
-        result = runner.run(setup(), context=context)
 
+        # ensure loop_factory gets ran before we start running...
+        if loop_factory:
+            asyncio.set_event_loop(loop_factory())
+
+        result = runner.run(setup(), context=context)
         # Copy the context vars modified by the setup task into the current
         # context, and (if needed) add a finalizer to reset them.
         #
@@ -372,8 +386,6 @@ def restore_contextvars():
 class PytestAsyncioFunction(Function):
     """Base class for all test functions managed by pytest-asyncio."""
 
-    loop_factory: Callable[[], AbstractEventLoop] | None
-
     @classmethod
     def item_subclass_for(cls, item: Function, /) -> type[PytestAsyncioFunction] | None:
         """
@@ -388,18 +400,12 @@ def item_subclass_for(cls, item: Function, /) -> type[PytestAsyncioFunction] | N
         return None
 
     @classmethod
-    def _from_function(
-        cls,
-        function: Function,
-        loop_factory: Callable[[], AbstractEventLoop] | None = None,
-        /,
-    ) -> Function:
+    def _from_function(cls, function: Function, /) -> Function:
         """
         Instantiates this specific PytestAsyncioFunction type from the specified
         Function item.
         """
         assert function.get_closest_marker("asyncio")
-
         subclass_instance = cls.from_parent(
             function.parent,
             name=function.name,
@@ -409,7 +415,6 @@ def _from_function(
             keywords=function.keywords,
             originalname=function.originalname,
         )
-        subclass_instance.loop_factory = loop_factory
         subclass_instance.own_markers = function.own_markers
         assert subclass_instance.own_markers == function.own_markers
         return subclass_instance
@@ -429,7 +434,8 @@ def _can_substitute(item: Function) -> bool:
         return inspect.iscoroutinefunction(func)
 
     def runtest(self) -> None:
-        synchronized_obj = wrap_in_sync(self.obj)
+        # print(self.obj.pytestmark[0].__dict__)
+        synchronized_obj = wrap_in_sync(self.obj, self.obj.pytestmark[0].kwargs.get('loop_factory', None))
         with MonkeyPatch.context() as c:
             c.setattr(self, "obj", synchronized_obj)
             super().runtest()
@@ -534,27 +540,9 @@ def pytest_pycollect_makeitem_convert_async_functions_to_subclass(
                     node.config
                 ) == Mode.AUTO and not node.get_closest_marker("asyncio"):
                     node.add_marker("asyncio")
-                if asyncio_marker := node.get_closest_marker("asyncio"):
-                    if loop_factory := asyncio_marker.kwargs.get("loop_factory", None):
-                        # multiply if loop_factory is an iterable object of factories
-                        if hasattr(loop_factory, "__iter__"):
-                            updated_item = [
-                                specialized_item_class._from_function(node, lf)
-                                for lf in loop_factory
-                            ]
-                        else:
-                            updated_item = specialized_item_class._from_function(
-                                node, loop_factory
-                            )
-                    else:
-                        updated_item = specialized_item_class._from_function(node)
-
-        # we could have multiple factroies to test if so,
-        # multiply the number of functions for us...
-        if isinstance(updated_item, list):
-            updated_node_collection.extend(updated_item)
-        else:
-            updated_node_collection.append(updated_item)
+                if node.get_closest_marker("asyncio"):
+                    updated_item = specialized_item_class._from_function(node)
+        updated_node_collection.append(updated_item)
     hook_result.force_result(updated_node_collection)
 
 
@@ -654,43 +642,46 @@ def pytest_pyfunc_call(pyfuncitem: Function) -> object | None:
 
 def wrap_in_sync(
     func: Callable[..., Awaitable[Any]],
+    loop_factory:Callable[[], AbstractEventLoop] | None = None
 ):
     """
     Return a sync wrapper around an async function executing it in the
     current event loop.
     """
-
     @functools.wraps(func)
     def inner(*args, **kwargs):
-        coro = func(*args, **kwargs)
-        _loop = _get_event_loop_no_warn()
-        task = asyncio.ensure_future(coro, loop=_loop)
+        _last_loop = asyncio.get_event_loop()
+        if loop_factory:
+            _loop = loop_factory()
+            asyncio.set_event_loop(_loop)
+        else:
+            _loop = asyncio.get_event_loop()
+        task = asyncio.ensure_future(func(*args, **kwargs), loop=_loop)
         try:
             _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
 
+        asyncio.set_event_loop(_last_loop)
     return inner
 
 
 def pytest_runtest_setup(item: pytest.Item) -> None:
     marker = item.get_closest_marker("asyncio")
     if marker is None:
         return
-    getattr(marker, "loop_factory", None)
     default_loop_scope = _get_default_test_loop_scope(item.config)
     loop_scope = _get_marked_loop_scope(marker, default_loop_scope)
     runner_fixture_id = f"_{loop_scope}_scoped_runner"
-    fixturenames: list[str] = item.fixturenames  # type: ignore[attr-defined]
-
+    fixturenames = item.fixturenames  # type: ignore[attr-defined]
     if runner_fixture_id not in fixturenames:
         fixturenames.append(runner_fixture_id)
-
     obj = getattr(item, "obj", None)
     if not getattr(obj, "hypothesis", False) and getattr(
         obj, "is_hypothesis_test", False
@@ -717,17 +708,12 @@ def pytest_fixture_setup(fixturedef: FixtureDef, request) -> object | None:
         or default_loop_scope
         or fixturedef.scope
     )
-    # XXX: Currently Confused as to where to debug and harvest and get the runner to use the loop_factory argument.
     loop_factory = getattr(fixturedef.func, "loop_factory", None)
 
-    print(f"LOOP FACTORY: {loop_factory} {fixturedef.func}")
-    sys.stdout.flush()
-
     runner_fixture_id = f"_{loop_scope}_scoped_runner"
-    runner: Runner = request.getfixturevalue(runner_fixture_id)
-
-    synchronizer = _fixture_synchronizer(fixturedef, runner, request)
-    _make_asyncio_fixture_function(synchronizer, loop_scope)
+    runner = request.getfixturevalue(runner_fixture_id)
+    synchronizer = _fixture_synchronizer(fixturedef, runner, request, loop_factory)
+    _make_asyncio_fixture_function(synchronizer, loop_scope, loop_factory)
     with MonkeyPatch.context() as c:
         c.setattr(fixturedef, "func", synchronizer)
         hook_result = yield
@@ -750,12 +736,9 @@ def _get_marked_loop_scope(
 ) -> _ScopeName:
     assert asyncio_marker.name == "asyncio"
     if asyncio_marker.args or (
-        asyncio_marker.kwargs
-        and set(asyncio_marker.kwargs) - {"loop_scope", "scope", "loop_factory"}
+        asyncio_marker.kwargs and set(asyncio_marker.kwargs) - {"loop_scope", "scope", "loop_factory"}
     ):
-        raise ValueError(
-            "mark.asyncio accepts only keyword arguments 'loop_scope', 'loop_factory'."
-        )
+        raise ValueError("mark.asyncio accepts only a keyword arguments 'loop_scope' or 'loop_factory'")
     if "scope" in asyncio_marker.kwargs:
         if "loop_scope" in asyncio_marker.kwargs:
             raise pytest.UsageError(_DUPLICATE_LOOP_SCOPE_DEFINITION_ERROR)