[KTNB] Wait for HL in kandy test

GitOrigin-RevId: 80d0cb58650d91c3e22ed4664a0cc96536b02374
This commit is contained in:
Nikolay Egorov
2025-06-13 06:49:37 +03:00
committed by intellij-monorepo-bot
parent 659ebae588
commit d384c5d8a3

View File

@@ -15,6 +15,7 @@ import com.intellij.driver.sdk.ui.components.elements.NotebookTableOutputUi
import com.intellij.driver.sdk.ui.pasteText
import com.intellij.driver.sdk.ui.ui
import com.intellij.driver.sdk.waitFor
import com.intellij.driver.sdk.waitForCodeAnalysis
import org.intellij.lang.annotations.Language
import kotlin.time.Duration
import kotlin.time.Duration.Companion.milliseconds
@@ -33,6 +34,10 @@ fun Finder.notebookEditor(action: NotebookEditorUiComponent.() -> Unit) {
return notebookEditor().action()
}
fun NotebookEditorUiComponent.waitForHighlighting() {
driver.waitForCodeAnalysis(file = editor.getVirtualFile())
}
typealias CellSelector = (List<UiComponent>) -> UiComponent
val FirstCell: CellSelector = { it.first() }