From aedae7f405c3aa9b56341ae19833a4d6c276dcb1 Mon Sep 17 00:00:00 2001
From: "vladislav.shishov" <vladislav.shishov@jetbrains.com>
Date: Fri, 31 Oct 2025 14:01:06 +0200
Subject: [PATCH] [remote-driver] test: flaky
 FloatingCodeToolbarUiTest#testAppearanceOnSelection

GitOrigin-RevId: 25574c5be8de57733b9176e38be86675c5c160cc
---
 .../src/com/intellij/driver/sdk/NotificationUtils.kt         | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/platform/remote-driver/test-sdk/src/com/intellij/driver/sdk/NotificationUtils.kt b/platform/remote-driver/test-sdk/src/com/intellij/driver/sdk/NotificationUtils.kt
index 8508bcc99f37..8baf205f6a99 100644
--- a/platform/remote-driver/test-sdk/src/com/intellij/driver/sdk/NotificationUtils.kt
+++ b/platform/remote-driver/test-sdk/src/com/intellij/driver/sdk/NotificationUtils.kt
@@ -7,6 +7,11 @@ fun Driver.disableAllBalloonNotifications() {
   utility(NotificationUtils::class).disableAllBalloonNotifications()
 }
 
+fun Driver.closeAndDisableAllBalloonNotifications() {
+  disableAllBalloonNotifications()
+  invokeAction("CloseAllNotifications", now = false)
+}
+
 @Remote("com.jetbrains.performancePlugin.utils.NotificationUtils")
 interface NotificationUtils {