test: cover features since 053a9c0d + repair test tooling

Add ~330 tests across server (Jest), client (Vitest), editor-ext (Vitest)
and packages/mcp (node:test) for the gitmost features added since
053a9c0d: AI chat, AI agent roles, public-share assistant, MCP per-user
auth, HTML embed, page templates/embed, realtime tree, tree
expand/collapse, and the AI-settings UI.

Test-tooling fixes (prerequisite, were silently hiding coverage):
- Repair 3 page-template specs broken by the 11-arg TransclusionService
  constructor; they never compiled, so template access-control / content
  -leak / unsync-strip coverage was fictitious.
- Build @docmost/editor-ext before server tests via a `pretest` hook;
  the stale dist omitted the new HtmlEmbed/PageEmbed exports (TS2305).
- Let jest resolve the .tsx email templates: add `tsx` to
  moduleFileExtensions and widen the ts-jest transform to (t|j)sx?.

Behaviour-preserving "extract pure core" refactors that the tests drive:
- server: resolveShareAssistantRequest + uiMessageTextLength
  (public-share controller), decideBasicGate + mapAuthResultToResponse
  (mcp), buildErrorAssistantRecord (ai-chat), jsonbObject export (roles).
- client: render-raw-html + shouldExecute/canEdit, decide-embed-state,
  page-embed picker utils, tree-socket reducers, open/close branch maps,
  isEndpointConfigured/resolveKeyField; buildTreeWithChildren now treats
  a permission-trimmed orphan as a root instead of crashing.

Deferred (need a test DB or HTTP harness, documented in the specs):
repo-level Postgres integration tests and the public-share XFF E2E.
Pre-existing DI/lib0-ESM suite failures are untouched and out of scope.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
claude_code
2026-06-20 23:40:40 +03:00
parent 692c0abe13
commit 90d3fab483
56 changed files with 5668 additions and 447 deletions

View File

