From c4373deeea245d0a8f206fc8d7d41de03aa005e9 Mon Sep 17 00:00:00 2001
From: Chenlei Hu <huchenlei@proton.me>
Date: Fri, 4 Apr 2025 14:57:19 -0400
Subject: [PATCH] [Test] Add test on moving output links (#887)

Ref: https://github.com/Comfy-Org/litegraph.js/issues/881

Verifies that moving output links does not cause link corruption.
---
 test/LinkConnector.integration.test.ts | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/test/LinkConnector.integration.test.ts b/test/LinkConnector.integration.test.ts
index c885e7117..c12d0c651 100644
--- a/test/LinkConnector.integration.test.ts
+++ b/test/LinkConnector.integration.test.ts
@@ -111,11 +111,13 @@ const test = baseTest.extend<TestContext>({
         for (const input of node.inputs) {
           if (input.link) {
             expect(graph.links.keys()).toContain(input.link)
+            expect(graph.links.get(input.link)?.target_id).toBe(node.id)
           }
         }
         for (const output of node.outputs) {
           for (const linkId of output.links ?? []) {
             expect(graph.links.keys()).toContain(linkId)
+            expect(graph.links.get(linkId)?.origin_id).toBe(node.id)
           }
         }
       }
@@ -330,6 +332,7 @@ describe("LinkConnector Integration", () => {
       expect(connector.outputLinks.length).toBe(0)
 
       expect(disconnectedNode.outputs[0].links).toEqual(nextLinkIds)
+      expect(hasOutputNode.outputs[0].links).toEqual([])
 
       const reroutesAfter = disconnectedNode.outputs[0].links
         ?.map(linkId => graph.links.get(linkId)!)