48 lines
2.8 KiB
Python
48 lines
2.8 KiB
Python
"""Controlled API delay benchmark; never connects to a real service."""
|
|
import ast
|
|
import json
|
|
import statistics
|
|
import sys
|
|
import tempfile
|
|
import threading
|
|
import time
|
|
from pathlib import Path
|
|
from types import SimpleNamespace
|
|
from unittest import mock
|
|
|
|
project=Path(sys.argv[1]);output=Path(sys.argv[2])
|
|
sys.path.insert(0,str(project));sys.dont_write_bytecode=True
|
|
import runtime_paths
|
|
with tempfile.TemporaryDirectory(prefix='local-first-benchmark-') as temp, mock.patch.object(runtime_paths,'application_data_dir',return_value=Path(temp)):
|
|
import wxwork_message_browser as browser
|
|
import backend_client
|
|
def method(path):
|
|
module=ast.parse(path.read_text(encoding='utf-8'))
|
|
cls=next(n for n in module.body if isinstance(n,ast.ClassDef) and n.name=='MainWindow')
|
|
node=next(n for n in cls.body if isinstance(n,ast.FunctionDef) and n.name=='_refresh_wechat_database')
|
|
node.decorator_list=[]
|
|
namespace={'threading':threading}
|
|
exec(compile(ast.Module(body=[node],type_ignores=[]),str(path),'exec'),namespace)
|
|
return namespace[node.name]
|
|
result={'project':str(project),'simulated_patient_api_ms':300,'iterations':3}
|
|
for label,path in [('before',project/'backups/loading-performance-20260917/wechat_gui_qt.py'),('after',project/'wechat_gui_qt.py')]:
|
|
refresh=method(path);samples=[];api_calls=[]
|
|
for iteration in range(3):
|
|
event=threading.Event();first=[]
|
|
host=SimpleNamespace(_wechat_db_state={},_wechat_db_selected_account='',_wechat_db_selected_conversation='',
|
|
_wechat_db_query='',_wechat_db_load_serial=0,_schedule_console_push=mock.Mock())
|
|
def emit(snapshot):
|
|
first.append(time.perf_counter());event.set()
|
|
host.wechatDatabaseLoaded=SimpleNamespace(emit=emit)
|
|
def patient(*a,**kw):
|
|
time.sleep(.3);return {'available':True}
|
|
snapshot={'loading':False,'selectedAccount':'synthetic-account','selectedConversation':'synthetic-chat','messages':[{'content':'synthetic'}]}
|
|
with mock.patch.object(browser,'load_browser_snapshot',side_effect=lambda **kw:dict(snapshot)), mock.patch.object(backend_client,'desktop_patient_context',side_effect=patient) as api:
|
|
started=time.perf_counter();refresh(host)
|
|
if not event.wait(3):raise RuntimeError('local snapshot timeout')
|
|
samples.append(round((first[0]-started)*1000,3));api_calls.append(api.call_count)
|
|
result[label]={'first_local_snapshot_ms':samples,'median_ms':statistics.median(samples),'api_calls_before_snapshot':api_calls}
|
|
assert result['after']['median_ms'] < result['before']['median_ms'] / 3
|
|
output.write_text(json.dumps(result,ensure_ascii=False,indent=2),encoding='utf-8')
|
|
print(json.dumps(result,ensure_ascii=False))
|