1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
|
import time
import pytest
from yalexs.exceptions import RateLimited
from yalexs.manager.ratelimit import RATE_LIMIT_WAKEUP_INTERVAL, _RateLimitChecker
@pytest.mark.asyncio
async def test_init_rate_limit():
_RateLimitChecker._client_wakeups.clear()
await _RateLimitChecker.check_rate_limit("test")
await _RateLimitChecker.register_wakeup("test")
with pytest.raises(RateLimited) as exc:
await _RateLimitChecker.check_rate_limit("test")
assert exc.value.next_allowed == pytest.approx(
time.monotonic() + RATE_LIMIT_WAKEUP_INTERVAL
)
|
