private static final String START_JUNIT_MSG = "Testing in progress. Please wait ...\n";

private static final String JUNIT_FINISHED_MSG = "All tests completed successfully.\n";

private static final String NO_TESTS_MSG = "";

private static final SimpleAttributeSet OUT_OF_SYNC_ATTRIBUTES = _getOutOfSyncAttributes();

private static final SimpleAttributeSet _getOutOfSyncAttributes() {

SimpleAttributeSet s = new SimpleAttributeSet();

s.addAttribute(StyleConstants.Foreground, Color.red.darker());

s.addAttribute(StyleConstants.Bold, Boolean.TRUE);

return s;

}

private static final SimpleAttributeSet TEST_PASS_ATTRIBUTES = _getTestPassAttributes();

private static final SimpleAttributeSet _getTestPassAttributes() {

SimpleAttributeSet s = new SimpleAttributeSet();

s.addAttribute(StyleConstants.Foreground, Color.green.darker());

return s;

}

private static final SimpleAttributeSet TEST_FAIL_ATTRIBUTES = _getTestFailAttributes();

private static final SimpleAttributeSet _getTestFailAttributes() {

SimpleAttributeSet s = new SimpleAttributeSet();

s.addAttribute(StyleConstants.Foreground, Color.red);

return s;

}

private static final String TEST_OUT_OF_SYNC =

"The documents being tested have been modified and should be recompiled!\n";

protected volatile JUnitErrorListPane _errorListPane;

private final MainFrame _mainFrame; // only used in assert statements

private volatile int _testCount;

private volatile boolean _testsSuccessful;

private final JUnitProgressBar _progressBar;

private Action _showStackTraceAction = new AbstractAction("Show Stack Trace") {

public void actionPerformed(ActionEvent ae) {

if (_error != null) {

_displayStackTrace(_error);

}

}

};

private volatile JButton _showStackTraceButton;

/** The currently selected error. */

private volatile JUnitError _error = null;

private volatile Window _stackFrame = null;

private volatile JTextArea _stackTextArea; // not final because it is rebound in _setupStackTraceFrame()

private final JLabel _errorLabel = new JLabel();

private final JLabel _testLabel = new JLabel();

private final JLabel _fileLabel = new JLabel();

/** Constructor.

public JUnitPanel(SingleDisplayModel model, MainFrame frame) {

super(model, frame, "Test Output", "Test Progress");

_mainFrame = frame;

_testCount = 0;

_testsSuccessful = true;

_progressBar = new JUnitProgressBar();

_progressBar.setUI(new javax.swing.plaf.basic.BasicProgressBarUI());

_showStackTraceButton = new JButton(_showStackTraceAction);

customPanel.add(_progressBar, BorderLayout.NORTH);

customPanel.add(_showStackTraceButton, BorderLayout.SOUTH);

_errorListPane = new JUnitErrorListPane();

setErrorListPane(_errorListPane);

}

/** Returns the JUnitErrorListPane that this panel manages. */

public JUnitErrorListPane getErrorListPane() { return _errorListPane; }

protected JUnitErrorModel getErrorModel() { return getModel().getJUnitModel().getJUnitErrorModel(); }

/** Updates all document styles with the attributes contained in newSet.

protected void _updateStyles(AttributeSet newSet) {

assert ! _mainFrame.isVisible() || EventQueue.isDispatchThread();

super._updateStyles(newSet);

OUT_OF_SYNC_ATTRIBUTES.addAttributes(newSet);

StyleConstants.setBold(OUT_OF_SYNC_ATTRIBUTES, true); // should always be bold

TEST_PASS_ATTRIBUTES.addAttributes(newSet);

TEST_FAIL_ATTRIBUTES.addAttributes(newSet);

}

/** called when work begins */

public void setJUnitInProgress() {

assert ! _mainFrame.isVisible() || EventQueue.isDispatchThread();

_errorListPane.setJUnitInProgress();

}

/** Closes this panel and resets the corresponding model. */

@Override

protected void _close() {

super._close();

getModel().getJUnitModel().resetJUnitErrors();

reset();

}

/** Reset the errors to the current error information. */

