浅念
22 lines
541 B
Python
22 lines
541 B
Python
from app.services.rate_limit import check, _reset_for_test
|
|
|
|
|
|
def test_allows_under_cap():
|
|
_reset_for_test()
|
|
for _ in range(30):
|
|
assert check("1.2.3.4", "dev1") is True
|
|
|
|
|
|
def test_blocks_over_cap():
|
|
_reset_for_test()
|
|
for _ in range(30):
|
|
assert check("1.2.3.4", "dev1") is True
|
|
assert check("1.2.3.4", "dev1") is False
|
|
|
|
|
|
def test_isolated_buckets():
|
|
_reset_for_test()
|
|
for _ in range(30):
|
|
check("1.2.3.4", "dev1")
|
|
assert check("5.6.7.8", "dev2") is True # different ip+device unaffected
|