65 lines
2.8 KiB
Python
65 lines
2.8 KiB
Python
import json, os
|
|
from pathlib import Path
|
|
os.environ['QT_QPA_PLATFORM'] = 'offscreen'
|
|
os.environ['QTWEBENGINE_CHROMIUM_FLAGS'] = '--disable-gpu --disable-gpu-compositing'
|
|
os.environ['QT_OPENGL'] = 'software'
|
|
from PySide6.QtCore import QTimer, QUrl, Qt
|
|
from PySide6.QtWidgets import QApplication
|
|
QApplication.setAttribute(Qt.ApplicationAttribute.AA_ShareOpenGLContexts)
|
|
from PySide6.QtWebEngineWidgets import QWebEngineView
|
|
from PySide6.QtWebEngineCore import QWebEnginePage
|
|
class DebugPage(QWebEnginePage):
|
|
def javaScriptConsoleMessage(self, level, message, line, source):
|
|
print(f'JS: {message} at {line}', flush=True)
|
|
app = QApplication([])
|
|
view = QWebEngineView()
|
|
view.setPage(DebugPage(view))
|
|
view.resize(1440, 1000)
|
|
view.show()
|
|
cases = [
|
|
{"loading": False, "accounts": [], "error": "", "manualSetupRequired": False},
|
|
{"loading": True, "accounts": [], "error": "", "manualSetupRequired": True},
|
|
{"loading": False, "accounts": [{"id": "test"}], "error": "", "manualSetupRequired": False},
|
|
{"loading": False, "accounts": [], "error": "自动解密未完成，请登录企业微信", "manualSetupRequired": True},
|
|
]
|
|
index = 0
|
|
|
|
def inspect(result):
|
|
global index
|
|
result = json.loads(result) if isinstance(result, str) and result else {}
|
|
expected = cases[index]
|
|
manual = bool(expected['manualSetupRequired'] and not expected['loading'])
|
|
valid = result.get('manualCount') == (3 if manual else 0) and result.get('refreshDisabled') == expected['loading']
|
|
if expected['error']:
|
|
valid = valid and result.get('unready') and not result.get('ready')
|
|
print(f"Database UI case {index + 1}: {'PASS' if valid else 'FAIL'} {result}", flush=True)
|
|
if not valid:
|
|
app.exit(2)
|
|
return
|
|
index += 1
|
|
if index == len(cases):
|
|
app.exit(0)
|
|
else:
|
|
render()
|
|
|
|
def render():
|
|
payload = {'ready': True, 'demo': False, 'view': 'wechatdb', 'wechatDb': cases[index]}
|
|
view.page().runJavaScript('window.applyState(' + json.dumps(payload) + '); window.showConsoleView("wechatdb");')
|
|
probe = """JSON.stringify({
|
|
manualCount: document.querySelectorAll('[data-action="wechat-db-directory"], [data-action="wechat-db-acquire"], [data-action="wechat-db-import"]').length,
|
|
refreshDisabled: document.querySelector('[data-action="wechat-db-refresh"]').disabled,
|
|
unready: document.body.innerText.includes('未就绪'),
|
|
ready: document.body.innerText.includes('已读取')
|
|
})"""
|
|
QTimer.singleShot(300, lambda: view.page().runJavaScript(probe, inspect))
|
|
|
|
def loaded(ok):
|
|
if not ok:
|
|
app.exit(3)
|
|
return
|
|
render()
|
|
|
|
view.loadFinished.connect(loaded)
|
|
view.load(QUrl.fromLocalFile(str(Path.cwd() / 'assets/ui/zhenyang-ai-console.html')))
|
|
QTimer.singleShot(25000, lambda: app.exit(4))
|
|
raise SystemExit(app.exec()) |