Added @TestOnly

This commit is contained in:
Valentin Kipyatkov
2016-09-05 17:45:24 +03:00
parent 82089da0cf
commit 29831f648d
@@ -28,6 +28,7 @@ import com.intellij.psi.search.LocalSearchScope
import com.intellij.psi.search.SearchScope import com.intellij.psi.search.SearchScope
import com.intellij.psi.search.searches.ReferencesSearch import com.intellij.psi.search.searches.ReferencesSearch
import com.intellij.util.Processor import com.intellij.util.Processor
import org.jetbrains.annotations.TestOnly
import org.jetbrains.kotlin.KtNodeTypes import org.jetbrains.kotlin.KtNodeTypes
import org.jetbrains.kotlin.asJava.classes.KtLightClass import org.jetbrains.kotlin.asJava.classes.KtLightClass
import org.jetbrains.kotlin.asJava.elements.KtLightMethod import org.jetbrains.kotlin.asJava.elements.KtLightMethod
@@ -69,6 +70,7 @@ class ExpressionsOfTypeProcessor(
private val suspiciousExpressionHandler: (KtExpression) -> Unit, private val suspiciousExpressionHandler: (KtExpression) -> Unit,
private val suspiciousScopeHandler: (SearchScope) -> Unit private val suspiciousScopeHandler: (SearchScope) -> Unit
) { ) {
@TestOnly
enum class Mode { enum class Mode {
ALWAYS_SMART, ALWAYS_SMART,
ALWAYS_PLAIN, ALWAYS_PLAIN,
@@ -76,8 +78,9 @@ class ExpressionsOfTypeProcessor(
} }
companion object { companion object {
// for tests @TestOnly
var mode = if (ApplicationManager.getApplication().isUnitTestMode) Mode.ALWAYS_SMART else Mode.PLAIN_WHEN_NEEDED var mode = if (ApplicationManager.getApplication().isUnitTestMode) Mode.ALWAYS_SMART else Mode.PLAIN_WHEN_NEEDED
@TestOnly
var testLog: MutableList<String>? = null var testLog: MutableList<String>? = null
fun logPresentation(element: PsiElement): String? { fun logPresentation(element: PsiElement): String? {