1324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver/* 2324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver [The "BSD licence"] 3324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver Copyright (c) 2007-2008 Leon Jen-Yuan Su 4324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver All rights reserved. 5324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 6324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver Redistribution and use in source and binary forms, with or without 7324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver modification, are permitted provided that the following conditions 8324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver are met: 9324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 1. Redistributions of source code must retain the above copyright 10324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver notice, this list of conditions and the following disclaimer. 11324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 2. Redistributions in binary form must reproduce the above copyright 12324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver notice, this list of conditions and the following disclaimer in the 13324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver documentation and/or other materials provided with the distribution. 14324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 3. The name of the author may not be used to endorse or promote products 15324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver derived from this software without specific prior written permission. 16324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 17324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 18324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 19324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 20324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 21324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 22324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 23324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 24324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 25324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 26324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 27324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver*/ 28324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverpackage org.antlr.gunit; 29324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport java.io.File; 30324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport java.io.IOException; 31324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 32324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverimport org.antlr.runtime.*; 33324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 34324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver/** The main gUnit interpreter entry point. 35324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * Read a gUnit script, run unit tests or generate a junit file. 36324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver */ 37324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverpublic class Interp { 38324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver static String testPackage; 39324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver static boolean genJUnit; 40324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver static String gunitFile; 41324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 42324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver public static void main(String[] args) throws IOException, ClassNotFoundException, RecognitionException { 43324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver /** Pull char from where? */ 44324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver CharStream input = null; 45324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver /** If the input source is a testsuite file, where is it? */ 46324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver String testsuiteDir = System.getProperty("user.dir"); 47324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 48324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver processArgs(args); 49324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 50324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if ( genJUnit ) { 51324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if ( gunitFile!=null ) { 52324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver input = new ANTLRFileStream(gunitFile); 53324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver File f = new File(gunitFile); 54324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver testsuiteDir = getTestsuiteDir(f.getCanonicalPath(), f.getName()); 55324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 56324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver else { 57324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver input = new ANTLRInputStream(System.in); 58324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 59324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver GrammarInfo grammarInfo = parse(input); 60324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver grammarInfo.setTestPackage(testPackage); 61324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver JUnitCodeGen generater = new JUnitCodeGen(grammarInfo, testsuiteDir); 62324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver generater.compile(); 63324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver return; 64324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 65324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 66324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if ( gunitFile!=null ) { 67324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver input = new ANTLRFileStream(gunitFile); 68324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver File f = new File(gunitFile); 69324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver testsuiteDir = getTestsuiteDir(f.getCanonicalPath(), f.getName()); 70324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 71324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver else 72324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver input = new ANTLRInputStream(System.in); 73324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 74324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver gUnitExecutor executer = new gUnitExecutor(parse(input), testsuiteDir); 75324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 76324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver System.out.print(executer.execTest()); // unit test result 77324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 78324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver //return an error code of the number of failures 79324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver System.exit(executer.failures.size() + executer.invalids.size()); 80324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 81324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 82324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver public static void processArgs(String[] args) { 83324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if (args == null || args.length == 0) return; 84324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver for (int i = 0; i < args.length; i++) { 85324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if (args[i].equals("-p")) { 86324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if (i + 1 >= args.length) { 87324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver System.err.println("missing library directory with -lib option; ignoring"); 88324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 89324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver else { 90324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver i++; 91324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver testPackage = args[i]; 92324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 93324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 94324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver else if (args[i].equals("-o")) genJUnit = true; 95324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver else gunitFile = args[i]; // Must be the gunit file 96324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 97324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 98324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 99324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver public static GrammarInfo parse(CharStream input) throws RecognitionException { 100324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver gUnitLexer lexer = new gUnitLexer(input); 101324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver CommonTokenStream tokens = new CommonTokenStream(lexer); 102324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 103324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver GrammarInfo grammarInfo = new GrammarInfo(); 104324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver gUnitParser parser = new gUnitParser(tokens, grammarInfo); 105324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver parser.gUnitDef(); // parse gunit script and save elements to grammarInfo 106324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver return grammarInfo; 107324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 108324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 109324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver public static String getTestsuiteDir(String fullPath, String fileName) { 110324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver return fullPath.substring(0, fullPath.length()-fileName.length()); 111324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 112324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 113324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver} 114