azalea/kotlin-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

AbstractUastTest: findUElementByTextFromPsi refactored for readability

Browse Source
This commit is contained in:
Nicolay Mitropolsky
2017-10-23 16:39:24 +03:00
committed by xiexed
parent 8e17bab56f
commit 8a34c48d56
1 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
plugins/uast-kotlin/tests/org/jetbrains/uast/test/env/AbstractUastTest.kt
Vendored
+12 -5
View File
@@ -67,8 +67,15 @@ inline fun <reified T : Any> UElement.findElementByText(refText: String): T = fi
inline fun <reified T : UElement> UElement.findElementByTextFromPsi(refText: String): T = (this.psi ?: fail("no psi for $this")).findUElementByTextFromPsi(refText)
inline fun <reified T : UElement> PsiElement.findUElementByTextFromPsi(refText: String): T =
(this.findElementAt(this.text.indexOf(refText)) ?: throw AssertionError("requested text '$refText' was not found in $this"))
.parentsWithSelf.dropWhile { !it.text.contains(refText) }.mapNotNull { it.toUElementOfType<T>() }.first().also {
if (it.psi != null && it.psi?.text != refText) throw AssertionError("requested text '$refText' found as '${it.psi?.text}' in $it")
}
inline fun <reified T : UElement> PsiElement.findUElementByTextFromPsi(refText: String): T {
val elementAtStart = this.findElementAt(this.text.indexOf(refText))
?: throw AssertionError("requested text '$refText' was not found in $this")
val uElementContainingText = elementAtStart.parentsWithSelf.
dropWhile { !it.text.contains(refText) }.
mapNotNull { it.toUElementOfType<T>() }.
first()
if (uElementContainingText.psi != null && uElementContainingText.psi?.text != refText) {
throw AssertionError("requested text '$refText' found as '${uElementContainingText.psi?.text}' in $uElementContainingText")
}
return uElementContainingText;
}