BaseTestTreeNode.java

package jetbrains.mps.baseLanguage.unitTest.execution.tool;

/*Generated by MPS */

import jetbrains.mps.ide.ui.tree.MPSTreeNode;
import jetbrains.mps.project.Project;
import org.jetbrains.annotations.NotNull;
import javax.swing.Icon;

public abstract class BaseTestTreeNode extends MPSTreeNode {
  protected final Project myProject;
  private TestState myState = TestState.NOT_RAN;
  public BaseTestTreeNode(@NotNull Project project) {
    super();
    myProject = project;
  }
  @Override
  protected void doUpdatePresentation() {
    Icon icon = myState.getIcon();
    if (icon != null) {
      setIcon(icon);
    }
  }
  public void setState(TestState state) {
    myState = state;
    myProject.getModelAccess().runReadInEDT(new Runnable() {
      public void run() {
        renewPresentation();
      }
    });
  }
  public TestState getState() {
    return myState;
  }
}