jlengrand/error-prone-support
1
0
Fork 0
mirror of https://github.com/jlengrand/error-prone-support.git synced 2026-03-10 08:11:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Filter out bug pattern tests

Browse Source
This commit is contained in:
Gijs de Jong
2022-09-30 08:24:41 +02:00
parent 2b8139bc39
commit aacde8e099
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
docgen/src/main/java/tech/picnic/errorprone/plugin/DocgenTaskListener.java
View File

@@ -6,6 +6,8 @@ import com.fasterxml.jackson.databind.SerializationFeature;
import com.google.errorprone.BugPattern;
import com.google.errorprone.util.ASTHelpers;
import com.sun.source.tree.ClassTree;
import com.sun.source.tree.Tree;
import com.sun.source.tree.VariableTree;
import com.sun.source.util.TaskEvent;
import com.sun.source.util.TaskEvent.Kind;
import com.sun.source.util.TaskListener;
@@ -40,7 +42,7 @@ final class DocgenTaskListener implements TaskListener {
}
ClassTree tree = JavacTrees.instance(context).getTree(taskEvent.getTypeElement());
if (tree == null || !isBugPattern(tree)) {
if (tree == null || !isBugPatternTest(tree)) {
return;
}
@@ -62,4 +64,11 @@ final class DocgenTaskListener implements TaskListener {
private static boolean isBugPattern(ClassTree tree) {
return ASTHelpers.hasDirectAnnotationWithSimpleName(tree, BugPattern.class.getSimpleName());
}
private static boolean isBugPatternTest(ClassTree tree) {
return tree.getMembers().stream()
.filter(VariableTree.class::isInstance)
.map(VariableTree.class::cast)
.anyMatch(member -> member.getType().toString().equals("BugCheckerRefactoringTestHelper"));
}
}