diff --git a/tests/test_annotated.py b/tests/test_annotated.py index beb4ce2293..657e2bd6d0 100644 --- a/tests/test_annotated.py +++ b/tests/test_annotated.py @@ -60,8 +60,10 @@ def cmd(force: Annotated[bool, typer.Option("--force")] = False): assert result.exit_code == 0, result.output assert "Forcing operation" in result.output + def test_runner_can_use_an_async_method(): app = typer.Typer() + @app.command() async def cmd(val: Annotated[int, typer.Argument()] = 0): print(f"hello {val}") @@ -74,10 +76,14 @@ async def cmd(val: Annotated[int, typer.Argument()] = 0): assert result.exit_code == 0, result.output assert "hello 42" in result.output + if sys.version_info >= (3, 11): + def test_runner_can_use_a_custom_async_loop(): import asyncio + app = typer.Typer(loop_factory=asyncio.new_event_loop) + @app.command() async def cmd(val: Annotated[int, typer.Argument()] = 0): print(f"hello {val}") @@ -89,4 +95,3 @@ async def cmd(val: Annotated[int, typer.Argument()] = 0): result = runner.invoke(app, ["42"]) assert result.exit_code == 0, result.output assert "hello 42" in result.output - diff --git a/typer/main.py b/typer/main.py index 2e644ccc4e..1c9884e95d 100644 --- a/typer/main.py +++ b/typer/main.py @@ -252,7 +252,9 @@ def runner(*args, **kwargs) -> Any: elif sys.version_info >= (3, 7): return asyncio.run(f(*args, **kwargs)) else: - asyncio.get_event_loop().run_until_complete(asyncio.wait(f(*args, **kwargs))) + asyncio.get_event_loop().run_until_complete( + asyncio.wait(f(*args, **kwargs)) + ) return runner @@ -280,7 +282,6 @@ def runner(*args, **kwargs) -> Any: ) ) - return decorator def add_typer(