Implemented AssertThatEnumerableIsEmptyInspection. Improved literal values by constant value calculation, more refactoring.

src/main/java/de/platon42/intellij/plugins/cajon/inspections/AbstractAssertJInspection.kt

@@ -27,19 +27,22 @@ open class AbstractAssertJInspection : AbstractBaseJavaLocalInspectionTool() {
         const val ABSTRACT_BOOLEAN_ASSERT_CLASSNAME = "org.assertj.core.api.AbstractBooleanAssert"
         const val ABSTRACT_STRING_ASSERT_CLASSNAME = "org.assertj.core.api.AbstractStringAssert"
         const val ABSTRACT_CHAR_SEQUENCE_ASSERT_CLASSNAME = "org.assertj.core.api.AbstractCharSequenceAssert"
-        const val IS_EQUAL_TO = "isEqualTo"
-        const val IS_NOT_EQUAL_TO = "isNotEqualTo"
-        const val HAS_SIZE = "hasSize"
+        const val ABSTRACT_ENUMERABLE_ASSERT_CLASSNAME = "org.assertj.core.api.EnumerableAssert"
 
-        val IS_EQUAL_TO_OBJECT = CallMatcher.instanceCall(ABSTRACT_ASSERT_CLASSNAME, IS_EQUAL_TO)
+        const val IS_EQUAL_TO_METHOD = "isEqualTo"
+        const val IS_NOT_EQUAL_TO_METHOD = "isNotEqualTo"
+        const val HAS_SIZE_METHOD = "hasSize"
+
+        val IS_EQUAL_TO_OBJECT = CallMatcher.instanceCall(ABSTRACT_ASSERT_CLASSNAME, IS_EQUAL_TO_METHOD)
             .parameterTypes(CommonClassNames.JAVA_LANG_OBJECT)!!
-        val IS_NOT_EQUAL_TO_OBJECT = CallMatcher.instanceCall(ABSTRACT_ASSERT_CLASSNAME, IS_NOT_EQUAL_TO)
+        val IS_NOT_EQUAL_TO_OBJECT = CallMatcher.instanceCall(ABSTRACT_ASSERT_CLASSNAME, IS_NOT_EQUAL_TO_METHOD)
             .parameterTypes(CommonClassNames.JAVA_LANG_OBJECT)!!
-        val IS_EQUAL_TO_BOOLEAN = CallMatcher.instanceCall(ABSTRACT_BOOLEAN_ASSERT_CLASSNAME, IS_EQUAL_TO)
+        val IS_EQUAL_TO_BOOLEAN = CallMatcher.instanceCall(ABSTRACT_BOOLEAN_ASSERT_CLASSNAME, IS_EQUAL_TO_METHOD)
             .parameterTypes("boolean")!!
-        val IS_NOT_EQUAL_TO_BOOLEAN = CallMatcher.instanceCall(ABSTRACT_BOOLEAN_ASSERT_CLASSNAME, IS_NOT_EQUAL_TO)
-            .parameterTypes("boolean")!!
-        val CHAR_SEQUENCE_HAS_SIZE = CallMatcher.instanceCall(ABSTRACT_CHAR_SEQUENCE_ASSERT_CLASSNAME, HAS_SIZE)
+        val IS_NOT_EQUAL_TO_BOOLEAN =
+            CallMatcher.instanceCall(ABSTRACT_BOOLEAN_ASSERT_CLASSNAME, IS_NOT_EQUAL_TO_METHOD)
+                .parameterTypes("boolean")!!
+        val HAS_SIZE = CallMatcher.instanceCall(ABSTRACT_ENUMERABLE_ASSERT_CLASSNAME, HAS_SIZE_METHOD)
             .parameterTypes("int")!!
     }
 
@@ -77,4 +80,10 @@ open class AbstractAssertJInspection : AbstractBaseJavaLocalInspectionTool() {
             ReplaceSimpleMethodCallQuickFix(description, replacementMethod)
         )
     }
+
+    protected fun calculateConstantParameterValue(expression: PsiMethodCallExpression, argIndex: Int): Any? {
+        val valueExpression = expression.argumentList.expressions[argIndex] ?: return null
+        val constantEvaluationHelper = JavaPsiFacade.getInstance(expression.project).constantEvaluationHelper
+        return constantEvaluationHelper.computeConstantExpression(valueExpression)
+    }
 }

src/main/java/de/platon42/intellij/plugins/cajon/inspections/AssertThatBooleanIsTrueOrFalseInspection.kt

@@ -31,22 +31,21 @@ class AssertThatBooleanIsTrueOrFalseInspection : AbstractAssertJInspection() {
                     return
                 }
 