@@ -0,0 +1,116 @@
import { afterEach, describe, expect, it } from "vitest";
import {
encodeHtmlEmbedSource,
decodeHtmlEmbedSource,
} from "./html-embed";
// Unit coverage for the base64 codec used by the htmlEmbed node's
// data-source attribute (html-embed.ts). The codec has two branches:
// - the BROWSER branch: btoa(encodeURIComponent(s)) / decodeURIComponent(atob(s));
// - the NODE fallback: Buffer.from(..).toString("base64") / Buffer.from(s,"base64").
// Server-side schema parsing (htmlToJson with no global btoa/atob) hits the
// fallback, so both branches must round-trip identically; otherwise an embed
// encoded in the browser would decode wrong on the server (or vice versa).
//
// We force the fallback by temporarily DELETING globalThis.btoa/atob (jsdom
// provides them in this env), restoring them after each test so the suite stays
// hermetic.
const realBtoa = globalThis.btoa;
const realAtob = globalThis.atob;
function deleteBase64Globals(): void {
// @ts-expect-error — intentionally removing the globals to exercise the
// `typeof btoa !== "function"` Node fallback branch in the codec.
delete globalThis.btoa;
// @ts-expect-error — see above.
delete globalThis.atob;
}
afterEach(() => {
// Always restore so one test's stubbing never leaks into another.
globalThis.btoa = realBtoa;
globalThis.atob = realAtob;
});
describe("html-embed codec — browser btoa/atob branch", () => {
it("round-trips ASCII source", () => {
const src = "<script>alert(1)</script>";
const enc = encodeHtmlEmbedSource(src);
expect(enc).not.toBe("");
// base64 of the encodeURIComponent form never contains a raw '<'.
expect(enc).not.toContain("<");
expect(decodeHtmlEmbedSource(enc)).toBe(src);
});
it("round-trips UTF-8 / non-Latin1 source (the reason for encodeURIComponent)", () => {
const src = '<p>héllo → 世界 𝕏</p>';
const enc = encodeHtmlEmbedSource(src);
expect(decodeHtmlEmbedSource(enc)).toBe(src);
});
});
describe("html-embed codec — Node Buffer fallback branch", () => {
it("encode uses the Buffer fallback when btoa is unavailable and still round-trips (UTF-8)", () => {
const src = '<div>héllo → 世界 𝕏</div>';
deleteBase64Globals();
// With the globals gone, encode must take the Buffer path...
const encFallback = encodeHtmlEmbedSource(src);
expect(encFallback).not.toBe("");
// ...and decode (also via Buffer) must recover the exact source.
expect(decodeHtmlEmbedSource(encFallback)).toBe(src);
});
it("the Buffer fallback produces the SAME bytes the browser branch does (cross-env parity)", () => {
const src = '<span>café — 日本語</span>';
// Browser branch (globals intact).
const encBrowser = encodeHtmlEmbedSource(src);
// Fallback branch.
deleteBase64Globals();
const encFallback = encodeHtmlEmbedSource(src);
// Identical base64 => an embed encoded in either environment decodes
// identically in the other (server <-> client losslessness).
expect(encFallback).toBe(encBrowser);
// And the fallback can decode what the browser produced.
expect(decodeHtmlEmbedSource(encBrowser)).toBe(src);
});
it("empty string -> '' on both encode and decode in the fallback (early return, branch never reached)", () => {
deleteBase64Globals();
expect(encodeHtmlEmbedSource("")).toBe("");
expect(decodeHtmlEmbedSource("")).toBe("");
});
it("decode of malformed base64 -> '' via the catch branch (fallback)", () => {
// In the Buffer fallback, Buffer.from(..,'base64') is lenient and never
// throws, so to hit the catch we need a payload whose DECODED bytes are an
// invalid percent-escape, which makes decodeURIComponent throw. base64 of a
// lone '%' decodes back to '%', and decodeURIComponent('%') is a URIError.
const badBase64 = Buffer.from("%", "utf-8").toString("base64"); // "JQ=="
deleteBase64Globals();
// Sanity: the raw decode really does throw, so we're exercising the catch.
expect(() =>
decodeURIComponent(Buffer.from(badBase64, "base64").toString("utf-8")),
).toThrow();
// The codec swallows it and returns "" rather than propagating.
expect(decodeHtmlEmbedSource(badBase64)).toBe("");
});
});
describe("html-embed codec — decode of malformed input (browser branch)", () => {
it("returns '' for input atob rejects (catch branch)", () => {
// atob throws on characters outside the base64 alphabet; the codec catches
// it and returns "" instead of throwing.
expect(decodeHtmlEmbedSource("@@not-base64@@")).toBe("");
});
it("empty string short-circuits to '' (never calls atob)", () => {
expect(decodeHtmlEmbedSource("")).toBe("");
});
});

View File

