UAST: utilize types test base

This commit is contained in:
Jinseong Jeon
2021-05-24 10:43:07 -07:00
committed by Ilya Kirillov
parent 06b4fe1613
commit 6afb68cec0
3 changed files with 57 additions and 42 deletions
@@ -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()
}
@@ -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
import org.jetbrains.uast.UElement
import org.jetbrains.uast.UExpression
import org.jetbrains.uast.UFile
import org.jetbrains.uast.test.env.kotlin.assertEqualsToFile
import org.jetbrains.uast.visitor.UastVisitor
import java.io.File
interface TypesTestBase {
fun getTypesFile(testName: String): File
private fun UFile.asLogTypes() = TypesLogger().apply {
this@asLogTypes.accept(this)
}.toString()
fun check(testName: String, file: UFile) {
val valuesFile = getTypesFile(testName)
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()
}
}
}