From b5ea94c4a7df98a1f9bc341a53e84d0860cb78d1 Mon Sep 17 00:00:00 2001
From: Ilya Kirillov <ilya.kirillov@jetbrains.com>
Date: Tue, 28 Oct 2025 11:24:36 +0100
Subject: [PATCH] [next edit] LLM-20838 do not try to remove the code that was
 just written by user

^LLM-20838 fixed

GitOrigin-RevId: d124d3aeff833671a159d3bb17e64ee62917d6b5
---
 .../InlineCompletionLifecycleTestDSL.kt       | 30 ++++++++++++-------
 1 file changed, 20 insertions(+), 10 deletions(-)

diff --git a/platform/testFramework/src/com/intellij/codeInsight/inline/completion/InlineCompletionLifecycleTestDSL.kt b/platform/testFramework/src/com/intellij/codeInsight/inline/completion/InlineCompletionLifecycleTestDSL.kt
index 440cb227b9a5..f9859894ec41 100644
--- a/platform/testFramework/src/com/intellij/codeInsight/inline/completion/InlineCompletionLifecycleTestDSL.kt
+++ b/platform/testFramework/src/com/intellij/codeInsight/inline/completion/InlineCompletionLifecycleTestDSL.kt
@@ -321,24 +321,34 @@ class InlineCompletionLifecycleTestDSL(val fixture: CodeInsightTestFixture) {
   inner class Caret {
 
     @ICUtil
-    suspend fun moveUp() {
-      callAction("EditorUp")
+    suspend fun moveUp(times: Int = 1) {
+      repeat(times) {
+        callAction("EditorUp")
+      }
+    }
+
+
+    @ICUtil
+    suspend fun moveDown(times: Int = 1) {
+      repeat(times) {
+        callAction("EditorDown")
+      }
     }
 
     @ICUtil
-    suspend fun moveDown() {
-      callAction("EditorDown")
+    suspend fun moveRight(times: Int = 1) {
+      repeat(times) {
+        callAction("EditorRight")
+      }
     }
 
     @ICUtil
-    suspend fun moveRight() {
-      callAction("EditorRight")
+    suspend fun moveLeft(times: Int = 1) {
+      repeat(times) {
+        callAction("EditorLeft")
+      }
     }
 
-    @ICUtil
-    suspend fun moveLeft() {
-      callAction("EditorLeft")
-    }
   }
 
   @ICUtil