001    /*
002     * $Id: GroovyTestSuite.java,v 1.9 2004/12/14 00:08:26 spullara Exp $
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 groovy.util;
036    
037    import groovy.lang.GroovyClassLoader;
038    import groovy.lang.Script;
039    import junit.framework.Test;
040    import junit.framework.TestSuite;
041    import junit.textui.TestRunner;
042    import org.codehaus.groovy.runtime.ScriptTestAdapter;
043    
044    import java.io.File;
045    
046    
047    /**
048     * A TestSuite which will run a Groovy unit test case inside any Java IDE
049     * either as a unit test case or as an application.
050     * <p/>
051     * You can specify the GroovyUnitTest to run by running this class as an appplication
052     * and specifying the script to run on the command line.
053     * <p/>
054     * <code>
055     * java groovy.util.GroovyTestSuite src/test/Foo.groovy
056     * </code>
057     * <p/>
058     * Or to run the test suite as a unit test suite in an IDE you can use
059     * the 'test' system property to define the test script to run.
060     * e.g. pass this into the JVM when the unit test plugin runs...
061     * <p/>
062     * <code>
063     * -Dtest=src/test/Foo.groovy
064     * </code>
065     *
066     * @author <a href="mailto:james@coredevelopers.net">James Strachan</a>
067     * @version $Revision: 1.9 $
068     */
069    public class GroovyTestSuite extends TestSuite {
070    
071        protected static String file = null;
072    
073        protected GroovyClassLoader loader = new GroovyClassLoader(GroovyTestSuite.class.getClassLoader());
074    
075        public static void main(String[] args) {
076            if (args.length > 0) {
077                file = args[0];
078            }
079            TestRunner.run(suite());
080        }
081    
082        public static Test suite() {
083            GroovyTestSuite suite = new GroovyTestSuite();
084            try {
085                suite.loadTestSuite();
086            } catch (Exception e) {
087                throw new RuntimeException("Could not create the test suite: " + e, e);
088            }
089            return suite;
090        }
091    
092        public void loadTestSuite() throws Exception {
093            String fileName = System.getProperty("test", file);
094            if (fileName == null) {
095                throw new RuntimeException("No filename given in the 'test' system property so cannot run a Groovy unit test");
096            }
097            System.out.println("Compiling: " + fileName);
098            Class type = compile(fileName);
099            String[] args = {};
100            if (!Test.class.isAssignableFrom(type) && Script.class.isAssignableFrom(type)) {
101                // lets treat the script as a Test
102                addTest(new ScriptTestAdapter(type, args));
103            } else {
104                addTestSuite(type);
105            }
106        }
107    
108        public Class compile(String fileName) throws Exception {
109            return loader.parseClass(new File(fileName));
110        }
111    }