mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-11-03 22:04:57 +08:00
fix(themoviedb): 让声明的 TMDB 请求超时真正生效 (#7123)
This commit is contained in:
@@ -139,6 +139,8 @@ class TMDb(object):
|
||||
"""
|
||||
读取运行配置并创建同步与异步请求客户端。
|
||||
|
||||
同步与异步客户端共用 15 秒请求超时,不随调用方会话传入。
|
||||
|
||||
未提供会话时创建自有会话。close 沿用历史行为,会关闭当前同步会话,
|
||||
包括调用方传入的会话;异步请求客户端不由该方法关闭。
|
||||
|
||||
@@ -157,6 +159,9 @@ class TMDb(object):
|
||||
self._total_pages: TMDbJson = None
|
||||
|
||||
self._owns_session = session is None
|
||||
# 超时必须在创建请求客户端之前赋值:_init_session 会读取它,
|
||||
# 放在客户端之后会让 timeout 静默回退到 RequestUtils 的默认值。
|
||||
self._timeout = 15
|
||||
self._init_session(session=session)
|
||||
|
||||
# TMDB 在部分代理和运营商链路下的 HTTP/2 长连接偶发卡死,识别路径优先保证稳定性。
|
||||
@@ -164,21 +169,26 @@ class TMDb(object):
|
||||
ua=get_runtime_setting('NORMAL_USER_AGENT'),
|
||||
proxies=self.proxies,
|
||||
http2=False,
|
||||
timeout=self._timeout,
|
||||
)
|
||||
|
||||
self._remaining = 40
|
||||
self._reset: int | None = None
|
||||
self._timeout = 15
|
||||
|
||||
def _init_session(self, session: requests.Session | None = None) -> None:
|
||||
"""
|
||||
初始化同步请求会话。
|
||||
|
||||
同步客户端沿用实例声明的请求超时。
|
||||
|
||||
:param session: 调用方提供的会话;为空时创建自有会话
|
||||
"""
|
||||
self._session = session or requests.Session()
|
||||
self._req = RequestUtils(
|
||||
ua=get_runtime_setting('NORMAL_USER_AGENT'),
|
||||
session=self._session,
|
||||
proxies=self.proxies,
|
||||
timeout=self._timeout,
|
||||
)
|
||||
|
||||
def _reset_owned_session(self) -> None:
|
||||
|
||||
91
tests/test_tmdb_request_timeout.py
Normal file
91
tests/test_tmdb_request_timeout.py
Normal file
@@ -0,0 +1,91 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
覆盖 TMDB 请求客户端超时契约的回归测试。
|
||||
|
||||
上游引入 `TMDb._timeout = 15` 时把它显式传给了当时的两处请求调用;后续将传输统一
|
||||
改走 RequestUtils/AsyncRequestUtils 的重构丢掉了这两个 `timeout=` 实参,并把赋值
|
||||
留在了同步与异步客户端构造之后,使声明的 15 秒静默回退成 20 秒默认值。识别链因此
|
||||
按更长的时间等待抖动链路,连接失败退避重试进一步放大了占用。
|
||||
"""
|
||||
from app.modules.themoviedb.tmdbv3api.tmdb import TMDb
|
||||
|
||||
|
||||
class _RecordingSession:
|
||||
"""
|
||||
记录传输层实际收到的请求参数,用于断言超时是否真正下发。
|
||||
"""
|
||||
|
||||
def __init__(self, payload: dict):
|
||||
"""
|
||||
保存预置响应载荷并初始化调用记录。
|
||||
"""
|
||||
self._payload = payload
|
||||
self.calls: list[dict] = []
|
||||
|
||||
def request(self, method: str, url: str, **kwargs):
|
||||
"""
|
||||
记录本次请求参数并返回预置响应。
|
||||
"""
|
||||
self.calls.append(kwargs)
|
||||
return _FakeResponse(payload=self._payload)
|
||||
|
||||
def close(self) -> None:
|
||||
"""
|
||||
兼容 requests.Session 的关闭接口,测试会话无需释放真实资源。
|
||||
"""
|
||||
|
||||
|
||||
class _FakeResponse:
|
||||
"""
|
||||
测试用响应对象,模拟 requests 响应的最小接口。
|
||||
"""
|
||||
|
||||
def __init__(self, payload: dict, headers: dict = None):
|
||||
"""
|
||||
初始化响应内容。
|
||||
"""
|
||||
self._payload = payload
|
||||
self.headers = headers or {}
|
||||
self.status_code = 200
|
||||
self.text = ""
|
||||
|
||||
def json(self) -> dict:
|
||||
"""
|
||||
返回预置JSON内容。
|
||||
"""
|
||||
return self._payload
|
||||
|
||||
|
||||
def test_sync_transport_receives_declared_timeout():
|
||||
"""
|
||||
同步请求必须把声明的 15 秒超时下发到传输层,而不是回退到 RequestUtils 默认的 20 秒。
|
||||
"""
|
||||
session = _RecordingSession(payload={"id": 1})
|
||||
tmdb = TMDb(session=session)
|
||||
|
||||
result = TMDb.request.__wrapped__(tmdb, "GET", "https://example.com/3/configuration", None, None)
|
||||
|
||||
assert result["json"] == {"id": 1}
|
||||
assert len(session.calls) == 1
|
||||
assert session.calls[0]["timeout"] == 15
|
||||
|
||||
|
||||
def test_async_transport_receives_declared_timeout():
|
||||
"""
|
||||
异步请求客户端与同步客户端共用同一超时声明,避免只有其中一条链路受约束。
|
||||
"""
|
||||
tmdb = TMDb()
|
||||
|
||||
assert tmdb._async_req._timeout == 15
|
||||
|
||||
|
||||
def test_reset_owned_session_keeps_declared_timeout():
|
||||
"""
|
||||
连接失败后重建自有会话时,超时声明必须保留,否则重试路径会退回默认值。
|
||||
"""
|
||||
tmdb = TMDb()
|
||||
assert tmdb._req._timeout == 15
|
||||
|
||||
tmdb._reset_owned_session()
|
||||
|
||||
assert tmdb._req._timeout == 15
|
||||
Reference in New Issue
Block a user