Simplify live video selector to main and sub streams

This commit is contained in:
Codex
2026-10-03 17:21:25 +08:00
parent d21170b41d
commit 2649854386
6 changed files with 15 additions and 7 deletions
+8
View File
@@ -4,6 +4,7 @@ import http.client
import io
import json
import os
import re
import tempfile
import threading
import time
@@ -63,6 +64,13 @@ class WebTests(unittest.TestCase):
self.assertEqual(headers['X-Frame-Options'], 'DENY')
self.assertIn("frame-ancestors 'none'", headers['Content-Security-Policy'])
def test_live_quality_has_only_main_and_sub(self):
code, _, body = self.request('/', False)
self.assertEqual(code, 200)
select = re.search(r'<select id="quality">(.*?)</select>', body.decode()).group(1)
self.assertEqual(re.findall(r'<option value="(.*?)">(.*?)</option>', select),
[('hd', '主码流'), ('compat', '子码流')])
def test_origin_rejected(self):
self.assertEqual(self.request('/api/storage', data={}, origin='https://example.org')[0], 403)