chore: apply flake8-pytest-style linter rules (#8307)

This commit is contained in:
Bowen Liang
2024-09-12 18:09:16 +08:00
committed by GitHub
parent 40fb4d16ef
commit 8815511ccb
7 changed files with 23 additions and 23 deletions

View File

@@ -1,6 +1,8 @@
import random
from unittest.mock import MagicMock, patch
import pytest
from core.helper.ssrf_proxy import SSRF_DEFAULT_MAX_RETRIES, STATUS_FORCELIST, make_request
@@ -22,11 +24,9 @@ def test_retry_exceed_max_retries(mock_request):
side_effects = [mock_response] * SSRF_DEFAULT_MAX_RETRIES
mock_request.side_effect = side_effects
try:
with pytest.raises(Exception) as e:
make_request("GET", "http://example.com", max_retries=SSRF_DEFAULT_MAX_RETRIES - 1)
raise AssertionError("Expected Exception not raised")
except Exception as e:
assert str(e) == f"Reached maximum retries ({SSRF_DEFAULT_MAX_RETRIES - 1}) for URL http://example.com"
assert str(e.value) == f"Reached maximum retries ({SSRF_DEFAULT_MAX_RETRIES - 1}) for URL http://example.com"
@patch("httpx.request")