UAST: utilize types test base
This commit is contained in:
committed by
Ilya Kirillov
parent
06b4fe1613
commit
6afb68cec0
@@ -0,0 +1,44 @@
|
|||||||
|
/*
|
||||||
|
* Copyright 2010-2021 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.uast.test.common.kotlin
|
||||||
|
|
||||||
|
import org.jetbrains.uast.UElement
|
||||||
|
import org.jetbrains.uast.UExpression
|
||||||
|
import org.jetbrains.uast.visitor.UastVisitor
|
||||||
|
|
||||||
|
class TypesLogger : UastVisitor {
|
||||||
|
|
||||||
|
val builder = StringBuilder()
|
||||||
|
|
||||||
|
var level = 0
|
||||||
|
|
||||||
|
override fun visitElement(node: UElement): Boolean {
|
||||||
|
val initialLine = node.asLogString() + " [" + run {
|
||||||
|
val renderString = node.asRenderString().lines()
|
||||||
|
if (renderString.size == 1) {
|
||||||
|
renderString.single()
|
||||||
|
} else {
|
||||||
|
renderString.first() + "..." + renderString.last()
|
||||||
|
}
|
||||||
|
} + "]"
|
||||||
|
|
||||||
|
(1..level).forEach { builder.append(" ") }
|
||||||
|
builder.append(initialLine)
|
||||||
|
if (node is UExpression) {
|
||||||
|
val value = node.getExpressionType()
|
||||||
|
value?.let { builder.append(" : ").append(it) }
|
||||||
|
}
|
||||||
|
builder.appendLine()
|
||||||
|
level++
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
override fun afterVisitElement(node: UElement) {
|
||||||
|
level--
|
||||||
|
}
|
||||||
|
|
||||||
|
override fun toString() = builder.toString()
|
||||||
|
}
|
||||||
+12
@@ -0,0 +1,12 @@
|
|||||||
|
/*
|
||||||
|
* Copyright 2010-2021 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.uast.test.common.kotlin
|
||||||
|
|
||||||
|
import org.jetbrains.uast.UFile
|
||||||
|
|
||||||
|
fun UFile.asLogTypes() = TypesLogger().apply {
|
||||||
|
this@asLogTypes.accept(this)
|
||||||
|
}.toString()
|
||||||
@@ -5,57 +5,16 @@
|
|||||||
|
|
||||||
package org.jetbrains.uast.test.common.kotlin
|
package org.jetbrains.uast.test.common.kotlin
|
||||||
|
|
||||||
import org.jetbrains.uast.UElement
|
|
||||||
import org.jetbrains.uast.UExpression
|
|
||||||
import org.jetbrains.uast.UFile
|
import org.jetbrains.uast.UFile
|
||||||
import org.jetbrains.uast.test.env.kotlin.assertEqualsToFile
|
import org.jetbrains.uast.test.env.kotlin.assertEqualsToFile
|
||||||
import org.jetbrains.uast.visitor.UastVisitor
|
|
||||||
import java.io.File
|
import java.io.File
|
||||||
|
|
||||||
interface TypesTestBase {
|
interface TypesTestBase {
|
||||||
fun getTypesFile(testName: String): File
|
fun getTypesFile(testName: String): File
|
||||||
|
|
||||||
private fun UFile.asLogTypes() = TypesLogger().apply {
|
|
||||||
this@asLogTypes.accept(this)
|
|
||||||
}.toString()
|
|
||||||
|
|
||||||
fun check(testName: String, file: UFile) {
|
fun check(testName: String, file: UFile) {
|
||||||
val valuesFile = getTypesFile(testName)
|
val valuesFile = getTypesFile(testName)
|
||||||
|
|
||||||
assertEqualsToFile("Log values", valuesFile, file.asLogTypes())
|
assertEqualsToFile("Log values", valuesFile, file.asLogTypes())
|
||||||
}
|
}
|
||||||
|
}
|
||||||
class TypesLogger : UastVisitor {
|
|
||||||
|
|
||||||
val builder = StringBuilder()
|
|
||||||
|
|
||||||
var level = 0
|
|
||||||
|
|
||||||
override fun visitElement(node: UElement): Boolean {
|
|
||||||
val initialLine = node.asLogString() + " [" + run {
|
|
||||||
val renderString = node.asRenderString().lines()
|
|
||||||
if (renderString.size == 1) {
|
|
||||||
renderString.single()
|
|
||||||
} else {
|
|
||||||
renderString.first() + "..." + renderString.last()
|
|
||||||
}
|
|
||||||
} + "]"
|
|
||||||
|
|
||||||
(1..level).forEach { builder.append(" ") }
|
|
||||||
builder.append(initialLine)
|
|
||||||
if (node is UExpression) {
|
|
||||||
val value = node.getExpressionType()
|
|
||||||
value?.let { builder.append(" : ").append(it) }
|
|
||||||
}
|
|
||||||
builder.appendLine()
|
|
||||||
level++
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
override fun afterVisitElement(node: UElement) {
|
|
||||||
level--
|
|
||||||
}
|
|
||||||
|
|
||||||
override fun toString() = builder.toString()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|||||||
Reference in New Issue
Block a user