Add tests and fix for package name override.

This commit is contained in:
A Pottinger
2019-10-13 14:37:15 -07:00
parent d57b4ac61b
commit 1fc326383e
5 changed files with 65 additions and 11 deletions

View File

@@ -771,15 +771,15 @@ public class PdeParseTreeListener extends ProcessingBaseListener {
protected void writeHeaderContents(PrintWriterWithEditGen decoratedWriter,
RewriteResultBuilder resultBuilder) {
if (!isTesting) {
writePreprocessorComment(decoratedWriter, resultBuilder);
}
if (destinationPackageName.isPresent()) {
decoratedWriter.addCodeLine("package " + destinationPackageName.get() + ";");
decoratedWriter.addEmptyLine();
}
if (!isTesting) {
writePreprocessorComment(decoratedWriter, resultBuilder);
}
writeImports(decoratedWriter, resultBuilder);
boolean requiresClassHeader = mode == PdePreprocessor.Mode.STATIC;

View File

@@ -191,7 +191,7 @@ public class PdePreprocessor {
tokens,
sketchName,
tabSize,
Optional.empty()
destinationPackage
);
listener.setTesting(isTesting);
listener.setCoreImports(coreImports);