Skip to content

Commit 7290766

Browse files
authored
[epilogue-processor] Fix naive tests in prep for huge refactor (#8255)
1 parent bd5141e commit 7290766

File tree

1 file changed

+14
-1
lines changed

1 file changed

+14
-1
lines changed

epilogue-processor/src/test/java/edu/wpi/first/epilogue/processor/AnnotationProcessorTest.java

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
import java.io.IOException;
1717
import java.util.List;
1818
import java.util.Locale;
19+
import java.util.regex.Pattern;
1920
import javax.tools.Diagnostic;
2021
import javax.tools.JavaFileObject;
2122
import org.junit.jupiter.api.Test;
@@ -2317,6 +2318,18 @@ private void assertCompilationError(
23172318
}
23182319

23192320
private void assertLoggerGenerates(String loggedClassContent, String loggerClassContent) {
2321+
// Extract the expected logger file name from the class declaration so we can find the correct
2322+
// generated file.
2323+
var pattern =
2324+
Pattern.compile(".*public class (.*) extends ClassSpecificLogger.*", Pattern.DOTALL);
2325+
var matcher = pattern.matcher(loggerClassContent);
2326+
var className = "ExampleLogger";
2327+
if (matcher.matches()) {
2328+
var result = matcher.toMatchResult();
2329+
className = result.group(1);
2330+
}
2331+
var loggerFileName = "/" + className + ".java";
2332+
23202333
Compilation compilation =
23212334
javac()
23222335
.withOptions(kJavaVersionOptions)
@@ -2329,7 +2342,7 @@ private void assertLoggerGenerates(String loggedClassContent, String loggerClass
23292342
var generatedFiles = compilation.generatedSourceFiles();
23302343
var generatedFile =
23312344
generatedFiles.stream()
2332-
.filter(jfo -> jfo.getName().contains("Example"))
2345+
.filter(jfo -> jfo.toUri().getPath().endsWith(loggerFileName))
23332346
.findFirst()
23342347
.orElseThrow(() -> new IllegalStateException("Logger file was not generated!"));
23352348
try {

0 commit comments

Comments
 (0)