fix: bug when getting CU from a parsed snippet (#2086)

INRIA · Jun 19, 2018 · 259de5c · 259de5c
1 parent 271be11
commit 259de5c
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 2 deletions.
diff --git a/src/main/java/spoon/reflect/factory/CompilationUnitFactory.java b/src/main/java/spoon/reflect/factory/CompilationUnitFactory.java
@@ -21,6 +21,7 @@
 import spoon.reflect.declaration.CtModule;
 import spoon.reflect.declaration.CtPackage;
 import spoon.reflect.declaration.CtType;
+import spoon.support.compiler.VirtualFile;
 import spoon.support.compiler.jdt.JDTSnippetCompiler;
 
 import java.io.File;
@@ -147,7 +148,11 @@ public CompilationUnit getOrCreate(String filePath) {
 				return cu;
 			}
 			cu = factory.Core().createCompilationUnit();
-			cu.setFile(new File(filePath));
+
+			if (!filePath.equals(VirtualFile.VIRTUAL_FILE_NAME)) {
+				cu.setFile(new File(filePath));
+			}
+
 			cachedCompilationUnits.put(filePath, cu);
 		}
 		return cu;

diff --git a/src/main/java/spoon/support/compiler/VirtualFile.java b/src/main/java/spoon/support/compiler/VirtualFile.java
@@ -24,10 +24,11 @@
 import java.io.InputStream;
 
 public class VirtualFile implements SpoonFile {
+	public static final String VIRTUAL_FILE_NAME = "virtual_file";
 
 	String content;
 
-	String name = "virtual_file";
+	String name = VIRTUAL_FILE_NAME;
 
 	public VirtualFile(String content) {
 		this.content = content;

diff --git a/src/test/java/spoon/test/api/APITest.java b/src/test/java/spoon/test/api/APITest.java
@@ -53,6 +53,7 @@
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertFalse;
 import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertSame;
 import static org.junit.Assert.assertTrue;
 import static org.junit.Assert.fail;
 
@@ -566,5 +567,15 @@ public void testOutputWithNoOutputProduceNoFolder() {
 		assertFalse("Output dir should not exist: "+outputDir.getAbsolutePath(), outputDir.exists());
 	}
 
+	@Test
+	public void testGetOneLinerMainClassFromCU() {
+		// contract: when using Spoon with a snippet, we can still use properly CompilationUnit methods
+		CtClass<?> l = Launcher.parseClass("class A { void m() { System.out.println(\"yeah\");} }");
+		CompilationUnit compilationUnit = l.getPosition().getCompilationUnit();
+
+		assertNotNull(compilationUnit);
+		CtType<?> mainType = compilationUnit.getMainType();
+		assertSame(l, mainType);
+	}
 
 }