mirror of
https://github.com/JefferyHcool/BiliNote.git
synced 2026-08-28 19:46:57 +08:00
pumpkinperson996andClaude Fable 5
b85d7bc1ff
从"稍后再看"（/list/watchlater/?bvid=BV...）、收藏夹播放页等场景 复制的链接会被判为无效链接，尽管其中包含完整 BV 号；校验通过的 链接也会带着追踪参数原样传给 yt-dlp。 在请求入口新增 normalize_video_url()：提取 BV 号重建标准 /video/BVxxx 链接，保留分 P 参数（?p=N），丢弃其余查询参数。 b23.tv 短链与其他平台行为不变，无 BV 号链接仍按原逻辑拒绝。 Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
68 lines
2.4 KiB
Python
68 lines
2.4 KiB
Python
import pytest
|
|
|
|
from app.utils.url_parser import normalize_video_url
|
|
from app.validators.video_url_validator import VideoRequest
|
|
|
|
|
|
def test_watchlater_url():
|
|
url = ("https://www.bilibili.com/list/watchlater/?bvid=BV1CPXpBYEui"
|
|
"&oid=116294762371214&spm_id_from=333.881.0.0&vd_source=abc")
|
|
assert normalize_video_url(url) == "https://www.bilibili.com/video/BV1CPXpBYEui"
|
|
|
|
|
|
def test_favlist_url():
|
|
url = "https://www.bilibili.com/list/ml123456?bvid=BV1xx411c7mD&oid=987"
|
|
assert normalize_video_url(url) == "https://www.bilibili.com/video/BV1xx411c7mD"
|
|
|
|
|
|
def test_tracking_params_stripped():
|
|
url = "https://www.bilibili.com/video/BV1xx411c7mD/?spm_id_from=333.881&vd_source=abc"
|
|
assert normalize_video_url(url) == "https://www.bilibili.com/video/BV1xx411c7mD"
|
|
|
|
|
|
def test_p_number_preserved():
|
|
url = "https://www.bilibili.com/video/BV1xx411c7mD?p=36&spm_id_from=333.881"
|
|
assert normalize_video_url(url) == "https://www.bilibili.com/video/BV1xx411c7mD?p=36"
|
|
|
|
|
|
def test_no_bv_returned_unchanged():
|
|
url = "https://www.bilibili.com/anime/timeline"
|
|
assert normalize_video_url(url) == url
|
|
|
|
|
|
def test_b23_short_url_unchanged():
|
|
url = "https://b23.tv/abc123"
|
|
assert normalize_video_url(url) == url
|
|
|
|
|
|
def test_video_request_accepts_watchlater():
|
|
req = VideoRequest(
|
|
url="https://www.bilibili.com/list/watchlater/?bvid=BV1CPXpBYEui&oid=116294762371214",
|
|
platform="bilibili",
|
|
)
|
|
assert str(req.url) == "https://www.bilibili.com/video/BV1CPXpBYEui"
|
|
|
|
|
|
def test_video_request_rejects_no_bv():
|
|
with pytest.raises(ValueError):
|
|
VideoRequest(url="https://www.bilibili.com/anime/timeline", platform="bilibili")
|
|
|
|
|
|
def test_note_router_request_accepts_watchlater():
|
|
# note.py 里的 VideoRequest 才是 /generate_note 实际使用的请求模型
|
|
from app.routers.note import VideoRequest as NoteVideoRequest
|
|
|
|
req = NoteVideoRequest(
|
|
video_url="https://www.bilibili.com/list/watchlater/?bvid=BV1CPXpBYEui&oid=116294762371214&spm_id_from=333.881.0.0",
|
|
platform="bilibili",
|
|
quality="fast",
|
|
model_name="test-model",
|
|
provider_id="test-provider",
|
|
)
|
|
assert req.video_url == "https://www.bilibili.com/video/BV1CPXpBYEui"
|
|
|
|
|
|
def test_video_request_youtube_unaffected():
|
|
req = VideoRequest(url="https://www.youtube.com/watch?v=dQw4w9WgXcQ", platform="youtube")
|
|
assert str(req.url) == "https://www.youtube.com/watch?v=dQw4w9WgXcQ"
|