public void reset() {

assert ! _mainFrame.isVisible() || EventQueue.isDispatchThread();

JUnitErrorModel junitErrorModel = getModel().getJUnitModel().getJUnitErrorModel();

boolean testsHaveRun = false;

if (junitErrorModel != null) {

_numErrors = junitErrorModel.getNumErrors();

testsHaveRun = junitErrorModel.haveTestsRun();

}

else _numErrors = 0;

_errorListPane.updateListPane(testsHaveRun); //changed!!

repaint();

}

/** Resets the progress bar to start counting the given number of tests.

public void progressReset(int numTests) {

assert ! _mainFrame.isVisible() || EventQueue.isDispatchThread();

_progressBar.reset();

_progressBar.start(numTests);

_testsSuccessful = true;

_testCount = 0;

}

/** Steps the progress bar forward by one test.

public void progressStep(boolean successful) {

assert ! _mainFrame.isVisible() || EventQueue.isDispatchThread();

_testCount++;

_testsSuccessful &= successful;

_progressBar.step(_testCount, _testsSuccessful);

}

public void testStarted(String className, String testName) { }

private void _displayStackTrace (JUnitError e) {

assert ! _mainFrame.isVisible() || EventQueue.isDispatchThread();

_errorLabel.setText((e.isWarning() ? "Error: " : "Failure: ") +

e.message());

_fileLabel.setText("File: " + (new File(e.fileName())).getName());

if (!e.testName().equals("")) {

_testLabel.setText("Test: " + e.testName());

}

else {

_testLabel.setText("");

}

_stackTextArea.setText(e.toString());

_stackTextArea.setCaretPosition(0);

_frame.setPopupLoc(_stackFrame);

_stackFrame.setVisible(true);

}

/** A pane to show JUnit errors. It acts like a listbox (clicking selects an item) but items can each wrap, etc. */

