Nicolay Mitropolsky
54285d328f
In case of result difference the `-ide`-suffixed testdata is used. Actually the difference in results is more likely to be a bug but currently we are fixating at least some behaviour to protect from unexpected regressions in Ultra Light Classes. `object Local` were removed because there is no local objects in actual Kotlin syntax
37 lines
1.6 KiB
Kotlin
37 lines
1.6 KiB
Kotlin
package org.jetbrains.uast.test.kotlin
|
|
|
|
import com.intellij.psi.PsiElement
|
|
import org.jetbrains.uast.*
|
|
import org.jetbrains.uast.test.common.UElementToParentMap
|
|
import org.jetbrains.uast.test.common.kotlin.IdentifiersTestBase
|
|
import org.jetbrains.uast.test.common.visitUFileAndGetResult
|
|
import org.jetbrains.uast.test.env.assertEqualsToFile
|
|
import java.io.File
|
|
import kotlin.test.assertNotNull
|
|
|
|
|
|
abstract class AbstractKotlinIdentifiersTest : AbstractKotlinUastTest(), IdentifiersTestBase {
|
|
|
|
private fun getTestFile(testName: String, ext: String) =
|
|
File(File(TEST_KOTLIN_MODEL_DIR, testName).canonicalPath + '.' + ext)
|
|
|
|
override fun getIdentifiersFile(testName: String): File = getTestFile(testName, "identifiers.txt")
|
|
|
|
override fun check(testName: String, file: UFile) {
|
|
super.check(testName, file)
|
|
assertEqualsToFile("refNames", getTestFile(testName, "refNames.txt"), file.asRefNames())
|
|
}
|
|
}
|
|
|
|
private fun refNameRetriever(psiElement: PsiElement): UElement? =
|
|
when (val uElement = psiElement.toUElementOfExpectedTypes(UCallExpression::class.java, UReferenceExpression::class.java)) {
|
|
is UReferenceExpression -> uElement.referenceNameElement
|
|
is UCallExpression -> uElement.classReference?.referenceNameElement
|
|
else -> null
|
|
}?.also {
|
|
assertNotNull(it.sourcePsi, "referenceNameElement should have physical source, origin = $psiElement")
|
|
}
|
|
|
|
fun UFile.asRefNames() = object : UElementToParentMap(::refNameRetriever) {
|
|
override fun renderSource(element: PsiElement): String = element.javaClass.simpleName
|
|
}.visitUFileAndGetResult(this) |