@@ -0,0 +1,105 @@
import { describe, expect, it } from "vitest";
import { htmlEmbedExtension } from "./utils/html-embed.marked";
import { markdownToHtml } from "./index";
import { encodeHtmlEmbedSource } from "../html-embed/html-embed";
// CONTRACT tests for the marked block tokenizer that rebuilds an htmlEmbed node
// from the `<!--html-embed:BASE64-->` marker (html-embed.marked.ts), plus the
// observable round-trip through markdownToHtml.
//
// These pin the REAL tokenizer behaviour the import path depends on:
// - the tokenizer rule is anchored (^) and only accepts the base64 alphabet
// [A-Za-z0-9+/=], so a marker with non-base64 chars is NOT tokenized and
// survives as a literal HTML comment (not silently turned into something the
// server's strip no longer recognizes);
// - start() reports the correct index of the next marker so marked invokes the
// tokenizer at the right offset when a marker sits mid-document / after text;
// - a marker with surrounding text on the SAME line is split out into its own
// embed div while the surrounding text becomes ordinary paragraphs.
//
// The contract is asserted against the actual exported extension and pipeline —
// no behaviour is invented; the expectations were read off the real tokenizer.
const SAMPLE = "<b>x</b>";
const ENC = encodeHtmlEmbedSource(SAMPLE);
describe("htmlEmbed marked tokenizer — start()", () => {
it("returns the index of a marker that sits mid-document", () => {
const src = `hello world <!--html-embed:${ENC}-->`;
expect(htmlEmbedExtension.start(src)).toBe(src.indexOf("<!--html-embed:"));
});
it("returns 0 when the marker is at the very start", () => {
expect(htmlEmbedExtension.start(`<!--html-embed:${ENC}-->`)).toBe(0);
});
it("returns -1 when there is no marker", () => {
expect(htmlEmbedExtension.start("no marker here")).toBe(-1);
});
});
describe("htmlEmbed marked tokenizer — tokenizer()", () => {
it("tokenizes a marker at the start of the input, capturing the base64 payload", () => {
const token = htmlEmbedExtension.tokenizer(`<!--html-embed:${ENC}-->`);
expect(token).toBeTruthy();
expect(token!.type).toBe("htmlEmbed");
expect(token!.raw).toBe(`<!--html-embed:${ENC}-->`);
expect(token!.encoded).toBe(ENC);
});
it("tokenizes an EMPTY marker (the [A-Za-z0-9+/=]* class allows zero chars)", () => {
const token = htmlEmbedExtension.tokenizer("<!--html-embed:-->");
expect(token).toBeTruthy();
expect(token!.encoded).toBe("");
expect(token!.raw).toBe("<!--html-embed:-->");
});
it("does NOT tokenize when text precedes the marker (rule is anchored ^)", () => {
// marked relies on start() to advance to the marker; the tokenizer itself
// only matches at offset 0, so a non-anchored call returns undefined.
expect(
htmlEmbedExtension.tokenizer(`hello <!--html-embed:${ENC}-->`),
).toBeUndefined();
});
it("does NOT tokenize a marker containing a non-base64 char ('$')", () => {
expect(
htmlEmbedExtension.tokenizer("<!--html-embed:ab$cd-->"),
).toBeUndefined();
});
it("does NOT tokenize a marker containing a space", () => {
expect(
htmlEmbedExtension.tokenizer("<!--html-embed:ab cd-->"),
).toBeUndefined();
});
it("renderer emits the embed div the node's parseHTML recognizes", () => {
const token = htmlEmbedExtension.tokenizer(`<!--html-embed:${ENC}-->`)!;
const html = htmlEmbedExtension.renderer(token as any);
expect(html).toBe(
`<div data-type="htmlEmbed" data-source="${ENC}"></div>`,
);
});
});
describe("htmlEmbed marked tokenizer — markdownToHtml round-trip", () => {
it("splits a marker out of surrounding same-line text into its own embed div", async () => {
const html = await markdownToHtml(`before <!--html-embed:${ENC}--> after`);
// The marker became the embed div...
expect(html).toContain(
`<div data-type="htmlEmbed" data-source="${ENC}"></div>`,
);
// ...and the surrounding text survived as ordinary paragraph content.
expect(html).toContain("before");
expect(html).toContain("after");
});
it("leaves a marker with non-base64 chars as a literal comment (NOT an embed div)", async () => {
const html = await markdownToHtml("<!--html-embed:ab$cd-->");
// It is NOT tokenized into an embed div the server would strip...
expect(html).not.toContain('data-type="htmlEmbed"');
// ...it passes through unchanged as a literal HTML comment.
expect(html).toContain("<!--html-embed:ab$cd-->");
});
});

View File

