新增PDF生成并发设置
This commit is contained in:
13
backend/tests/test_pdf_limiter.py
Normal file
13
backend/tests/test_pdf_limiter.py
Normal file
@@ -0,0 +1,13 @@
|
||||
import pytest
|
||||
|
||||
from app.services.pdf import PdfGenerationBusy, PdfGenerationLimiter
|
||||
|
||||
|
||||
def test_pdf_generation_limiter_times_out_when_full():
|
||||
limiter = PdfGenerationLimiter()
|
||||
limiter.acquire(limit=1)
|
||||
with pytest.raises(PdfGenerationBusy):
|
||||
limiter.acquire(limit=1, timeout_seconds=0)
|
||||
limiter.release()
|
||||
limiter.acquire(limit=1, timeout_seconds=0)
|
||||
limiter.release()
|
||||
Reference in New Issue
Block a user