Update BlockStatementTemplateGeneratorAdd with support for templating while-loop condition expressions. (#2213)

Author: pway99

rewrite-java-tck/src/main/kotlin/org/openrewrite/java/JavaTemplateTest.kt

@@ -25,10 +25,7 @@ import org.junit.jupiter.api.Test
 import org.openrewrite.ExecutionContext
 import org.openrewrite.Issue
 import org.openrewrite.java.Assertions.java
-import org.openrewrite.java.tree.J
-import org.openrewrite.java.tree.JavaType
-import org.openrewrite.java.tree.Space
-import org.openrewrite.java.tree.TypeUtils
+import org.openrewrite.java.tree.*
 import org.openrewrite.test.RewriteTest
 import java.util.Comparator.comparing
 
@@ -2436,4 +2433,44 @@ interface JavaTemplateTest : RewriteTest, JavaRecipeTest {
             """
         )
     )
+
+    @Suppress("LoopConditionNotUpdatedInsideLoop")
+    @Test
+    fun templatingWhileLoopCondition() = rewriteRun(
+        { spec ->
+            spec.recipe(toRecipe {
+                object : JavaVisitor<ExecutionContext>() {
+                    override fun visitBinary(binary: J.Binary, p: ExecutionContext): J {
+                        if (binary.left is J.MethodInvocation) {
+                            val mi = binary.left as J.MethodInvocation
+                            return binary.withTemplate(
+                                JavaTemplate.builder(this::getCursor, "!#{any(java.util.List)}.isEmpty()")
+                                    .build(), mi.coordinates.replace(), mi.select
+                            )
+                        } else if (binary.left is J.Unary) {
+                            return binary.left
+                        }
+                        return binary
+                    }
+                }
+            })
+            spec.expectedCyclesThatMakeChanges(2)
+        },
+        java("""
+            import java.util.List;
+            class T {
+                void m(List<?> l) {
+                    while (l.size() != 0) {}
+                }
+            }
+        """,
+        """
+            import java.util.List;
+            class T {
+                void m(List<?> l) {
+                    while (!l.isEmpty()) {}
+                }
+            }
+        """)
+    )
 }

rewrite-java/src/main/java/org/openrewrite/java/internal/template/BlockStatementTemplateGenerator.java

@@ -337,6 +337,12 @@ else if (m.getSelect() == prior) {
                 before.insert(0, "Object __b" + cursor.getPathAsStream().count() + "__ =");
                 after.append(";");
             }
+        } else if (j instanceof J.WhileLoop) {
+            J.WhileLoop wl = (J.WhileLoop)j;
+            if(referToSameElement(prior, wl.getCondition())) {
+                before.insert(0, "Object __b" + cursor.getPathAsStream().count() + "__ =");
+                after.append(";");
+            }
         } else if(j instanceof J.Assignment) {
             J.Assignment as = (J.Assignment)j;
             if(referToSameElement(prior, as.getAssignment())) {