fix: correct Emby dashboard links

This commit is contained in:
jxxghp
2026-05-28 14:33:50 +08:00
parent cc3e062262
commit 543bee9ad5
4 changed files with 148 additions and 13 deletions

View File

@@ -0,0 +1,118 @@
import unittest
from typing import Any
from unittest.mock import Mock, patch
from app import schemas
from app.api.endpoints.mediaserver import play_item
from app.modules.emby.emby import Emby
class _FakeResponse:
"""提供 Emby 接口响应的最小 json 封装。"""
def __init__(self, payload: Any):
"""保存测试预置的响应体。"""
self._payload = payload
def json(self) -> Any:
"""返回测试预置的响应体。"""
return self._payload
class EmbyDashboardLinksTest(unittest.TestCase):
"""验证 Emby 仪表盘条目使用真实媒体服务器标识生成跳转链接。"""
@staticmethod
def _build_client() -> Emby:
"""构造绕过真实初始化的 Emby 实例。"""
client = Emby.__new__(Emby)
client._host = "http://emby.local/"
client._playhost = None
client._apikey = "api-key"
client._sync_libraries = []
client.user = "user-id"
client.serverid = "server-id"
return client
def test_get_server_id_falls_back_to_emby_prefixed_system_info(self):
"""
兼容 Emby 反代只暴露 /emby/System/Info 的场景,避免生成 serverId=None。
"""
client = self._build_client()
client.serverid = None
with patch("app.modules.emby.emby.RequestUtils") as request_utils_cls:
request_utils_cls.return_value.get_res.side_effect = [
None,
_FakeResponse({"Id": "server-id"}),
]
server_id = client.get_server_id()
self.assertEqual(server_id, "server-id")
self.assertEqual(
request_utils_cls.return_value.get_res.call_args_list[0].args[0],
"http://emby.local/System/Info",
)
self.assertEqual(
request_utils_cls.return_value.get_res.call_args_list[1].args[0],
"http://emby.local/emby/System/Info",
)
def test_get_play_url_omits_missing_server_id(self):
"""serverId 为空时不应把 None 字符串拼入播放链接。"""
client = self._build_client()
client.serverid = None
play_url = client.get_play_url("item-id")
self.assertEqual(
play_url,
"http://emby.local/web/index.html#!/item?id=item-id&context=home",
)
def test_get_latest_returns_item_and_server_ids(self):
"""最近入库条目需要显式返回 Emby item_id 和 server_id 供前端纠偏链接。"""
client = self._build_client()
client.get_user_library_folders = Mock(return_value=[])
with patch("app.modules.emby.emby.RequestUtils") as request_utils_cls:
request_utils_cls.return_value.get_res.return_value = _FakeResponse([
{
"Id": "emby-item-id",
"Name": "测试电影",
"Type": "Movie",
"ProductionYear": 2026,
}
])
items = client.get_latest()
self.assertEqual(items[0].id, "emby-item-id")
self.assertEqual(items[0].item_id, "emby-item-id")
self.assertEqual(items[0].server_id, "server-id")
self.assertIn("id=emby-item-id", items[0].link)
self.assertIn("serverId=server-id", items[0].link)
def test_play_item_returns_server_type(self):
"""播放地址接口需要返回 server_type供前端跳转时选择正确媒体服务器类型。"""
item = schemas.MediaServerItem(server="emby", item_id="emby-item-id")
with (
patch("app.api.endpoints.mediaserver.MediaServerHelper") as helper_cls,
patch("app.api.endpoints.mediaserver.MediaServerChain") as chain_cls,
):
helper_cls.return_value.get_configs.return_value = {"Emby": object()}
chain = chain_cls.return_value
chain.iteminfo.return_value = item
chain.get_play_url.return_value = "http://emby.local/web/index.html#!/item?id=emby-item-id"
response = play_item("emby-item-id")
self.assertTrue(response.success)
self.assertEqual(response.data["url"], "http://emby.local/web/index.html#!/item?id=emby-item-id")
self.assertEqual(response.data["server_type"], "emby")
if __name__ == "__main__":
unittest.main()