From c86e157a48757b8d7aa251618cf16c1fd07d5840 Mon Sep 17 00:00:00 2001
From: Nikolay Krasko <nikolay.krasko@jetbrains.com>
Date: Wed, 18 Apr 2018 17:34:23 +0300
Subject: [PATCH] Fix KotlinFileStructureTestGenerated in 181

Use common code for 173 and 181.
---
 .../kotlin/idea/structureView/KotlinFileStructureTestBase.kt   | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/idea/tests/org/jetbrains/kotlin/idea/structureView/KotlinFileStructureTestBase.kt b/idea/tests/org/jetbrains/kotlin/idea/structureView/KotlinFileStructureTestBase.kt
index 423522b63dc..051a7c87e37 100644
--- a/idea/tests/org/jetbrains/kotlin/idea/structureView/KotlinFileStructureTestBase.kt
+++ b/idea/tests/org/jetbrains/kotlin/idea/structureView/KotlinFileStructureTestBase.kt
@@ -22,7 +22,7 @@ import com.intellij.openapi.util.text.StringUtil
 import com.intellij.testFramework.FileStructureTestFixture
 import com.intellij.testFramework.PlatformTestUtil
 import com.intellij.testFramework.UsefulTestCase
-import com.intellij.testFramework.fixtures.LightCodeInsightFixtureTestCase
+import com.intellij.util.ui.tree.TreeUtil
 import org.jetbrains.kotlin.idea.test.KotlinLightCodeInsightFixtureTestCase
 
 abstract class KotlinFileStructureTestBase : KotlinLightCodeInsightFixtureTestCase() {
@@ -76,6 +76,7 @@ abstract class KotlinFileStructureTestBase : KotlinLightCodeInsightFixtureTestCa
 
     protected fun checkResult() {
         val printInfo = Queryable.PrintInfo(arrayOf("text"), arrayOf("location"))
+        TreeUtil.expandAll(popupFixture.tree);
         val popupText = StructureViewUtil.print(popupFixture.tree, false, printInfo, null).trim { it <= ' ' }
         UsefulTestCase.assertSameLinesWithFile(testDataPath + "/" + treeFileName, popupText)
     }