diff options
Diffstat (limited to 'src/test/java/com/puppycrawl/tools/checkstyle/utils/AnnotationUtilityTest.java')
-rw-r--r-- | src/test/java/com/puppycrawl/tools/checkstyle/utils/AnnotationUtilityTest.java | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/src/test/java/com/puppycrawl/tools/checkstyle/utils/AnnotationUtilityTest.java b/src/test/java/com/puppycrawl/tools/checkstyle/utils/AnnotationUtilityTest.java index 090b2d3b8..a000ea583 100644 --- a/src/test/java/com/puppycrawl/tools/checkstyle/utils/AnnotationUtilityTest.java +++ b/src/test/java/com/puppycrawl/tools/checkstyle/utils/AnnotationUtilityTest.java @@ -173,6 +173,7 @@ public class AnnotationUtilityTest { final DetailAST annotationName = new DetailAST(); annotations.setType(TokenTypes.ANNOTATIONS); annotation.setType(TokenTypes.ANNOTATION); + annotationNameHolder.setType(TokenTypes.AT); annotationName.setText("Annotation"); annotationNameHolder.setNextSibling(annotationName); @@ -184,4 +185,31 @@ public class AnnotationUtilityTest { assertTrue("Annotation should contain " + astForTest, AnnotationUtility.containsAnnotation(astForTest, "Annotation")); } + + @Test + public void testContainsAnnotationWithComment() { + final DetailAST astForTest = new DetailAST(); + astForTest.setType(TokenTypes.PACKAGE_DEF); + final DetailAST child = new DetailAST(); + final DetailAST annotations = new DetailAST(); + final DetailAST annotation = new DetailAST(); + final DetailAST annotationNameHolder = new DetailAST(); + final DetailAST annotationName = new DetailAST(); + final DetailAST comment = new DetailAST(); + annotations.setType(TokenTypes.ANNOTATIONS); + annotation.setType(TokenTypes.ANNOTATION); + annotationNameHolder.setType(TokenTypes.AT); + comment.setType(TokenTypes.BLOCK_COMMENT_BEGIN); + annotationName.setText("Annotation"); + + annotationNameHolder.setNextSibling(annotationName); + annotation.setFirstChild(comment); + comment.setNextSibling(annotationNameHolder); + annotations.setFirstChild(annotation); + child.setNextSibling(annotations); + astForTest.setFirstChild(child); + + assertTrue("Annotation should contain " + astForTest, + AnnotationUtility.containsAnnotation(astForTest, "Annotation")); + } } |