[Tests] Extract AbstractDecompiledKnmFileTest from the stub test
KTIJ-28668
This commit is contained in:
Pavel Kirpichenkov
committed by
Space Team
Space Team
parent
23e558d172
commit
f6a35ef0e5
+100
@@ -0,0 +1,100 @@
|
||||
/*
|
||||
* Copyright 2010-2024 JetBrains s.r.o. and Kotlin Programming Language contributors.
|
||||
* Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
|
||||
*/
|
||||
|
||||
package org.jetbrains.kotlin.analysis.decompiler.konan
|
||||
|
||||
import com.intellij.openapi.vfs.VfsUtilCore
|
||||
import com.intellij.openapi.vfs.VirtualFile
|
||||
import com.intellij.openapi.vfs.impl.jar.CoreJarFileSystem
|
||||
import org.jetbrains.kotlin.analysis.decompiler.stub.files.findMainTestKotlinFile
|
||||
import org.jetbrains.kotlin.cli.jvm.compiler.EnvironmentConfigFiles
|
||||
import org.jetbrains.kotlin.cli.jvm.compiler.KotlinCoreEnvironment
|
||||
import org.jetbrains.kotlin.library.KLIB_METADATA_FILE_EXTENSION_WITH_DOT
|
||||
import org.jetbrains.kotlin.test.*
|
||||
import org.jetbrains.kotlin.test.directives.model.Directive
|
||||
import org.jetbrains.kotlin.test.util.KtTestUtil
|
||||
import org.jetbrains.kotlin.utils.addIfNotNull
|
||||
import java.io.File
|
||||
import java.nio.file.Files
|
||||
import java.nio.file.Path
|
||||
import java.nio.file.Paths
|
||||
import java.util.stream.Collectors
|
||||
import kotlin.io.path.absolutePathString
|
||||
import kotlin.io.path.extension
|
||||
|
||||
abstract class AbstractDecompiledKnmFileTest : KotlinTestWithEnvironment() {
|
||||
abstract val ignoreDirective: Directive
|
||||
|
||||
protected abstract fun doTest(testDirectoryPath: Path)
|
||||
|
||||
override fun createEnvironment(): KotlinCoreEnvironment {
|
||||
return KotlinCoreEnvironment.createForTests(
|
||||
ApplicationEnvironmentDisposer.ROOT_DISPOSABLE,
|
||||
KotlinTestUtils.newConfiguration(
|
||||
ConfigurationKind.JDK_NO_RUNTIME,
|
||||
TestJdkKind.MOCK_JDK,
|
||||
),
|
||||
EnvironmentConfigFiles.METADATA_CONFIG_FILES,
|
||||
)
|
||||
}
|
||||
|
||||
fun runTest(testDirectory: String) {
|
||||
val testDirectoryPath = Paths.get(testDirectory)
|
||||
withKnmIgnoreDirective(testDirectoryPath) { doTest(testDirectoryPath) }
|
||||
}
|
||||
|
||||
protected fun compileToKnmFiles(testDirectoryPath: Path): List<VirtualFile> {
|
||||
val commonKlib = compileCommonKlib(testDirectoryPath)
|
||||
return getKnmFilesFromKlib(commonKlib)
|
||||
}
|
||||
|
||||
private fun compileCommonKlib(testDirectory: Path): File {
|
||||
val ktFiles = Files.list(testDirectory).filter { it.extension == "kt" }.collect(Collectors.toList())
|
||||
val testKlib = KtTestUtil.tmpDir("testLibrary").resolve("library.klib")
|
||||
KlibTestUtil.compileCommonSourcesToKlib(
|
||||
ktFiles.map(Path::toFile),
|
||||
libraryName = "library",
|
||||
testKlib,
|
||||
)
|
||||
|
||||
return testKlib
|
||||
}
|
||||
|
||||
private fun getKnmFilesFromKlib(klib: File): List<VirtualFile> {
|
||||
val path = klib.toPath()
|
||||
val jarFileSystem = environment.projectEnvironment.environment.jarFileSystem as CoreJarFileSystem
|
||||
val root = jarFileSystem.refreshAndFindFileByPath(path.absolutePathString() + "!/")!!
|
||||
val files = mutableListOf<VirtualFile>()
|
||||
VfsUtilCore.iterateChildrenRecursively(
|
||||
root,
|
||||
{ virtualFile -> virtualFile.isDirectory || virtualFile.name.endsWith(KLIB_METADATA_FILE_EXTENSION_WITH_DOT) },
|
||||
{ virtualFile ->
|
||||
if (!virtualFile.isDirectory) {
|
||||
files.addIfNotNull(virtualFile)
|
||||
}
|
||||
true
|
||||
})
|
||||
|
||||
return files
|
||||
}
|
||||
|
||||
private fun isTestIgnored(testDirectory: Path): Boolean {
|
||||
val mainKotlinFile = findMainTestKotlinFile(testDirectory).toFile()
|
||||
return InTextDirectivesUtils.isDirectiveDefined(mainKotlinFile.readText(), "// ${ignoreDirective.name}")
|
||||
}
|
||||
|
||||
private fun withKnmIgnoreDirective(testDirectory: Path, block: () -> Unit) {
|
||||
val isIgnored = isTestIgnored(testDirectory)
|
||||
|
||||
try {
|
||||
block()
|
||||
} catch (e: Throwable) {
|
||||
if (isIgnored) return
|
||||
else throw e
|
||||
}
|
||||
|
||||
if (isIgnored) error("The test is passing. Please, remove the `// ${ignoreDirective.name}` directive")
|
||||
}
|
||||
}
|
||||
+3
-70
@@ -64,22 +64,9 @@ abstract class AbstractDecompiledKnmStubConsistencyK2Test : AbstractDecompiledKn
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
abstract class AbstractDecompiledKnmStubConsistencyTest : KotlinTestWithEnvironment() {
|
||||
abstract val ignoreDirective: Directive
|
||||
abstract class AbstractDecompiledKnmStubConsistencyTest : AbstractDecompiledKnmFileTest() {
|
||||
abstract fun createDecompiler(): KlibMetadataDecompiler<*>
|
||||
|
||||
override fun createEnvironment(): KotlinCoreEnvironment {
|
||||
return KotlinCoreEnvironment.createForTests(
|
||||
ApplicationEnvironmentDisposer.ROOT_DISPOSABLE,
|
||||
KotlinTestUtils.newConfiguration(
|
||||
ConfigurationKind.JDK_NO_RUNTIME,
|
||||
TestJdkKind.MOCK_JDK,
|
||||
),
|
||||
EnvironmentConfigFiles.METADATA_CONFIG_FILES,
|
||||
)
|
||||
}
|
||||
|
||||
override fun setUp() {
|
||||
super.setUp()
|
||||
|
||||
@@ -88,68 +75,14 @@ abstract class AbstractDecompiledKnmStubConsistencyTest : KotlinTestWithEnvironm
|
||||
)
|
||||
}
|
||||
|
||||
fun runTest(testDirectory: String) {
|
||||
val testDirectoryPath = Paths.get(testDirectory)
|
||||
withKnmIgnoreDirective(testDirectoryPath) { doTest(testDirectoryPath) }
|
||||
}
|
||||
|
||||
private fun isTestIgnored(testDirectory: Path): Boolean {
|
||||
val mainKotlinFile = findMainTestKotlinFile(testDirectory).toFile()
|
||||
return InTextDirectivesUtils.isDirectiveDefined(mainKotlinFile.readText(), "// ${ignoreDirective.name}")
|
||||
}
|
||||
|
||||
private fun withKnmIgnoreDirective(testDirectory: Path, block: () -> Unit) {
|
||||
val isIgnored = isTestIgnored(testDirectory)
|
||||
|
||||
try {
|
||||
block()
|
||||
} catch (e: Throwable) {
|
||||
if (isIgnored) return
|
||||
else throw e
|
||||
}
|
||||
|
||||
if (isIgnored) error("The test is passing. Please, remove the `// ${ignoreDirective.name}` directive")
|
||||
}
|
||||
|
||||
private fun doTest(testDirectoryPath: Path) {
|
||||
val commonKlib = compileCommonKlib(testDirectoryPath)
|
||||
val files = getKnmFilesFromKlib(commonKlib)
|
||||
override fun doTest(testDirectoryPath: Path) {
|
||||
val files = compileToKnmFiles(testDirectoryPath)
|
||||
|
||||
for (knmFile in files) {
|
||||
checkKnmStubConsistency(knmFile)
|
||||
}
|
||||
}
|
||||
|
||||
private fun compileCommonKlib(testDirectory: Path): File {
|
||||
val ktFiles = Files.list(testDirectory).filter { it.extension == "kt" }.collect(Collectors.toList())
|
||||
val testKlib = KtTestUtil.tmpDir("testLibrary").resolve("library.klib")
|
||||
KlibTestUtil.compileCommonSourcesToKlib(
|
||||
ktFiles.map(Path::toFile),
|
||||
libraryName = "library",
|
||||
testKlib,
|
||||
)
|
||||
|
||||
return testKlib
|
||||
}
|
||||
|
||||
private fun getKnmFilesFromKlib(klib: File): List<VirtualFile> {
|
||||
val path = klib.toPath()
|
||||
val jarFileSystem = environment.projectEnvironment.environment.jarFileSystem as CoreJarFileSystem
|
||||
val root = jarFileSystem.refreshAndFindFileByPath(path.absolutePathString() + "!/")!!
|
||||
val files = mutableListOf<VirtualFile>()
|
||||
VfsUtilCore.iterateChildrenRecursively(
|
||||
root,
|
||||
{ virtualFile -> virtualFile.isDirectory || virtualFile.name.endsWith(KLIB_METADATA_FILE_EXTENSION_WITH_DOT) },
|
||||
{ virtualFile ->
|
||||
if (!virtualFile.isDirectory) {
|
||||
files.addIfNotNull(virtualFile)
|
||||
}
|
||||
true
|
||||
})
|
||||
|
||||
return files
|
||||
}
|
||||
|
||||
private fun checkKnmStubConsistency(knmFile: VirtualFile) {
|
||||
val stubTreeBinaryFile = createDecompiler().stubBuilder.buildFileStub(FileContentImpl.createByFile(knmFile, environment.project))!!
|
||||
|
||||
|
||||
Reference in New Issue
Block a user