Make :kotlin-dataframe-compiler-plugin:test cacheable

^KTI-2586

Author: CristianGM

analysis/analysis-test-framework/testFixtures/org/jetbrains/kotlin/analysis/test/framework/base/AbstractAnalysisApiBasedTest.kt

@@ -30,6 +30,7 @@ import org.jetbrains.kotlin.analysis.test.framework.test.configurators.FrontendK
 import org.jetbrains.kotlin.analysis.test.framework.test.configurators.registerAllServices
 import org.jetbrains.kotlin.analysis.test.framework.utils.SkipTestException
 import org.jetbrains.kotlin.analysis.test.framework.utils.singleOrZeroValue
+import org.jetbrains.kotlin.codegen.forTestCompile.ForTestCompileRuntime
 import org.jetbrains.kotlin.psi.KtElement
 import org.jetbrains.kotlin.psi.KtFile
 import org.jetbrains.kotlin.test.TestConfiguration
@@ -55,7 +56,6 @@ import org.junit.jupiter.api.TestInfo
 import org.junit.jupiter.api.extension.ExtendWith
 import java.io.IOException
 import java.nio.file.Path
-import java.nio.file.Paths
 import kotlin.io.path.exists
 import kotlin.io.path.nameWithoutExtension
 
@@ -404,7 +404,7 @@ abstract class AbstractAnalysisApiBasedTest : TestWithDisposable() {
     }
 
     protected fun runTest(@TestDataFile path: String, block: (TestServices) -> Unit) {
-        testDataPath = configurator.computeTestDataPath(Paths.get(path))
+        testDataPath = configurator.computeTestDataPath(ForTestCompileRuntime.transformTestDataPath(path).toPath())
         val testConfiguration = createTestConfiguration()
         testServices = testConfiguration.testServices
         createAndRegisterTestModuleStructure(testConfiguration)

plugins/kotlin-dataframe/build.gradle.kts

@@ -38,17 +38,19 @@ sourceSets {
 }
 
 projectTests {
+    testData(isolated, "testData")
+    withJvmStdlibAndReflect()
+    withScriptRuntime()
+    withTestJar()
+    withMockJdkAnnotationsJar()
+
     testTask(jUnitMode = JUnitMode.JUnit5) {
-        dependsOn(":dist")
-        workingDir = rootDir
         val classpathProvider = objects.newInstance<DataFramePluginClasspathProvider>()
         classpathProvider.classpath.from(dataframeRuntimeClasspath)
         jvmArgumentProviders.add(classpathProvider)
     }
 
     testGenerator("org.jetbrains.kotlin.fir.dataframe.TestGeneratorKt")
-
-    withJvmStdlibAndReflect()
 }
 
 abstract class DataFramePluginClasspathProvider : CommandLineArgumentProvider {

plugins/kotlin-dataframe/testFixtures/org/jetbrains/kotlin/fir/dataframe/AbstractDataFrameBlackBoxCodegenTest.kt

@@ -43,15 +43,16 @@ open class AbstractDataFrameBlackBoxCodegenTest : AbstractFirLightTreeBlackBoxCo
 
     class SelectionDslUtilsSourceProvider(testServices: TestServices) : AdditionalSourceProvider(testServices) {
         companion object {
-            const val SELECTION_DSL_UTILS = "plugins/kotlin-dataframe/testData/selectionDslTestUtils.kt"
+            const val SELECTION_DSL_UTILS = "selectionDslTestUtils.kt"
         }
 
         override fun produceAdditionalFiles(
             globalDirectives: RegisteredDirectives,
             module: TestModule,
             testModuleStructure: TestModuleStructure,
         ): List<TestFile> {
-            return listOf(File(SELECTION_DSL_UTILS).toTestFile())
+            val classLoader = this::class.java.classLoader
+            return listOf(classLoader.getResource(SELECTION_DSL_UTILS)!!.toTestFile())
         }
     }
 }

plugins/kotlin-dataframe/testFixtures/org/jetbrains/kotlin/fir/dataframe/services/TestUtilsSourceProvider.kt

@@ -11,11 +11,10 @@ import org.jetbrains.kotlin.test.model.TestModule
 import org.jetbrains.kotlin.test.services.AdditionalSourceProvider
 import org.jetbrains.kotlin.test.services.TestModuleStructure
 import org.jetbrains.kotlin.test.services.TestServices
-import java.io.File
 
 class TestUtilsSourceProvider(testServices: TestServices) : AdditionalSourceProvider(testServices) {
     companion object {
-        const val COMMON_SOURCE_PATH = "plugins/kotlin-dataframe/testData/testUtils.kt"
+        const val COMMON_SOURCE_PATH = "testUtils.kt"
     }
 
     override fun produceAdditionalFiles(
@@ -24,10 +23,11 @@ class TestUtilsSourceProvider(testServices: TestServices) : AdditionalSourceProv
         testModuleStructure: TestModuleStructure,
     ): List<TestFile> {
         return buildList {
-            add(File(COMMON_SOURCE_PATH).toTestFile())
+            add(this::class.java.classLoader.getResource(COMMON_SOURCE_PATH)!!.toTestFile())
+
             if (DataFrameDirectives.WITH_SCHEMA_READER in module.directives) {
-                add(File("plugins/kotlin-dataframe/testData/schemaReaderDeclaration.kt").toTestFile())
-                add(File("plugins/kotlin-dataframe/testData/dataSchemaSourceDeclaration.kt").toTestFile())
+                add(this::class.java.classLoader.getResource("schemaReaderDeclaration.kt")!!.toTestFile())
+                add(this::class.java.classLoader.getResource("dataSchemaSourceDeclaration.kt")!!.toTestFile())
             }
         }
     }