add test

Author: Gawor270

fishjam/api/_client.py

@@ -11,7 +11,11 @@
 class Client:
     def __init__(self, fishjam_id: str, management_token: str):
         self._fishjam_url = get_fishjam_url(fishjam_id)
-        self.client = AuthenticatedClient(self._fishjam_url, token=management_token, headers={"x-fishjam-api-client": f"python-server-{get_version()}"})
+        self.client = AuthenticatedClient(
+            self._fishjam_url,
+            token=management_token,
+            headers={"x-fishjam-api-client": f"python-server-{get_version()}"},
+        )
 
     def _request(self, method, **kwargs):
         response = method.sync_detailed(client=self.client, **kwargs)

tests/test_room_api.py

@@ -1,5 +1,7 @@
 import os
+from unittest.mock import Mock, patch
 
+import httpx
 import pytest
 
 from fishjam import (
@@ -26,6 +28,7 @@
     RoomType,
     VideoCodec,
 )
+from fishjam.version import get_version
 
 HOST = "proxy" if os.getenv("DOCKER_TEST") == "TRUE" else "localhost"
 FISHJAM_ID = f"http://{HOST}:5555"
@@ -54,6 +57,36 @@ def test_valid_token(self):
         assert room in all_rooms
 
 
+class TestAPIClientHeader:
+    def test_x_fishjam_api_client_header_is_sent(self):
+        mock_response = Mock(spec=httpx.Response)
+        mock_response.status_code = 200
+        mock_response.headers = httpx.Headers({})
+        mock_response.json.return_value = {"data": []}
+
+        captured_headers = None
+
+        def mock_send(request, **kwargs):
+            nonlocal captured_headers
+            captured_headers = dict(request.headers)
+            return mock_response
+
+        room_api = FishjamClient(FISHJAM_ID, MANAGEMENT_TOKEN)
+
+        with patch.object(httpx.HTTPTransport, "handle_request", side_effect=mock_send):
+            try:
+                room_api.get_all_rooms()
+            except Exception:
+                # We don't care if the request fails, we just want to check the headers
+                pass
+
+        assert captured_headers is not None
+        assert "x-fishjam-api-client" in captured_headers
+
+        expected_header_value = f"python-server-{get_version()}"
+        assert captured_headers["x-fishjam-api-client"] == expected_header_value
+
+
 @pytest.fixture
 def room_api():
     return FishjamClient(FISHJAM_ID, MANAGEMENT_TOKEN)