/*
* Hibernate, Relational Persistence for Idiomatic Java
*
* Copyright (c) 2008-2011, Red Hat Inc. or third-party contributors as
* indicated by the @author tags or express copyright attribution
* statements applied by the authors. All third-party contributions are
* distributed under license by Red Hat Inc.
*
* This copyrighted material is made available to anyone wishing to use, modify,
* copy, or redistribute it subject to the terms and conditions of the GNU
* Lesser General Public License, as published by the Free Software Foundation.
*
* This program 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 distribution; if not, write to:
* Free Software Foundation, Inc.
* 51 Franklin Street, Fifth Floor
* Boston, MA 02110-1301 USA
*/
package org.hibernate.dialect.resolver;
import java.sql.Connection;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
import java.util.Properties;
import org.junit.Before;
import org.junit.Test;
import org.hibernate.HibernateException;
import org.hibernate.cfg.Environment;
import org.hibernate.dialect.DB2Dialect;
import org.hibernate.dialect.DerbyDialect;
import org.hibernate.dialect.DerbyTenFiveDialect;
import org.hibernate.dialect.DerbyTenSevenDialect;
import org.hibernate.dialect.DerbyTenSixDialect;
import org.hibernate.dialect.H2Dialect;
import org.hibernate.dialect.HSQLDialect;
import org.hibernate.dialect.InformixDialect;
import org.hibernate.dialect.IngresDialect;
import org.hibernate.dialect.Mocks;
import org.hibernate.dialect.MySQLDialect;
import org.hibernate.dialect.Oracle10gDialect;
import org.hibernate.dialect.Oracle8iDialect;
import org.hibernate.dialect.Oracle9iDialect;
import org.hibernate.dialect.PostgreSQLDialect;
import org.hibernate.dialect.SQLServerDialect;
import org.hibernate.dialect.SybaseASE15Dialect;
import org.hibernate.dialect.SybaseAnywhereDialect;
import org.hibernate.dialect.TestingDialects;
import org.hibernate.service.classloading.internal.ClassLoaderServiceImpl;
import org.hibernate.service.classloading.spi.ClassLoadingException;
import org.hibernate.service.jdbc.dialect.internal.DialectFactoryImpl;
import org.hibernate.service.jdbc.dialect.internal.DialectResolverSet;
import org.hibernate.service.jdbc.dialect.internal.StandardDialectResolver;
import org.hibernate.service.jdbc.dialect.spi.DialectResolver;
import org.hibernate.testing.junit4.BaseUnitTestCase;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.fail;
/**
* @author Steve Ebersole
*/
public class DialectFactoryTest extends BaseUnitTestCase {
private DialectFactoryImpl dialectFactory;
@Before
public void setUp() {
dialectFactory = new DialectFactoryImpl();
dialectFactory.setClassLoaderService( new ClassLoaderServiceImpl( getClass().getClassLoader() ) );
dialectFactory.setDialectResolver( new StandardDialectResolver() );
}
@Test
public void testExplicitlySuppliedDialectClassName() {
final Map<String, String> configValues = new HashMap<String, String>();
configValues.put( Environment.DIALECT, "org.hibernate.dialect.HSQLDialect" );
assertEquals( HSQLDialect.class, dialectFactory.buildDialect( configValues, null ).getClass() );
configValues.put( Environment.DIALECT, "org.hibernate.dialect.NoSuchDialect" );
try {
dialectFactory.buildDialect( configValues, null );
fail();
}
catch ( HibernateException e ) {
assertEquals( "unexpected exception type", ClassLoadingException.class, e.getCause().getClass() );
}
configValues.put( Environment.DIALECT, "java.lang.Object" );
try {
dialectFactory.buildDialect( configValues, null );
fail();
}
catch ( HibernateException e ) {
assertEquals( "unexpected exception type", ClassCastException.class, e.getCause().getClass() );
}
}
@Test
public void testBuildDialectByProperties() {
Properties props = new Properties();
try {
dialectFactory.buildDialect( props, null );
fail();
}
catch ( HibernateException e ) {
assertNull( e.getCause() );
}
props.setProperty( Environment.DIALECT, "org.hibernate.dialect.HSQLDialect" );
assertEquals( HSQLDialect.class, dialectFactory.buildDialect( props, null ).getClass() );
}
@Test
public void testPreregisteredDialects() {
DialectResolver resolver = new StandardDialectResolver();
testDetermination( "HSQL Database Engine", HSQLDialect.class, resolver );
testDetermination( "H2", H2Dialect.class, resolver );
testDetermination( "MySQL", MySQLDialect.class, resolver );
testDetermination( "PostgreSQL", PostgreSQLDialect.class, resolver );
testDetermination( "Apache Derby", 10, 4, DerbyDialect.class, resolver );
testDetermination( "Apache Derby", 10, 5, DerbyTenFiveDialect.class, resolver );
testDetermination( "Apache Derby", 10, 6, DerbyTenSixDialect.class, resolver );
testDetermination( "Apache Derby", 11, 5, DerbyTenSevenDialect.class, resolver );
testDetermination( "Ingres", IngresDialect.class, resolver );
testDetermination( "ingres", IngresDialect.class, resolver );
testDetermination( "INGRES", IngresDialect.class, resolver );
testDetermination( "Microsoft SQL Server Database", SQLServerDialect.class, resolver );
testDetermination( "Microsoft SQL Server", SQLServerDialect.class, resolver );
testDetermination( "Sybase SQL Server", SybaseASE15Dialect.class, resolver );
testDetermination( "Adaptive Server Enterprise", SybaseASE15Dialect.class, resolver );
testDetermination( "Adaptive Server Anywhere", SybaseAnywhereDialect.class, resolver );
testDetermination( "Informix Dynamic Server", InformixDialect.class, resolver );
testDetermination( "DB2/NT", DB2Dialect.class, resolver );
testDetermination( "DB2/LINUX", DB2Dialect.class, resolver );
testDetermination( "DB2/6000", DB2Dialect.class, resolver );
testDetermination( "DB2/HPUX", DB2Dialect.class, resolver );
testDetermination( "DB2/SUN", DB2Dialect.class, resolver );
testDetermination( "DB2/LINUX390", DB2Dialect.class, resolver );
testDetermination( "DB2/AIX64", DB2Dialect.class, resolver );
testDetermination( "Oracle", 8, Oracle8iDialect.class, resolver );
testDetermination( "Oracle", 9, Oracle9iDialect.class, resolver );
testDetermination( "Oracle", 10, Oracle10gDialect.class, resolver );
testDetermination( "Oracle", 11, Oracle10gDialect.class, resolver );
}
@Test
public void testCustomDialects() {
DialectResolverSet resolvers = new DialectResolverSet();
resolvers.addResolver( new TestingDialects.MyDialectResolver1() );
resolvers.addResolver( new TestingDialects.MyDialectResolver2() );
resolvers.addResolver( new TestingDialects.ErrorDialectResolver1() );
resolvers.addResolver( new TestingDialects.ErrorDialectResolver2() );
resolvers.addResolver( new TestingDialects.MyOverridingDialectResolver1() );
//DialectFactory.registerDialectResolver( "org.hibernate.dialect.NoSuchDialectResolver" );
//DialectFactory.registerDialectResolver( "java.lang.Object" );
testDetermination( "MyDatabase1", TestingDialects.MyDialect1.class, resolvers );
testDetermination( "MyDatabase2", 1, TestingDialects.MyDialect21.class, resolvers );
testDetermination( "MyTrickyDatabase1", TestingDialects.MyDialect1.class, resolvers );
// This should be mapped to DB2Dialect by default, but actually it will be
// my custom dialect because I have registered MyOverridingDialectResolver1.
testDetermination( "DB2/MySpecialPlatform", TestingDialects.MySpecialDB2Dialect.class, resolvers );
try {
testDetermination( "ErrorDatabase1", Void.TYPE, resolvers );
fail();
}
catch ( HibernateException e ) {
}
try {
testDetermination( "ErrorDatabase2", Void.TYPE, resolvers );
fail();
}
catch ( HibernateException e ) {
}
}
@Test
public void testDialectNotFound() {
Map properties = Collections.EMPTY_MAP;
try {
dialectFactory.buildDialect( properties, Mocks.createConnection( "NoSuchDatabase", 666 ) );
fail();
}
catch ( HibernateException e ) {
assertNull( e.getCause() );
}
}
private void testDetermination(String databaseName, Class clazz, DialectResolver resolver) {
testDetermination( databaseName, -9999, clazz, resolver );
}
private void testDetermination(String databaseName, int databaseMajorVersion, Class clazz, DialectResolver resolver) {
testDetermination( databaseName, databaseMajorVersion, -9999, clazz, resolver );
}
private void testDetermination(String databaseName, int majorVersion, int minorVersion, Class clazz, DialectResolver resolver) {
dialectFactory.setDialectResolver( resolver );
Properties properties = new Properties();
Connection conn = Mocks.createConnection( databaseName, majorVersion, minorVersion );
assertEquals( clazz, dialectFactory.buildDialect( properties, conn ).getClass() );
}
}