Skip to content

Rate limiting

provide.testkit.time.rate_limiting

Rate limiting mock fixtures for testing.

Fixtures for testing rate-limited code.

Classes

Functions

rate_limiter_mock 

rate_limiter_mock() -> MockRateLimiter

Mock for testing rate-limited code.

Returns:

Type Description
MockRateLimiter

Mock rate limiter that can be controlled in tests.
Example

def test_rate_limiting(rate_limiter_mock): ... rate_limiter_mock.set_limit(after_calls=3) ... assert rate_limiter_mock.check() is True # Call 1 ... assert rate_limiter_mock.check() is True # Call 2 ... assert rate_limiter_mock.check() is True # Call 3 ... assert rate_limiter_mock.check() is False # Rate limited!

Source code in provide/testkit/time/rate_limiting.py 
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
@pytest.fixture
def rate_limiter_mock() -> MockRateLimiter:
    """Mock for testing rate-limited code.

    Returns:
        Mock rate limiter that can be controlled in tests.

    Example:
        >>> def test_rate_limiting(rate_limiter_mock):
        ...     rate_limiter_mock.set_limit(after_calls=3)
        ...     assert rate_limiter_mock.check() is True  # Call 1
        ...     assert rate_limiter_mock.check() is True  # Call 2
        ...     assert rate_limiter_mock.check() is True  # Call 3
        ...     assert rate_limiter_mock.check() is False  # Rate limited!
    """
    return MockRateLimiter()