FlexUnitMethodInspectionBase.java

package com.intellij.lang.javascript.flex.flexunit.inspections;

import com.intellij.codeInspection.LocalInspectionTool;
import com.intellij.codeInspection.ProblemsHolder;
import com.intellij.lang.javascript.flex.FlexBundle;
import com.intellij.lang.javascript.flex.flexunit.FlexUnitSupport;
import com.intellij.lang.javascript.psi.JSFunction;
import com.intellij.lang.javascript.psi.ecmal4.JSClass;
import com.intellij.psi.PsiElementVisitor;
import org.jetbrains.annotations.Nls;
import org.jetbrains.annotations.NotNull;

public abstract class FlexUnitMethodInspectionBase extends LocalInspectionTool {
  @Nls
  @NotNull
  public String getGroupDisplayName() {
    return FlexBundle.message("flexunit.inspections.group");
  }

  public boolean isEnabledByDefault() {
    return true;
  }

  @NotNull
  @Override
  public PsiElementVisitor buildVisitor(@NotNull final ProblemsHolder holder, boolean isOnTheFly) {
    return new FlexUnitInspectionVisitor() {
      @Override
      public void visitJSFunctionDeclaration(JSFunction function) {
        if (!(function.getParent() instanceof JSClass)) return;
        FlexUnitSupport support = getFlexUnitSupport(function);
        if (support == null || !support.isPotentialTestMethod(function)) return;
        visitPotentialTestMethod(function, holder, support);
      }
    };
  }

  protected abstract void visitPotentialTestMethod(JSFunction function, ProblemsHolder holder, FlexUnitSupport support);

}