-                val equalToExpression = expression.argumentList.expressions[0]!!
+                val equalToExpression = expression.argumentList.expressions[0] ?: return
                 if (!TypeConversionUtil.isBooleanType(equalToExpression.type)) {
                     return
                 }
-                val constantEvaluationHelper = JavaPsiFacade.getInstance(holder.project).constantEvaluationHelper
-                var result = constantEvaluationHelper.computeConstantExpression(equalToExpression)
-                if (result == null) {
+                var value = calculateConstantParameterValue(expression, 0)
+                if (value == null) {
                     val field = (equalToExpression as? PsiReferenceExpression)?.resolve() as? PsiField
                     if (field?.containingClass?.qualifiedName == CommonClassNames.JAVA_LANG_BOOLEAN) {
                         when {
-                            field.name == "TRUE" -> result = true
-                            field.name == "FALSE" -> result = false
+                            field.name == "TRUE" -> value = true
+                            field.name == "FALSE" -> value = false
                         }
                     }
                 }
-                val expectedResult = result as? Boolean ?: return
+                val expectedResult = value as? Boolean ?: return
 
                 val replacementMethod = if (expectedResult xor flippedBooleanTest) "isTrue()" else "isFalse()"
                 registerSimplifyMethod(holder, expression, replacementMethod)

src/main/java/de/platon42/intellij/plugins/cajon/inspections/AssertThatEnumerableIsEmptyInspection.kt

@@ -0,0 +1,34 @@
+package de.platon42.intellij.plugins.cajon.inspections
+
+import com.intellij.codeInspection.ProblemsHolder
+import com.intellij.psi.JavaElementVisitor
+import com.intellij.psi.PsiElementVisitor
+import com.intellij.psi.PsiMethodCallExpression
+import org.jetbrains.annotations.NonNls
+
+class AssertThatEnumerableIsEmptyInspection : AbstractAssertJInspection() {
+
+    companion object {
+        @NonNls
+        private val DISPLAY_NAME = "Asserting an enumerable is empty"
+    }
+
+    override fun getDisplayName() = DISPLAY_NAME
+
+    override fun buildVisitor(holder: ProblemsHolder, isOnTheFly: Boolean): PsiElementVisitor {
+        return object : JavaElementVisitor() {
+            override fun visitMethodCallExpression(expression: PsiMethodCallExpression) {
+                super.visitMethodCallExpression(expression)
+                val hasSize = HAS_SIZE.test(expression)
+                if (!hasSize) {
+                    return
+                }
+
+                val value = calculateConstantParameterValue(expression, 0) ?: return
+                if (value == 0) {
+                    registerSimplifyMethod(holder, expression, "isEmpty()")
+                }
+            }
+        }
+    }
+}

src/main/java/de/platon42/intellij/plugins/cajon/inspections/AssertThatStringIsEmptyInspection.kt

@@ -3,7 +3,6 @@ package de.platon42.intellij.plugins.cajon.inspections
 import com.intellij.codeInspection.ProblemsHolder
 import com.intellij.psi.JavaElementVisitor
 import com.intellij.psi.PsiElementVisitor
-import com.intellij.psi.PsiLiteralExpression
 import com.intellij.psi.PsiMethodCallExpression
 import org.jetbrains.annotations.NonNls
 
@@ -21,7 +20,7 @@ class AssertThatStringIsEmptyInspection : AbstractAssertJInspection() {
             override fun visitMethodCallExpression(expression: PsiMethodCallExpression) {
                 super.visitMethodCallExpression(expression)
                 val isEqual = IS_EQUAL_TO_OBJECT.test(expression)
-                val hasSize = CHAR_SEQUENCE_HAS_SIZE.test(expression)
+                val hasSize = HAS_SIZE.test(expression)
                 if (!(isEqual || hasSize)) {
                     return
                 }
@@ -30,17 +29,9 @@ class AssertThatStringIsEmptyInspection : AbstractAssertJInspection() {
                     return
                 }
 
-                if (isEqual) {
-                    val psiExpression = expression.argumentList.expressions[0] as? PsiLiteralExpression ?: return
-
-                    if (psiExpression.value == "") {
-                        registerSimplifyMethod(holder, expression, "isEmpty()")
-                    }
-                } else {
-                    val psiExpression = expression.argumentList.expressions[0] as? PsiLiteralExpression ?: return
-                    if (psiExpression.value == 0) {
-                        registerSimplifyMethod(holder, expression, "isEmpty()")
-                    }
+                val value = calculateConstantParameterValue(expression, 0) ?: return
+                if ((isEqual && (value == "")) || (hasSize && (value == 0))) {
+                    registerSimplifyMethod(holder, expression, "isEmpty()")
                 }
             }
         }