From 9da904b277f847e457a39b28a5ef6f9861ec430d Mon Sep 17 00:00:00 2001
From: Daniil Stankevych <daniil.stankevych@jetbrains.com>
Date: Mon, 15 Dec 2025 15:28:05 +0200
Subject: [PATCH] [jupyter-ui-tests] PY-84438 fix `outputs preserve after
 reopening` test

GitOrigin-RevId: 9f287997f98ec137dc04e794b1eb8b977259b4c9
---
 .../sdk/ui/components/notebooks/NotebookEditorComponent.kt   | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

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 06ba11f43f37..4b101bc25ad3 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
@@ -355,7 +355,7 @@ fun Driver.withNotebookEditor(testBody: NotebookEditorUiComponent.() -> Unit): I
   }
 }
 
-fun Driver.openFileWithProjectPanel(fileName: String): IdeaFrameUI = ideFrame {
+fun Driver.openNotebookWithProjectPanel(fileName: String): IdeaFrameUI = ideFrame {
   leftToolWindowToolbar.projectButton.open()
   projectView {
     projectViewTree.run {
@@ -363,6 +363,7 @@ fun Driver.openFileWithProjectPanel(fileName: String): IdeaFrameUI = ideFrame {
     }
   }
   waitFor("the editor is present", timeout = 30.seconds) {
-    editor().present()
+    notebookEditor().present()
+
   }
 }