mirror of
https://github.com/unshackle-dl/unshackle.git
synced 2026-11-12 12:05:07 +00:00
imSp4rky
2cc5e44957
The client sends its credentials, cookies and cache, but none for an anonymous-search service. The API relay uses the same path.
72 lines
2.6 KiB
Python
72 lines
2.6 KiB
Python
"""RemoteService.search: the body ``search --remote`` posts, and the results it yields."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from types import SimpleNamespace
|
|
from typing import Any
|
|
|
|
import pytest
|
|
|
|
from unshackle.core import remote_service
|
|
from unshackle.core.remote_service import RemoteService
|
|
|
|
pytestmark = pytest.mark.unit
|
|
|
|
|
|
class FakeClient:
|
|
server_url = "https://server.example"
|
|
|
|
def __init__(self) -> None:
|
|
self.sent: dict[str, Any] = {}
|
|
|
|
def post(self, endpoint: str, data: dict[str, Any]) -> dict[str, Any]:
|
|
self.sent = {"endpoint": endpoint, **data}
|
|
return {"results": [{"id": "t1", "title": "Show", "label": "2019"}], "cache": {"token": "x"}}
|
|
|
|
|
|
def make_service(params: dict[str, Any], server_accounts: Any = None) -> tuple[RemoteService, FakeClient, list]:
|
|
svc = RemoteService.__new__(RemoteService)
|
|
client = FakeClient()
|
|
saved: list = []
|
|
svc.service_tag = "TEST"
|
|
svc.title_id = "query"
|
|
svc.client = client # type: ignore[assignment]
|
|
svc.ctx = SimpleNamespace(parent=SimpleNamespace(params=params), obj=None) # type: ignore[assignment]
|
|
svc._server_accounts = server_accounts
|
|
svc._anonymous_search = False
|
|
svc.load_cache_files = lambda profile=None: {"token": "old"} # type: ignore[method-assign]
|
|
svc.save_returned_cache = lambda data, profile=None: saved.append(data) # type: ignore[method-assign]
|
|
return svc, client, saved
|
|
|
|
|
|
def test_search_sends_client_login_and_yields_results(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.setattr(remote_service, "load_credentials_for_transport", lambda tag, profile: {"username": "u"})
|
|
monkeypatch.setattr(remote_service, "load_cookies_for_transport", lambda tag, profile: None)
|
|
svc, client, saved = make_service({"profile": "p1", "no_proxy": True})
|
|
|
|
results = list(svc.search())
|
|
|
|
assert client.sent == {
|
|
"endpoint": "/api/search",
|
|
"service": "TEST",
|
|
"query": "query",
|
|
"profile": "p1",
|
|
"credentials": {"username": "u"},
|
|
"cache": {"token": "old"},
|
|
"no_proxy": True,
|
|
}
|
|
assert [(r.id, r.title, r.label) for r in results] == [("t1", "Show", "2019")]
|
|
assert saved == [{"token": "x"}]
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("server_accounts", "anonymous"), [({"global": True}, False), (None, True)], ids=["server-account", "anonymous"]
|
|
)
|
|
def test_search_sends_no_login(server_accounts: Any, anonymous: bool) -> None:
|
|
svc, client, _ = make_service({}, server_accounts)
|
|
svc._anonymous_search = anonymous
|
|
|
|
list(svc.search())
|
|
|
|
assert client.sent == {"endpoint": "/api/search", "service": "TEST", "query": "query"}
|