review: fix: prevent renaming CtRootPackage

src/main/java/spoon/reflect/CtModelImpl.java

@@ -18,6 +18,7 @@
 
 import spoon.processing.Processor;
 import spoon.reflect.declaration.CtElement;
+import spoon.reflect.declaration.CtNamedElement;
 import spoon.reflect.declaration.CtPackage;
 import spoon.reflect.declaration.CtType;
 import spoon.reflect.declaration.ParentNotInitializedException;
@@ -64,6 +65,19 @@ public String getSimpleName() {
 			return super.getSimpleName();
 		}
 
+		@Override
+		public <T extends CtNamedElement> T setSimpleName(String name) {
+			if (name == null) {
+				return (T) this;
+			}
+
+			if (name.equals(CtPackage.TOP_LEVEL_PACKAGE_NAME)) {
+				return super.setSimpleName(name);
+			}
+
+			return (T) this;
+		}
+
 		@Override
 		public String getQualifiedName() {
 			return "";

src/test/java/spoon/test/pkg/PackageTest.java

@@ -4,6 +4,7 @@
 import org.junit.Test;
 import spoon.Launcher;
 import spoon.OutputType;
+import spoon.SpoonException;
 import spoon.SpoonModelBuilder;
 import spoon.compiler.Environment;
 import spoon.compiler.SpoonResourceHelper;
@@ -12,10 +13,17 @@
 import spoon.reflect.declaration.CtClass;
 import spoon.reflect.declaration.CtPackage;
 import spoon.reflect.factory.Factory;
+import spoon.reflect.visitor.DefaultJavaPrettyPrinter;
+import spoon.reflect.visitor.PrettyPrinter;
+import spoon.reflect.visitor.filter.NamedElementFilter;
 import spoon.test.pkg.name.PackageTestClass;
+import spoon.test.pkg.testclasses.ElementProcessor;
 import spoon.testing.utils.ModelUtils;
 
+import java.io.BufferedReader;
 import java.io.File;
+import java.io.FileReader;
+import java.util.Collections;
 
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertTrue;
@@ -119,4 +127,94 @@ public void testAnnotationInPackageInfoWhenTemplatesCompiled() throws Exception
 		launcher.prettyprint();
 		canBeBuilt("./target/spooned/packageAndTemplate/spoon/test/pkg/package-info.java", 8);
 	}
