Aman - Unused step inspection is now working for groovy code as well.

kumaraman21 · Dec 18, 2011 · d2ed38a · d2ed38a
1 parent 72dbebc
commit d2ed38a
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/com/github/kumaraman21/intellijbehave/codeInspector/UnusedStepsInspection.java b/src/com/github/kumaraman21/intellijbehave/codeInspector/UnusedStepsInspection.java
@@ -80,9 +80,14 @@ public PsiElementVisitor buildVisitor(@NotNull final ProblemsHolder holder, bool
     return new JavaElementVisitor() {
 
       @Override
-      public void visitAnnotation(PsiAnnotation annotation) {
-        super.visitAnnotation(annotation);
+      public void visitElement(PsiElement element) {
+        super.visitElement(element);
 
+        if(! (element instanceof PsiAnnotation)) {
+          return;
+        }
+
+        PsiAnnotation annotation = (PsiAnnotation)element;
         if(! JBEHAVE_ANNOTATIONS.contains(annotation.getQualifiedName())) {
           return;
         }