chore: checkpoint pi lab rust integration work
This commit is contained in:
Agent Board
@@ -0,0 +1,302 @@
|
||||
#!/usr/bin/env node
|
||||
"use strict";
|
||||
|
||||
const assert = require("node:assert");
|
||||
const fs = require("node:fs");
|
||||
const os = require("node:os");
|
||||
const path = require("node:path");
|
||||
const { chromium } = require("playwright");
|
||||
const {
|
||||
setupWorkspaceAccess,
|
||||
seedAiPolicy,
|
||||
} = require("./lib/control-plane-dev-seed");
|
||||
|
||||
const BASE = process.env.MNOTE_UI_BASE_URL || "http://127.0.0.1:3000";
|
||||
const STAMP = Date.now();
|
||||
const OUT = process.env.MNOTE_PI_MODEL_CONTROLS_OUT || path.join(os.tmpdir(), `mnote-pi-model-controls-${STAMP}`);
|
||||
const TIMEOUT = Number.parseInt(process.env.UI_TIMEOUT_MS || "240000", 10);
|
||||
const ACTOR_ID = process.env.MNOTE_E2E_ACTOR_ID || "mnote-e2e";
|
||||
const DEFAULT_E2E_ROOT = "/mnt/Data1T/Mnote_data/users/mnote-e2e/workspaces/my-space";
|
||||
const ROOT_FROM_ENV = process.env.MNOTE_PI_MODEL_ROOT_PATH;
|
||||
const ROOT_PATH = ROOT_FROM_ENV || (ACTOR_ID === "mnote-e2e" && fs.existsSync(DEFAULT_E2E_ROOT) ? DEFAULT_E2E_ROOT : path.join(OUT, "workspace"));
|
||||
const USING_DEFAULT_E2E_ROOT = !ROOT_FROM_ENV && ACTOR_ID === "mnote-e2e" && ROOT_PATH === DEFAULT_E2E_ROOT;
|
||||
const WORKSPACE_ID = process.env.MNOTE_PI_MODEL_WORKSPACE_ID || (USING_DEFAULT_E2E_ROOT ? "local-ws:mnote-e2e:my-space" : `local-ws:${ACTOR_ID}:pi-model-controls`);
|
||||
const ROOT_URI = process.env.MNOTE_PI_MODEL_ROOT_URI || `file://${ROOT_PATH}`;
|
||||
const PAGE_PATH = `pi-model-controls-${STAMP}/pi-model-controls-${STAMP}.md`;
|
||||
const MODEL_PROVIDER = process.env.MNOTE_PI_MODEL_PROVIDER || "omniroute";
|
||||
const MODEL_ID = process.env.MNOTE_PI_MODEL_ID || "gpt-5.4-mini";
|
||||
const CHROMIUM_EXECUTABLE = process.env.PLAYWRIGHT_CHROMIUM_EXECUTABLE
|
||||
|| (fs.existsSync("/usr/bin/chromium-browser") ? "/usr/bin/chromium-browser" : "")
|
||||
|| (fs.existsSync("/usr/bin/chromium") ? "/usr/bin/chromium" : "")
|
||||
|| (fs.existsSync("/usr/bin/google-chrome") ? "/usr/bin/google-chrome" : "");
|
||||
|
||||
function mkdirp(dir) {
|
||||
fs.mkdirSync(dir, { recursive: true });
|
||||
}
|
||||
|
||||
async function quickLogin(page) {
|
||||
await page.goto(`${BASE}/auth`, { waitUntil: "commit", timeout: TIMEOUT });
|
||||
const quickLoginButton = page.getByRole("button", { name: "测试账号快速登录" });
|
||||
if (!(await quickLoginButton.isVisible({ timeout: 4000 }).catch(() => false))) return;
|
||||
await Promise.all([
|
||||
page.waitForURL((url) => !url.toString().includes("/auth"), { timeout: TIMEOUT }).catch(() => null),
|
||||
quickLoginButton.click(),
|
||||
]);
|
||||
}
|
||||
|
||||
async function requestJson(page, url, options = {}) {
|
||||
const response = await page.request.fetch(`${BASE}${url}`, {
|
||||
...options,
|
||||
headers: {
|
||||
accept: "application/json",
|
||||
"content-type": "application/json",
|
||||
...(options.headers || {}),
|
||||
},
|
||||
timeout: options.timeout || TIMEOUT,
|
||||
});
|
||||
const text = await response.text();
|
||||
let body = {};
|
||||
try {
|
||||
body = text ? JSON.parse(text) : {};
|
||||
} catch {
|
||||
body = { raw: text };
|
||||
}
|
||||
assert(response.ok(), `${options.method || "GET"} ${url} failed: ${response.status()} ${text.slice(0, 800)}`);
|
||||
return body;
|
||||
}
|
||||
|
||||
function policy() {
|
||||
return {
|
||||
defaultModel: `${MODEL_PROVIDER}/${MODEL_ID}`,
|
||||
allowedModels: [`${MODEL_PROVIDER}/${MODEL_ID}`],
|
||||
tools: {
|
||||
"mnote.current_page.read": "allow",
|
||||
"mnote.selection.read": "allow",
|
||||
"mnote.allowed_roots.describe": "allow",
|
||||
"mnote.local_file.read": "allow",
|
||||
"mnote.local_file.patch": "ask",
|
||||
"mnote.knowledge_rag.status": "allow",
|
||||
"mnote.knowledge_rag.query": "allow",
|
||||
"mnote.knowledge_rag.section_context": "allow",
|
||||
"mnote.knowledge_rag.open_reference": "allow",
|
||||
"mnote.reference.open": "allow",
|
||||
"mnote.tool_receipt.write": "allow",
|
||||
"mnote.codex_rescue.request": "ask",
|
||||
},
|
||||
skills: {},
|
||||
mcpServers: {},
|
||||
};
|
||||
}
|
||||
|
||||
async function seedWorkspace(page) {
|
||||
mkdirp(path.dirname(path.join(ROOT_PATH, PAGE_PATH)));
|
||||
fs.writeFileSync(path.join(ROOT_PATH, PAGE_PATH), "# Pi model controls\n\nMODEL_CONTROLS_OK\n", "utf8");
|
||||
if (USING_DEFAULT_E2E_ROOT) return;
|
||||
await setupWorkspaceAccess(page.request, BASE, {
|
||||
actorId: ACTOR_ID,
|
||||
email: "mnote.e2e@example.com",
|
||||
username: ACTOR_ID,
|
||||
displayName: ACTOR_ID,
|
||||
role: "admin",
|
||||
workspaceId: WORKSPACE_ID,
|
||||
workspaceName: "Pi model controls smoke",
|
||||
rootPath: ROOT_PATH,
|
||||
rootUri: ROOT_URI,
|
||||
permission: "write",
|
||||
capabilities: ["ai", "read", "write"],
|
||||
timeoutMs: TIMEOUT,
|
||||
});
|
||||
await seedAiPolicy(page.request, BASE, {
|
||||
id: `pi-model-controls-policy-${ACTOR_ID}-${WORKSPACE_ID}`,
|
||||
userId: ACTOR_ID,
|
||||
workspaceId: WORKSPACE_ID,
|
||||
allowedRootsJson: [{ rootUri: ROOT_URI, rootPath: ROOT_PATH, permission: "write" }],
|
||||
modelPolicyJson: policy(),
|
||||
quotaJson: { daily: 200 },
|
||||
timeoutMs: TIMEOUT,
|
||||
});
|
||||
}
|
||||
|
||||
async function main() {
|
||||
mkdirp(OUT);
|
||||
const browser = await chromium.launch({
|
||||
headless: process.env.MNOTE_PI_MODEL_CONTROLS_HEADED === "1" ? false : true,
|
||||
executablePath: CHROMIUM_EXECUTABLE || undefined,
|
||||
});
|
||||
const context = await browser.newContext({ viewport: { width: 1440, height: 980 } });
|
||||
await context.addInitScript(() => {
|
||||
window.__MNOTE_PI_LAB_TEST__ = true;
|
||||
});
|
||||
const page = await context.newPage();
|
||||
const result = {
|
||||
base: BASE,
|
||||
outputDir: OUT,
|
||||
rootUri: ROOT_URI,
|
||||
pagePath: PAGE_PATH,
|
||||
screenshots: {},
|
||||
checks: {},
|
||||
};
|
||||
|
||||
try {
|
||||
await quickLogin(page);
|
||||
await seedWorkspace(page);
|
||||
const start = await requestJson(page, "/api/page-ai/pi/start", {
|
||||
method: "POST",
|
||||
data: {
|
||||
sessionId: `pi-model-controls-${STAMP}`,
|
||||
rootUri: ROOT_URI,
|
||||
workspaceId: WORKSPACE_ID,
|
||||
pagePath: PAGE_PATH,
|
||||
pageTitle: "Pi model controls",
|
||||
modelProvider: MODEL_PROVIDER,
|
||||
modelId: MODEL_ID,
|
||||
thinkingLevel: "medium",
|
||||
},
|
||||
});
|
||||
assert.equal(start.ok, true, "start should succeed");
|
||||
|
||||
await page.goto(`${BASE}/page-ai/pi`, { waitUntil: "commit", timeout: TIMEOUT });
|
||||
await page.locator('[data-page-ai-pi-lab="drawer"]').waitFor({ state: "visible", timeout: TIMEOUT });
|
||||
await page.waitForFunction(() => !!window.__mnotePiLabTest, null, { timeout: TIMEOUT });
|
||||
const modelToggle = page.locator("[data-page-ai-pi-lab-model-menu-toggle]");
|
||||
const modelLabel = page.locator("[data-page-ai-pi-lab-model-label]");
|
||||
const thinkingToggle = page.locator("[data-page-ai-pi-lab-thinking-menu-toggle]");
|
||||
const thinkingLabel = page.locator("[data-page-ai-pi-lab-thinking-label]");
|
||||
await modelToggle.waitFor({ state: "visible", timeout: TIMEOUT });
|
||||
await thinkingToggle.waitFor({ state: "visible", timeout: TIMEOUT });
|
||||
await modelLabel.waitFor({ state: "visible", timeout: TIMEOUT });
|
||||
await thinkingLabel.waitFor({ state: "visible", timeout: TIMEOUT });
|
||||
|
||||
result.checks.modelToggleDisabled = await modelToggle.isDisabled();
|
||||
result.checks.thinkingToggleDisabled = await thinkingToggle.isDisabled();
|
||||
assert.equal(result.checks.modelToggleDisabled, false, "model toggle button should be enabled");
|
||||
assert.equal(result.checks.thinkingToggleDisabled, false, "thinking toggle button should be enabled");
|
||||
|
||||
// Open thinking menu via toggle button
|
||||
await thinkingToggle.click();
|
||||
const thinkingMenu = page.locator("[data-page-ai-pi-lab-thinking-menu]");
|
||||
await thinkingMenu.waitFor({ state: "visible", timeout: TIMEOUT });
|
||||
|
||||
// Assert menu position: bounding box above toggle, horizontally adjacent
|
||||
const thinkingMenuBox = await thinkingMenu.boundingBox();
|
||||
const thinkingToggleBox = await thinkingToggle.boundingBox();
|
||||
result.checks.thinkingMenuBox = thinkingMenuBox;
|
||||
result.checks.thinkingToggleBox = thinkingToggleBox;
|
||||
if (thinkingMenuBox && thinkingToggleBox) {
|
||||
result.checks.thinkingMenuAboveToggle = thinkingMenuBox.y + thinkingMenuBox.height <= thinkingToggleBox.y + 1;
|
||||
result.checks.thinkingMenuHorizAdjacent = Math.abs(thinkingMenuBox.x - thinkingToggleBox.x) <= 100;
|
||||
assert.equal(result.checks.thinkingMenuAboveToggle, true, "thinking menu should be above toggle button");
|
||||
assert.equal(result.checks.thinkingMenuHorizAdjacent, true, "thinking menu should be horizontally adjacent to toggle");
|
||||
}
|
||||
|
||||
// Assert mutual exclusion: open thinking menu, model menu should be closed
|
||||
const modelWrap = page.locator("[data-page-ai-pi-lab-model-menu-wrap]");
|
||||
result.checks.thinkingOpenModelClosed = await modelWrap.getAttribute("data-open");
|
||||
assert.equal(result.checks.thinkingOpenModelClosed, "false", "model menu should close when thinking menu opens");
|
||||
|
||||
const configureRequestPromise = page.waitForRequest(
|
||||
(request) => request.url().includes("/api/page-ai/pi/configure") && request.method() === "POST",
|
||||
{ timeout: TIMEOUT },
|
||||
);
|
||||
const offOption = page.locator('[data-page-ai-pi-lab-thinking-option="off"]');
|
||||
await offOption.click();
|
||||
const configureRequest = await configureRequestPromise;
|
||||
result.checks.configureRequest = configureRequest.postDataJSON();
|
||||
assert.equal(result.checks.configureRequest.sessionId, start.session.sessionId, "configure should keep current session");
|
||||
assert.equal(result.checks.configureRequest.modelProvider, MODEL_PROVIDER);
|
||||
assert.equal(result.checks.configureRequest.modelId, MODEL_ID);
|
||||
assert.equal(result.checks.configureRequest.thinkingLevel, "off");
|
||||
|
||||
// Verify label updated
|
||||
await page.waitForFunction(
|
||||
() => document.querySelector("[data-page-ai-pi-lab-thinking-label]")?.textContent?.includes("思考 关"),
|
||||
null,
|
||||
{ timeout: TIMEOUT },
|
||||
);
|
||||
// Open model menu and verify thinking menu closed (mutual exclusion reverse)
|
||||
await modelToggle.click();
|
||||
const modelMenu = page.locator("[data-page-ai-pi-lab-model-menu]");
|
||||
await modelMenu.waitFor({ state: "visible", timeout: TIMEOUT });
|
||||
const thinkingWrap = page.locator("[data-page-ai-pi-lab-thinking-menu-wrap]");
|
||||
result.checks.modelOpenThinkingClosed = await thinkingWrap.getAttribute("data-open");
|
||||
assert.equal(result.checks.modelOpenThinkingClosed, "false", "thinking menu should close when model menu opens");
|
||||
|
||||
// Close menu by clicking outside (click on drawer body outside the controls)
|
||||
const drawer = page.locator('[data-page-ai-pi-lab="drawer"]');
|
||||
const drawerBox = await drawer.boundingBox();
|
||||
if (drawerBox) {
|
||||
const closeX = drawerBox.x + drawerBox.width - 10;
|
||||
const closeY = drawerBox.y + 10;
|
||||
await page.mouse.click(closeX, closeY);
|
||||
await page.waitForTimeout(200);
|
||||
}
|
||||
result.checks.modelMenuClosedOutsideClick = await modelWrap.getAttribute("data-open");
|
||||
assert.equal(result.checks.modelMenuClosedOutsideClick, "false", "model menu should close on outside click");
|
||||
|
||||
// Re-open model menu to verify it still anchors correctly
|
||||
await modelToggle.click();
|
||||
await modelMenu.waitFor({ state: "visible", timeout: TIMEOUT });
|
||||
const modelMenuBox = await modelMenu.boundingBox();
|
||||
const modelToggleBox = await modelToggle.boundingBox();
|
||||
result.checks.modelMenuBox = modelMenuBox;
|
||||
result.checks.modelToggleBox = modelToggleBox;
|
||||
if (modelMenuBox && modelToggleBox) {
|
||||
result.checks.modelMenuAboveToggle = modelMenuBox.y + modelMenuBox.height <= modelToggleBox.y + 1;
|
||||
result.checks.modelMenuHorizAdjacent = Math.abs(modelMenuBox.x - modelToggleBox.x) <= 100;
|
||||
assert.equal(result.checks.modelMenuAboveToggle, true, "model menu should be above toggle button after re-open");
|
||||
assert.equal(result.checks.modelMenuHorizAdjacent, true, "model menu should be horizontally adjacent to toggle after re-open");
|
||||
}
|
||||
|
||||
// + menu should replace model menu.
|
||||
await page.locator("[data-page-ai-pi-lab-action-menu-toggle]").click();
|
||||
result.checks.plusOpenedModelClosed = {
|
||||
actionMenuOpen: await page.locator(".wolai-page-ai-pi-lab-composer").getAttribute("data-menu-open"),
|
||||
modelMenuOpen: await modelWrap.getAttribute("data-open"),
|
||||
};
|
||||
assert.equal(result.checks.plusOpenedModelClosed.actionMenuOpen, "true", "+ menu should open");
|
||||
assert.equal(result.checks.plusOpenedModelClosed.modelMenuOpen, "false", "model menu should close when + menu opens");
|
||||
|
||||
// Access control should replace + menu.
|
||||
await page.locator("[data-page-ai-pi-lab-permission]").click();
|
||||
const permissionWrap = page.locator("[data-page-ai-pi-lab-permission-wrap]");
|
||||
result.checks.permissionOpenedPlusClosed = {
|
||||
actionMenuOpen: await page.locator(".wolai-page-ai-pi-lab-composer").getAttribute("data-menu-open"),
|
||||
permissionMenuOpen: await permissionWrap.getAttribute("data-open"),
|
||||
};
|
||||
assert.equal(result.checks.permissionOpenedPlusClosed.actionMenuOpen, "false", "+ menu should close when access control opens");
|
||||
assert.equal(result.checks.permissionOpenedPlusClosed.permissionMenuOpen, "true", "access control menu should open");
|
||||
|
||||
// Model menu should replace access control and remain anchored.
|
||||
await modelToggle.click();
|
||||
await modelMenu.waitFor({ state: "visible", timeout: TIMEOUT });
|
||||
result.checks.modelOpenedPermissionClosed = {
|
||||
modelMenuOpen: await modelWrap.getAttribute("data-open"),
|
||||
permissionMenuOpen: await permissionWrap.getAttribute("data-open"),
|
||||
};
|
||||
assert.equal(result.checks.modelOpenedPermissionClosed.modelMenuOpen, "true", "model menu should reopen");
|
||||
assert.equal(result.checks.modelOpenedPermissionClosed.permissionMenuOpen, "false", "access control should close when model menu opens");
|
||||
|
||||
// Close model menu via outside click again.
|
||||
if (drawerBox) {
|
||||
const closeX = drawerBox.x + drawerBox.width - 10;
|
||||
const closeY = drawerBox.y + 10;
|
||||
await page.mouse.click(closeX, closeY);
|
||||
await page.waitForTimeout(200);
|
||||
}
|
||||
|
||||
await page.locator("[data-page-ai-pi-lab-status-text]").waitFor({ state: "attached", timeout: TIMEOUT });
|
||||
result.checks.statusText = (await page.locator("[data-page-ai-pi-lab-status-text]").textContent() || "").trim();
|
||||
await page.screenshot({ path: path.join(OUT, "01-model-thinking-enabled-configured.png"), fullPage: false });
|
||||
result.screenshots.modelThinkingConfigured = path.join(OUT, "01-model-thinking-enabled-configured.png");
|
||||
|
||||
fs.writeFileSync(path.join(OUT, "result.json"), JSON.stringify(result, null, 2), "utf8");
|
||||
console.log(`✅ Pi Lab model/thinking controls smoke passed. Output: ${OUT}`);
|
||||
} finally {
|
||||
await browser.close();
|
||||
}
|
||||
}
|
||||
|
||||
main().catch((error) => {
|
||||
console.error(error);
|
||||
process.exit(1);
|
||||
});
|
||||
Reference in New Issue
Block a user