From 6f654aabd4ab1bdaa53677e491ca13e6db69be32 Mon Sep 17 00:00:00 2001
From: Daniil Stankevych <daniil.stankevych@jetbrains.com>
Date: Mon, 1 Sep 2025 13:59:31 +0300
Subject: [PATCH] fix-restart-kernel-jupyter-nightly-test

GitOrigin-RevId: 850201e2fb3d2e632eb95dcc3c50b6666d00ef3b
---
 .../ui/components/notebooks/NotebookEditorComponent.kt | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/platform/remote-driver/test-sdk/src/com/intellij/driver/sdk/ui/components/notebooks/NotebookEditorComponent.kt b/platform/remote-driver/test-sdk/src/com/intellij/driver/sdk/ui/components/notebooks/NotebookEditorComponent.kt
index aae03d06f779..d0f997824e68 100644
--- a/platform/remote-driver/test-sdk/src/com/intellij/driver/sdk/ui/components/notebooks/NotebookEditorComponent.kt
+++ b/platform/remote-driver/test-sdk/src/com/intellij/driver/sdk/ui/components/notebooks/NotebookEditorComponent.kt
@@ -9,6 +9,7 @@ import com.intellij.driver.sdk.invokeActionWithRetries
 import com.intellij.driver.sdk.singleProject
 import com.intellij.driver.sdk.step
 import com.intellij.driver.sdk.ui.Finder
+import com.intellij.driver.sdk.ui.UiText.Companion.asString
 import com.intellij.driver.sdk.ui.components.ComponentData
 import com.intellij.driver.sdk.ui.components.UiComponent
 import com.intellij.driver.sdk.ui.components.common.EditorComponentImpl
@@ -71,6 +72,8 @@ class NotebookEditorUiComponent(private val data: ComponentData) : JEditorUiComp
     get() = x("//div[@myicon='restartKernel.svg']")
   private val interruptKernel
     get() = x("//div[@myicon='stop.svg']")
+  private val deleteCell
+    get() = x("//div[@myicon='delete.svg']")
   val notebookCellOutputs: List<UiComponent>
     get() = xx("//div[@class='FullEditorWidthRenderer']//div[@class='EditorComponentImpl']").list()
   val jcefOffScreens: List<JcefOffScreenViewComponent>
@@ -85,6 +88,8 @@ class NotebookEditorUiComponent(private val data: ComponentData) : JEditorUiComp
     get() = x("//div[@class='JupyterFileEditorToolbar']")
   val imagePanel: List<UiComponent>
     get() = xx("//div[@class='FullEditorWidthRenderer']//div[@class='ImagePanel']").list()
+  val lastNotebookOutput: String
+    get() = notebookCellOutputs.last().getAllTexts().asString()
 
   override val editorComponent: EditorComponentImpl
     get() = when {
@@ -122,6 +127,11 @@ class NotebookEditorUiComponent(private val data: ComponentData) : JEditorUiComp
 
   fun interruptKernel(): Unit = interruptKernel.click()
 
+  fun deleteFirstCell() {
+    notebookCellEditors.first().click()
+    deleteCell.click()
+  }
+
   fun restartHighlighting() {
     driver.withContext {
       invokeActionWithRetries("RestartKotlinNotebookHighlighting")