Adds test for async function

plathanus-tech · Jun 24, 2022 · f493841 · f493841
1 parent d6b82be
commit f493841
Show file tree

Hide file tree

Showing 4 changed files with 29 additions and 2 deletions.
diff --git a/pdm.lock b/pdm.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -20,6 +20,7 @@ dev = [
     "black>=22.3.0",
     "mypy>=0.961",
     "pytest>=7.1.2",
+    "pytest-asyncio>=0.18.3",
 ]
 
 [build-system]

diff --git a/pytest.ini b/pytest.ini
@@ -0,0 +1,2 @@
+[pytest]
+asyncio_mode = strict
diff --git a/tests/test_decorators.py b/tests/test_decorators.py
@@ -28,7 +28,7 @@ def test_requires_decorator_must_have_dependency_type_annotated_on_decorated_sig
         def f(a: str):
             pass
 
-        f("a")
+        f("abc")
 
 
 def test_requires_decorator_does_not_allow_duplicated_annotation_for_dependency_type():
@@ -86,6 +86,16 @@ def f(t: T):
     f()
 
 
+@pytest.mark.asyncio
+async def test_requires_can_also_wraps_coroutine():
+    @requires(T)
+    async def f(t: T):
+        pass
+
+    register_dependency(T())
+    await f()
+
+
 def test_provider_decorator_must_be_called_on_function():
     with pytest.raises(exc.InvalidFunctionDecoration):