wehub-resource-sync
fed8b2eed7
Backend release / release (push) Waiting to run
Bandit Security Scan / bandit_scan (push) Waiting to run
Build and push multi-arch DocsGPT Docker image / build (linux/amd64, ubuntu-latest, amd64) (push) Waiting to run
Build and push multi-arch DocsGPT Docker image / build (linux/arm64, ubuntu-24.04-arm, arm64) (push) Waiting to run
Build and push multi-arch DocsGPT Docker image / manifest (push) Blocked by required conditions
Build and push DocsGPT FE Docker image for development / build (linux/amd64, ubuntu-latest, amd64) (push) Waiting to run
Build and push DocsGPT FE Docker image for development / build (linux/arm64, ubuntu-24.04-arm, arm64) (push) Waiting to run
Build and push DocsGPT FE Docker image for development / manifest (push) Blocked by required conditions
Python linting / ruff (push) Waiting to run
Run python tests with pytest / Run tests and count coverage (3.12) (push) Waiting to run
React Widget Build / build (push) Waiting to run
84 lines
2.7 KiB
Python
84 lines
2.7 KiB
Python
import pytest
|
|
from application.error import bad_request, response_error, sanitize_api_error
|
|
from flask import Flask
|
|
|
|
|
|
@pytest.fixture
|
|
def app():
|
|
app = Flask(__name__)
|
|
return app
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_bad_request_with_message(app):
|
|
with app.app_context():
|
|
message = "Invalid input"
|
|
response = bad_request(status_code=400, message=message)
|
|
assert response.status_code == 400
|
|
assert response.json == {"error": "Bad Request", "message": message}
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_bad_request_without_message(app):
|
|
with app.app_context():
|
|
response = bad_request(status_code=400)
|
|
assert response.status_code == 400
|
|
assert response.json == {"error": "Bad Request"}
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_response_error_with_message(app):
|
|
with app.app_context():
|
|
message = "Something went wrong"
|
|
response = response_error(code_status=500, message=message)
|
|
assert response.status_code == 500
|
|
assert response.json == {"error": "Internal Server Error", "message": message}
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_response_error_without_message(app):
|
|
with app.app_context():
|
|
response = response_error(code_status=500)
|
|
assert response.status_code == 500
|
|
assert response.json == {"error": "Internal Server Error"}
|
|
|
|
|
|
@pytest.mark.unit
|
|
class TestSanitizeApiError:
|
|
|
|
def test_503_unavailable(self):
|
|
assert "temporarily unavailable" in sanitize_api_error("503 Service Unavailable")
|
|
|
|
def test_high_demand(self):
|
|
assert "temporarily unavailable" in sanitize_api_error("high demand")
|
|
|
|
def test_429_rate_limit(self):
|
|
assert "Rate limit" in sanitize_api_error("429 Too Many Requests")
|
|
|
|
def test_quota_exceeded(self):
|
|
assert "Rate limit" in sanitize_api_error("Quota exceeded")
|
|
|
|
def test_401_unauthorized(self):
|
|
assert "Authentication" in sanitize_api_error("401 Unauthorized")
|
|
|
|
def test_invalid_api_key(self):
|
|
assert "Authentication" in sanitize_api_error("Invalid API key provided")
|
|
|
|
def test_timeout(self):
|
|
assert "timed out" in sanitize_api_error("Request timed out")
|
|
|
|
def test_connection_error(self):
|
|
assert "Network" in sanitize_api_error("Connection refused")
|
|
|
|
def test_long_message_sanitized(self):
|
|
assert "error occurred" in sanitize_api_error("x" * 201)
|
|
|
|
def test_traceback_sanitized(self):
|
|
assert "error occurred" in sanitize_api_error("Traceback (most recent call)")
|
|
|
|
def test_json_sanitized(self):
|
|
assert "error occurred" in sanitize_api_error('{"error": "something"}')
|
|
|
|
def test_short_safe_message_passed_through(self):
|
|
assert sanitize_api_error("Something broke") == "Something broke"
|