"""Unit tests for SDK HTTP error decoding helpers."""
from __future__ import annotations
import httpx
import pytest
from omnigent_client._errors import OmnigentError, require_json_object, response_body
def test_response_body_returns_text_for_non_json_error() -> None:
"""Non-JSON error bodies are preserved as text for status handling."""
resp = httpx.Response(
502,
headers={"content-type": "text/html"},
text="Bad gateway",
)
assert response_body(resp) == "Bad gateway"
def test_response_body_returns_json_error_object() -> None:
"""Structured server error envelopes stay structured."""
body = {"error": {"code": "invalid_input", "message": "bad request"}}
resp = httpx.Response(400, json=body)
assert response_body(resp) == body
def test_require_json_object_rejects_html_success_response() -> None:
"""Successful HTML fallback pages raise ``OmnigentError`` with context."""
resp = httpx.Response(
200,
headers={"content-type": "text/html; charset=utf-8"},
text="Omnigent",
)
with pytest.raises(OmnigentError) as exc_info:
require_json_object(resp, "GET /v1/conversations")
message = str(exc_info.value)
assert "GET /v1/conversations returned non-JSON response" in message
assert "status=200" in message
assert "text/html" in message
assert "Omnigent" in message
assert exc_info.value.status_code == 200
def test_require_json_object_rejects_json_array_success_response() -> None:
"""Success endpoints expecting objects reject other JSON types."""
resp = httpx.Response(200, json=[])
with pytest.raises(OmnigentError) as exc_info:
require_json_object(resp, "GET /v1/conversations")
assert str(exc_info.value) == "GET /v1/conversations returned JSON list, expected object"
assert exc_info.value.status_code == 200
def test_require_json_object_reports_empty_body_without_content_type() -> None:
"""Empty non-JSON bodies report the missing response details explicitly."""
resp = httpx.Response(200, content=b"")
with pytest.raises(OmnigentError) as exc_info:
require_json_object(resp, "GET /v1/conversations")
message = str(exc_info.value)
assert "status=200" in message
assert "no content-type header" in message
assert "" in message