chore: import upstream snapshot with attribution
Build and push multi-arch DocsGPT Docker image / build (linux/amd64, ubuntu-latest, amd64) (push) Has been cancelled
Backend release / release (push) Has been cancelled
Bandit Security Scan / bandit_scan (push) Has been cancelled
Build and push multi-arch DocsGPT Docker image / build (linux/arm64, ubuntu-24.04-arm, arm64) (push) Has been cancelled
Build and push multi-arch DocsGPT Docker image / manifest (push) Has been cancelled
Build and push DocsGPT FE Docker image for development / build (linux/amd64, ubuntu-latest, amd64) (push) Has been cancelled
Build and push DocsGPT FE Docker image for development / build (linux/arm64, ubuntu-24.04-arm, arm64) (push) Has been cancelled
Build and push DocsGPT FE Docker image for development / manifest (push) Has been cancelled
Python linting / ruff (push) Has been cancelled
Run python tests with pytest / Run tests and count coverage (3.12) (push) Has been cancelled
React Widget Build / build (push) Has been cancelled
Build and push multi-arch DocsGPT Docker image / build (linux/amd64, ubuntu-latest, amd64) (push) Has been cancelled
Backend release / release (push) Has been cancelled
Bandit Security Scan / bandit_scan (push) Has been cancelled
Build and push multi-arch DocsGPT Docker image / build (linux/arm64, ubuntu-24.04-arm, arm64) (push) Has been cancelled
Build and push multi-arch DocsGPT Docker image / manifest (push) Has been cancelled
Build and push DocsGPT FE Docker image for development / build (linux/amd64, ubuntu-latest, amd64) (push) Has been cancelled
Build and push DocsGPT FE Docker image for development / build (linux/arm64, ubuntu-24.04-arm, arm64) (push) Has been cancelled
Build and push DocsGPT FE Docker image for development / manifest (push) Has been cancelled
Python linting / ruff (push) Has been cancelled
Run python tests with pytest / Run tests and count coverage (3.12) (push) Has been cancelled
React Widget Build / build (push) Has been cancelled
This commit is contained in:
@@ -0,0 +1,83 @@
|
||||
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"
|
||||
Reference in New Issue
Block a user