包含在junit的test方法中的print语句将不再提示错误

Author: wuhao000

src/com/wuhao/code/check/CommonConstants.kt

@@ -8,6 +8,11 @@ import com.intellij.psi.PsiElement
 import com.intellij.psi.impl.PsiElementFactoryImpl
 import com.intellij.psi.impl.PsiManagerEx
 
+
+/**
+ * junit的Test注解的类名称
+ */
+const val JUNIT_TEST_ANNOTATION_CLASS_NAME = "org.junit.Test"
 /**
  * 默认的vue项目模板git地址
  */

src/com/wuhao/code/check/Ext.kt

@@ -17,6 +17,21 @@ inline fun <reified T> PsiElement.ancestorOfType(): T? {
   return el as T?
 }
 
+/**
+ * 获取指定类型的所有的祖先元素
+ */
+inline fun <reified T> PsiElement.ancestorsOfType(): ArrayList<T> {
+  val result = arrayListOf<T>()
+  var el: PsiElement? = this.parent
+  while (el != null) {
+    if (el is T) {
+      result.add(el)
+    }
+    el = el.parent
+  }
+  return result
+}
+
 /**
  * 获取psi元素的所有祖先元素，按距离从近到远
  */

src/com/wuhao/code/check/inspection/visitor/JavaCodeFormatVisitor.kt

@@ -15,8 +15,7 @@ import com.intellij.openapi.project.Project
 import com.intellij.psi.*
 import com.intellij.psi.PsiPrimitiveType.*
 import com.intellij.psi.javadoc.PsiDocComment
-import com.wuhao.code.check.Messages
-import com.wuhao.code.check.getPsiElementFactory
+import com.wuhao.code.check.*
 import com.wuhao.code.check.inspection.CodeFormatInspection
 import com.wuhao.code.check.inspection.fix.ExtractToVariableFix
 import com.wuhao.code.check.inspection.fix.JavaBlockCommentFix
@@ -59,32 +58,40 @@ class JavaCodeFormatVisitor(holder: ProblemsHolder) : BaseCodeFormatVisitor(hold
         if (element.parent is PsiExpressionList
             && element.text != "0"
             && element.text != "0L" && element.type in PRIMITIVE_TYPES) {
-          holder.registerProblem(element, "不得直接使用数字作为方法参数",
+          holder.registerProblem(element, "不允许直接使用数字作为方法参数",
               ProblemHighlightType.GENERIC_ERROR,
               ExtractToVariableFix())
         }
       }
       is PsiMethodCallExpression -> {
         // 使用日志输入代替System.out
         if (element.text.startsWith("System.out") || element.text.startsWith("System.err")) {
-          holder.registerProblem(element, "使用日志向控制台输出", ProblemHighlightType.GENERIC_ERROR, object : LocalQuickFix {
+          if (element.ancestorOfType<PsiMethod>() == null
+              || !element.ancestorsOfType<PsiMethod>().any { func ->
+                func.annotations.any { annotation ->
+                  annotation.qualifiedName == JUNIT_TEST_ANNOTATION_CLASS_NAME
+                }
+              }
+          ) {
+            holder.registerProblem(element, "使用日志向控制台输出", ProblemHighlightType.GENERIC_ERROR, object : LocalQuickFix {
 
-            override fun getFamilyName(): String {
-              return "替换为日志输出"
-            }
+              override fun getFamilyName(): String {
+                return "替换为日志输出"
+              }
 
-            override fun applyFix(project: Project, descriptor: ProblemDescriptor) {
-              val el = descriptor.endElement
-              val factory = getPsiElementFactory(element)
-              if (el.firstChild is PsiReferenceExpression) {
-                if (el.firstChild.text.startsWith("System.out.print")) {
-                  el.firstChild.replace(factory.createExpressionFromText("LOG.info", null))
-                } else if (el.firstChild.text.startsWith("System.err.print")) {
-                  el.firstChild.replace(factory.createExpressionFromText("LOG.error", null))
+              override fun applyFix(project: Project, descriptor: ProblemDescriptor) {
+                val el = descriptor.endElement
+                val factory = getPsiElementFactory(element)
+                if (el.firstChild is PsiReferenceExpression) {
+                  if (el.firstChild.text.startsWith("System.out.print")) {
+                    el.firstChild.replace(factory.createExpressionFromText("LOG.info", null))
+                  } else if (el.firstChild.text.startsWith("System.err.print")) {
+                    el.firstChild.replace(factory.createExpressionFromText("LOG.error", null))
+                  }
                 }
               }
-            }
-          })
+            })
+          }
         }
       }
       is PsiMethod -> {

src/com/wuhao/code/check/inspection/visitor/KotlinCodeFormatVisitor.kt

@@ -12,11 +12,10 @@ import com.intellij.lang.Language
 import com.intellij.openapi.project.Project
 import com.intellij.psi.PsiElement
 import com.intellij.psi.impl.source.tree.LeafPsiElement
-import com.wuhao.code.check.Messages
-import com.wuhao.code.check.hasDocComment
+import com.wuhao.code.check.*
 import com.wuhao.code.check.inspection.CodeFormatInspection
 import com.wuhao.code.check.inspection.fix.KotlinCommentQuickFix
-import com.wuhao.code.check.isFirstLevelProperty
+import org.jetbrains.kotlin.asJava.toLightAnnotation
 import org.jetbrains.kotlin.idea.KotlinLanguage
 import org.jetbrains.kotlin.idea.refactoring.getLineCount
 import org.jetbrains.kotlin.kdoc.psi.api.KDoc
@@ -70,8 +69,19 @@ class KotlinCodeFormatVisitor(holder: ProblemsHolder) : BaseCodeFormatVisitor(ho
       }
       is KtReferenceExpression -> {
         // 使用日志输入代替System.out
+
         if (element.text == "println") {
-          holder.registerProblem(element, "使用日志向控制台输出", ProblemHighlightType.GENERIC_ERROR)
+          if (element.ancestorOfType<KtFunction>() == null
+              || !element.ancestorsOfType<KtFunction>().any { func ->
+                func.annotationEntries.map { annoEntry ->
+                  annoEntry.toLightAnnotation()
+                }.any { lightAnnotation ->
+                  lightAnnotation?.qualifiedName == JUNIT_TEST_ANNOTATION_CLASS_NAME
+                }
+              }
+          ) {
+            holder.registerProblem(element, "使用日志向控制台输出", ProblemHighlightType.GENERIC_ERROR)
+          }
         }
       }
       is LeafPsiElement -> {