001: /*
002: * $Id: FindAllTestsSuite.java 1924 2005-03-02 08:23:38Z jstrachan $
003: *
004: * Copyright 2003 (C) James Strachan and Bob Mcwhirter. All Rights Reserved.
005: *
006: * Redistribution and use of this software and associated documentation
007: * ("Software"), with or without modification, are permitted provided that the
008: * following conditions are met:
009: * 1. Redistributions of source code must retain copyright statements and
010: * notices. Redistributions must also contain a copy of this document.
011: * 2. Redistributions in binary form must reproduce the above copyright
012: * notice, this list of conditions and the following disclaimer in the
013: * documentation and/or other materials provided with the distribution.
014: * 3. The name "groovy" must not be used to endorse or promote products
015: * derived from this Software without prior written permission of The Codehaus.
016: * For written permission, please contact info@codehaus.org.
017: * 4. Products derived from this Software may not be called "groovy" nor may
018: * "groovy" appear in their names without prior written permission of The
019: * Codehaus. "groovy" is a registered trademark of The Codehaus.
020: * 5. Due credit should be given to The Codehaus - http://groovy.codehaus.org/
021: *
022: * THIS SOFTWARE IS PROVIDED BY THE CODEHAUS AND CONTRIBUTORS ``AS IS'' AND ANY
023: * EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
024: * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
025: * DISCLAIMED. IN NO EVENT SHALL THE CODEHAUS OR ITS CONTRIBUTORS BE LIABLE FOR
026: * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
027: * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
028: * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
029: * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
030: * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
031: * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH
032: * DAMAGE.
033: *
034: */
035: package org.codehaus.groovy.tools;
036:
037: import junit.framework.Test;
038: import junit.framework.TestSuite;
039: import junit.textui.TestRunner;
040:
041: import java.io.File;
042: import java.util.ArrayList;
043: import java.util.Iterator;
044: import java.util.List;
045:
046: /**
047: * A TestSuite which will run a Groovy unit test case inside any Java IDE
048: * either as a unit test case or as an application.
049: * <p/>
050: * You can specify the GroovyUnitTest to run by running this class as an appplication
051: * and specifying the script to run on the command line.
052: * <p/>
053: * <code>
054: * java groovy.util.GroovyTestSuite src/test/Foo.groovy
055: * </code>
056: * <p/>
057: * Or to run the test suite as a unit test suite in an IDE you can use
058: * the 'test' system property to define the test script to run.
059: * e.g. pass this into the JVM when the unit test plugin runs...
060: * <p/>
061: * <code>
062: * -Dtest=src/test/Foo.groovy
063: * </code>
064: *
065: * @author <a href="mailto:james@coredevelopers.net">James Strachan</a>
066: * @version $Revision: 1924 $
067: */
068: public class FindAllTestsSuite extends TestSuite {
069:
070: protected static String testDirectory = "target/test-classes";
071:
072: public static void main(String[] args) {
073: TestRunner.run(suite());
074: }
075:
076: public static Test suite() {
077: FindAllTestsSuite suite = new FindAllTestsSuite();
078: try {
079: suite.loadTestSuite();
080: } catch (Exception e) {
081: throw new RuntimeException(
082: "Could not create the test suite: " + e, e);
083: }
084: return suite;
085: }
086:
087: public void loadTestSuite() throws Exception {
088: recurseDirectory(new File(testDirectory));
089: }
090:
091: protected void recurseDirectory(File dir) throws Exception {
092: File[] files = dir.listFiles();
093: List traverseList = new ArrayList();
094: for (int i = 0; i < files.length; i++) {
095: File file = files[i];
096: if (file.isDirectory()) {
097: traverseList.add(file);
098: } else {
099: String name = file.getName();
100: if (name.endsWith("Test.class")
101: || name.endsWith("Bug.class")) {
102: addTest(file);
103: }
104: }
105: }
106: for (Iterator iter = traverseList.iterator(); iter.hasNext();) {
107: recurseDirectory((File) iter.next());
108: }
109: }
110:
111: protected void addTest(File file) throws Exception {
112: String name = file.getPath();
113:
114: name = name.substring(testDirectory.length() + 1, name.length()
115: - ".class".length());
116: name = name.replace(File.separatorChar, '.');
117:
118: //System.out.println("Found: " + name);
119: Class type = loadClass(name);
120: addTestSuite(type);
121: }
122:
123: protected Class loadClass(String name)
124: throws ClassNotFoundException {
125: try {
126: return Thread.currentThread().getContextClassLoader()
127: .loadClass(name);
128: } catch (ClassNotFoundException e) {
129: try {
130: return getClass().getClassLoader().loadClass(name);
131: } catch (ClassNotFoundException e1) {
132: return Class.forName(name);
133: }
134: }
135: }
136: }
|