plugins/testng/src/com/theoryinpractice/testng/model/TestTreeBuilder.java

   1 /*
   2  * Copyright 2000-2009 JetBrains s.r.o.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  * http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16 package com.theoryinpractice.testng.model;
  17
  18 import com.intellij.execution.testframework.ui.AbstractTestTreeBuilder;
  19 import com.intellij.ide.util.treeView.AbstractTreeStructure;
  20 import com.intellij.ide.util.treeView.IndexComparator;
  21 import com.intellij.ide.util.treeView.NodeDescriptor;
  22
  23 import javax.swing.*;
  24 import javax.swing.tree.DefaultMutableTreeNode;
  25 import javax.swing.tree.DefaultTreeModel;
  26
  27 /**
  28  * @author Hani Suleiman Date: Jul 28, 2005 Time: 10:49:36 PM
  29  */
  30 public class TestTreeBuilder extends AbstractTestTreeBuilder
  31 {
  32     public TestTreeBuilder(JTree tree, AbstractTreeStructure structure) {
  33         super(tree, new DefaultTreeModel(new DefaultMutableTreeNode(structure.getRootElement())), structure, IndexComparator.INSTANCE);
  34         initRootNode();
  35     }
  36
  37     @Override
  38     protected boolean isSmartExpand() {
  39         return false;
  40     }
  41
  42     @Override
  43     protected boolean isAlwaysShowPlus(NodeDescriptor descriptor) {
  44         return false;
  45     }
  46
  47     @Override
  48     protected boolean isAutoExpandNode(NodeDescriptor descriptor) {
  49         return descriptor.getElement() == getTreeStructure().getRootElement();
  50     }
  51
  52     public void addItem(TestProxy parent, TestProxy proxy) {
  53         parent.addResult(proxy);
  54         DefaultMutableTreeNode parentNode = getNodeForElement(parent);
  55         if (parentNode != null)
  56             updateSubtree(parentNode);
  57     }
  58
  59     public TestProxy getRoot() {
  60         return (TestProxy) getTreeStructure().getRootElement();
  61     }
  62 }