diff --git a/internal/platform/canvas_ai_model_integration_test.go b/internal/platform/canvas_ai_model_integration_test.go index a93e9b6..0e9ac66 100644 --- a/internal/platform/canvas_ai_model_integration_test.go +++ b/internal/platform/canvas_ai_model_integration_test.go @@ -508,6 +508,14 @@ func TestCanonicalAIAdmissionAndLifecycle(t *testing.T) { model.Tombstones[node] = 2 model.TombstoneOwners[node] = child delete(model.Elements, node) + for id, element := range model.Elements { + if element["structure"] == "connection" && (element["source"].(map[string]any)["elementId"] == node || element["target"].(map[string]any)["elementId"] == node) { + model.DeadRecords[id] = element + model.Tombstones[id] = modelInt(element["incarnation"]) + model.TombstoneOwners[id] = modelString(element["ownerCanvasId"]) + delete(model.Elements, id) + } + } }) execute(doomed["run_id"].(string)) if result := call("GET", base(child)+"/ai/runs/"+doomed["run_id"].(string), nil, 200, 0, ""); result["status"] != "cancelled" {