ai-kefu-admin/tests/test_rate_limit.py

from app.engine.services.rate_limit import check, _reset_for_test


def test_allows_under_cap():
    _reset_for_test()
    for _ in range(30):
        assert check("t1", "1.2.3.4", "dev1") is True


def test_blocks_over_cap():
    _reset_for_test()
    for _ in range(30):
        assert check("t1", "1.2.3.4", "dev1") is True
    assert check("t1", "1.2.3.4", "dev1") is False


def test_isolated_buckets():
    _reset_for_test()
    for _ in range(30):
        check("t1", "1.2.3.4", "dev1")
    assert check("t1", "5.6.7.8", "dev2") is True  # different ip+device unaffected