+
+	@Test
+	public void testRenamePackageAndPrettyPrint() throws Exception {
+		final Launcher spoon = new Launcher();
+		spoon.addInputResource("./src/test/java/spoon/test/pkg/testclasses/Foo.java");
+		spoon.buildModel();
+
+		CtPackage ctPackage = spoon.getModel().getElements(new NamedElementFilter<CtPackage>(CtPackage.class, "spoon")).get(0);
+		ctPackage.setSimpleName("otherName");
+
+		CtClass foo = spoon.getModel().getElements(new NamedElementFilter<CtClass>(CtClass.class, "Foo")).get(0);
+		assertEquals("otherName.test.pkg.testclasses.Foo", foo.getQualifiedName());
+
+		PrettyPrinter prettyPrinter = new DefaultJavaPrettyPrinter(spoon.getEnvironment());
+		prettyPrinter.calculate(spoon.getFactory().CompilationUnit().create("./src/test/java/spoon/test/pkg/testclasses/Foo.java"), Collections.singletonList(foo));
+		String result = prettyPrinter.getResult();
+
+		assertTrue(result.contains("package otherName.test.pkg.testclasses;"));
+	}
+
+	@Test
+	public void testRenamePackageAndPrettyPrintNoclasspath() throws Exception {
+		final Launcher spoon = new Launcher();
+		spoon.addInputResource("./src/test/resources/noclasspath/app/Test.java");
+		spoon.getEnvironment().setNoClasspath(true);
+		spoon.buildModel();
+
+		CtPackage ctPackage = spoon.getModel().getElements(new NamedElementFilter<CtPackage>(CtPackage.class, "app")).get(0);
+		ctPackage.setSimpleName("otherName");
+
+		CtClass foo = spoon.getModel().getElements(new NamedElementFilter<CtClass>(CtClass.class, "Test")).get(0);
+		assertEquals("otherName.Test", foo.getQualifiedName());
+
+		PrettyPrinter prettyPrinter = new DefaultJavaPrettyPrinter(spoon.getEnvironment());
+		prettyPrinter.calculate(spoon.getFactory().CompilationUnit().create("./src/test/resources/noclasspath/app/Test.java"), Collections.singletonList(foo));
+		String result = prettyPrinter.getResult();
+
+		assertTrue(result.contains("package otherName;"));
+	}
+
+	@Test
+	public void testRenamePackageAndPrettyPrintWithProcessor() throws Exception {
+		String destPath = "./target/spoon-rename-processor";
+		final Launcher spoon = new Launcher();
+		spoon.addInputResource("./src/test/resources/noclasspath/app/Test.java");
+		spoon.getEnvironment().setNoClasspath(true);
+		spoon.addProcessor(new ElementProcessor());
+		spoon.setSourceOutputDirectory(destPath);
+		spoon.run();
+
+		String fileDir = destPath+"/newtest/Test.java";
+		File f = new File(fileDir);
+		assertTrue(f.exists());
+
+		BufferedReader reader = new BufferedReader(new FileReader(f));
+		assertTrue(reader.lines().anyMatch((s) -> {
+			return s.equals("package newtest;");
+		}));
+	}
+
+	@Test
+	public void testRenameRootPackage() throws Exception {
+		final Launcher spoon = new Launcher();
+		spoon.addInputResource("./src/test/resources/noclasspath/app/Test.java");
+		spoon.getEnvironment().setNoClasspath(true);
+		spoon.buildModel();
+
+		CtPackage rootPackage = spoon.getFactory().Package().getRootPackage();
+		String rootPackageName = rootPackage.getSimpleName();
+		rootPackage.setSimpleName("test");
+		assertEquals(rootPackageName, rootPackage.getSimpleName());
+	}
+
+	@Test
+	public void testRenameRootPackageWithNullOrEmpty() throws Exception {
+		final Launcher spoon = new Launcher();
+		spoon.addInputResource("./src/test/resources/noclasspath/app/Test.java");
+		spoon.getEnvironment().setNoClasspath(true);
+		spoon.buildModel();
+
+		CtPackage rootPackage = spoon.getFactory().Package().getRootPackage();
+		String rootPackageName = rootPackage.getSimpleName();
+		assertEquals(CtPackage.TOP_LEVEL_PACKAGE_NAME, rootPackageName);
+
+		rootPackage.setSimpleName("");
+		assertEquals(CtPackage.TOP_LEVEL_PACKAGE_NAME, rootPackageName);
+
+		rootPackage.setSimpleName(null);
+		assertEquals(CtPackage.TOP_LEVEL_PACKAGE_NAME, rootPackageName);
+	}
 }

src/test/java/spoon/test/pkg/testclasses/ElementProcessor.java

@@ -0,0 +1,17 @@
+package spoon.test.pkg.testclasses;
+
+import spoon.processing.AbstractProcessor;
+import spoon.reflect.declaration.CtElement;
+import spoon.reflect.declaration.CtPackage;
+
+/**
+ * Created by urli on 11/09/2017.
+ */
+public class ElementProcessor extends AbstractProcessor<CtElement> {
+    public void process(CtElement element) {
+        if(element instanceof CtPackage){
+            ((CtPackage) element).setSimpleName("newtest");
+        }
+    }
+}
+

src/test/resources/noclasspath/app/Test.java

@@ -0,0 +1,7 @@
+package app;
+
+public class Test {
+    public static void main(String[] args){
+        System.out.println("test");
+    }
+}