TestDuplicateClass.java

package dev.bitflippers;

import org.junit.jupiter.api.Test;
import org.openrewrite.java.JavaParser;
import org.openrewrite.test.RecipeSpec;
import org.openrewrite.test.RewriteTest;

import static org.openrewrite.java.Assertions.java;

public class TestDuplicateClass implements RewriteTest {
    @Override
    public void defaults(RecipeSpec spec) {
        spec.parser(JavaParser.fromJavaVersion().classpath("lombok"));
    }

    @Test
    public void classIsDuplicatedWithNewPrefix() {
        rewriteRun(
            spec -> spec.recipe(new DuplicateClass("dev.bitflippers.example", "TestClass")),
            java(
                //language=java
                """
                    package dev.bitflippers.example;
                    
                    import lombok.Getter;
                                                    
                    public class TestClass {
                        @Getter
                        public String foo = "foo";
                        @Getter
                        public int a = 10;
                        
                        private TestClass() {}
                    }                                                    
                    """,
                spec -> spec.path("foo/bar/src/main/dev/bitflippers/example/TestClass.java")
            ),
            java(null,
                //language=java
                """
                    package dev.bitflippers.example;
                    
                    import lombok.Getter;
                                                    
                    public class CopyOfTestClass {
                        @Getter
                        public String foo = "foo";
                        @Getter
                        public int a = 10;
                        
                        private CopyOfTestClass() {}
                    }                                                    
                    """,
                spec -> spec.path("foo/bar/src/main/dev/bitflippers/example/CopyOfTestClass.java")
            )
        );
    }

    @Test
    public void classNotDuplicatedIfDoesNotMatch() {
        rewriteRun(
                spec -> spec.recipe(new DuplicateClass("dev.bitflippers.example", "SomeOtherClass")),
                java(
                        //language=java
                        """
                            package dev.bitflippers.example;
                            
                            import lombok.Getter;
                                                            
                            public class TestClass {
                                @Getter
                                public String foo = "foo";
                                @Getter
                                public int a = 10;
                            }                                                    
                            """,
                        spec -> spec.path("foo/bar/src/main/dev/bitflippers/example/TestClass.java")
                )
        );
    }

    @Test
    public void classIsDuplicatedNotDuplicatedIfClassWithSameNameAlreadyExists() {
        rewriteRun(
                spec -> spec.recipe(new DuplicateClass("dev.bitflippers.example", "TestClass")),
                java(
                        //language=java
                        """
                            package dev.bitflippers.example;
                            
                            import lombok.Getter;
                                                            
                            public class TestClass {
                                @Getter
                                public String foo = "foo";
                                @Getter
                                public int a = 10;
                            }                                                    
                            """,
                        spec -> spec.path("foo/bar/src/main/dev/bitflippers/example/TestClass.java")
                ),
                java(
                        //language=java
                        """
                            package dev.bitflippers.example;
                            
                            import lombok.Getter;
                                                            
                            public class CopyOfTestClass {
                                @Getter
                                public String foo = "foo";
                                @Getter
                                public int a = 10;
                            }                                                    
                            """,
                        spec -> spec.path("foo/bar/src/main/dev/bitflippers/example/CopyOfTestClass.java")
                )
        );
    }
}