@@ -0,0 +1,88 @@
import { describe, expect, it } from "vitest";
import { getSchema } from "@tiptap/core";
import { generateHTML, generateJSON } from "@tiptap/html";
import { Document } from "@tiptap/extension-document";
import { Paragraph } from "@tiptap/extension-paragraph";
import { Text } from "@tiptap/extension-text";
import { PageEmbed } from "./page-embed";
// CONTRACT tests for the PageEmbed node's parse/render round-trip
// (page-embed.ts). The whole-page live embed stores ONLY a `sourcePageId`
// reference; renderHTML must serialize it as `data-source-page-id` and parseHTML
// must recover it. If this attribute mapping drifts, an embed saved to HTML loses
// its target page on reload (the node view would have nothing to fetch).
//
// We assert at the editor-ext schema level using the same Tiptap utilities the
// other editor-ext tests use (getSchema + @tiptap/html generateHTML/generateJSON
// over a jsdom DOM), driving a real HTML -> node JSON -> HTML round-trip through
// the node's actual addAttributes()/parseHTML()/renderHTML().
// Minimal schema: a doc of blocks, plus the PageEmbed block node under test.
const extensions = [Document, Paragraph, Text, PageEmbed];
describe("PageEmbed schema", () => {
it("registers the pageEmbed node in the schema", () => {
const schema = getSchema(extensions);
expect(schema.nodes.pageEmbed).toBeTruthy();
});
});
describe("PageEmbed parse/render round-trip", () => {
it("recovers sourcePageId from data-source-page-id on parse (HTML -> JSON)", () => {
const html = `<div data-type="pageEmbed" data-source-page-id="pg-123"></div>`;
const json = generateJSON(html, extensions);
const node = json.content?.[0];
expect(node?.type).toBe("pageEmbed");
expect(node?.attrs?.sourcePageId).toBe("pg-123");
});
it("emits data-source-page-id on render (JSON -> HTML)", () => {
const json = {
type: "doc",
content: [{ type: "pageEmbed", attrs: { sourcePageId: "pg-456" } }],
};
const html = generateHTML(json, extensions);
expect(html).toContain('data-type="pageEmbed"');
expect(html).toContain('data-source-page-id="pg-456"');
});
it("survives a full HTML -> node -> HTML round-trip (attribute preserved)", () => {
const start = `<div data-type="pageEmbed" data-source-page-id="pg-789"></div>`;
// HTML -> node JSON -> HTML.
const json = generateJSON(start, extensions);
const html = generateHTML(json, extensions);
// The id survived the round-trip in the serialized HTML...
expect(html).toContain('data-source-page-id="pg-789"');
// ...and re-parsing the round-tripped HTML yields the same id (stable across
// an extra pass — no loss, no duplication).
const json2 = generateJSON(html, extensions);
expect(json2.content?.[0]?.attrs?.sourcePageId).toBe("pg-789");
});
it("omits data-source-page-id entirely when sourcePageId is null (renderHTML guard)", () => {
// The renderHTML maps a null/empty id to {} (no attribute), so an embed
// without a target page does not emit a stray empty attribute.
const json = {
type: "doc",
content: [{ type: "pageEmbed", attrs: { sourcePageId: null } }],
};
const html = generateHTML(json, extensions);
expect(html).toContain('data-type="pageEmbed"');
expect(html).not.toContain("data-source-page-id");
});
it("parses a div without the attribute to a null sourcePageId (default)", () => {
const html = `<div data-type="pageEmbed"></div>`;
const json = generateJSON(html, extensions);
expect(json.content?.[0]?.type).toBe("pageEmbed");
// getAttribute returns null when absent; parseHTML returns it verbatim.
expect(json.content?.[0]?.attrs?.sourcePageId).toBeNull();
});
});

View File

