chore: more tests

tests/test_error.py

@@ -1,5 +1,5 @@
 import pytest
-from application.error import bad_request, response_error
+from application.error import bad_request, response_error, sanitize_api_error
 from flask import Flask
 
 
@@ -41,3 +41,43 @@ def test_response_error_without_message(app):
         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"