private void terminateStillRunning(@NotNull JUnitRunningModel model) {
   if (model.getRoot() != null) {
     final List<AbstractTestProxy> runningTests =
         TestStateUpdater.RUNNING_LEAF.select(myModel.getRoot().getAllTests());
     for (final AbstractTestProxy runningTest : runningTests) {
       final TestProxy testProxy = (TestProxy) runningTest;
       final TestState terminated = NotFailedState.createTerminated();
       testProxy.setState(terminated);
       TestProxy parent = testProxy.getParent();
       while (parent != null) {
         parent.setState(terminated);
         parent = parent.getParent();
       }
     }
   }
 }
  private TestProxy getDynamicParent(JUnitRunningModel model, final TestProxy currentTest) {
    if (myKnownDynamicParents == null) {
      myKnownDynamicParents = new HashMap<String, TestProxy>();
    }
    final String parentClass = currentTest.getInfo().getComment();
    TestProxy dynamicParent = myKnownDynamicParents.get(parentClass);
    if (dynamicParent == null) {
      final TestProxy root = model.getRoot();
      if (Comparing.strEqual(
          parentClass, StringUtil.getQualifiedName(root.getInfo().getComment(), root.getName()))) {
        dynamicParent = root;
      } else {
        dynamicParent =
            new TestProxy(
                new ClassBasedInfo(DisplayTestInfoExtractor.FOR_CLASS) {
                  {
                    setClassName(parentClass);
                  }

                  @Override
                  public void readFrom(ObjectReader reader) {}
                });
        root.addChild(dynamicParent);
      }
      myKnownDynamicParents.put(parentClass, dynamicParent);
    }
    return dynamicParent;
  }
 public void notifyTestStart(ObjectReader reader) {
   final TestProxy testProxy = reader.readObject();
   final JUnitRunningModel model = getModel();
   if (model != null && testProxy.getParent() == null && model.getRoot() != testProxy) {
     getDynamicParent(model, testProxy).addChild(testProxy);
   }
 }
 public void notifyFinish(ObjectReader reader) {
   myIsTerminated = true;
   synchronized (myCurrentTests) {
     myCurrentTests.clear();
   }
   final JUnitRunningModel model = getModel();
   if (model != null) {
     model.getNotifier().fireRunnerStateChanged(new CompletionEvent(true, reader.readInt()));
     TestStatusListener.notifySuiteFinished(model.getRoot(), model.getProject());
     terminateStillRunning(model);
   }
 }
예제 #5
0
 protected void notifyByBalloon(
     JUnitRunningModel model, boolean started, JUnitConsoleProperties consoleProperties) {
   TestsUIUtil.notifyByBalloon(
       myProject, started, model != null ? model.getRoot() : null, consoleProperties);
 }