From 354915d12c7742e9efe90d0facf34aae852d3e25 Mon Sep 17 00:00:00 2001
From: Anton Kozub <anton.kozub@jetbrains.com>
Date: Thu, 16 Oct 2025 17:25:54 +0200
Subject: [PATCH] KTIJ-35470 Kotlin Routing DSL: logical structure view is
 empty for routers registered via programmatic bean registration

(cherry picked from commit 0f9b0d2f8ae0cf240b80ac5264413191d3c1ce32)

IJ-CR-179523

GitOrigin-RevId: c21bec064dbc4fc65115ea81f31f10b0a848426e
---
 .../LogicalStructureTestUtils.kt              | 91 ++++++++++---------
 1 file changed, 48 insertions(+), 43 deletions(-)

diff --git a/java/testFramework/src/com/intellij/testFramework/LogicalStructureTestUtils.kt b/java/testFramework/src/com/intellij/testFramework/LogicalStructureTestUtils.kt
index b20a37e8be92..7d84de8ef4e1 100644
--- a/java/testFramework/src/com/intellij/testFramework/LogicalStructureTestUtils.kt
+++ b/java/testFramework/src/com/intellij/testFramework/LogicalStructureTestUtils.kt
@@ -7,6 +7,7 @@ import com.intellij.ide.structureView.impl.common.PsiTreeElementBase
 import com.intellij.ide.structureView.logical.impl.LogicalStructureViewService
 import com.intellij.ide.structureView.newStructureView.StructureViewComponent
 import com.intellij.ide.util.treeView.PresentableNodeDescriptor
+import com.intellij.openapi.util.Disposer
 import com.intellij.pom.PomTargetPsiElement
 import com.intellij.psi.PsiElement
 import com.intellij.psi.PsiFile
@@ -51,52 +52,56 @@ fun assertLogicalStructure(
   val builder = LogicalStructureViewService.getInstance(project).getLogicalStructureBuilder(psiFile)
   assertNotNull(builder)
   val structureView = builder!!.createStructureView(null, project)
-  var targetStructureElement = structureView.treeModel.root
-  nodePath?.split("/")?.forEach { pathPart ->
-    val child = targetStructureElement.children.firstOrNull {
-      val presentation = it.presentation
-      presentation.presentableText == pathPart || (presentation as? PresentationData)?.coloredText?.firstOrNull()?.text == pathPart
-    } as? StructureViewTreeElement
-    assertNotNull("Can't find a child '$pathPart'", child)
-    targetStructureElement = child!!
-  }
-  var actualRoot = createActualNode(targetStructureElement)
-  if (nodePath != null) {
-    actualRoot = LogicalStructureNode(null, "", "").also {
-      it.subNode(actualRoot)
+  try {
+    var targetStructureElement = structureView.treeModel.root
+    nodePath?.split("/")?.forEach { pathPart ->
+      val child = targetStructureElement.children.firstOrNull {
+        val presentation = it.presentation
+        presentation.presentableText == pathPart || (presentation as? PresentationData)?.coloredText?.firstOrNull()?.text == pathPart
+      } as? StructureViewTreeElement
+      assertNotNull("Can't find a child '$pathPart'", child)
+      targetStructureElement = child!!
     }
-  }
-  val expectedRoot = LogicalStructureNode(null, "", "")
-  expectedRoot.expectedStructureInitializer()
-  if (!expectedRoot.isEqualTo(actualRoot, false)) {
-    expectedRoot.synchronizeImportantElements(actualRoot)
-    throw ComparisonFailure("The models are not equal: ",
-                            expectedRoot.print("", false),
-                            actualRoot.print("", false))
-  }
-  if (selectedElement != null) {
-    val expectedSelectedPath = expectedRoot.getSelectedNodePath().let { it.subList(1, it.size) }
-    val select = (structureView as StructureViewComponent).select(selectedElement, true)
-    val actualPaths = try {
-      val treePath = PlatformTestUtil.waitForPromise(select)
-      treePath!!.path.toList().let { it.subList(1, it.size) }
-    } catch (e: Throwable) {
-      e.printStackTrace()
-      throw e
-    }
-    val nodePaths = nodePath?.split("/") ?: emptyList()
-    for ((index, any) in actualPaths.withIndex()) {
-      if (index < nodePaths.size) {
-        assertTrue("Selected node is different: ", any.toString().startsWith(nodePaths[index]))
-      }
-      else {
-        val expectedNode = expectedSelectedPath.getOrNull(index - nodePaths.size)!!
-        val expectedName = if (expectedNode.coloredTextElements.isNotEmpty()) {
-          expectedNode.coloredTextElements.joinToString("") { it.text }
-        } else expectedNode.name
-        assertEquals("Selected node is different: ", expectedName, any.toString())
+    var actualRoot = createActualNode(targetStructureElement)
+    if (nodePath != null) {
+      actualRoot = LogicalStructureNode(null, "", "").also {
+        it.subNode(actualRoot)
       }
     }
+    val expectedRoot = LogicalStructureNode(null, "", "")
+    expectedRoot.expectedStructureInitializer()
+    if (!expectedRoot.isEqualTo(actualRoot, false)) {
+      expectedRoot.synchronizeImportantElements(actualRoot)
+      throw ComparisonFailure("The models are not equal: ",
+                              expectedRoot.print("", false),
+                              actualRoot.print("", false))
+    }
+    if (selectedElement != null) {
+      val expectedSelectedPath = expectedRoot.getSelectedNodePath().let { it.subList(1, it.size) }
+      val select = (structureView as StructureViewComponent).select(selectedElement, true)
+      val actualPaths = try {
+        val treePath = PlatformTestUtil.waitForPromise(select)
+        treePath!!.path.toList().let { it.subList(1, it.size) }
+      } catch (e: Throwable) {
+        e.printStackTrace()
+        throw e
+      }
+      val nodePaths = nodePath?.split("/") ?: emptyList()
+      for ((index, any) in actualPaths.withIndex()) {
+        if (index < nodePaths.size) {
+          assertTrue("Selected node is different: ", any.toString().startsWith(nodePaths[index]))
+        }
+        else {
+          val expectedNode = expectedSelectedPath.getOrNull(index - nodePaths.size)!!
+          val expectedName = if (expectedNode.coloredTextElements.isNotEmpty()) {
+            expectedNode.coloredTextElements.joinToString("") { it.text }
+          } else expectedNode.name
+          assertEquals("Selected node is different: ", expectedName, any.toString())
+        }
+      }
+    }
+  } finally {
+    Disposer.dispose(structureView)
   }
 }