[Tests] Implement integration test runner for IJ

#JBRes-6600 Done

Author: ivandev0

buildSrc/src/main/kotlin/TraceAgentIntegrationTestsTasks.kt

@@ -49,6 +49,11 @@ private val projectsToTest = listOf(
         repositoryName = "kotlinx-datetime",
         commitHash = "deb897d7c6df11734d0468527cc7ebfc41679a94"
     ),
+    GithubProjectSnapshot(
+        organization = "ivandev0",
+        repositoryName = "intellij-community",
+        commitHash = "8f65df6d733acefffc32bbeb86538ffd45ca260a"
+    ),
 )
 
 lateinit var traceAgentIntegrationTestsPrerequisites: TaskProvider<Task>

integration-test/common/src/main/AbstractGradleTraceIntegrationTest.kt

@@ -65,9 +65,6 @@ abstract class AbstractGradleTraceIntegrationTest: AbstractTraceIntegrationTest(
         extraJvmArgs: List<String>,
         extraAgentArgs: Map<String, String>,
     ): String {
-        fun String.escapeDollar() = replace("$", "\\$")
-
-        val pathToFatJar = File(Paths.get("build", "libs", fatJarName).toString()).absolutePath.escape()
         // We need to escape it twice, as our argument parser will de-escape it when split into array
         val pathToOutput = fileToDump.absolutePath.escape().escape()
         val agentArgs =
@@ -94,6 +91,4 @@ abstract class AbstractGradleTraceIntegrationTest: AbstractTraceIntegrationTest(
             }
         """.trimIndent()
     }
-
-    private fun String.escape(): String = this.replace("\\", "\\\\")
 }

integration-test/common/src/main/AbstractIntellijTraceIntegrationTest.kt

@@ -0,0 +1,43 @@
+/*
+ * Lincheck
+ *
+ * Copyright (C) 2019 - 2025 JetBrains s.r.o.
+ *
+ * This Source Code Form is subject to the terms of the
+ * Mozilla Public License, v. 2.0. If a copy of the MPL was not distributed
+ * with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
+ */
+
+import java.io.File
+
+abstract class AbstractIntellijTraceIntegrationTest: AbstractTraceIntegrationTest() {
+    override val fatJarName: String = "trace-recorder-fat.jar"
+
+    override fun runTestImpl(
+        testClassName: String,
+        testMethodName: String,
+        extraJvmArgs: List<String>,
+        extraAgentArgs: Map<String, String>,
+        commands: List<String>,
+        outputFile: File
+    ) {
+        if (!File("$projectPath/android").exists()) {
+            ProcessBuilder("$projectPath/getPlugins.sh")
+                .directory(File(projectPath))
+                .inheritIO().start().waitFor()
+        }
+
+        // We need to escape it twice, as our argument parser will de-escape it when split into array
+        val pathToOutput = outputFile.absolutePath.escape().escape()
+        ProcessBuilder(
+            "./tests.cmd",
+            "-Dintellij.build.test.patterns=${testClassName.escapeDollar()}",
+            "-Dintellij.build.test.trace.recorder.enabled=true",
+            "-Dintellij.build.test.trace.recorder.className=${testClassName.escapeDollar()}",
+            "-Dintellij.build.test.trace.recorder.methodName=${testMethodName.escapeDollar()}",
+            "-Dintellij.build.test.trace.recorder.traceDump=$pathToOutput",
+            "-Dintellij.build.test.trace.recorder.agentJar=$pathToFatJar"
+        ).directory(File(projectPath))
+            .inheritIO().start().waitFor()
+    }
+}

integration-test/common/src/main/AbstractTraceIntegrationTest.kt

@@ -21,6 +21,9 @@ abstract class AbstractTraceIntegrationTest {
     abstract val fatJarName: String
     abstract val projectPath: String
 
+    protected val pathToFatJar: String
+        get() = File(Paths.get("build", "libs", fatJarName).toString()).absolutePath.escape()
+
     private fun getGoldenDataFileFor(
         testClassName: String,
         testMethodName: String,

integration-test/common/src/main/Utils.kt

@@ -0,0 +1,12 @@
+/*
+ * Lincheck
+ *
+ * Copyright (C) 2019 - 2025 JetBrains s.r.o.
+ *
+ * This Source Code Form is subject to the terms of the
+ * Mozilla Public License, v. 2.0. If a copy of the MPL was not distributed
+ * with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
+ */
+
+fun String.escape(): String = this.replace("\\", "\\\\")
+fun String.escapeDollar() = replace("$", "\\$")

integration-test/trace-recorder/src/main/org/jetbrains/trace/recorder/test/impl/IntellijTraceRecorderIntegrationTest.kt

@@ -0,0 +1,63 @@
+/*
+ * Lincheck
+ *
+ * Copyright (C) 2019 - 2025 JetBrains s.r.o.
+ *
+ * This Source Code Form is subject to the terms of the
+ * Mozilla Public License, v. 2.0. If a copy of the MPL was not distributed
+ * with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
+ */
+
+package org.jetbrains.trace.recorder.test.impl
+
+import AbstractIntellijTraceIntegrationTest
+import AbstractTraceIntegrationTest
+import org.jetbrains.trace.recorder.test.runner.AbstractTraceRecorderIntegrationTest
+import org.jetbrains.trace.recorder.test.runner.ExtendedTraceRecorderTest
+import org.jetbrains.trace.recorder.test.runner.loadResourceText
+import org.jetbrains.trace.recorder.test.runner.parseJsonEntries
+import org.jetbrains.trace.recorder.test.runner.transformEntriesToArray
+import org.junit.Test
+import org.junit.experimental.categories.Category
+import org.junit.experimental.runners.Enclosed
+import org.junit.runner.RunWith
+import org.junit.runners.Parameterized
+import java.nio.file.Paths
+
+@RunWith(Enclosed::class)
+class IntellijTraceRecorderIntegrationTest {
+    @Category(ExtendedTraceRecorderTest::class)
+    @RunWith(Parameterized::class)
+    class Parametrized(
+        private val testClassName: String,
+        private val testMethodName: String,
+        @Suppress("unused") private val gradleCommand: String,
+        private val perEntryJvmArgs: List<String>,
+        private val perEntryCheckRepresentation: Boolean,
+    ) : AbstractIntellijTraceIntegrationTest() {
+        override val projectPath: String = Paths.get("build", "integrationTestProjects", "intellij-community").toAbsolutePath().toString()
+
+        @Test(timeout = 30 * 60 * 1000L)
+        fun runIntellijTest() = runTest(
+            testClassName = testClassName,
+            testMethodName = testMethodName,
+            extraJvmArgs = perEntryJvmArgs,
+            extraAgentArgs = emptyMap(),
+            commands = emptyList(),
+            checkRepresentation = perEntryCheckRepresentation
+        )
+
+        companion object {
+            @JvmStatic
+            @Parameterized.Parameters(name = "{index}: {0}::{1}")
+            fun data(): Collection<Array<Any>> {
+                val json = loadResourceText(
+                    "/integrationTestData/intellijTests.json",
+                    IntellijTraceRecorderIntegrationTest::class.java
+                )
+                val entries = parseJsonEntries(json)
+                return entries.transformEntriesToArray()
+            }
+        }
+    }
+}

integration-test/trace-recorder/src/main/resources/integrationTestData/intellij-community/com.intellij.util.ArrayUtilTest_testInsertString.txt

@@ -0,0 +1,2 @@
+# Thread 1
+ArrayUtilTest.testInsertString() at :0

integration-test/trace-recorder/src/main/resources/integrationTestData/intellijTests.json

@@ -0,0 +1,10 @@
+[
+  {
+    "class": "com.intellij.util.ArrayUtilTest",
+    "gradleCommand": ":kotlinx-collections-immutable:jvmTest",
+    "methods": [
+      "testInsertString"
+    ],
+    "checkRepresentation": true
+  }
+]