Prepare DiffusionGemma Director 0.2.0

tests/test_model_loader_ui.mjs

@@ -0,0 +1,89 @@
+import assert from "node:assert/strict";
+
+let registeredExtension = null;
+globalThis.window = {
+  comfyAPI: {
+    app: {
+      app: {
+        registerExtension(extension) {
+          registeredExtension = extension;
+        },
+      },
+    },
+  },
+};
+
+await import(`../web/js/model_loader.js?test=${Date.now()}`);
+assert.ok(registeredExtension, "the model-loader frontend extension should register");
+
+class UnrelatedNode {}
+await registeredExtension.beforeRegisterNodeDef(UnrelatedNode, { name: "OtherNode" });
+assert.equal(
+  UnrelatedNode.prototype.onNodeCreated,
+  undefined,
+  "unrelated node types should not be patched",
+);
+
+let createdCalls = 0;
+let configuredCalls = 0;
+
+class FakeLoaderNode {}
+FakeLoaderNode.prototype.onNodeCreated = function () {
+  createdCalls += 1;
+};
+FakeLoaderNode.prototype.onConfigure = function () {
+  configuredCalls += 1;
+};
+
+await registeredExtension.beforeRegisterNodeDef(FakeLoaderNode, {
+  name: "DiffusionGemmaModelLoader",
+});
+
+const hiddenWidgetNames = new Set([
+  "backend",
+  "dtype",
+  "quantization",
+  "local_files_only",
+  "unload_policy",
+  "max_memory_gb",
+]);
+const node = new FakeLoaderNode();
+node.widgets = [
+  { name: "model_path", value: "models/diffusiongemma" },
+  { name: "temperature", value: "invalid" },
+  ...[...hiddenWidgetNames].map((name) => ({ name, value: "advanced" })),
+];
+node.size = [520, 640];
+node.computeSize = () => [360, 240];
+node.setSize = (size) => {
+  node.size = [...size];
+};
+
+node.onNodeCreated();
+assert.equal(createdCalls, 1, "the original creation hook should still run");
+assert.equal(
+  node.widgets.find((entry) => entry.name === "temperature").value,
+  0.45,
+  "invalid saved temperatures should migrate to the stable default",
+);
+assert.deepEqual(node.size, [520, 240], "hiding widgets should preserve the user's width");
+
+for (const entry of node.widgets) {
+  if (hiddenWidgetNames.has(entry.name)) {
+    assert.equal(entry.hidden, true, `${entry.name} should be hidden`);
+    assert.equal(entry.__dgHidden, true, `${entry.name} should only be hidden once`);
+    assert.deepEqual(entry.computeSize(), [0, -4]);
+  } else {
+    assert.notEqual(entry.hidden, true, `${entry.name} should remain visible`);
+  }
+}
+
+node.widgets.find((entry) => entry.name === "temperature").value = 0.7;
+node.size = [610, 700];
+node.computeSize = () => [380, 260];
+node.onConfigure({});
+assert.equal(configuredCalls, 1, "the original configuration hook should still run");
+assert.equal(node.widgets.find((entry) => entry.name === "temperature").value, 0.7);
+assert.deepEqual(node.size, [610, 260], "configuration should keep the saved width");
+
+console.log("DiffusionGemma model-loader UI checks passed");