@@ -0,0 +1,273 @@
// Unit tests for createMcpHttpHandler's idle-session eviction (http.ts).
//
// http.ts keeps one transport per MCP session alive between requests, keyed by
// the mcp-session-id header, and runs a periodic sweep (setInterval, every 5
// min) that closes any transport idle longer than the idle TTL
// (MCP_SESSION_IDLE_MS, default 30 min) and drops its lastSeen + sessionIdentity
// bookkeeping. Routing a request to an existing transport refreshes its
// lastSeen.
//
// We drive this DETERMINISTICALLY rather than waiting wall-clock: the env knob
// MCP_SESSION_IDLE_MS is read ONCE when the handler is created, so we set it
// small; and node:test's mock.timers lets us mock both `setInterval` (the sweep)
// and `Date` (the lastSeen comparison clock) so ticking advances the clock and
// fires the sweep on demand.
//
// IMPORTANT mock.timers semantics: when a tick spans MULTIPLE timer fires (or
// overshoots a fire), the callbacks all observe Date.now() == the FINAL ticked
// time, not their individual scheduled times. So to make the sweep's
// `now - lastSeen` comparison meaningful we tick EXACTLY to a sweep boundary
// (a multiple of the sweep interval): then Date.now() inside the sweep equals
// that boundary. The mocked clock starts at 0, so sweeps fire at SWEEP, 2*SWEEP,
// ... We pin each session's lastSeen by establishing/touching it at a known
// pre-boundary clock, then tick the remaining delta to land exactly on the
// boundary.
//
// Sessions are established over a real loopback http server (so the SDK's
// StreamableHTTPServerTransport gets genuine Node req/res and a real
// mcp-session-id), exactly like http-resolver.test.mjs, and the server is closed
// in a finally.
//
// Eviction is asserted via its OBSERVABLE effect: once a session is evicted its
// transport is gone from the handler's internal map, so a subsequent non-init
// request replaying that session id is treated as unknown (400 "no valid
// session ID") — the same response an id that was never established would get.
// An active (recently-seen) session is retained and its subsequent request is
// NOT a 400.
import { test, mock } from "node:test";
import assert from "node:assert/strict";
const INIT_BODY = {
jsonrpc: "2.0",
id: 1,
method: "initialize",
params: {
protocolVersion: "2025-03-26",
capabilities: {},
clientInfo: { name: "test", version: "0.0.0" },
},
};
const SWEEP_MS = 5 * 60 * 1000; // setInterval cadence in http.ts.
// Spin a loopback http server bridging every request into the MCP handler with
// its JSON body parsed, mirroring the embedding host. Returns { call, close }.
async function startLoopback(handler) {
const http = await import("node:http");
const server = http.createServer((req, res) => {
let raw = "";
req.on("data", (c) => (raw += c));
req.on("end", () => {
const body = raw ? JSON.parse(raw) : undefined;
handler.handleRequest(req, res, body).catch(() => {
if (!res.headersSent) {
res.statusCode = 500;
res.end();
}
});
});
});
await new Promise((r) => server.listen(0, "127.0.0.1", r));
const { port } = server.address();
const call = (headers, body) =>
new Promise((resolve) => {
const r = http.request(
{
host: "127.0.0.1",
port,
method: "POST",
path: "/mcp",
headers: {
"Content-Type": "application/json",
Accept: "application/json, text/event-stream",
...headers,
},
},
(resp) => {
let data = "";
resp.on("data", (c) => (data += c));
resp.on("end", () =>
resolve({
statusCode: resp.statusCode,
sessionId: resp.headers["mcp-session-id"],
body: data,
}),
);
},
);
r.end(JSON.stringify(body));
});
return { call, close: () => new Promise((r) => server.close(r)) };
}
// The sweep closes transports asynchronously (void transport.close()), whose
// onclose then removes the entry from the internal map. Yield to the event loop
// so those microtasks settle before we assert the observable effect.
const settle = () => new Promise((r) => setImmediate(r));
// Set the idle TTL env knob (read once at handler creation) and enable mocked
// setInterval + Date BEFORE creating the handler, so the sweep interval and
// every Date.now() (lastSeen at init, lastSeen on routing, and the sweep's
// comparison) all run on the same mocked clock. Returns restore() to undo it.
function withMockedTimers(idleMs) {
const prevIdle = process.env.MCP_SESSION_IDLE_MS;
process.env.MCP_SESSION_IDLE_MS = String(idleMs);
mock.timers.enable({ apis: ["setInterval", "Date"] });
return () => {
mock.timers.reset();
if (prevIdle === undefined) delete process.env.MCP_SESSION_IDLE_MS;
else process.env.MCP_SESSION_IDLE_MS = prevIdle;
};
}
test("idle session is evicted by the sweep; an active session is retained", async () => {
// A small TTL: idle longer than 1s triggers eviction. Both sessions start at
// clock 0; we keep one fresh (touch it just before the sweep) and leave the
// other idle, then fire ONE sweep exactly on its boundary.
const idleMs = 1000;
const restore = withMockedTimers(idleMs);
const { createMcpHttpHandler } = await import("../../build/http.js");
const handler = createMcpHttpHandler(() => ({
apiUrl: "http://127.0.0.1:3000/api",
getToken: async () => "t",
}));
const lb = await startLoopback(handler);
try {
// T0 (clock 0): establish both sessions; lastSeen(A) = lastSeen(B) = 0.
const a = await lb.call({}, INIT_BODY);
const b = await lb.call({}, INIT_BODY);
assert.ok(a.sessionId, "session A must get an mcp-session-id");
assert.ok(b.sessionId, "session B must get an mcp-session-id");
assert.notEqual(a.sessionId, b.sessionId, "distinct sessions");
// Advance to just before the first sweep boundary (SWEEP - 1ms): no sweep
// fires yet (boundary not reached). lastSeen(A) stays 0.
mock.timers.tick(SWEEP_MS - 1);
// Touch ONLY B here, refreshing lastSeen(B) to SWEEP-1 (active); A is left
// idle since clock 0.
const touchB = await lb.call(
{ "mcp-session-id": b.sessionId },
{ jsonrpc: "2.0", method: "ping", id: 5 },
);
assert.notEqual(touchB.statusCode, 400, "B alive right before the sweep");
// Land EXACTLY on the sweep boundary (clock = SWEEP). Inside the sweep
// Date.now() == SWEEP, so:
// idle(A) = SWEEP - 0 = SWEEP > TTL(1s) -> A EVICTED
// idle(B) = SWEEP - (SWEEP-1) = 1ms < TTL(1s) -> B RETAINED
mock.timers.tick(1);
await settle();
// OBSERVABLE EFFECT 1 — A evicted: replaying its session id on a non-init
// request is now treated as unknown (400, no valid session).
const aAfter = await lb.call(
{ "mcp-session-id": a.sessionId },
{ jsonrpc: "2.0", method: "ping", id: 10 },
);
assert.equal(aAfter.statusCode, 400, "evicted session id is unknown -> 400");
assert.match(aAfter.body, /no valid session ID/);
// OBSERVABLE EFFECT 2 — B retained: a subsequent request on its session id
// is routed to the live transport, NOT rejected as an unknown session.
const bAfter = await lb.call(
{ "mcp-session-id": b.sessionId },
{ jsonrpc: "2.0", method: "ping", id: 11 },
);
assert.notEqual(
bAfter.statusCode,
400,
"active session must survive the sweep (not 400)",
);
} finally {
await lb.close();
restore();
}
});
test("a session left idle past the TTL is dropped so its id becomes unknown", async () => {
// Simplest single-session eviction: establish a session, let it go idle past
// the TTL, fire the sweep on its boundary, and confirm its id is now unknown
// (400). Pins the core "lastSeen older than TTL -> closed and dropped" path.
const idleMs = 1000;
const restore = withMockedTimers(idleMs);
const { createMcpHttpHandler } = await import("../../build/http.js");
const handler = createMcpHttpHandler(() => ({
apiUrl: "http://127.0.0.1:3000/api",
getToken: async () => "t",
}));
const lb = await startLoopback(handler);
try {
const s = await lb.call({}, INIT_BODY);
assert.ok(s.sessionId, "session must get an mcp-session-id");
// Fire the first sweep exactly on its boundary: Date.now() == SWEEP, idle =
// SWEEP - 0 = SWEEP > TTL, so the untouched session is evicted.
mock.timers.tick(SWEEP_MS);
await settle();
const after = await lb.call(
{ "mcp-session-id": s.sessionId },
{ jsonrpc: "2.0", method: "ping", id: 30 },
);
assert.equal(after.statusCode, 400, "idle session id is unknown -> 400");
assert.match(after.body, /no valid session ID/);
} finally {
await lb.close();
restore();
}
});
test("activity refreshes lastSeen so a busy session is never evicted", async () => {
// A session kept busy (a request just before the sweep) refreshes its
// lastSeen, so even though it was created long ago the sweep must not evict
// it. Pins the "routing to an existing transport refreshes its idle
// timestamp" branch of http.ts.
const idleMs = 1000;
const restore = withMockedTimers(idleMs);
const { createMcpHttpHandler } = await import("../../build/http.js");
const handler = createMcpHttpHandler(() => ({
apiUrl: "http://127.0.0.1:3000/api",
getToken: async () => "t",
}));
const lb = await startLoopback(handler);
try {
const s = await lb.call({}, INIT_BODY);
assert.ok(s.sessionId, "session must get an mcp-session-id");
// Age to just before the sweep boundary, then touch the session so its
// lastSeen is refreshed to SWEEP-1 (well within the TTL of the imminent
// sweep).
mock.timers.tick(SWEEP_MS - 1);
const touch = await lb.call(
{ "mcp-session-id": s.sessionId },
{ jsonrpc: "2.0", method: "ping", id: 40 },
);
assert.notEqual(touch.statusCode, 400, "session still alive before sweep");
// Land exactly on the sweep boundary: idle = SWEEP - (SWEEP-1) = 1ms < TTL,
// so the busy session is retained.
mock.timers.tick(1);
await settle();
const after = await lb.call(
{ "mcp-session-id": s.sessionId },
{ jsonrpc: "2.0", method: "ping", id: 41 },
);
assert.notEqual(
after.statusCode,
400,
"a session touched just before the sweep must not be evicted",
);
} finally {
await lb.close();
restore();
}
});