Hibernate
/
hibernate-orm
mirror of https://github.com/hibernate/hibernate-orm
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Provide a way to override global env properties in a test ConnectionProviderBuilder

This commit is contained in:
marko-bekhta 2024-01-03 17:51:21 +01:00 committed by Christian Beikov
parent 7321ca4ecd
commit 641a7ddb4b
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
hibernate-testing/src/main/java/org/hibernate/testing/env/ConnectionProviderBuilder.java vendored
View File

@ -12,6 +12,7 @@ import java.lang.reflect.Proxy;
import java.sql.Connection; import java.sql.Connection;
import java.sql.SQLException; import java.sql.SQLException;
import java.util.Collections; import java.util.Collections;
import java.util.Map;
import java.util.Properties; import java.util.Properties;
import javax.sql.DataSource; import javax.sql.DataSource;
@ -26,7 +27,6 @@ import org.hibernate.internal.util.ReflectHelper;
import org.hibernate.testing.DialectCheck; import org.hibernate.testing.DialectCheck;
import org.hibernate.testing.jdbc.ConnectionProviderDelegate; import org.hibernate.testing.jdbc.ConnectionProviderDelegate;
import org.hibernate.testing.jdbc.SharedDriverManagerConnectionProviderImpl;
import org.hibernate.testing.util.ServiceRegistryUtil; import org.hibernate.testing.util.ServiceRegistryUtil;
/** /**
@ -46,7 +46,14 @@ public class ConnectionProviderBuilder implements DialectCheck {
public static final String PASS = ""; public static final String PASS = "";
public static Properties getConnectionProviderProperties(String dbName) { public static Properties getConnectionProviderProperties(String dbName) {
return getConnectionProviderProperties( dbName, Collections.emptyMap() );
}
public static Properties getConnectionProviderProperties(String dbName, Map<String, String> environmentOverrides) {
final Properties globalProperties = Environment.getProperties(); final Properties globalProperties = Environment.getProperties();
// since returned global properties are a copy, we just add our overrides to them:
globalProperties.putAll( environmentOverrides );
assert globalProperties.getProperty( Environment.URL ).startsWith( "jdbc:h2:" ) assert globalProperties.getProperty( Environment.URL ).startsWith( "jdbc:h2:" )
: "Connection provider properties are only usable when running against H2"; : "Connection provider properties are only usable when running against H2";
final Properties props = new Properties( null ); final Properties props = new Properties( null );
@ -95,6 +102,10 @@ public class ConnectionProviderBuilder implements DialectCheck {
return buildConnectionProvider( getConnectionProviderProperties( dbName ), false ); return buildConnectionProvider( getConnectionProviderProperties( dbName ), false );
} }
public static ConnectionProvider buildConnectionProvider(String dbName, Map<String, String> environmentOverrides) {
return buildConnectionProvider( getConnectionProviderProperties( dbName, environmentOverrides ), false );
}
public static ConnectionProvider buildDataSourceConnectionProvider(String dbName) { public static ConnectionProvider buildDataSourceConnectionProvider(String dbName) {
final Properties globalProperties = Environment.getProperties(); final Properties globalProperties = Environment.getProperties();
assert globalProperties.getProperty( Environment.URL ).startsWith( "jdbc:h2:" ) assert globalProperties.getProperty( Environment.URL ).startsWith( "jdbc:h2:" )