Context without event_loop (#186)

aioworkers · Dec 26, 2024 · 74749e0 · 74749e0
1 parent 34b4c4c
commit 74749e0
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/pytest_aioworkers/plugin.py b/pytest_aioworkers/plugin.py
@@ -1,3 +1,4 @@
+import asyncio
 from unittest import mock
 
 import pytest
@@ -36,14 +37,18 @@ def config(aioworkers, config_yaml):
 
 
 @pytest.fixture
-def context(event_loop, groups, config):
+async def context(aioworkers, config):
     from aioworkers.core.context import Context, GroupResolver
 
     gr = GroupResolver(
-        include=groups.include,
-        exclude=groups.exclude,
+        include=aioworkers.groups.include,
+        exclude=aioworkers.groups.exclude,
     )
-    with Context(config, group_resolver=gr, loop=event_loop) as ctx:
+    async with Context(
+        config,
+        group_resolver=gr,
+        loop=asyncio.get_running_loop(),
+    ) as ctx:
         yield ctx