1   /* Version 1.0 based on Apache Software License 1.1
2    *
3    * Copyright (c) 2003 Piotr Maj and DBMonster developers. All rights
4    * reserved.
5    *
6    * Redistribution and use in source and binary forms, with or without
7    * modification, are permitted provided that the following conditions are
8    * met:
9    *
10   * 1. Redistributions of source code must retain the above copyright
11   *    notice, this list of conditions and the following disclaimer.
12   *
13   * 2. Redistributions in binary form must reproduce the above copyright
14   *    notice, this list of conditions and the following disclaimer in the
15   *    documentation and/or other materials provided with the distribution.
16   *
17   * 3. The end-user documentation included with the redistribution, if any,
18   *    must include the following acknowledgment:
19   *
20   *    "This product includes software developed by DBMonster developers
21   *    (http://dbmonster.kernelpanic.pl/)."
22   *
23   *  Alternately, this acknowledgment may appear in the software itself,
24   *  if and wherever such third-party acknowledgments normally appear.
25   *
26   * 4. The name "DBMonster" must not be used to endorse or promote products
27   *    derived from this software without prior written permission. For
28   *    written permission, please contact piotr.maj@kernelpanic.pl.
29   *
30   * 5. Products derived from this software may not be called "DBMonster",
31   *    nor may "DBMonster" appear in their name, without prior written
32   *    permission of Piotr Maj.
33   *
34   * THIS SOFTWARE IS PROVIDED "AS IS" AND ANY EXPRESSED OR IMPLIED
35   * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
36   * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
37   * IN NO EVENT SHALL THE DBMONSTER DEVELOPERS BE LIABLE FOR ANY DIRECT,
38   * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
39   * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
40   * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
41   * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
42   * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
43   * IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
44   * POSSIBILITY OF SUCH DAMAGE.
45   */
46  
47  package pl.kernelpanic.dbmonster.test;
48  
49  import java.net.URL;
50  import java.sql.Connection;
51  import java.sql.Statement;
52  import java.util.Properties;
53  
54  import junit.framework.TestCase;
55  
56  import pl.kernelpanic.dbmonster.DBMonster;
57  import pl.kernelpanic.dbmonster.connection.ConnectionProvider;
58  import pl.kernelpanic.dbmonster.connection.DBCPConnectionProvider;
59  import pl.kernelpanic.dbmonster.schema.Schema;
60  import pl.kernelpanic.dbmonster.schema.SchemaUtil;
61  
62  /***
63   * DBMonster test.
64   *
65   * @author Piotr Maj <piotr.maj@kernelpanic.pl>
66   *
67   * @version $Id: PostgreSQLTest.html,v 1.1 2007/06/21 08:38:14 sbahloul Exp $
68   */
69  public class PostgreSQLTest extends TestCase {
70  
71  	private static final String TEST_NAME = "postgresql";
72  
73      public final void testDBMonster() throws Exception {
74      	if (!Utils.performTest(TEST_NAME)) {
75              Utils.skipTest(TEST_NAME);
76      		return;
77      	}
78      	String driver = "org.postgresql.Driver";
79          String url    = "jdbc:postgresql://127.0.0.1/dbmonster";
80          String user   = "dbmonster";
81          String pass   = "dbmonster";
82          
83          Properties props = new Properties();
84          props.put("dbmonster.jdbc.driver", driver);
85          props.put("dbmonster.jdbc.url", url);
86          props.put("dbmonster.jdbc.username", user);
87          props.put("dbmonster.jdbc.password", pass);
88          props.put("dbmonster.max-tries", "1000");
89  
90          ConnectionProvider cp = 
91              new DBCPConnectionProvider(driver, url, user, pass);
92  
93          Connection conn = cp.getConnection();
94          Statement stmt = conn.createStatement();
95          
96          try {
97              stmt.executeUpdate("DROP TABLE DEFAULT_TEST");
98          } catch (Exception e) {
99          }
100         try {
101             stmt.executeUpdate("DROP TABLE USERS");
102         } catch (Exception e) {
103         }
104         try {
105             stmt.executeUpdate("DROP TABLE GROUPS");
106         } catch (Exception e) {
107         }
108         try {
109             stmt.executeUpdate(
110                 "CREATE TABLE GROUPS (id int not null primary key,"
111                 + "name varchar(255) not null, description varchar)");
112 
113             stmt.executeUpdate(
114                 "CREATE TABLE USERS (id int not null primary key,"
115                 + "login varchar(255) not null, id_group int not null references groups(id))");
116 
117             stmt.executeUpdate(
118                 "CREATE TABLE DEFAULT_TEST (id serial not null primary key,"
119                 + "name varchar(255) not null)");
120             stmt.close();
121             conn.commit();
122             conn.close();
123         } catch (Exception e) {
124             System.out.println(e.getMessage());
125             throw e;
126         }
127 
128         DBMonster dbm = new DBMonster();
129         dbm.setConnectionProvider(cp);
130         dbm.setProperties(props);
131         String u = "/pl/kernelpanic/dbmonster/test/resources/dbmonster-schema.xml";
132         URL ur = getClass().getResource(u);
133         Schema schema = SchemaUtil.loadSchema(ur, dbm.getLogger());
134         dbm.addSchema(schema);
135         try {
136             dbm.doTheJob();
137         } catch (Exception e) {
138             e.printStackTrace();
139             throw e;
140         }
141     }
142 }