Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 4 additions & 4 deletions src/trio/_core/_tests/test_ki.py
Original file line number Diff line number Diff line change
Expand Up @@ -164,7 +164,7 @@ def protected_manager() -> Iterator[None]:
@pytest.mark.skipif(async_generator is None, reason="async_generator not installed")
async def test_async_generator_agen_protection() -> None:
@_core.enable_ki_protection
@async_generator # type: ignore[misc] # untyped generator
@async_generator # type: ignore[untyped-decorator]
async def agen_protected1() -> None: # type: ignore[misc] # untyped generator
assert _core.currently_ki_protected()
try:
Expand All @@ -173,7 +173,7 @@ async def agen_protected1() -> None: # type: ignore[misc] # untyped generator
assert _core.currently_ki_protected()

@_core.disable_ki_protection
@async_generator # type: ignore[misc] # untyped generator
@async_generator # type: ignore[untyped-decorator]
async def agen_unprotected1() -> None: # type: ignore[misc] # untyped generator
assert not _core.currently_ki_protected()
try:
Expand All @@ -182,7 +182,7 @@ async def agen_unprotected1() -> None: # type: ignore[misc] # untyped generator
assert not _core.currently_ki_protected()

# Swap the order of the decorators:
@async_generator # type: ignore[misc] # untyped generator
@async_generator # type: ignore[untyped-decorator]
@_core.enable_ki_protection
async def agen_protected2() -> None: # type: ignore[misc] # untyped generator
assert _core.currently_ki_protected()
Expand All @@ -191,7 +191,7 @@ async def agen_protected2() -> None: # type: ignore[misc] # untyped generator
finally:
assert _core.currently_ki_protected()

@async_generator # type: ignore[misc] # untyped generator
@async_generator # type: ignore[untyped-decorator]
@_core.disable_ki_protection
async def agen_unprotected2() -> None: # type: ignore[misc] # untyped generator
assert not _core.currently_ki_protected()
Expand Down