/* * 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.collection; import java.util.Arrays; import java.util.Collections; import javax.persistence.EntityManager; import org.hibernate.envers.test.BaseEnversJPAFunctionalTestCase; import org.hibernate.envers.test.Priority; import org.hibernate.envers.test.entities.collection.StringSetEntity; import org.hibernate.envers.test.tools.TestTools; import org.junit.Test; /** * @author Adam Warski (adam at warski dot org) */ public class StringSet extends BaseEnversJPAFunctionalTestCase { private Integer sse1_id; private Integer sse2_id; @Override protected Class<?>[] getAnnotatedClasses() { return new Class[] {StringSetEntity.class}; } @Test @Priority(10) public void initData() { EntityManager em = getEntityManager(); StringSetEntity sse1 = new StringSetEntity(); StringSetEntity sse2 = new StringSetEntity(); // Revision 1 (sse1: initialy empty, sse2: initialy 2 elements) em.getTransaction().begin(); sse2.getStrings().add( "sse2_string1" ); sse2.getStrings().add( "sse2_string2" ); em.persist( sse1 ); em.persist( sse2 ); em.getTransaction().commit(); // Revision 2 (sse1: adding 2 elements, sse2: adding an existing element) em.getTransaction().begin(); sse1 = em.find( StringSetEntity.class, sse1.getId() ); sse2 = em.find( StringSetEntity.class, sse2.getId() ); sse1.getStrings().add( "sse1_string1" ); sse1.getStrings().add( "sse1_string2" ); sse2.getStrings().add( "sse2_string1" ); em.getTransaction().commit(); // Revision 3 (sse1: removing a non-existing element, sse2: removing one element) em.getTransaction().begin(); sse1 = em.find( StringSetEntity.class, sse1.getId() ); sse2 = em.find( StringSetEntity.class, sse2.getId() ); sse1.getStrings().remove( "sse1_string3" ); sse2.getStrings().remove( "sse2_string1" ); em.getTransaction().commit(); // sse1_id = sse1.getId(); sse2_id = sse2.getId(); } @Test public void testRevisionsCounts() { assert Arrays.asList( 1, 2 ).equals( getAuditReader().getRevisions( StringSetEntity.class, sse1_id ) ); assert Arrays.asList( 1, 3 ).equals( getAuditReader().getRevisions( StringSetEntity.class, sse2_id ) ); } @Test public void testHistoryOfSse1() { StringSetEntity rev1 = getAuditReader().find( StringSetEntity.class, sse1_id, 1 ); StringSetEntity rev2 = getAuditReader().find( StringSetEntity.class, sse1_id, 2 ); StringSetEntity rev3 = getAuditReader().find( StringSetEntity.class, sse1_id, 3 ); assert rev1.getStrings().equals( Collections.EMPTY_SET ); assert rev2.getStrings().equals( TestTools.makeSet( "sse1_string1", "sse1_string2" ) ); assert rev3.getStrings().equals( TestTools.makeSet( "sse1_string1", "sse1_string2" ) ); } @Test public void testHistoryOfSse2() { StringSetEntity rev1 = getAuditReader().find( StringSetEntity.class, sse2_id, 1 ); StringSetEntity rev2 = getAuditReader().find( StringSetEntity.class, sse2_id, 2 ); StringSetEntity rev3 = getAuditReader().find( StringSetEntity.class, sse2_id, 3 ); assert rev1.getStrings().equals( TestTools.makeSet( "sse2_string1", "sse2_string2" ) ); assert rev2.getStrings().equals( TestTools.makeSet( "sse2_string1", "sse2_string2" ) ); assert rev3.getStrings().equals( TestTools.makeSet( "sse2_string2" ) ); } }