HHH-4808 Add test for issue

This commit is contained in:
Andrea Boriero 2018-07-05 12:54:02 +01:00 committed by Sanne Grinovero
parent 8f4450c433
commit 179c1d1da0
1 changed files with 365 additions and 0 deletions

View File

@ -0,0 +1,365 @@
/*
* Hibernate, Relational Persistence for Idiomatic Java
*
* License: GNU Lesser General Public License (LGPL), version 2.1 or later.
* See the lgpl.txt file in the root directory or <http://www.gnu.org/licenses/lgpl-2.1.html>.
*/
package org.hibernate.test.connections;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Properties;
import java.util.Set;
import javax.persistence.Entity;
import javax.persistence.EntityManager;
import javax.persistence.FetchType;
import javax.persistence.Id;
import javax.persistence.JoinColumn;
import javax.persistence.ManyToOne;
import javax.persistence.OneToMany;
import javax.persistence.Query;
import javax.sql.DataSource;
import org.hibernate.annotations.Fetch;
import org.hibernate.annotations.FetchMode;
import org.hibernate.annotations.LazyCollection;
import org.hibernate.annotations.LazyCollectionOption;
import org.hibernate.annotations.LazyToOne;
import org.hibernate.annotations.LazyToOneOption;
import org.hibernate.cfg.AvailableSettings;
import org.hibernate.cfg.Environment;
import org.hibernate.engine.jdbc.connections.internal.UserSuppliedConnectionProviderImpl;
import org.hibernate.jpa.test.BaseEntityManagerFunctionalTestCase;
import org.hibernate.jpa.test.connection.BaseDataSource;
import org.hibernate.resource.jdbc.spi.PhysicalConnectionHandlingMode;
import org.hibernate.testing.TestForIssue;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import static junit.framework.TestCase.assertTrue;
import static org.hibernate.testing.transaction.TransactionUtil.doInJPA;
import static org.junit.Assert.assertEquals;
import static org.mockito.Mockito.spy;
/**
* @author Selaron
*/
@TestForIssue(jiraKey = "HHH-4808")
public class LazyLoadingConnectionCloseTest extends BaseEntityManagerFunctionalTestCase {
private ConnectionProviderDecorator connectionProvider;
@Override
protected Class<?>[] getAnnotatedClasses() {
return new Class[] { SimpleEntity.class, ChildEntity.class };
}
@Override
protected Map getConfig() {
Map config = super.getConfig();
config.put( AvailableSettings.ENABLE_LAZY_LOAD_NO_TRANS, "true" );
config.put( AvailableSettings.CONNECTION_HANDLING, PhysicalConnectionHandlingMode.DELAYED_ACQUISITION_AND_RELEASE_AFTER_STATEMENT );
config.put( AvailableSettings.AUTOCOMMIT, "false" );
connectionProvider = new ConnectionProviderDecorator( getDataSource() );
config.put( AvailableSettings.CONNECTION_PROVIDER, connectionProvider );
return config;
}
@Before
public void setUp() {
doInJPA( this::entityManagerFactory, entityManager -> {
final SimpleEntity entity = new SimpleEntity();
entity.setId( 1L );
entity.setName( "TheParent" );
final ChildEntity c1 = new ChildEntity();
c1.setId( 1L );
c1.setParent( entity );
c1.setName( "child1" );
final ChildEntity c2 = new ChildEntity();
c2.setId( 2L );
c2.setParent( entity );
c2.setName( "child2" );
entityManager.persist( entity );
entityManager.persist( c1 );
entityManager.persist( c2 );
} );
}
/**
* Tests connections get closed after transaction commit.
*/
@Test
public void testConnectionCloseAfterTx() {
connectionProvider.clear();
EntityManager entityManager = getOrCreateEntityManager();
try {
entityManager.getTransaction().begin();
try {
final Query qry = entityManager.createQuery( "FROM SimpleEntity" );
final List<SimpleEntity> entities = qry.getResultList();
final SimpleEntity entity = entities.get( 0 );
assertEquals( 1, connectionProvider.getCurrentOpenConnections() );
}
catch (Exception e) {
if ( entityManager.getTransaction().isActive() ) {
entityManager.getTransaction().rollback();
}
throw e;
}
finally {
if ( entityManager.getTransaction().isActive() ) {
entityManager.getTransaction().commit();
}
}
assertTrue( connectionProvider.areAllConnectionClosed() );
}
finally {
entityManager.close();
}
}
/**
* Tests connections get closed after lazy collection initialization.
*/
@Test
public void testConnectionCloseAfterLazyCollectionInit() {
connectionProvider.clear();
EntityManager entityManager = getOrCreateEntityManager();
try {
final Query qry = entityManager.createQuery( "FROM SimpleEntity" );
final List<SimpleEntity> entities = qry.getResultList();
final SimpleEntity entity = entities.get( 0 );
// assert no connection is open
assertTrue( connectionProvider.areAllConnectionClosed() );
final int oldOpenedConnections = connectionProvider.getTotalOpenedConnectionCount();
final Set<ChildEntity> lazyChildren = entity.getChildren();
// this will initialize the collection and such trigger a query
lazyChildren.stream().findAny();
// assert a connection had been opened
Assert.assertTrue( oldOpenedConnections < connectionProvider.getTotalOpenedConnectionCount() );
// assert there's no remaining connection left.
assertTrue( connectionProvider.areAllConnectionClosed() );
}
finally {
entityManager.close();
}
}
/**
* Tests connections get closed after transaction commit.
*/
@Test
public void testConnectionCloseAfterLazyPojoPropertyInit() {
connectionProvider.clear();
EntityManager entityManager = getOrCreateEntityManager();
try {
final Query qry = entityManager.createQuery( "FROM ChildEntity" );
final List<ChildEntity> entities = qry.getResultList();
final ChildEntity entity = entities.get( 0 );
// assert no connection is open
assertTrue( connectionProvider.areAllConnectionClosed() );
final int oldOpenedConnections = connectionProvider.getTotalOpenedConnectionCount();
final SimpleEntity parent = entity.getParent();
// this will initialize the collection and such trigger a query
parent.getName();
// assert a connection had been opened
Assert.assertTrue( oldOpenedConnections < connectionProvider.getTotalOpenedConnectionCount() );
// assert there's no remaining connection left.
assertTrue( connectionProvider.areAllConnectionClosed() );
}
finally {
entityManager.close();
}
}
/**
* Tests connections get closed after transaction commit.
*/
@Test
public void testConnectionCloseAfterQueryWithoutTx() {
connectionProvider.clear();
EntityManager entityManager = getOrCreateEntityManager();
try {
final int oldOpenedConnections = connectionProvider.getTotalOpenedConnectionCount();
final List<ChildEntity> childrenByQuery = entityManager.createQuery( "FROM ChildEntity" ).getResultList();
assertTrue( childrenByQuery.size() > 0 );
// assert a connection had been opened
assertTrue( oldOpenedConnections < connectionProvider.getTotalOpenedConnectionCount() );
// assert there's no remaining connection left.
assertTrue( connectionProvider.areAllConnectionClosed() );
}
finally {
entityManager.close();
}
}
@Entity(name = "SimpleEntity")
public static class SimpleEntity {
private Long id;
private String name;
Set<ChildEntity> children = new HashSet<>();
@Id
public Long getId() {
return id;
}
public void setId(final Long id) {
this.id = id;
}
public String getName() {
return name;
}
public void setName(final String name) {
this.name = name;
}
@OneToMany(targetEntity = ChildEntity.class, mappedBy = "parent")
@LazyCollection(LazyCollectionOption.EXTRA)
@Fetch(FetchMode.SELECT)
public Set<ChildEntity> getChildren() {
return children;
}
public void setChildren(final Set<ChildEntity> children) {
this.children = children;
}
}
@Entity(name = "ChildEntity")
public static class ChildEntity {
private Long id;
private String name;
private SimpleEntity parent;
@Id
public Long getId() {
return id;
}
public void setId(final Long id) {
this.id = id;
}
public String getName() {
return name;
}
public void setName(final String name) {
this.name = name;
}
@ManyToOne(fetch = FetchType.LAZY)
@JoinColumn
@LazyToOne(LazyToOneOption.PROXY)
public SimpleEntity getParent() {
return parent;
}
public void setParent(final SimpleEntity parent) {
this.parent = parent;
}
}
private BaseDataSource getDataSource() {
final Properties connectionProps = new Properties();
connectionProps.put( "user", Environment.getProperties().getProperty( Environment.USER ) );
connectionProps.put( "password", Environment.getProperties().getProperty( Environment.PASS ) );
final String url = Environment.getProperties().getProperty( Environment.URL );
return new BaseDataSource() {
@Override
public Connection getConnection() throws SQLException {
return DriverManager.getConnection( url, connectionProps );
}
@Override
public Connection getConnection(String username, String password) throws SQLException {
return DriverManager.getConnection( url, connectionProps );
}
};
}
public static class ConnectionProviderDecorator extends UserSuppliedConnectionProviderImpl {
private final DataSource dataSource;
private int connectionCount;
private int openConnections;
private Connection connection;
public ConnectionProviderDecorator(DataSource dataSource) {
this.dataSource = dataSource;
}
@Override
public Connection getConnection() throws SQLException {
connectionCount++;
openConnections++;
connection = spy( dataSource.getConnection() );
return connection;
}
@Override
public void closeConnection(Connection connection) throws SQLException {
connection.close();
openConnections--;
}
public int getTotalOpenedConnectionCount() {
return this.connectionCount;
}
public int getCurrentOpenConnections() {
return openConnections;
}
public boolean areAllConnectionClosed() {
return openConnections == 0;
}
public void clear() {
connectionCount = 0;
openConnections = 0;
}
}
}