/*
* 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.envers.test.integration.reventity;
import java.util.Arrays;
import java.util.Date;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import javax.persistence.EntityManager;
import org.hibernate.envers.AuditReader;
import org.hibernate.envers.exception.RevisionDoesNotExistException;
import org.hibernate.envers.test.BaseEnversJPAFunctionalTestCase;
import org.hibernate.envers.test.Priority;
import org.hibernate.envers.test.entities.StrTestEntity;
import org.junit.Test;
/**
* @author Adam Warski (adam at warski dot org)
*/
public class CustomBoxed extends BaseEnversJPAFunctionalTestCase {
private Integer id;
private long timestamp1;
private long timestamp2;
private long timestamp3;
@Override
protected Class<?>[] getAnnotatedClasses() {
return new Class[] {StrTestEntity.class, CustomBoxedRevEntity.class};
}
@Test
@Priority(10)
public void initData() throws InterruptedException {
timestamp1 = System.currentTimeMillis();
Thread.sleep( 100 );
// Revision 1
EntityManager em = getEntityManager();
em.getTransaction().begin();
StrTestEntity te = new StrTestEntity( "x" );
em.persist( te );
id = te.getId();
em.getTransaction().commit();
timestamp2 = System.currentTimeMillis();
Thread.sleep( 100 );
// Revision 2
em.getTransaction().begin();
te = em.find( StrTestEntity.class, id );
te.setStr( "y" );
em.getTransaction().commit();
timestamp3 = System.currentTimeMillis();
}
@Test(expected = RevisionDoesNotExistException.class)
public void testTimestamps1() {
getAuditReader().getRevisionNumberForDate( new Date( timestamp1 ) );
}
@Test
public void testTimestamps() {
assert getAuditReader().getRevisionNumberForDate( new Date( timestamp2 ) ).intValue() == 1;
assert getAuditReader().getRevisionNumberForDate( new Date( timestamp3 ) ).intValue() == 2;
}
@Test
public void testDatesForRevisions() {
AuditReader vr = getAuditReader();
assert vr.getRevisionNumberForDate( vr.getRevisionDate( 1 ) ).intValue() == 1;
assert vr.getRevisionNumberForDate( vr.getRevisionDate( 2 ) ).intValue() == 2;
}
@Test
public void testRevisionsForDates() {
AuditReader vr = getAuditReader();
assert vr.getRevisionDate( vr.getRevisionNumberForDate( new Date( timestamp2 ) ) ).getTime() <= timestamp2;
assert vr.getRevisionDate( vr.getRevisionNumberForDate( new Date( timestamp2 ) ).intValue() + 1 )
.getTime() > timestamp2;
assert vr.getRevisionDate( vr.getRevisionNumberForDate( new Date( timestamp3 ) ) ).getTime() <= timestamp3;
}
@Test
public void testFindRevision() {
AuditReader vr = getAuditReader();
long rev1Timestamp = vr.findRevision( CustomBoxedRevEntity.class, 1 ).getCustomTimestamp();
assert rev1Timestamp > timestamp1;
assert rev1Timestamp <= timestamp2;
long rev2Timestamp = vr.findRevision( CustomBoxedRevEntity.class, 2 ).getCustomTimestamp();
assert rev2Timestamp > timestamp2;
assert rev2Timestamp <= timestamp3;
}
@Test
public void testFindRevisions() {
AuditReader vr = getAuditReader();
Set<Number> revNumbers = new HashSet<Number>();
revNumbers.add( 1 );
revNumbers.add( 2 );
Map<Number, CustomBoxedRevEntity> revisionMap = vr.findRevisions( CustomBoxedRevEntity.class, revNumbers );
assert (revisionMap.size() == 2);
assert (revisionMap.get( 1 ).equals( vr.findRevision( CustomBoxedRevEntity.class, 1 ) ));
assert (revisionMap.get( 2 ).equals( vr.findRevision( CustomBoxedRevEntity.class, 2 ) ));
}
@Test
public void testRevisionsCounts() {
assert Arrays.asList( 1, 2 ).equals( getAuditReader().getRevisions( StrTestEntity.class, id ) );
}
@Test
public void testHistoryOfId1() {
StrTestEntity ver1 = new StrTestEntity( "x", id );
StrTestEntity ver2 = new StrTestEntity( "y", id );
assert getAuditReader().find( StrTestEntity.class, id, 1 ).equals( ver1 );
assert getAuditReader().find( StrTestEntity.class, id, 2 ).equals( ver2 );
}
}