// jTDS JDBC Driver for Microsoft SQL Server and Sybase
// Copyright (C) 2004 The jTDS Project
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
//
// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
// Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
//
package net.sourceforge.jtds.jdbc;
import junit.framework.Test;
import junit.framework.TestCase;
import junit.framework.TestSuite;
import net.sourceforge.jtds.jdbc.DefaultProperties;
import net.sourceforge.jtds.jdbc.Driver;
import net.sourceforge.jtds.jdbc.Messages;
import net.sourceforge.jtds.jdbc.TdsCore;
import java.sql.DriverPropertyInfo;
import java.sql.SQLException;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.Properties;
/**
* Unit tests for the {@link Driver} class.
*
* @author David D. Kilzer
* @version $Id: DriverUnitTest.java,v 1.20 2007-07-08 18:08:54 bheineman Exp $
*/
public class DriverUnitTest extends UnitTestBase {
/**
* Construct a test suite for this class.
* <p/>
* The test suite includes the tests in this class, and adds tests
* from {@link DefaultPropertiesTestLibrary} after creating
* anonymous {@link DefaultPropertiesTester} objects.
*
* @return The test suite to run.
*/
public static Test suite() {
TestSuite testSuite = new TestSuite(DriverUnitTest.class);
testSuite.addTest(
Test_Driver_setupConnectProperties.suite("test_setupConnectProperties_DefaultProperties"));
testSuite.addTest(
Test_Driver_getPropertyInfo.suite("test_getPropertyInfo_DefaultProperties"));
addParseURLCorrectTests(testSuite);
return testSuite;
}
/**
* Test to ensure that the version reported by the driver matches the JAR
* file's name.
*/
public void testDriverVersion()
throws Exception
{
String file = Driver.class.getResource( '/' + Driver.class.getName().replace( '.', '/' ) + ".class" ).toString();
// only check if jTDS has been loaded from a jar
if( file.startsWith( "jar" ) )
{
// parse path, e.g. jar:file:/lib/jtds-1.3.0.jar!/net/sourceforge/jtds/jdbc/Driver.class
file = file.substring( 0, file.indexOf( ".jar!" ) );
file = file.substring( file.lastIndexOf( '/' ) + 1 );
assertEquals( Driver.getVersion(), file.substring( file.lastIndexOf( '-' ) + 1 ) );
}
}
/**
* Constructor.
*
* @param name The name of the test.
*/
public DriverUnitTest(final String name) {
super(name);
}
/**
* Tests that passing in a null properties argument to
* {@link Driver#getPropertyInfo(String, Properties)}
* causes the url to be parsed, which then throws a {@link SQLException}.
*/
public void test_getPropertyInfo_ThrowsSQLExceptionWithNullProperties() {
try {
new Driver().getPropertyInfo("wxyz:", null);
fail("Expected SQLException to be throw");
}
catch (SQLException e) {
// Expected
}
}
/**
* Tests that passing in a non-null properties argument to
* {@link Driver#getPropertyInfo(String, Properties)}
* causes the url to be parsed, which then throws a {@link SQLException}.
*/
public void test_getPropertyInfo_ThrowsSQLExceptionWithNonNullProperties() {
try {
new Driver().getPropertyInfo("wxyz:", new Properties());
fail("Expected SQLException to be throw");
}
catch (SQLException e) {
// Expected
}
}
/**
* Tests that the {@link DriverPropertyInfo} array returned from
* {@link Driver#getPropertyInfo(String, Properties)}
* matches the list of properties defined in <code>Messages.properties</code>.
*/
public void test_getPropertyInfo_MatchesMessagesProperties() {
final Map infoMap = new HashMap();
loadDriverPropertyInfoMap(infoMap);
final Map propertyMap = new HashMap();
final Map descriptionMap = new HashMap();
invokeStaticMethod(
Messages.class, "loadDriverProperties",
new Class[]{Map.class, Map.class},
new Object[]{propertyMap, descriptionMap});
assertEquals(
"Properties list size (expected) does not equal DriverPropertyInfo array length (actual)",
propertyMap.size(), infoMap.keySet().size());
assertEquals(
"Description list size (expected) does not equal DriverPropertyInfo array length (actual)",
descriptionMap.size(), infoMap.keySet().size());
for (Iterator iterator = propertyMap.keySet().iterator(); iterator.hasNext();) {
final String key = (String) iterator.next();
final DriverPropertyInfo driverPropertyInfo =
(DriverPropertyInfo) infoMap.get(propertyMap.get(key));
assertNotNull("No DriverPropertyInfo object exists for property '" + key + "'", driverPropertyInfo);
assertEquals(
"Property description (expected) does not match DriverPropertyInfo description (actual)",
descriptionMap.get(key), driverPropertyInfo.description);
}
}
/**
* Tests that the {@link DriverPropertyInfo} array returned from
* {@link Driver#getPropertyInfo(String, Properties)} contains
* the correct <code>choices</code> value on each of the objects.
*/
public void test_getPropertyInfo_Choices() {
String[] expectedBooleanChoices = new String[]{
String.valueOf(true),
String.valueOf(false),
};
String[] expectedPrepareSqlChoices = new String[]{
String.valueOf(TdsCore.UNPREPARED),
String.valueOf(TdsCore.TEMPORARY_STORED_PROCEDURES),
String.valueOf(TdsCore.EXECUTE_SQL),
String.valueOf(TdsCore.PREPARE)
};
String[] expectedServerTypeChoices = new String[]{
String.valueOf(Driver.SQLSERVER),
String.valueOf(Driver.SYBASE),
};
String[] expectedTdsChoices = new String[]{
DefaultProperties.TDS_VERSION_42,
DefaultProperties.TDS_VERSION_50,
DefaultProperties.TDS_VERSION_70,
DefaultProperties.TDS_VERSION_80,
};
Map expectedChoicesMap = new HashMap();
expectedChoicesMap.put(Messages.get(Driver.LASTUPDATECOUNT), expectedBooleanChoices);
expectedChoicesMap.put(Messages.get(Driver.NAMEDPIPE), expectedBooleanChoices);
expectedChoicesMap.put(Messages.get(Driver.PREPARESQL), expectedPrepareSqlChoices);
expectedChoicesMap.put(Messages.get(Driver.SERVERTYPE), expectedServerTypeChoices);
expectedChoicesMap.put(Messages.get(Driver.TDS), expectedTdsChoices);
expectedChoicesMap.put(Messages.get(Driver.SENDSTRINGPARAMETERSASUNICODE), expectedBooleanChoices);
expectedChoicesMap.put(Messages.get(Driver.CACHEMETA), expectedBooleanChoices);
expectedChoicesMap.put(Messages.get(Driver.USECURSORS), expectedBooleanChoices);
expectedChoicesMap.put(Messages.get(Driver.USELOBS), expectedBooleanChoices);
final Map infoMap = new HashMap();
loadDriverPropertyInfoMap(infoMap);
final Iterator iterator = infoMap.keySet().iterator();
while (iterator.hasNext()) {
String key = (String) iterator.next();
DriverPropertyInfo info = (DriverPropertyInfo) infoMap.get(key);
if (expectedChoicesMap.containsKey(key)) {
assertEquals("Choices did not match for key " + key,
((String[]) expectedChoicesMap.get(key)), info.choices);
}
else {
assertNull("Expected choices to be null for key " + key,
expectedChoicesMap.get(key));
}
}
}
/**
* Tests that the {@link DriverPropertyInfo} array returned from
* {@link Driver#getPropertyInfo(String, Properties)} contains
* the correct <code>required</code> value on each of the objects.
*/
public void test_getPropertyInfo_Required() {
Map requiredTrueMap = new HashMap();
requiredTrueMap.put(Messages.get(Driver.SERVERNAME), Boolean.TRUE);
requiredTrueMap.put(Messages.get(Driver.SERVERTYPE), Boolean.TRUE);
final Map infoMap = new HashMap();
loadDriverPropertyInfoMap(infoMap);
final Iterator iterator = infoMap.keySet().iterator();
while (iterator.hasNext()) {
String key = (String) iterator.next();
DriverPropertyInfo info = (DriverPropertyInfo) infoMap.get(key);
if (requiredTrueMap.containsKey(key)) {
assertTrue("The 'required' field is not true for key " + key, info.required);
}
else {
assertFalse("The 'required' field is not false for key " + key, info.required);
}
}
}
/**
* Retrieve the {@link DriverPropertyInfo} array from
* {@link Driver#getPropertyInfo(String, Properties)} and convert it
* into a {@link Map} using the <code>name</code> property for the keys.
*
* @param driverPropertyInfoMap The map of {@link DriverPropertyInfo} objects to be populated.
*/
private void loadDriverPropertyInfoMap(final Map driverPropertyInfoMap) {
try {
final DriverPropertyInfo[] driverPropertyInfoArray = new Driver().getPropertyInfo(
"jdbc:jtds:sqlserver://servername/databasename", new Properties());
for (int i = 0; i < driverPropertyInfoArray.length; i++) {
DriverPropertyInfo driverPropertyInfo = driverPropertyInfoArray[i];
driverPropertyInfoMap.put(driverPropertyInfo.name, driverPropertyInfo);
}
}
catch (SQLException e) {
throw new RuntimeException(e.getMessage());
}
}
/**
* Class used to test <code>Driver.setupConnectProperties(String, java.util.Properties)</code>.
*/
public static class Test_Driver_setupConnectProperties extends DefaultPropertiesTestLibrary {
/**
* Construct a test suite for this library.
*
* @param name The name of the tests.
* @return The test suite.
*/
public static Test suite(String name) {
return new TestSuite(Test_Driver_setupConnectProperties.class, name);
}
/**
* Default constructor.
*/
public Test_Driver_setupConnectProperties() {
setTester(
new DefaultPropertiesTester() {
public void assertDefaultProperty(
String message, String url, Properties properties, String fieldName,
String key, String expected) {
Properties results =
(Properties) invokeInstanceMethod(
new Driver(), "setupConnectProperties",
new Class[]{String.class, Properties.class},
new Object[]{url, properties});
assertEquals(message, expected, results.getProperty(Messages.get(key)));
}
}
);
}
}
/**
* Class used to test {@link Driver#getPropertyInfo(String, Properties)}.
*/
public static class Test_Driver_getPropertyInfo extends DefaultPropertiesTestLibrary {
/**
* Construct a test suite for this library.
*
* @param name The name of the tests.
* @return The test suite.
*/
public static Test suite(String name) {
return new TestSuite(Test_Driver_getPropertyInfo.class, name);
}
/**
* Default constructor.
*/
public Test_Driver_getPropertyInfo() {
setTester(
new DefaultPropertiesTester() {
public void assertDefaultProperty(
String message, String url, Properties properties, String fieldName,
String key, String expected) {
try {
boolean found = false;
String messageKey = Messages.get(key);
DriverPropertyInfo[] infoArray = new Driver().getPropertyInfo(url, properties);
for (int i = 0; i < infoArray.length; i++) {
DriverPropertyInfo info = infoArray[i];
if (info.name.equals(messageKey)) {
assertEquals(message, expected, info.value);
found = true;
}
}
if (!found) {
fail("DriverPropertyInfo for '" + messageKey + "' not found!");
}
}
catch (SQLException e) {
throw new RuntimeException(e.getMessage());
}
}
}
);
}
}
/**
* Creates tests for passing all variants of correct URLs.
* {@link Driver#parseURL(String, Properties)}
*/
public static void addParseURLCorrectTests(TestSuite suite) {
final String[] SERVERTYPES = new String[] {
"sqlserver", "sybase"
};
final String[] HOSTS = new String[] {
"local", "local.com.aa", "100.2.3.400", "fc00::36A", "::ffff:192.168.0.1"
};
final String[] PORTS = new String[] {
null, "1433", "2332"
};
final String[] DATABASES = new String[] {
null, "instance", "my_name"
};
final String[][][] PROPERTIES = new String[][][] {
null,
{ { "key", "value" } },
{ { "key1", "value" } , { "ke_y2", "valu.e_|22" } },
{ { "keyn", null } }
};
Map<String, Properties> urls = generateAllURLVariants(SERVERTYPES, HOSTS, PORTS, DATABASES, PROPERTIES);
Properties prop = new Properties();
for (Map.Entry<String, Properties> e : urls.entrySet()) {
String url = e.getKey();
Properties expectedProp = e.getValue();
suite.addTest(new ParseURLTest(url, expectedProp));
}
}
private static class ParseURLTest extends TestCase {
private final String url;
private final Properties expectedProp;
public ParseURLTest(String url, Properties expectedProp) {
super("testParseUrl \"" + url + "\"");
this.url = url;
this.expectedProp = expectedProp;
}
public void runTest() {
testParseUrl();
}
public void testParseUrl() {
Properties parsedProp = invokeDriverParseURL(url, new Properties());
assertNotNull("URL '" + url + "' cannot be parsed", parsedProp);
if (!expectedProp.equals(parsedProp)) {
assertEquals("Result from URL '" + url + "' from Driver is not what expected", expectedProp, parsedProp);
}
}
}
private static Map<String, Properties> generateAllURLVariants(String[] serverTypes,
String[] hosts, String[] ports, String[] databases,
String[][][] properties)
{
Map<String, Properties> res = new HashMap<String, Properties>();
for (int is = 0; is < serverTypes.length; is++) {
String serverType = serverTypes[is];
for (int ih = 0; ih < hosts.length; ih++) {
String host = hosts[ih];
for (int ip = 0; ip < ports.length; ip++) {
String port = ports[ip];
for (int id = 0; id < databases.length; id++) {
String database = databases[id];
for (int ir = 0; ir < properties.length; ir++) {
String[][] property = properties[ir];
StringBuilder url = new StringBuilder();
url.append("jdbc:jtds:").append(serverType);
url.append("://");
if (host.indexOf(':') >= 0) {
url.append('[').append(host).append(']');
} else {
url.append(host);
}
if (port != null) {
url.append(':').append(port);
}
if (database != null) {
url.append('/').append(database);
}
Properties p = new Properties();
p.put("SERVERTYPE", String.valueOf(DefaultProperties.getServerType(serverType)));
p.put("SERVERNAME", host);
if (port != null) {
p.put("PORTNUMBER", String.valueOf(Integer.parseInt(port)));
}
if (database != null) {
p.put("DATABASENAME", database);
}
for (int j = 0, n = (property == null ? 0 : property.length); j < n; j++) {
url.append(';').append(property[j][0]);
if (property[j][1] != null) {
url.append('=').append(property[j][1]);
}
p.put(property[j][0].toUpperCase(), (property[j][1] == null ? "" : property[j][1]));
}
res.put(url.toString(), p);
}
}
}
}
}
return res;
}
private static Properties invokeDriverParseURL(String url, Properties prop) {
return (Properties) invokeStaticMethod(
Driver.class, "parseURL",
new Class[]{String.class, Properties.class},
new Object[]{url, prop});
}
}