/*
* Hibernate, Relational Persistence for Idiomatic Java
*
* Copyright (c) 2006-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.test.interceptor;
import java.io.Serializable;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;
import org.hibernate.AssertionFailure;
import org.hibernate.EmptyInterceptor;
import org.hibernate.Interceptor;
import org.hibernate.Session;
import org.hibernate.Transaction;
import org.hibernate.type.Type;
import org.junit.Test;
import org.hibernate.testing.TestForIssue;
import org.hibernate.testing.junit4.BaseCoreFunctionalTestCase;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertSame;
import static org.junit.Assert.assertTrue;
/**
* @author Gavin King
* @author Lukasz Antoniak (lukasz dot antoniak at gmail dot com)
*/
public class InterceptorTest extends BaseCoreFunctionalTestCase {
@Override
public String[] getMappings() {
return new String[] { "interceptor/User.hbm.xml", "interceptor/Image.hbm.xml" };
}
@Test
public void testCollectionIntercept() {
Session s = openSession( new CollectionInterceptor() );
Transaction t = s.beginTransaction();
User u = new User("Gavin", "nivag");
s.persist(u);
u.setPassword("vagni");
t.commit();
s.close();
s = openSession();
t = s.beginTransaction();
u = (User) s.get(User.class, "Gavin");
assertEquals( 2, u.getActions().size() );
s.delete(u);
t.commit();
s.close();
}
@Test
public void testPropertyIntercept() {
Session s = openSession( new PropertyInterceptor() );
Transaction t = s.beginTransaction();
User u = new User("Gavin", "nivag");
s.persist(u);
u.setPassword("vagni");
t.commit();
s.close();
s = openSession();
t = s.beginTransaction();
u = (User) s.get(User.class, "Gavin");
assertNotNull( u.getCreated() );
assertNotNull( u.getLastUpdated() );
s.delete(u);
t.commit();
s.close();
}
/**
* Test case from HHH-1921. Here the interceptor resets the
* current-state to the same thing as the current db state; this
* causes EntityPersister.findDirty() to return no dirty properties.
*/
@Test
@TestForIssue( jiraKey = "HHH-1921" )
public void testPropertyIntercept2() {
Session s = openSession();
Transaction t = s.beginTransaction();
User u = new User("Josh", "test");
s.persist( u );
t.commit();
s.close();
s = openSession(
new EmptyInterceptor() {
public boolean onFlushDirty(Object entity, Serializable id, Object[] currentState, Object[] previousState, String[] propertyNames, Type[] types) {
currentState[0] = "test";
return true;
}
}
);
t = s.beginTransaction();
u = ( User ) s.get( User.class, u.getName() );
u.setPassword( "nottest" );
t.commit();
s.close();
s = openSession();
t = s.beginTransaction();
u = (User) s.get(User.class, "Josh");
assertEquals("test", u.getPassword());
s.delete(u);
t.commit();
s.close();
}
@Test
public void testComponentInterceptor() {
final int checkPerm = 500;
final String checkComment = "generated from interceptor";
Session s = openSession(
new EmptyInterceptor() {
public boolean onSave(Object entity, Serializable id, Object[] state, String[] propertyNames, Type[] types) {
if ( state[0] == null ) {
Image.Details detail = new Image.Details();
detail.setPerm1( checkPerm );
detail.setComment( checkComment );
state[0] = detail;
}
return true;
}
}
);
s.beginTransaction();
Image i = new Image();
i.setName( "compincomp" );
i = ( Image ) s.merge( i );
assertNotNull( i.getDetails() );
assertEquals( checkPerm, i.getDetails().getPerm1() );
assertEquals( checkComment, i.getDetails().getComment() );
s.getTransaction().commit();
s.close();
s = openSession();
s.beginTransaction();
i = ( Image ) s.get( Image.class, i.getId() );
assertNotNull( i.getDetails() );
assertEquals( checkPerm, i.getDetails().getPerm1() );
assertEquals( checkComment, i.getDetails().getComment() );
s.delete( i );
s.getTransaction().commit();
s.close();
}
@Test
public void testStatefulIntercept() {
final StatefulInterceptor statefulInterceptor = new StatefulInterceptor();
Session s = openSession( statefulInterceptor );
statefulInterceptor.setSession(s);
Transaction t = s.beginTransaction();
User u = new User("Gavin", "nivag");
s.persist(u);
u.setPassword("vagni");
t.commit();
s.close();
s = openSession();
t = s.beginTransaction();
List logs = s.createCriteria(Log.class).list();
assertEquals( 2, logs.size() );
s.delete(u);
s.createQuery( "delete from Log" ).executeUpdate();
t.commit();
s.close();
}
@Test
public void testInitiateIntercept() {
final String injectedString = "******";
final InstantiateInterceptor initiateInterceptor = new InstantiateInterceptor( injectedString );
Session s = openSession( initiateInterceptor );
Transaction t = s.beginTransaction();
User u = new User( "Gavin", "nivag" );
s.persist( u );
t.commit();
s.close();
assertNull( u.getInjectedString() );
u.setPassword( "blah" );
s = openSession( initiateInterceptor );
t = s.beginTransaction();
User merged = ( User ) s.merge( u );
assertEquals( injectedString, merged.getInjectedString() );
assertEquals( u.getName(), merged.getName() );
assertEquals( u.getPassword(), merged.getPassword() );
merged.setInjectedString( null );
User loaded = ( User ) s.load(User.class, merged.getName());
// the session-bound instance was not instantiated by the interceptor, load simply returns it
assertSame( merged, loaded );
assertNull( merged.getInjectedString() );
// flush the session and evict the merged instance from session to force an actual load
s.flush();
s.evict( merged );
User reloaded = ( User ) s.load( User.class, merged.getName() );
// Interceptor IS called for instantiating the persistent instance associated to the session when using load
assertEquals( injectedString, reloaded.getInjectedString() );
assertEquals( u.getName(), reloaded.getName() );
assertEquals( u.getPassword(), reloaded.getPassword() );
s.delete( reloaded );
t.commit();
s.close();
}
@Test
@TestForIssue( jiraKey = "HHH-6594" )
public void testPrepareStatementIntercept() {
final Queue<String> expectedSQLs = new LinkedList<String>();
// Transaction 1
expectedSQLs.add( "insert" );
// Transaction 2
expectedSQLs.add( "select" );
expectedSQLs.add( "select" );
// Transaction 3
expectedSQLs.add( "select" );
expectedSQLs.add( "select" );
expectedSQLs.add( "update" );
// Transaction 4
expectedSQLs.add( "select" );
expectedSQLs.add( "delete" );
final Interceptor interceptor = new EmptyInterceptor() {
@Override
public String onPrepareStatement(String sql) {
assertNotNull( sql );
assertTrue( sql.toLowerCase().startsWith( expectedSQLs.poll().toLowerCase() ) );
return sql;
}
};
Session s = openSession(interceptor);
Transaction t = s.beginTransaction();
User u = new User( "Lukasz", "Antoniak" );
s.persist( u );
t.commit();
s.close();
s = openSession(interceptor);
t = s.beginTransaction();
s.get( User.class, "Lukasz" );
s.createQuery( "from User u" ).list();
t.commit();
s.close();
u.setPassword( "Kinga" );
s = openSession(interceptor);
t = s.beginTransaction();
s.merge( u );
t.commit();
s.close();
s = openSession(interceptor);
t = s.beginTransaction();
s.delete( u );
t.commit();
s.close();
assertTrue( expectedSQLs.isEmpty() );
}
@Test(expected = AssertionFailure.class)
public void testPrepareStatementFaultIntercept() {
final Interceptor interceptor = new EmptyInterceptor() {
@Override
public String onPrepareStatement(String sql) {
return null;
}
};
Session s = openSession(interceptor);
Transaction t = s.beginTransaction();
User u = new User( "Kinga", "Mroz" );
s.persist( u );
t.commit();
s.close();
}
}