mirror of
https://github.com/unshackle-dl/unshackle.git
synced 2026-06-10 19:22:08 +00:00
746b573711
Covers RemoteClient/RemoteService, REST routes, handlers, SessionStore, InputBridge, DownloadQueueManager, errors, compression, and serve CLI. E2e tier opts in via --live and can auto-spawn its own serve.
112 lines
3.1 KiB
Python
112 lines
3.1 KiB
Python
"""Unit tests for unshackle.core.remote_service.RemoteClient."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
|
|
import pytest
|
|
import responses
|
|
|
|
from unshackle.core.remote_service import RemoteClient
|
|
|
|
pytestmark = pytest.mark.unit
|
|
|
|
|
|
@pytest.fixture
|
|
def client() -> RemoteClient:
|
|
return RemoteClient(server_url="http://srv:8786", api_key="secret-xyz")
|
|
|
|
|
|
def test_session_sets_secret_key_header(client: RemoteClient) -> None:
|
|
s = client.session
|
|
assert s.headers.get("X-Secret-Key") == "secret-xyz"
|
|
assert s.headers["User-Agent"].startswith("unshackle/")
|
|
|
|
|
|
def test_session_omits_secret_key_when_empty() -> None:
|
|
c = RemoteClient(server_url="http://srv:8786", api_key="")
|
|
assert "X-Secret-Key" not in c.session.headers
|
|
|
|
|
|
def test_server_url_trailing_slash_stripped() -> None:
|
|
c = RemoteClient(server_url="http://srv:8786/", api_key="")
|
|
assert c.server_url == "http://srv:8786"
|
|
|
|
|
|
@responses.activate
|
|
def test_get_returns_json(client: RemoteClient) -> None:
|
|
responses.add(
|
|
responses.GET,
|
|
"http://srv:8786/api/health",
|
|
json={"status": "ok"},
|
|
status=200,
|
|
)
|
|
assert client.get("/api/health") == {"status": "ok"}
|
|
|
|
|
|
@responses.activate
|
|
def test_post_sends_json_body(client: RemoteClient) -> None:
|
|
captured = {}
|
|
|
|
def cb(request):
|
|
captured["body"] = json.loads(request.body)
|
|
return (200, {}, json.dumps({"session_id": "abc"}))
|
|
|
|
responses.add_callback(
|
|
responses.POST, "http://srv:8786/api/session/create", callback=cb, content_type="application/json"
|
|
)
|
|
result = client.post("/api/session/create", {"service": "ATV"})
|
|
assert result == {"session_id": "abc"}
|
|
assert captured["body"] == {"service": "ATV"}
|
|
|
|
|
|
@responses.activate
|
|
def test_delete_returns_json(client: RemoteClient) -> None:
|
|
responses.add(
|
|
responses.DELETE,
|
|
"http://srv:8786/api/session/abc",
|
|
json={"status": "deleted"},
|
|
status=200,
|
|
)
|
|
assert client.delete("/api/session/abc") == {"status": "deleted"}
|
|
|
|
|
|
@responses.activate
|
|
def test_4xx_raises_systemexit_with_logged_error(client: RemoteClient, caplog: pytest.LogCaptureFixture) -> None:
|
|
responses.add(
|
|
responses.GET,
|
|
"http://srv:8786/api/session/none",
|
|
json={"error_code": "SESSION_NOT_FOUND", "message": "no such session"},
|
|
status=404,
|
|
)
|
|
with caplog.at_level("ERROR"), pytest.raises(SystemExit):
|
|
client.get("/api/session/none")
|
|
assert "SESSION_NOT_FOUND" in caplog.text
|
|
assert "no such session" in caplog.text
|
|
|
|
|
|
@responses.activate
|
|
def test_connection_error_raises_systemexit(client: RemoteClient) -> None:
|
|
import requests
|
|
|
|
responses.add(
|
|
responses.GET,
|
|
"http://srv:8786/api/health",
|
|
body=requests.ConnectionError("boom"),
|
|
)
|
|
with pytest.raises(SystemExit):
|
|
client.get("/api/health")
|
|
|
|
|
|
@responses.activate
|
|
def test_timeout_raises_systemexit(client: RemoteClient) -> None:
|
|
import requests
|
|
|
|
responses.add(
|
|
responses.GET,
|
|
"http://srv:8786/api/health",
|
|
body=requests.Timeout("slow"),
|
|
)
|
|
with pytest.raises(SystemExit):
|
|
client.get("/api/health")
|