public class JUnitErrorListPane extends ErrorPanel.ErrorListPane {

private JPopupMenu _popMenu;

private String _runningTestName;

private boolean _warnedOutOfSync;

private static final String JUNIT_WARNING = "junit.framework.TestSuite$1.warning";

/** Maps any test names in the currently running suite to the position that they appear in the list pane. */

private final HashMap<String, Position> _runningTestNamePositions;

/** Constructs the JUnitErrorListPane. */

public JUnitErrorListPane() {

removeMouseListener(defaultMouseListener);

_popMenu = new JPopupMenu();

_popMenu.add(_showStackTraceAction);

_error = null;

_setupStackTraceFrame();

addMouseListener(new PopupAdapter());

_runningTestName = null;

_runningTestNamePositions = new HashMap<String, Position>();

_showStackTraceButton.setEnabled(false);

}

private String _getTestFromName(String name) {

int paren = name.indexOf('(');

if (paren > 0) return name.substring(0, paren);

else throw new IllegalArgumentException("Name does not contain any parens: " + name);

}

private String _getClassFromName(String name) {

int paren = name.indexOf('(');

if ((paren > -1) && (paren < name.length())) return name.substring(paren + 1, name.length() - 1);

else throw new IllegalArgumentException("Name does not contain any parens: " + name);

}

/** Provides the the name of the test being run to the JUnitPanel.

public void testStarted(String name) {

assert ! _mainFrame.isVisible() || EventQueue.isDispatchThread();

if (name.indexOf('(') < 0) return;

String testName = _getTestFromName(name);

String className = _getClassFromName(name);

String fullName = className + "." + testName;

if (fullName.equals(JUNIT_WARNING)) return;

ErrorDocument doc = getErrorDocument();

// Converted this GUI operation to a Runnable and use invokeLater

Utilities.invokeLater(new Runnable() {

public void run() {

try {

int len = doc.getLength();

// Insert the classname if it has changed

if (! className.equals(_runningTestName)) {

_runningTestName = className;

doc.insertString(len, " " + className + "\n", NORMAL_ATTRIBUTES);

len = doc.getLength();

}

// Insert the test name, remembering its position

doc.insertString(len, " ", NORMAL_ATTRIBUTES);

len = doc.getLength();

doc.insertString(len, testName + "\n", NORMAL_ATTRIBUTES);

Position pos = doc.createPosition(len);

_runningTestNamePositions.put(fullName, pos);

setCaretPosition(len);

}

catch (BadLocationException ble) {

// Inserting at end, shouldn't happen

throw new UnexpectedException(ble);

}

}

});

}

/** Displays the results of a test that has finished in the JUnitPanel.

public void testEnded(String name, boolean wasSuccessful, boolean causedError) {

assert ! _mainFrame.isVisible() || EventQueue.isDispatchThread();

if (name.indexOf('(')<0) return;

String testName = _getTestFromName(name);

String fullName = _getClassFromName(name) + "." + testName;

if (fullName.equals(JUNIT_WARNING)) return;

// TODO: convert this GUI operation to a Runnable and use invokeLater

ErrorDocument doc = getErrorDocument();

Utilities.invokeLater(new Runnable() {

public void run() {

Position namePos = _runningTestNamePositions.get(fullName);

AttributeSet set;

if (! wasSuccessful || causedError) set = TEST_FAIL_ATTRIBUTES;

else set = TEST_PASS_ATTRIBUTES;

if (namePos != null) {

int index = namePos.getOffset();

int length = testName.length();

doc.setCharacterAttributes(index, length, set, false);

}

}

});

}

/** Puts the error pane into "junit in progress" state. Only runs in event thread. */

public void setJUnitInProgress() {

assert EventQueue.isDispatchThread();

_errorListPositions = new Position[0];

progressReset(0);

_runningTestNamePositions.clear();

_runningTestName = null;

_warnedOutOfSync = false;

ErrorDocument doc = new ErrorDocument(getErrorDocumentTitle());

doc.append(START_JUNIT_MSG, BOLD_ATTRIBUTES);

setDocument(doc);

selectNothing();

}

/** Used to show that testing was unsuccessful. */

protected void _updateWithErrors() throws BadLocationException {

assert ! _mainFrame.isVisible() || EventQueue.isDispatchThread();

//DefaultStyledDocument doc = new DefaultStyledDocument();

ErrorDocument doc = getErrorDocument();

_updateWithErrors("test", "failed", doc);

}

/** Gets the message indicating the number of errors and warnings.*/

protected String _getNumErrorsMessage(String failureName, String failureMeaning) {

StringBuilder numErrMsg;

/** Used for display purposes only */

int numCompErrs = getErrorModel().getNumCompilerErrors();

int numWarnings = getErrorModel().getNumWarnings();

if (! getErrorModel().hasOnlyWarnings()) {

numErrMsg = new StringBuilder(numCompErrs + " " + failureName); //failureName = error or test (for compilation and JUnit testing respectively)

if (numCompErrs > 1) numErrMsg.append("s");

numErrMsg.append(" " + failureMeaning);

if (numWarnings > 0) numErrMsg.append(" and " + numWarnings + " warning");

}

else numErrMsg = new StringBuilder(numWarnings + " warning");

if (numWarnings > 1) numErrMsg.append("s");

if (numWarnings > 0) numErrMsg.append(" found");

numErrMsg.append(":\n");

return numErrMsg.toString();

}

protected void _updateWithErrors(String failureName, String failureMeaning, ErrorDocument doc)

throws BadLocationException {

// Print how many errors

_replaceInProgressText(_getNumErrorsMessage(failureName, failureMeaning));

_insertErrors(doc);

// Select the first error

switchToError(0);

}

/** Replaces the "Testing in progress..." text with the given message. Only runs in event thread.

public void _replaceInProgressText(String msg) throws BadLocationException {

assert ! _mainFrame.isVisible() || EventQueue.isDispatchThread();

int start = 0;

if (_warnedOutOfSync) { start = TEST_OUT_OF_SYNC.length(); }

int len = START_JUNIT_MSG.length();

ErrorDocument doc = getErrorDocument();

if (doc.getLength() >= len + start) {

doc.remove(start, len);

doc.insertString(start, msg, BOLD_ATTRIBUTES);

}

}

/** Returns the string to identify a warning. In JUnit, warnings (the odd case) indicate errors/exceptions.

protected String _getWarningText() { return "Error: "; }

/** Returns the string to identify an error. In JUnit, errors (the normal case) indicate TestFailures. */

protected String _getErrorText() { return "Failure: "; }

/** Updates the list pane with no errors. */

protected void _updateNoErrors(boolean haveTestsRun) throws BadLocationException {

//DefaultStyledDocument doc = new DefaultStyledDocument();

_replaceInProgressText(haveTestsRun ? JUNIT_FINISHED_MSG : NO_TESTS_MSG);

selectNothing();

setCaretPosition(0);

}

private void _setupStackTraceFrame() {

//DrJava.consoleOut().println("Stack Trace for Error: \n" + e.stackTrace());

assert ! _mainFrame.isVisible() || EventQueue.isDispatchThread();

JDialog _dialog = new JDialog(_frame,"JUnit Error Stack Trace",false);

_stackFrame = _dialog;

_stackTextArea = new JTextArea();

_stackTextArea.setEditable(false);

_stackTextArea.setLineWrap(false);

JScrollPane scroll = new BorderlessScrollPane(_stackTextArea,

ScrollPaneConstants.VERTICAL_SCROLLBAR_AS_NEEDED,

ScrollPaneConstants.HORIZONTAL_SCROLLBAR_AS_NEEDED);

ActionListener closeListener = new ActionListener() {

public void actionPerformed(ActionEvent e) {

_stackFrame.setVisible(false);

}

};

JButton closeButton = new JButton("Close");

closeButton.addActionListener(closeListener);

JPanel closePanel = new JPanel(new BorderLayout());

closePanel.setBorder(new EmptyBorder(5,5,0,0));

closePanel.add(closeButton, BorderLayout.EAST);

JPanel cp = new JPanel(new BorderLayout());

_dialog.setContentPane(cp);

cp.setBorder(new EmptyBorder(5,5,5,5));

cp.add(scroll, BorderLayout.CENTER);

cp.add(closePanel, BorderLayout.SOUTH);

JPanel topPanel = new JPanel(new GridLayout(0,1,0,5));

topPanel.setBorder(new EmptyBorder(0,0,5,0));

topPanel.add(_fileLabel);

topPanel.add(_testLabel);

topPanel.add(_errorLabel);

cp.add(topPanel, BorderLayout.NORTH);

_dialog.setSize(600, 500);

// initial location is relative to parent (MainFrame)

}

/** Overrides selectItem in ErrorListPane to update the current _error selected

public void selectItem(DJError error) {

assert ! _mainFrame.isVisible() || EventQueue.isDispatchThread();

super.selectItem(error);

_error = (JUnitError) error;

_showStackTraceButton.setEnabled(true);

}

/** Overrides _removeListHighlight in ErrorListPane to disable the _showStackTraceButton. */

protected void _removeListHighlight() {

assert ! _mainFrame.isVisible() || EventQueue.isDispatchThread();

super._removeListHighlight();

_showStackTraceButton.setEnabled(false);

}

private class PopupAdapter extends RightClickMouseAdapter {

/** Show popup if the click is on an error.

public void mousePressed(MouseEvent e) {

if (_selectError(e)) {

super.mousePressed(e);

}

}

/** Show popup if the click is on an error.

public void mouseReleased(MouseEvent e) {

if (_selectError(e)) super.mouseReleased(e);

}

/** Select the error at the given mouse event.

private boolean _selectError(MouseEvent e) {

assert ! _mainFrame.isVisible() || EventQueue.isDispatchThread();

//TODO: get rid of cast in the next line, if possible

_error = (JUnitError)_errorAtPoint(e.getPoint());

if (_isEmptySelection() && _error != null) {

_errorListPane.switchToError(_error);

return true;

}

else {

selectNothing();

return false;

}

}

/** Shows the popup menu for this mouse adapter.

protected void _popupAction(MouseEvent e) {

assert ! _mainFrame.isVisible() || EventQueue.isDispatchThread();

_popMenu.show(e.getComponent(), e.getX(), e.getY());

}

}

public String getErrorDocumentTitle() { return "Javadoc Errors"; }

}

/** A progress bar showing the status of JUnit tests.

static class JUnitProgressBar extends JProgressBar {

private boolean _hasError = false;

public JUnitProgressBar() {

super();

setForeground(getStatusColor());

}

private Color getStatusColor() {

assert EventQueue.isDispatchThread();

if (_hasError) {

return Color.red;

}

else {

return Color.green;

}

}

public void reset() {

assert EventQueue.isDispatchThread();

_hasError = false;

setForeground(getStatusColor());

setValue(0);

}

public void start(int total) {

assert EventQueue.isDispatchThread();

setMaximum(total);

reset();

}

public void step(int value, boolean successful) {

assert EventQueue.isDispatchThread();

setValue(value);

if (!_hasError && !successful) {

_hasError= true;

setForeground(getStatusColor());

}

}

}