155 lines
8.6 KiB
Python
155 lines
8.6 KiB
Python
"""Isolated real-browser regression for gallery -> ComfyUI LoRA-stack actions.
|
|
|
|
Build the gallery, start `python scripts/preview.py --host 0.0.0.0 --port 8189`,
|
|
then `python scripts/browser_lora_stack.py http://127.0.0.1:8189`.
|
|
The metadata and ComfyUI receiver are intercepted/injected; no real workflow
|
|
or image files are changed. The actual bridge is separately tested by Node.
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import asyncio
|
|
import sys
|
|
from playwright.async_api import async_playwright, expect
|
|
|
|
BASE = sys.argv[1].rstrip("/") if len(sys.argv) > 1 else "http://127.0.0.1:8189"
|
|
|
|
RECEIVER = r"""
|
|
(() => {
|
|
localStorage.setItem('universal-extractor-locale', 'zh-CN');
|
|
localStorage.setItem('universal-extractor:onboarding-tour-v1-completed', 'true');
|
|
localStorage.setItem('universal-extractor:ui-preferences', JSON.stringify({
|
|
enableImagePrefetch: false, enableLiveGalleryRefresh: false
|
|
}));
|
|
const channel = new BroadcastChannel('universal-extractor-workflow');
|
|
const info = window.__ueTest = { received: [], wrongSent: false, replyError: '' };
|
|
channel.addEventListener('message', ({data}) => {
|
|
if (data.type === 'universal-extractor:workflow-probe') {
|
|
channel.postMessage({type: 'universal-extractor:workflow-ack',
|
|
instanceId:'background',probeId:data.probeId,focused:false,visibilityState:'hidden',loraStackProtocol:2});
|
|
channel.postMessage({type: 'universal-extractor:workflow-ack',
|
|
instanceId:'active',probeId:data.probeId,focused:true,visibilityState:'visible',loraStackProtocol:2});
|
|
}
|
|
if (data.type === 'universal-extractor:lora-stack-message') {
|
|
info.received.push(data);
|
|
const id = data.payload.id;
|
|
// A background ComfyUI tab tries to claim delivery first: it must not
|
|
// finish the gallery operation that is addressed to the active tab.
|
|
setTimeout(() => {
|
|
channel.postMessage({type:'universal-extractor:lora-stack-delivered',
|
|
instanceId:'background',payloadId:id,ok:true,loraStackProtocol:2,summary:{target:'WRONG',applied:50,added:50,updated:0,removed:0}});
|
|
info.wrongSent = true;
|
|
}, 40);
|
|
setTimeout(() => {
|
|
const replace = data.payload.mode === 'replace';
|
|
channel.postMessage({type:'universal-extractor:lora-stack-delivered',
|
|
instanceId:'active',payloadId:id,ok:!info.replyError,loraStackProtocol:2,
|
|
error:info.replyError || null,
|
|
summary:{target:'LoRA target',applied:2,added:replace?2:1,updated:replace?0:1,removed:replace?3:0}});
|
|
}, 800);
|
|
}
|
|
});
|
|
})();
|
|
"""
|
|
|
|
|
|
async def main():
|
|
errors = []
|
|
state = {"valid": True}
|
|
async with async_playwright() as playwright:
|
|
browser = await playwright.chromium.launch(args=["--no-sandbox"])
|
|
page = await browser.new_page(viewport={"width": 1100, "height": 780}, reduced_motion="reduce")
|
|
page.on("pageerror", lambda error: errors.append(str(error)))
|
|
await page.add_init_script(RECEIVER)
|
|
|
|
async def metadata(route):
|
|
response = await route.fetch()
|
|
data = await response.json()
|
|
recipe = data.get("recipe") or {}
|
|
stack = [
|
|
{"name": "A", "strength_model": .8, "strength_clip": .8, "enabled": True},
|
|
{"name": "A", "strength_model": 1.2, "strength_clip": 0, "enabled": True},
|
|
{"name": "B", "strength_model": .7, "strength_clip": .7, "enabled": True},
|
|
] if state["valid"] else []
|
|
recipe["lora_manager"] = {"detected": state["valid"], "raw_stack": "", "loras": stack}
|
|
recipe["loras"] = stack
|
|
data["recipe"] = recipe
|
|
await route.fulfill(json=data)
|
|
|
|
await page.route("**/universal_gallery/api/metadata?*", metadata)
|
|
await page.goto(BASE + "/gallery/", wait_until="networkidle")
|
|
card = page.locator(".ue-gallery-card").first
|
|
await expect(card).to_be_visible()
|
|
|
|
async def open_from_context():
|
|
await card.click(button="right")
|
|
await page.get_by_role("button", name="应用 LoRA 堆到当前工作流").click()
|
|
dialog = page.get_by_role("alertdialog")
|
|
await expect(dialog).to_be_visible()
|
|
return dialog
|
|
|
|
dialog = await open_from_context()
|
|
text = await dialog.inner_text()
|
|
assert "2 个 LoRA" in text, text # deduped before confirming
|
|
assert "CLIP 强度与主强度不同" in text and "同步覆盖 CLIP" in text, text
|
|
assert await dialog.locator("#ue-confirm-body").evaluate("el => getComputedStyle(el).whiteSpace") == "pre-line"
|
|
for label in ["取消", "追加到原堆", "覆盖旧堆"]:
|
|
await expect(dialog.get_by_role("button", name=label)).to_be_visible()
|
|
await dialog.get_by_role("button", name="取消").click()
|
|
assert await page.evaluate("window.__ueTest.received.length") == 0
|
|
|
|
dialog = await open_from_context()
|
|
await dialog.get_by_role("button", name="追加到原堆").click()
|
|
# The preview's CSP intentionally disallows eval-based polling.
|
|
await asyncio.sleep(1.2) # probe (900 ms) + the wrong receipt (40 ms)
|
|
assert await page.evaluate("window.__ueTest.wrongSent")
|
|
assert await page.locator(".ue-operation-card--success").filter(has_text="WRONG").count() == 0
|
|
assert await page.locator(".ue-operation-card--pending").filter(has_text="应用 LoRA").count() == 1, "a background tab must not end the operation"
|
|
await expect(page.locator(".ue-operation-card--success").filter(has_text="已将 2 个 LoRA 合并到“LoRA target”")).to_be_visible(timeout=6000)
|
|
await expect(page.locator(".ue-operation-card--success").filter(has_text="新增 1 项，同名更新 1 项")).to_be_visible()
|
|
received = await page.evaluate("window.__ueTest.received")
|
|
assert len(received) == 1 and received[0]["targetInstanceId"] == "active", received
|
|
assert received[0]["payload"]["mode"] == "append", received
|
|
|
|
# The detail-page button must have exactly the same three choices.
|
|
await card.locator("img").first.click()
|
|
detail = page.locator(".ue-lightbox-shell")
|
|
await expect(detail).to_be_visible()
|
|
detail_button = detail.get_by_role("button", name="应用 LoRA 堆到当前工作流")
|
|
await expect(detail_button).to_be_visible()
|
|
await detail_button.click()
|
|
dialog = page.get_by_role("alertdialog")
|
|
await expect(dialog).to_be_visible()
|
|
await page.set_viewport_size({"width": 390, "height": 780})
|
|
await expect(dialog.get_by_role("button", name="覆盖旧堆")).to_be_visible()
|
|
assert not await page.evaluate("document.documentElement.scrollWidth > innerWidth"), "mobile horizontal overflow"
|
|
for label in ["取消", "追加到原堆", "覆盖旧堆"]:
|
|
rect = await dialog.get_by_role("button", name=label).bounding_box()
|
|
assert rect and rect["x"] >= 0 and rect["x"] + rect["width"] <= 391, (label, rect)
|
|
await dialog.get_by_role("button", name="覆盖旧堆").click()
|
|
await expect(page.locator(".ue-operation-card--success").filter(has_text="已用 2 个 LoRA 覆盖“LoRA target”")).to_be_visible(timeout=6000)
|
|
await expect(page.locator(".ue-operation-card--success").filter(has_text="移除旧条目 3 项")).to_be_visible()
|
|
received = await page.evaluate("window.__ueTest.received")
|
|
assert len(received) == 2 and received[-1]["payload"]["mode"] == "replace", received
|
|
|
|
# A receiver failure stays visible; an empty recipe must never open
|
|
# a destructive confirmation or dispatch a third stack.
|
|
state["valid"] = False
|
|
await detail_button.click()
|
|
await expect(page.locator(".ue-operation-card--error").filter(has_text="未识别到可用")).to_be_visible()
|
|
await expect(page.get_by_role("alertdialog")).to_have_count(0)
|
|
assert await page.evaluate("window.__ueTest.received.length") == 2
|
|
|
|
state["valid"] = True
|
|
await page.evaluate("window.__ueTest.replyError = 'receiver refused the stack'")
|
|
await detail_button.click()
|
|
await page.get_by_role("alertdialog").get_by_role("button", name="追加到原堆").click()
|
|
await expect(page.locator(".ue-operation-card--error").filter(has_text="receiver refused the stack")).to_be_visible(timeout=6000)
|
|
assert await page.evaluate("window.__ueTest.received.length") == 3
|
|
assert not errors, errors
|
|
await browser.close()
|
|
print("PASS: context and detail append/replace/cancel, deduplicated count, CLIP warning, mobile actions, targeted async receipts, missing-stack and error feedback. No page errors.")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main())
|