treetestcase.java

基于Junit的 功能和单元测试的的测试工具。只支持Swing.

package org.uispec4j;

import org.uispec4j.utils.DummyTreeCellRenderer;
import org.uispec4j.utils.UnitTestCase;
import org.uispec4j.xml.EventLogger;

import javax.swing.*;
import javax.swing.tree.DefaultMutableTreeNode;
import java.awt.*;

public abstract class TreeTestCase extends UnitTestCase {
  protected JTree jTree;
  protected Tree tree;
  protected DefaultMutableTreeNode rootNode;
  protected DefaultMutableTreeNode child1Node;
  protected DefaultMutableTreeNode child1_1Node;
  protected DefaultMutableTreeNode child2Node;
  protected DummyTreeCellRenderer.UserObject root =
    new DummyTreeCellRenderer.UserObject("root");
  protected DummyTreeCellRenderer.UserObject child1 =
    new DummyTreeCellRenderer.UserObject("child1");
  protected DummyTreeCellRenderer.UserObject child1_1 =
    new DummyTreeCellRenderer.UserObject("child1_1");
  protected DummyTreeCellRenderer.UserObject child2 =
    new DummyTreeCellRenderer.UserObject("child2");

  protected void setUp() throws Exception {
    super.setUp();
    rootNode = new DefaultMutableTreeNode(root);
    child1Node = new DefaultMutableTreeNode(child1);
    child1_1Node = new DefaultMutableTreeNode(child1_1);
    child2Node = new DefaultMutableTreeNode(child2);

    rootNode.add(child1Node);
    child1Node.add(child1_1Node);
    rootNode.add(child2Node);
    initTree();
  }

  protected void initTree() {
    jTree = new JTree(rootNode);
    jTree.setCellRenderer(new DummyTreeCellRenderer(jTree));
    jTree.setName("myTree");
    tree = new Tree(jTree);
  }

  protected static class DummyTreeCellValueConverter extends EventLogger implements TreeCellValueConverter {
    private String boldPattern;
    private String redFontPattern;

    public String getValue(Component renderedComponent, Object modelObject) {
      log("getValue")
        .add("component", renderedComponent.getClass().getName())
        .add("object", modelObject.toString());
      return '_' + modelObject.toString() + '_';
    }

    public void setRedFontPattern(String redFontPattern) {
      this.redFontPattern = redFontPattern;
    }

    public Color getForeground(Component renderedComponent, Object modelObject) {
      if ((redFontPattern != null) &&
          (modelObject.toString().indexOf(redFontPattern) >= 0)) {
        return Color.RED;
      }

      return Color.BLACK;
    }

    public void setBoldPattern(String boldPattern) {
      this.boldPattern = boldPattern;
    }

    public boolean isBold(Component renderedComponent, Object modelObject) {
      if (boldPattern == null) {
        return false;
      }
      return modelObject.toString().indexOf(boldPattern) >= 0;
    }
  }
}