01: /*
02: * JBoss, Home of Professional Open Source.
03: * Copyright 2006, Red Hat Middleware LLC, and individual contributors
04: * as indicated by the @author tags. See the copyright.txt file in the
05: * distribution for a full listing of individual contributors.
06: *
07: * This is free software; you can redistribute it and/or modify it
08: * under the terms of the GNU Lesser General Public License as
09: * published by the Free Software Foundation; either version 2.1 of
10: * the License, or (at your option) any later version.
11: *
12: * This software is distributed in the hope that it will be useful,
13: * but WITHOUT ANY WARRANTY; without even the implied warranty of
14: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15: * Lesser General Public License for more details.
16: *
17: * You should have received a copy of the GNU Lesser General Public
18: * License along with this software; if not, write to the Free
19: * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
20: * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
21: */
22: package org.jboss.test.cmp2.dbschema.util;
23:
24: import org.jboss.test.util.ejb.EJBTestCase;
25:
26: import javax.sql.DataSource;
27: import javax.naming.InitialContext;
28: import javax.naming.NamingException;
29: import java.sql.Connection;
30: import java.sql.SQLException;
31:
32: /**
33: *
34: * @author <a href="mailto:alex@jboss.org">Alex Loubyansky</a>
35: */
36: public class AbstractDBSchemaTest extends EJBTestCase {
37: protected String datasourceName = "java:/DefaultDS";
38: protected DataSource datasource;
39:
40: public AbstractDBSchemaTest(String s) {
41: super (s);
42: }
43:
44: protected Connection getConnection() throws NamingException,
45: SQLException {
46: return getDataSource().getConnection();
47: }
48:
49: protected DataSource getDataSource() throws NamingException {
50: if (datasource == null) {
51: datasource = getDataSource(datasourceName);
52: }
53: return datasource;
54: }
55:
56: protected DataSource getDataSource(String datasource)
57: throws NamingException {
58: InitialContext ic = null;
59: try {
60: ic = new InitialContext();
61: return (DataSource) ic.lookup(datasource);
62: } finally {
63: if (ic != null)
64: try {
65: ic.close();
66: } catch (Exception ignore) {
67: }
68: }
69: }
70: }
|