org.openrewrite.java.tree.ForLoopTest.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of rewrite-test Show documentation
Eliminate tech-debt. Automatically.
There is a newer version: 8.42.0
/*
 * Copyright 2020 the original author or authors.
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 

 * https://www.apache.org/licenses/LICENSE-2.0
 * 

 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.openrewrite.java.tree

import org.junit.jupiter.api.Assertions.assertEquals
import org.junit.jupiter.api.Assertions.assertTrue
import org.junit.jupiter.api.Test
import org.openrewrite.java.JavaParser
import org.openrewrite.java.firstMethodStatement

interface ForLoopTest {
    
    @Test
    fun forLoop(jp: JavaParser) {
        val a = jp.parse("""
            public class A {
                public void test() {
                    for(int i = 0; i < 10; i++) {
                    }
                }
            }
        """)[0]
        
        val forLoop = a.firstMethodStatement() as J.ForLoop
        assertTrue(forLoop.control.init is J.VariableDecls)
        assertTrue(forLoop.control.condition is J.Binary)
        assertEquals(1, forLoop.control.update.size)
    }

    @Test
    fun infiniteLoop(jp: JavaParser) {
        val a = jp.parse("""
            public class A {
                public void test() {
                    for(;;) {
                    }
                }
            }
        """)[0]

        val forLoop = a.firstMethodStatement() as J.ForLoop
        assertTrue(forLoop.control.init is J.Empty)
        assertTrue(forLoop.control.condition is J.Empty)
        assertTrue(forLoop.control.update[0] is J.Empty)
    }

    @Test
    fun format(jp: JavaParser) {
        val a = jp.parse("""
            public class A {
                public void test() {
                    for ( int i = 0 ; i < 10 ; i++ ) {
                    }
                }
            }
        """)[0]

        val forLoop = a.firstMethodStatement() as J.ForLoop
        assertEquals("for ( int i = 0 ; i < 10 ; i++ ) {\n}", forLoop.printTrimmed())
    }

    @Test
    fun formatInfiniteLoop(jp: JavaParser) {
        val a = jp.parse("""
            public class A {
                public void test() {
                    for ( ; ; ) {}
                }
            }
        """)[0]

        val forLoop = a.firstMethodStatement() as J.ForLoop
        assertEquals("for ( ; ; ) {}", forLoop.printTrimmed())
    }

    @Test
    fun formatLoopNoInit(jp: JavaParser) {
        val a = jp.parse("""
            public class A {
                public void test() {
                    int i = 0;
                    for ( ; i < 10 ; i++ ) {}
                }
            }
        """)[0]

        val forLoop = a.classes[0].methods[0].body!!.statements[1] as J.ForLoop
        assertEquals("for ( ; i < 10 ; i++ ) {}", forLoop.printTrimmed())
    }

    @Test
    fun formatLoopNoCondition(jp: JavaParser) {
        val a = jp.parse("""
            public class A {
                public void test() {
                    int i = 0;
                    for(; i < 10; i++) {}
                }
            }
        """)[0]

        val forLoop = a.classes[0].methods[0].body!!.statements[1] as J.ForLoop
        assertEquals("for(; i < 10; i++) {}", forLoop.printTrimmed())
    }

    @Test
    fun statementTerminatorForSingleLineForLoops(jp: JavaParser) {
        val a = jp.parse("""
            public class A {
                public void test() {
                    for(;;) test();
                }
            }
        """)[0]

        val forLoop = a.classes[0].methods[0].body!!.statements[0] as J.ForLoop
        assertEquals("for(;;) test();", forLoop.printTrimmed())
    }

    @Test
    fun initializerIsAnAssignment(jp: JavaParser) {
        val a = jp.parse("""
            public class A {
                int[] a;
                public void test() {
                    int i=0;
                    for(i=0; i