diff --git a/platform/remote-driver/test-sdk/src/com/intellij/driver/sdk/ui/components/MainToolbarUI.kt b/platform/remote-driver/test-sdk/src/com/intellij/driver/sdk/ui/components/MainToolbarUI.kt
index 12937dfeb424..694c61b018d1 100644
--- a/platform/remote-driver/test-sdk/src/com/intellij/driver/sdk/ui/components/MainToolbarUI.kt
+++ b/platform/remote-driver/test-sdk/src/com/intellij/driver/sdk/ui/components/MainToolbarUI.kt
@@ -32,4 +32,11 @@ class MainToolbarUI(data: ComponentData) : UiComponent(data) {
 
   fun projectWidget(projectName: String): UiComponent =
     x("//div[@class='ToolbarComboButton' and @visible_text='$projectName']")
-}
\ No newline at end of file
+}
+
+val MainToolbarUI.rerunButton get() = x { contains(byAccessibleName("Rerun")) }
+val MainToolbarUI.resumeButton get() = x { contains(byAccessibleName("Resume")) }
+val MainToolbarUI.pauseButton get() = x { contains(byAccessibleName("Pause")) }
+val MainToolbarUI.restartDebugButton get() = x { contains(byAccessibleName("Restart Debug")) }
+val MainToolbarUI.stopButton get() = x { contains(byAccessibleName("Stop")) }
+val MainToolbarUI.stopButtonWith2Configs get() = x { and(contains(byAccessibleName("Stop")), byVisibleText("2")) }
\ No newline at end of file