Support annotations in TypeNameCheck

Author: tsjensen

src/main/java/com/puppycrawl/tools/checkstyle/checks/naming/TypeNameCheck.java

@@ -68,6 +68,7 @@ public int[] getDefaultTokens()
         return new int[] {TokenTypes.CLASS_DEF,
                           TokenTypes.INTERFACE_DEF,
                           TokenTypes.ENUM_DEF,
+                          TokenTypes.ANNOTATION_DEF,
         };
     }
 }

src/test/java/com/puppycrawl/tools/checkstyle/checks/naming/TypeNameCheckTest.java

@@ -26,6 +26,7 @@
 
 import com.puppycrawl.tools.checkstyle.BaseCheckTestSupport;
 import com.puppycrawl.tools.checkstyle.DefaultConfiguration;
+import com.puppycrawl.tools.checkstyle.api.TokenTypes;
 
 public class TypeNameCheckTest
     extends BaseCheckTestSupport
@@ -69,6 +70,8 @@ public void testDefault()
                         TypeNameCheck.DEFAULT_PATTERN),
                 buildMesssage(7, 17, "inputHeaderEnum",
                         TypeNameCheck.DEFAULT_PATTERN),
+                buildMesssage(9, 23, "inputHeaderAnnotation",
+                    TypeNameCheck.DEFAULT_PATTERN),
         };
         verify(checkConfig, inputFilename, expected);
     }
@@ -79,7 +82,7 @@ public void testClassSpecific()
     {
         final DefaultConfiguration checkConfig =
             createCheckConfig(TypeNameCheck.class);
-        checkConfig.addAttribute("tokens", "CLASS_DEF");
+        checkConfig.addAttribute("tokens", TokenTypes.getTokenName(TokenTypes.CLASS_DEF));
         final String[] expected = {
                 buildMesssage(3, 7, "inputHeaderClass",
                         TypeNameCheck.DEFAULT_PATTERN),
@@ -93,7 +96,7 @@ public void testInterfaceSpecific()
     {
         final DefaultConfiguration checkConfig =
             createCheckConfig(TypeNameCheck.class);
-        checkConfig.addAttribute("tokens", "INTERFACE_DEF");
+        checkConfig.addAttribute("tokens", TokenTypes.getTokenName(TokenTypes.INTERFACE_DEF));
         final String[] expected = {
                 buildMesssage(5, 22, "inputHeaderInterface",
                         TypeNameCheck.DEFAULT_PATTERN),
@@ -107,14 +110,28 @@ public void testEnumSpecific()
     {
         final DefaultConfiguration checkConfig =
             createCheckConfig(TypeNameCheck.class);
-        checkConfig.addAttribute("tokens", "ENUM_DEF");
+        checkConfig.addAttribute("tokens", TokenTypes.getTokenName(TokenTypes.ENUM_DEF));
         final String[] expected = {
                 buildMesssage(7, 17, "inputHeaderEnum",
                         TypeNameCheck.DEFAULT_PATTERN),
         };
         verify(checkConfig, inputFilename, expected);
     }
 
+    @Test
+    public void testAnnotationSpecific()
+        throws Exception
+    {
+        final DefaultConfiguration checkConfig =
+            createCheckConfig(TypeNameCheck.class);
+        checkConfig.addAttribute("tokens", TokenTypes.getTokenName(TokenTypes.ANNOTATION_DEF));
+        final String[] expected = {
+            buildMesssage(9, 23, "inputHeaderAnnotation",
+                        TypeNameCheck.DEFAULT_PATTERN),
+        };
+        verify(checkConfig, inputFilename, expected);
+    }
+
     private String buildMesssage(int lineNumber, int colNumber, String name,
             String pattern)
     {

src/test/resources/com/puppycrawl/tools/checkstyle/naming/InputTypeName.java

@@ -6,4 +6,6 @@ public interface inputHeaderInterface {};
 
     public enum inputHeaderEnum { one, two };
 
+    public @interface inputHeaderAnnotation {};
+
 }