[Update] 2026-02-12 20:23:09, 18 files

tests/conftest.py

@@ -2,10 +2,25 @@
 
 from __future__ import annotations
 
+import base64
 import json
 import struct
 
 import pytest
+import respx
+
+from tensors.generate import SDClient
+
+BASE_URL = "http://127.0.0.1:1234"
+
+# 1x1 red PNG for image response stubs
+TINY_PNG = (
+    b"\x89PNG\r\n\x1a\n\x00\x00\x00\rIHDR\x00\x00\x00\x01"
+    b"\x00\x00\x00\x01\x08\x02\x00\x00\x00\x90wS\xde\x00"
+    b"\x00\x00\x0cIDATx\x9cc\xf8\x0f\x00\x00\x01\x01\x00"
+    b"\x05\x18\xd8N\x00\x00\x00\x00IEND\xaeB`\x82"
+)
+TINY_PNG_B64 = base64.b64encode(TINY_PNG).decode()
 
 
 @pytest.fixture
@@ -28,3 +43,18 @@ def temp_safetensor(tmp_path):
         f.write(header_bytes)
 
     return file_path
+
+
+@pytest.fixture()
+def mock_api():
+    """Activate respx mock for the sd-server base URL."""
+    with respx.mock(base_url=BASE_URL, assert_all_called=False) as rsps:
+        yield rsps
+
+
+@pytest.fixture()
+def client(mock_api: respx.MockRouter) -> SDClient:  # noqa: ARG001
+    """SDClient wired to the mocked transport."""
+    c = SDClient()
+    yield c  # type: ignore[misc]
+    c.close()