IDEA-332876 UI tests: Add base for HeaderStateTest

GitOrigin-RevId: 32e34d1667332ff76f0f317fea616fa2181b8eef
JetBrains · Nov 24, 2023 · 815b4bd · 815b4bd
1 parent c8a9bb3
commit 815b4bd
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/intellij.tools.ide.starter/src/com/intellij/ide/starter/utils/utils.kt b/intellij.tools.ide.starter/src/com/intellij/ide/starter/utils/utils.kt
@@ -47,6 +47,10 @@ catch (t: Throwable) {
 fun String.withIndent(indent: String = "  "): String = lineSequence().map { "$indent$it" }.joinToString(System.lineSeparator())
 
 fun takeScreenshot(logsDir: Path) {
+  takeScreenshot(logsDir, "screenshot_beforeKill.jpg")
+}
+
+fun takeScreenshot(logsDir: Path, screenshotName: String) {
   val toolsDir = GlobalPaths.instance.getCacheDirectoryFor("tools")
   val toolName = "TakeScreenshot"
   val screenshotTool = toolsDir / toolName / "$toolName.jar"
@@ -55,7 +59,7 @@ fun takeScreenshot(logsDir: Path) {
     val screenshotJar = File(IDERunContext::class.java.classLoader.getResource("tools/$toolName.jar")!!.toURI())
     screenshotJar.copyTo(screenshotTool.toFile(), true)
   }
-  val screenshotFile = logsDir.resolve("screenshot_beforeKill.jpg")
+  val screenshotFile = logsDir.resolve(screenshotName)
 
   val javaPath = ProcessHandle.current().info().command().orElseThrow().toString()
   ProcessExecutor(