Relax content-type check (#22)

Co-authored-by: Daniel Buades Marcos <daniel@buad.es>

src/httpx_sse/_api.py

@@ -14,9 +14,9 @@ def __init__(self, response: httpx.Response) -> None:
 
  def _check_content_type(self) -> None:
  content_type = self._response.headers.get("content-type", "").partition(";")[0]
- if content_type != "text/event-stream":
+ if "text/event-stream" not in content_type:
  raise SSEError(
- "Expected response header Content-Type to be 'text/event-stream', "
+ "Expected response header Content-Type to contain 'text/event-stream', "
  f"got {content_type!r}"
  )
 

tests/test_api.py

@@ -4,15 +4,24 @@
 from httpx_sse import SSEError, aconnect_sse, connect_sse
 
 
-def test_connect_sse() -> None:
+@pytest.mark.parametrize(
+ "content_type",
+ [
+ pytest.param("text/event-stream", id="exact"),
+ pytest.param(
+ "application/json, text/event-stream; charset=utf-8", id="contains"
+ ),
+ ],
+)
+def test_connect_sse(content_type: str) -> None:
  def handler(request: httpx.Request) -> httpx.Response:
  if request.url.path == "/":
  return httpx.Response(200, text="Hello, world!")
  else:
  assert request.url.path == "/sse"
  text = "data: test\n\n"
  return httpx.Response(
- 200, headers={"content-type": "text/event-stream"}, text=text
+ 200, headers={"content-type": content_type}, text=text
  )
 
  with httpx.Client(transport=httpx.MockTransport(handler)) as client: