TestSuitePanel.java revision 58a8b0aba2dec5695628a2bf25a3fae42c2c3533 
1package junit.swingui;
2
3import java.awt.BorderLayout;
4import java.awt.Component;
5import java.awt.Dimension;
6import java.util.Vector;
7
8import javax.swing.Icon;
9import javax.swing.JComponent;
10import javax.swing.JPanel;
11import javax.swing.JScrollPane;
12import javax.swing.JTree;
13import javax.swing.SwingUtilities;
14import javax.swing.ToolTipManager;
15import javax.swing.tree.DefaultTreeCellRenderer;
16import javax.swing.tree.TreeModel;
17import javax.swing.tree.TreePath;
18
19import junit.framework.AssertionFailedError;
20import junit.framework.Test;
21import junit.framework.TestListener;
22
23/**
24 * A Panel showing a test suite as a tree.
25 */
26class TestSuitePanel extends JPanel implements TestListener {
27	private JTree fTree;
28	private JScrollPane fScrollTree;
29	private TestTreeModel fModel;
30
31	static class TestTreeCellRenderer extends DefaultTreeCellRenderer {
32		private Icon fErrorIcon;
33		private Icon fOkIcon;
34		private Icon fFailureIcon;
35
36		TestTreeCellRenderer() {
37	    		super();
38	    		loadIcons();
39		}
40
41		void loadIcons() {
42			fErrorIcon= TestRunner.getIconResource(getClass(), "icons/error.gif");
43	    		fOkIcon= TestRunner.getIconResource(getClass(), "icons/ok.gif");
44	    		fFailureIcon= TestRunner.getIconResource(getClass(), "icons/failure.gif");
45		}
46
47		String stripParenthesis(Object o) {
48			String text= o.toString ();
49    			int pos= text.indexOf('(');
50    			if (pos < 1)
51    				return text;
52    			return text.substring (0, pos);
53  		}
54
55		public Component getTreeCellRendererComponent(JTree tree, Object value,
56				boolean sel, boolean expanded, boolean leaf, int row, boolean hasFocus) {
57
58			Component c= super.getTreeCellRendererComponent(tree, value, sel, expanded, leaf, row, hasFocus);
59	  		TreeModel model= tree.getModel();
60	  		if (model instanceof TestTreeModel) {
61				TestTreeModel testModel= (TestTreeModel)model;
62				Test t= (Test)value;
63				String s= "";
64	    		if (testModel.isFailure(t)) {
65	    			if (fFailureIcon != null)
66	    				setIcon(fFailureIcon);
67	    			s= " - Failed";
68	    		}
69	    		else if (testModel.isError(t)) {
70	    			if (fErrorIcon != null)
71	    				setIcon(fErrorIcon);
72	    			s= " - Error";
73	    		}
74	    		else if (testModel.wasRun(t)) {
75	    			if (fOkIcon != null)
76	    				setIcon(fOkIcon);
77	    			s= " - Passed";
78	    		}
79	    		if (c instanceof JComponent)
80	    			((JComponent)c).setToolTipText(getText()+s);
81	  		}
82	  		setText(stripParenthesis(value));
83			return c;
84		}
85	}
86
87	public TestSuitePanel() {
88		super(new BorderLayout());
89		setPreferredSize(new Dimension(300, 100));
90		fTree= new JTree();
91		fTree.setModel(null);
92		fTree.setRowHeight(20);
93		ToolTipManager.sharedInstance().registerComponent(fTree);
94		fTree.putClientProperty("JTree.lineStyle", "Angled");
95		fScrollTree= new JScrollPane(fTree);
96		add(fScrollTree, BorderLayout.CENTER);
97	}
98
99	public void addError(final Test test, final Throwable t) {
100 		fModel.addError(test);
101		fireTestChanged(test, true);
102	}
103
104	public void addFailure(final Test test, final AssertionFailedError t) {
105 		fModel.addFailure(test);
106		fireTestChanged(test, true);
107	}
108
109	/**
110 	 * A test ended.
111 	 */
112 	public void endTest(Test test) {
113		fModel.addRunTest(test);
114	 	fireTestChanged(test, false);
115 	}
116
117	/**
118 	 * A test started.
119 	 */
120 	public void startTest(Test test) {
121 	}
122
123	/**
124	 * Returns the selected test or null if multiple or none is selected
125	 */
126	public Test getSelectedTest() {
127		TreePath[] paths= fTree.getSelectionPaths();
128		if (paths != null && paths.length == 1)
129			return (Test)paths[0].getLastPathComponent();
130		return null;
131	}
132
133	/**
134	 * Returns the Tree
135	 */
136	public JTree getTree() {
137		return fTree;
138	}
139
140	/**
141	 * Shows the test hierarchy starting at the given test
142	 */
143	public void showTestTree(Test root) {
144		fModel= new TestTreeModel(root);
145		fTree.setModel(fModel);
146		fTree.setCellRenderer(new TestTreeCellRenderer());
147	}
148
149	private void fireTestChanged(final Test test, final boolean expand) {
150		SwingUtilities.invokeLater(
151			new Runnable() {
152				public void run() {
153					Vector vpath= new Vector();
154					int index= fModel.findTest(test, (Test)fModel.getRoot(), vpath);
155					if (index >= 0) {
156						Object[] path= new Object[vpath.size()];
157						vpath.copyInto(path);
158						TreePath treePath= new TreePath(path);
159						fModel.fireNodeChanged(treePath, index);
160						if (expand) {
161							Object[] fullPath= new Object[vpath.size()+1];
162							vpath.copyInto(fullPath);
163							fullPath[vpath.size()]= fModel.getChild(treePath.getLastPathComponent(), index);;
164							TreePath fullTreePath= new TreePath(fullPath);
165							fTree.scrollPathToVisible(fullTreePath);
166						}
167					}
168				}
169			}
170		);
171	}
172}