Raw FIR builder: add performance test to build FIR for all Kotlin

#KT-24086 Fixed
This commit is contained in:
Mikhail Glukhikh
2018-03-16 14:26:16 +03:00
parent 72be9ef738
commit 5716bda15d
2 changed files with 68 additions and 9 deletions
@@ -17,6 +17,7 @@ import org.jetbrains.kotlin.KtNodeTypes
import org.jetbrains.kotlin.cli.common.script.CliScriptDefinitionProvider
import org.jetbrains.kotlin.fir.FirRenderer
import org.jetbrains.kotlin.fir.FirSessionBase
import org.jetbrains.kotlin.fir.declarations.FirFile
import org.jetbrains.kotlin.parsing.KotlinParserDefinition
import org.jetbrains.kotlin.psi.KtFile
import org.jetbrains.kotlin.psi.KtPsiFactory
@@ -50,20 +51,29 @@ abstract class AbstractRawFirBuilderTestCase : KtParsingTestCase(
}
protected fun doRawFirTest(filePath: String) {
doBaseTest(filePath, KtNodeTypes.KT_FILE)
doRawFirTest(filePath, checkNeeded = true)
}
private fun doBaseTest(filePath: String, fileType: IElementType) {
protected fun doRawFirTest(filePath: String, checkNeeded: Boolean) {
val file = createKtFile(filePath)
val firFile = file.toFirFile()
if (checkNeeded) {
val firFileDump = StringBuilder().also { FirRenderer(it).visitFile(firFile) }.toString()
val expectedPath = filePath.replace(".kt", ".txt")
KotlinTestUtils.assertEqualsToFile(File(expectedPath), firFileDump)
}
}
protected fun createKtFile(filePath: String): KtFile {
myFileExt = FileUtilRt.getExtension(PathUtil.getFileName(filePath))
val file = createFile(filePath, fileType) as KtFile
myFile = file
val firFile = RawFirBuilder(object : FirSessionBase() {}).buildFirFile(file)
val firFileDump = StringBuilder().also { FirRenderer(it).visitFile(firFile) }.toString()
val expectedPath = filePath.replace(".kt", ".txt")
KotlinTestUtils.assertEqualsToFile(File(expectedPath), firFileDump)
return (createFile(filePath, KtNodeTypes.KT_FILE) as KtFile).apply {
myFile = this
}
}
protected fun KtFile.toFirFile(): FirFile =
RawFirBuilder(object : FirSessionBase() {}).buildFirFile(this)
override fun tearDown() {
super.tearDown()
FileTypeRegistry.ourInstanceGetter = Getter<FileTypeRegistry> { FileTypeManager.getInstance() }