pythonasyncioapiasyncfastapiframeworkjsonjson-schemaopenapiopenapi3pydanticpython-typespython3redocreststarletteswaggerswagger-uiuvicornweb
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
68 lines
2.0 KiB
68 lines
2.0 KiB
from collections.abc import Iterator
|
|
from typing import Any
|
|
|
|
import anyio
|
|
import pytest
|
|
from fastapi import concurrency
|
|
|
|
|
|
@pytest.fixture
|
|
def _reset_capacity_limiter() -> Iterator[None]:
|
|
# Reset the capacity limiter before each test to ensure a clean slate
|
|
concurrency._anti_deadlock_capacity_limiter = None
|
|
concurrency._global_capacity_limiter = None
|
|
yield
|
|
concurrency._anti_deadlock_capacity_limiter = None
|
|
concurrency._global_capacity_limiter = None
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_run_in_threadpool(_reset_capacity_limiter: None) -> None:
|
|
def blocking_function(x: int, y: int) -> int:
|
|
return x + y
|
|
|
|
result = await concurrency.run_in_threadpool(blocking_function, 1, y=2)
|
|
assert result == 3
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_iterate_in_threadpool(_reset_capacity_limiter: None) -> None:
|
|
result = []
|
|
async for item in concurrency.iterate_in_threadpool(range(5)):
|
|
result.append(item)
|
|
|
|
assert result == [0, 1, 2, 3, 4]
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"limit, anti_deadlock_reserve, error_kind",
|
|
[
|
|
("not an int", 2, TypeError),
|
|
(10, "not an int", TypeError),
|
|
(1, 0, ValueError),
|
|
(10, 0, ValueError),
|
|
(10, 9, ValueError),
|
|
],
|
|
)
|
|
def test_set_thread_limit_invalid_args(
|
|
limit: Any, anti_deadlock_reserve: Any, error_kind: type[Exception]
|
|
) -> None:
|
|
with pytest.raises(error_kind):
|
|
concurrency.set_thread_limit(limit, anti_deadlock_reserve)
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_set_thread_limit(_reset_capacity_limiter: None) -> None:
|
|
original_default_thread_limit = (
|
|
anyio.to_thread.current_default_thread_limiter().total_tokens
|
|
)
|
|
|
|
concurrency.set_thread_limit(10, anti_deadlock_reserve=2)
|
|
assert concurrency._anti_deadlock_capacity_limiter.total_tokens == 8
|
|
assert concurrency._global_capacity_limiter.total_tokens == 10
|
|
|
|
# default anyio token limiter not affected
|
|
assert (
|
|
anyio.to_thread.current_default_thread_limiter().total_tokens
|
|
== original_default_thread_limit
|
|
)
|
|
|