sorrow404null
a7412824e0
Import the pre-repair source tree as the history baseline. Runtime data (data/), virtualenvs, bytecode caches and logs are gitignored so local secrets and user state stay out of the repo.
65 lines
1.8 KiB
Python
65 lines
1.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""验证 probeStream 探针在真实 WebEngine 页返回正确数据"""
|
|
import os, sys
|
|
os.environ["QTWEBENGINE_CHROMIUM_FLAGS"] = "--disable-gpu"
|
|
os.environ.setdefault("QT_QPA_PLATFORM", "offscreen")
|
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
|
|
from PyQt6.QtWidgets import QApplication
|
|
import PyQt6.QtWebEngineWidgets # noqa
|
|
from PyQt6.QtCore import QTimer
|
|
from ui.views.main_window import MainWindow
|
|
|
|
PASS = 0
|
|
FAIL = 0
|
|
|
|
def check(name, cond, detail=""):
|
|
global PASS, FAIL
|
|
if cond:
|
|
PASS += 1
|
|
print(f" PASS {name}")
|
|
else:
|
|
FAIL += 1
|
|
print(f" FAIL {name} {detail}")
|
|
|
|
app = QApplication(sys.argv)
|
|
window = MainWindow()
|
|
window.resize(1280, 800)
|
|
window.show()
|
|
|
|
PROBE_JS = r"""
|
|
(function() {
|
|
var m = 'probe-test-' + Date.now();
|
|
createMessage(m, 'assistant', '', 'LG');
|
|
appendReasoning(m, '思考探针内容');
|
|
appendToken(m, '正文探针 ');
|
|
appendToken(m, '第二段');
|
|
// 同步强制渲染
|
|
if (typeof forceRenderNow === 'function') forceRenderNow(m);
|
|
var res = probeStream(m);
|
|
finishMessage(m);
|
|
return res;
|
|
})()
|
|
"""
|
|
|
|
def step1():
|
|
window.browser.page().runJavaScript(PROBE_JS, on_probe)
|
|
|
|
def on_probe(res):
|
|
import json
|
|
d = json.loads(str(res))
|
|
print(" probe:", str(d)[:300])
|
|
check("probe 返回缓冲段", len(d.get("segs", [])) >= 1, str(d))
|
|
if d.get("segs"):
|
|
s0 = d["segs"][0]
|
|
check("缓冲长度>0", s0["b"] > 0, str(s0))
|
|
check("DOM 已写入", s0["d"] > 0, str(s0))
|
|
check("思考段探针", len(d.get("thinks", [])) >= 1, str(d))
|
|
check("无 JS 异常", "err" not in d, str(d))
|
|
print(f"===== {'ALL PASS' if FAIL == 0 else 'HAS FAILURES'}: {PASS}/{PASS+FAIL} =====")
|
|
app.quit()
|
|
|
|
QTimer.singleShot(1500, step1)
|
|
app.exec()
|
|
sys.exit(1 if FAIL else 0)
|