KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > org > jboss > test > cmp2 > dbschema > util > AbstractDBSchemaTest


1 /*
2   * JBoss, Home of Professional Open Source
3   * Copyright 2005, JBoss Inc., and individual contributors as indicated
4   * by the @authors tag. See the copyright.txt in the distribution for a
5   * full listing of individual contributors.
6   *
7   * This is free software; you can redistribute it and/or modify it
8   * under the terms of the GNU Lesser General Public License as
9   * 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 net.sourceforge.junitejb.EJBTestCase;
25
26 import javax.sql.DataSource JavaDoc;
27 import javax.naming.InitialContext JavaDoc;
28 import javax.naming.NamingException JavaDoc;
29 import java.sql.Connection JavaDoc;
30 import java.sql.SQLException JavaDoc;
31
32
33 /**
34  *
35  * @author <a HREF="mailto:alex@jboss.org">Alex Loubyansky</a>
36  */

37 public class AbstractDBSchemaTest
38    extends EJBTestCase
39 {
40    protected String JavaDoc datasourceName = "java:/DefaultDS";
41    protected DataSource JavaDoc datasource;
42
43    public AbstractDBSchemaTest(String JavaDoc s)
44    {
45       super(s);
46    }
47
48    protected Connection JavaDoc getConnection()
49       throws NamingException JavaDoc, SQLException JavaDoc
50    {
51       return getDataSource().getConnection();
52    }
53
54    protected DataSource JavaDoc getDataSource() throws NamingException JavaDoc
55    {
56       if(datasource == null)
57       {
58          datasource = getDataSource(datasourceName);
59       }
60       return datasource;
61    }
62
63    protected DataSource JavaDoc getDataSource(String JavaDoc datasource) throws NamingException JavaDoc
64    {
65       InitialContext JavaDoc ic = null;
66       try
67       {
68          ic = new InitialContext JavaDoc();
69          return (DataSource JavaDoc)ic.lookup(datasource);
70       }
71       finally
72       {
73          if(ic != null)
74             try
75             {
76                ic.close();
77             }
78             catch(Exception JavaDoc ignore){}
79       }
80    }
81 }
82
Popular Tags