String (boolean withCollapseStatus) { Editor topEditor = getHostEditor(); return EdtTestUtil.runInEdtAndGet(() -> { IdeaTestExecutionPolicy policy = IdeaTestExecutionPolicy.current(); if (policy != null) { policy.waitForHighlighting(getProject(), topEditor); } EditorTestUtil.buildInitialFoldingsInBackground(topEditor); return getFoldingData(topEditor, withCollapseStatus); }); }