xinjian0101-continuity-director/tests/frontend_smoke.mjs

import fs from "node:fs";
import vm from "node:vm";
import path from "node:path";
import { fileURLToPath } from "node:url";

const root = path.resolve(path.dirname(fileURLToPath(import.meta.url)), "..");
let source = fs.readFileSync(path.join(root, "js", "continuity_director.js"), "utf8");
source = source.replace('import { app } from "../../scripts/app.js";', "const app = globalThis.__app;");
source = source.replace('new URL("./continuity_director.css",import.meta.url).href', '"continuity_director.css"');

class FakeClassList {
  constructor() { this.values = new Set(); }
  add(...items) { items.forEach((item) => this.values.add(item)); }
  toggle(item, enabled) { enabled ? this.values.add(item) : this.values.delete(item); }
}

class FakeElement {
  constructor(tag = "div") {
    this.tagName = tag.toUpperCase();
    this.children = [];
    this.dataset = {};
    this.classList = new FakeClassList();
    this.hidden = false;
    this.style = {};
    this.textContent = "";
  }
  append(...items) { this.children.push(...items); }
  appendChild(item) { this.children.push(item); return item; }
  replaceChildren(...items) { this.children = [...items]; }
  querySelectorAll(selector) {
    const output = [];
    const visit = (node) => {
      if (!(node instanceof FakeElement)) return;
      if (selector === "button" && node.tagName === "BUTTON") output.push(node);
      node.children.forEach(visit);
    };
    visit(this);
    return output;
  }
  set className(value) { this._className = value; }
  get className() { return this._className || ""; }
  set innerHTML(value) { this._innerHTML = value; }
  get innerHTML() { return this._innerHTML || ""; }
}

const storage = new Map();
const sidebar = {};
const graphNodes = [];
let nextId = 1;
const graph = {
  _nodes: graphNodes,
  links: {},
  add(node) { node.id = nextId++; graphNodes.push(node); },
};
const makeNode = (type) => ({
  type,
  comfyClass: type,
  title: type,
  outputs: [],
  inputs: [],
  size: [200, 120],
  connect() { return true; },
});
let extension;

globalThis.window = globalThis;
globalThis.localStorage = {
  getItem: (key) => storage.get(key) ?? null,
  setItem: (key, value) => storage.set(key, String(value)),
};
globalThis.document = {
  head: new FakeElement("head"),
  body: new FakeElement("body"),
  querySelector: () => null,
  createElement: (tag) => new FakeElement(tag),
  createTextNode: (text) => ({ textContent: String(text) }),
};
globalThis.LiteGraph = { createNode: makeNode, ROUND_SHAPE: 2 };
globalThis.__app = {
  graph,
  canvas: {
    graph_mouse: [100, 100],
    setDirty() {},
    selectNodes() {},
    fitViewToSelection() {},
    centerOnNode() {},
  },
  extensionManager: {
    registerSidebarTab(config) { sidebar.config = config; },
    toast: { add() {} },
  },
  registerExtension(config) { extension = config; },
};

vm.runInThisContext(source, { filename: "continuity_director.js" });
if (!extension) throw new Error("frontend extension was not registered");
if (extension.name !== "continuity-director.dashboard") throw new Error("unexpected extension name");
await extension.setup();
if (!sidebar.config || sidebar.config.type !== "custom") throw new Error("sidebar tab was not registered");
const host = new FakeElement("aside");
sidebar.config.render(host);
if (host.children.length === 0) throw new Error("sidebar render produced no content");
const starter = extension.commands.find((item) => item.id === "continuity-director.add-starter");
if (!starter) throw new Error("starter command missing");
starter.function();
if (graphNodes.length !== 8) throw new Error(`starter chain created ${graphNodes.length} nodes instead of 8`);
console.log("frontend smoke passed: sidebar rendered and starter chain created");