From a74f71c3c53c6eb24695611fda50928a71c1bbfc Mon Sep 17 00:00:00 2001
From: Sergey Malenkov <sergey.malenkov@jetbrains.com>
Date: Thu, 21 Dec 2017 19:59:15 +0300
Subject: [PATCH] fix async tree test

---
 python/testSrc/com/jetbrains/python/PyStructureViewTest.java | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/python/testSrc/com/jetbrains/python/PyStructureViewTest.java b/python/testSrc/com/jetbrains/python/PyStructureViewTest.java
index 351902759de3..059c014d5377 100644
--- a/python/testSrc/com/jetbrains/python/PyStructureViewTest.java
+++ b/python/testSrc/com/jetbrains/python/PyStructureViewTest.java
@@ -3,6 +3,7 @@ package com.jetbrains.python;
 
 import com.intellij.navigation.ItemPresentation;
 import com.intellij.openapi.command.WriteCommandAction;
+import com.intellij.testFramework.PlatformTestUtil;
 import com.jetbrains.python.fixtures.PyTestCase;
 import com.jetbrains.python.psi.PyFunction;
 import com.jetbrains.python.structureView.PyStructureViewElement;
@@ -134,6 +135,7 @@ public class PyStructureViewTest extends PyTestCase {
   private void doTest(final String expected, final boolean inherited) {
     myFixture.testStructureView(component -> {
       component.setActionActive("SHOW_INHERITED", !inherited);
+      PlatformTestUtil.waitWhileBusy(component.getTree());
       assertTreeEqual(component.getTree(), expected);
     });
   }