1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
|
package org.postgresql.test;
import java.sql.*;
import junit.framework.TestCase;
/*
* Utility class for JDBC tests
*/
public class TestUtil
{
/*
* Returns the Test database JDBC URL
*/
public static String getURL()
{
return System.getProperty("database");
}
/*
* Returns the Postgresql username
*/
public static String getUser()
{
return System.getProperty("username");
}
/*
* Returns the user's password
*/
public static String getPassword()
{
return System.getProperty("password");
}
/*
* Helper - opens a connection.
*/
public static java.sql.Connection openDB()
{
try
{
Class.forName("org.postgresql.Driver");
return java.sql.DriverManager.getConnection(getURL(), getUser(), getPassword());
}
catch (ClassNotFoundException ex)
{
TestCase.fail(ex.getMessage());
}
catch (SQLException ex)
{
TestCase.fail(ex.getMessage());
}
return null;
}
/*
* Helper - closes an open connection. This rewrites SQLException to a failed
* assertion. It's static so other classes can use it.
*/
public static void closeDB(Connection con)
{
try
{
if (con != null)
con.close();
}
catch (SQLException ex)
{
TestCase.fail(ex.getMessage());
}
}
/*
* Helper - creates a test table for use by a test
*/
public static void createTable(Connection con,
String table,
String columns)
{
try
{
Statement st = con.createStatement();
try
{
// Drop the table
dropTable(con, table);
// Now create the table
st.executeUpdate("create table " + table + " (" + columns + ")");
}
finally
{
st.close();
}
}
catch (SQLException ex)
{
TestCase.fail(ex.getMessage());
}
}
/*
* Helper - drops a table
*/
public static void dropTable(Connection con, String table)
{
try
{
Statement stmt = con.createStatement();
try
{
stmt.executeUpdate("DROP TABLE " + table);
}
catch (SQLException ex)
{
// ignore
}
}
catch (SQLException ex)
{
TestCase.fail(ex.getMessage());
}
}
/*
* Helper - generates INSERT SQL - very simple
*/
public static String insertSQL(String table, String values)
{
return insertSQL(table, null, values);
}
public static String insertSQL(String table, String columns, String values)
{
String s = "INSERT INTO " + table;
if (columns != null)
s = s + " (" + columns + ")";
return s + " VALUES (" + values + ")";
}
/*
* Helper - generates SELECT SQL - very simple
*/
public static String selectSQL(String table, String columns)
{
return selectSQL(table, columns, null, null);
}
public static String selectSQL(String table, String columns, String where)
{
return selectSQL(table, columns, where, null);
}
public static String selectSQL(String table, String columns, String where, String other)
{
String s = "SELECT " + columns + " FROM " + table;
if (where != null)
s = s + " WHERE " + where;
if (other != null)
s = s + " " + other;
return s;
}
/*
* Helper to prefix a number with leading zeros - ugly but it works...
* @param v value to prefix
* @param l number of digits (0-10)
*/
public static String fix(int v, int l)
{
String s = "0000000000".substring(0, l) + Integer.toString(v);
return s.substring(s.length() - l);
}
}
|
