001: // You can redistribute this software and/or modify it under the terms of
002: // the Ozone Library License version 1 published by ozone-db.org.
003: //
004: // The original code and portions created by SMB are
005: // Copyright (C) 1997-2000 by SMB GmbH. All rights reserved.
006: //
007: // $Id: OzoneTestRunner.java,v 1.2 2001/12/30 16:57:49 per_nyfelt Exp $
008:
009: package org.ozoneDB.test;
010:
011: import java.io.File;
012: import java.io.PrintWriter;
013:
014: import junit.framework.AssertionFailedError;
015: import junit.framework.Test;
016: import junit.framework.TestResult;
017: import junit.runner.BaseTestRunner;
018: import junit.runner.TestSuiteLoader;
019: import org.apache.log4j.BasicConfigurator;
020: import org.apache.log4j.Category;
021: import org.apache.log4j.Priority;
022: import org.ozoneDB.ExternalDatabase;
023: import org.ozoneDB.Setup;
024: import org.ozoneDB.tools.Install;
025:
026: /**
027: * OzoneTestRunner is the JUnit test runner for Ozone
028: * environment.
029: *
030: * @author <a href="http://www.softwarebuero.de/">SMB</a>
031: * @author <a href="mailto:david@d11e.com">David Li</a>
032: * @version $Revision: 1.2 $Date: 2001/12/30 16:57:49 $
033: */
034: public class OzoneTestRunner extends BaseTestRunner {
035:
036: /**
037: * log4j logger
038: */
039: private static Category fLog = Category
040: .getInstance(OzoneTestRunner.class);
041:
042: /**
043: * Print the usage of the program
044: */
045: private static void printUsage() {
046: System.out
047: .println("usage: ojvm "
048: + OzoneTestRunner.class.getName()
049: + " {-local | -remote [-host=HOST] [-port=PORT]} [-debug]"
050: + " TestCase");
051: System.exit(1);
052: }
053:
054: /**
055: * The database
056: */
057: private ExternalDatabase fDatabase;
058:
059: /**
060: * whether to use remote database
061: */
062: private boolean fIsRemote = false;
063:
064: /**
065: * whether to use local database
066: */
067: private boolean fIsLocal = false;
068:
069: /**
070: * whether to do debugging print out.
071: */
072: private boolean fDebug;
073:
074: /**
075: * database server hostname
076: */
077: private String fDBHost = "localhost";
078:
079: /**
080: * database server port
081: */
082: private int fDBPort = 3333;
083:
084: /**
085: * default dir to create test database
086: */
087: private File fLocalTestDir = new File(System
088: .getProperty("java.io.tmpdir")
089: + File.separator + "OzoneTest");
090:
091: /**
092: * name of the test suite to run
093: */
094: private String fTestSuite;
095:
096: /**
097: * Constructor
098: */
099: private OzoneTestRunner(String[] args) {
100: for (int i = 0; i < args.length; i++) {
101: if (args[i].equals("-local")) {
102: fIsLocal = true;
103: } else if (args[i].equals("-remote")) {
104: fIsRemote = true;
105: } else if (args[i].startsWith("-host=")) {
106: fDBHost = args[i].substring(6);
107: } else if (args[i].startsWith("-port=")) {
108: fDBPort = Integer.parseInt(args[i].substring(6));
109: } else if (args[i].equals("-debug")) {
110: fDebug = true;
111: Category.getRoot().setPriority(Priority.DEBUG);
112: } else {
113: fTestSuite = args[i];
114: }
115: }
116: }
117:
118: /**
119: * initialize the database
120: * @param dbURL the database url
121: */
122: private void initDB(String dbURL) throws Exception {
123: fLog.debug("initDB(): open database at '" + dbURL + "'");
124: fDatabase = ExternalDatabase.openDatabase(dbURL);
125: fDatabase.reloadClasses();
126: }
127:
128: /**
129: * Initialize a remote database.
130: * @param host hostname of the database machine
131: * @param port port number of the db server
132: */
133: private void initRemoteDB(String host, int port) throws Exception {
134: initDB("ozonedb:remote://" + fDBHost + ":" + fDBPort);
135: }
136:
137: /**
138: * The test class loader for the runner.
139: */
140: private static TestSuiteLoader fTestSuiteLoader = new OzoneTestSuiteLoader();
141:
142: /**
143: * overrideing this method from BaseTestRunner is necessary
144: * because of the default behavior of getLoader creates new
145: * classloader everytime it's called. This cause problem with
146: * Ozone.
147: */
148: public TestSuiteLoader getLoader() {
149: fLog.debug("getLoader ()");
150: return fTestSuiteLoader;
151: }
152:
153: /**
154: * initialize a local database
155: * if one doesn't exists, create it.
156: */
157: private void initLocalDB() throws Exception {
158: fLocalTestDir.mkdirs();
159: File dbDir = File.createTempFile("OZT", null, fLocalTestDir);
160: Setup defaults = new Setup(null);
161: defaults.addProperties(System.getProperties(), "ozoneDB.");
162: Install.createDB(dbDir.getPath() + ".dir", defaults,
163: new PrintWriter(System.out, true));
164:
165: initDB("ozonedb:local://" + dbDir.getPath() + ".dir");
166: }
167:
168: /**
169: * set up the database
170: */
171: private void setUpDatabase() throws Exception {
172: fLog.info("Test: remote=" + fIsRemote + " local=" + fIsLocal);
173: if (fIsLocal)
174: initLocalDB();
175: else if (fIsRemote)
176: initRemoteDB(fDBHost, fDBPort);
177: }
178:
179: /**
180: * run the test suite
181: */
182: private TestResult doRun() throws Exception {
183: TestResult result = new TestResult();
184: Test suite = getTest(fTestSuite);
185: result.addListener(this );
186: try {
187: setUpDatabase();
188: suite.run(result);
189: } finally {
190: if (fDatabase != null)
191: fDatabase.close();
192: }
193: return result;
194: }
195:
196: /**
197: * main
198: */
199: public static void main(String[] args) {
200:
201: BasicConfigurator.configure();
202: Category.getRoot().setPriority(Priority.INFO);
203:
204: if (args.length == 0) {
205: printUsage();
206: }
207:
208: try {
209: OzoneTestRunner runner = new OzoneTestRunner(args);
210: TestResult result = runner.doRun();
211: fLog.info("run: " + result.runCount() + " error: "
212: + result.errorCount() + " failure: "
213: + result.failureCount());
214: } catch (Exception e) {
215: fLog.error("run fails", e);
216: }
217: }
218:
219: /*
220: * junit.runner.BaseTestRunner methods
221: */
222: protected void runFailed(String message) {
223: }
224:
225: /*
226: * junit.framework.TestListener methods
227: */
228: public void addError(Test test, Throwable t) {
229: fLog.info("addError(): " + test, t);
230: }
231:
232: public void addFailure(Test test, AssertionFailedError t) {
233: fLog.info("addFailure(): " + test, t);
234: }
235:
236: public void endTest(Test test) {
237: fLog.debug("endTest(): " + test);
238: }
239:
240: public void startTest(Test test) {
241: fLog.info("startTest(): " + test);
242: if (test instanceof OzoneTestCase) {
243: ((OzoneTestCase) test).setDB(fDatabase);
244: }
245: }
246: }
|