Archived
feat(routes): 迁移音频流代理与结果交付端点(流式)
- web/routers/stream.py:/api/v1/audio_stream 音频范围流代理 · 签名校验(core.security,密钥同源两栈互认);HEAD 直接 200 不请求上游 · Range 原样透传上游(206 + Content-Range,拖动进度可用) · 编码透明:上游带 Content-Encoding 时不沿用压缩前长度(同 Flask F92) · StreamingResponse 16KB 分块,finally 关闭上游(下游中断亦释放) - core/http_client.py 增 open():返回未关闭流式响应,便于先读状态码再转发 - web/routers/delivery.py:/download/result、/download/batch_zip/result · 归属 403 / 进行中 202 / 暂停 200 / 取消 200 / 失败 500 / 缺失 404 分支逐一对齐 · FileResponse 流式交付 + no-store + X-Download-Filename;清理由监控线程统一回收 验证:pytest 51/51(新增 5 项流代理:无效 token/HEAD/全量/Range 透传/上游关闭); ruff All checks passed;pyright 0 errors;奇偶校验 26/26;实测 5.2MB 流与 Range 206 正确。
This commit is contained in:
1 parent
f588d313f0
commit
0104ae55e5
5 files changed
+326
No files matched your search
@@ -0,0 +1,88 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
"""音频流代理测试:签名校验、HEAD、Range 透传、无效 token。"""
|
||||
import pytest
|
||||
from fastapi import FastAPI
|
||||
from fastapi.testclient import TestClient
|
||||
|
||||
from core import security
|
||||
from web.routers import stream as stream_router
|
||||
|
||||
SECRET = 'stream-test-secret'
|
||||
PAYLOAD = b'AUDIO-DATA-' * 1000 # 11000 字节
|
||||
|
||||
|
||||
@pytest.fixture()
|
||||
def client(monkeypatch):
|
||||
security.init_secret_key(SECRET)
|
||||
app = FastAPI()
|
||||
app.include_router(stream_router.router, prefix='/api/v1')
|
||||
|
||||
class _Upstream:
|
||||
def __init__(self):
|
||||
self.status = 200
|
||||
self.headers = {'Content-Type': 'audio/mpeg', 'Content-Length': str(len(PAYLOAD)),
|
||||
'Accept-Ranges': 'bytes'}
|
||||
self.closed = False
|
||||
self.content = self
|
||||
|
||||
async def iter_chunked(self, n):
|
||||
for i in range(0, len(PAYLOAD), n):
|
||||
yield PAYLOAD[i:i + n]
|
||||
|
||||
def close(self):
|
||||
self.closed = True
|
||||
|
||||
state = {'last_range': None, 'upstream': None}
|
||||
|
||||
async def fake_open(method, url, *, headers=None, timeout=None, **kw):
|
||||
state['last_range'] = (headers or {}).get('Range')
|
||||
up = _Upstream()
|
||||
if state['last_range']:
|
||||
up.status = 206
|
||||
up.headers = dict(up.headers)
|
||||
up.headers['Content-Range'] = f'bytes 0-99/{len(PAYLOAD)}'
|
||||
state['upstream'] = up
|
||||
return up
|
||||
|
||||
monkeypatch.setattr('web.routers.stream.http_client.open', fake_open)
|
||||
return TestClient(app), state
|
||||
|
||||
|
||||
def test_invalid_token(client):
|
||||
c, _ = client
|
||||
r = c.get('/api/v1/audio_stream?token=bogus')
|
||||
assert r.status_code == 403
|
||||
|
||||
|
||||
def test_head_no_upstream(client):
|
||||
c, state = client
|
||||
r = c.head('/api/v1/audio_stream?token=x')
|
||||
assert r.status_code == 200
|
||||
assert r.headers['accept-ranges'] == 'bytes'
|
||||
assert state['upstream'] is None # HEAD 不应请求上游
|
||||
|
||||
|
||||
def test_stream_full(client):
|
||||
c, _ = client
|
||||
token = security.make_audio_token('http://m801.music.126.net/a.mp3')
|
||||
r = c.get(f'/api/v1/audio_stream?token={token}')
|
||||
assert r.status_code == 200
|
||||
assert r.content == PAYLOAD
|
||||
assert r.headers['cache-control'] == 'no-store'
|
||||
assert r.headers['content-length'] == str(len(PAYLOAD))
|
||||
|
||||
|
||||
def test_range_forwarded_and_206(client):
|
||||
c, state = client
|
||||
token = security.make_audio_token('http://m801.music.126.net/a.mp3')
|
||||
r = c.get(f'/api/v1/audio_stream?token={token}', headers={'Range': 'bytes=0-99'})
|
||||
assert state['last_range'] == 'bytes=0-99' # Range 已透传上游
|
||||
assert r.status_code == 206
|
||||
assert 'content-range' in r.headers
|
||||
|
||||
|
||||
def test_upstream_closed_on_complete(client):
|
||||
c, state = client
|
||||
token = security.make_audio_token('http://m801.music.126.net/a.mp3')
|
||||
c.get(f'/api/v1/audio_stream?token={token}')
|
||||
assert state['upstream'].closed is True
|
||||
Reference in new issue
Block a user