HHH-7949 - Added test case.

This commit is contained in:
Chris Cranford 2016-12-04 15:07:18 -05:00
parent b0f6f77d74
commit 1081c02c6b
1 changed files with 186 additions and 0 deletions

View File

@ -0,0 +1,186 @@
/*
* 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.modifiedflags;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import javax.persistence.Entity;
import javax.persistence.Id;
import javax.persistence.ManyToOne;
import javax.persistence.OneToMany;
import org.hibernate.envers.Audited;
import org.hibernate.envers.test.Priority;
import org.hibernate.testing.TestForIssue;
import org.hibernate.testing.transaction.TransactionUtil;
import org.junit.Test;
import static org.hibernate.envers.test.tools.TestTools.extractRevisionNumbers;
import static org.junit.Assert.assertEquals;
/**
* @author Chris Cranford
*/
@TestForIssue(jiraKey = "HHH-7949")
public class HasChangedBidirectionalTest extends AbstractModifiedFlagsEntityTest {
@Override
protected Class<?>[] getAnnotatedClasses() {
return new Class<?>[] { Ticket.class, Comment.class };
}
@Test
@Priority(10)
public void initData() throws Exception {
// Revision 1 | Create ticket with comments
TransactionUtil.doInJPA( this::entityManagerFactory, entityManager -> {
final Ticket ticket = new Ticket( 1, "data-t1" );
final Comment comment = new Comment( 1, "Initial comment-t1" );
ticket.addComment( comment );
entityManager.persist( comment );
entityManager.persist( ticket );
} );
// Revision 2 | Create ticket without comments
TransactionUtil.doInJPA( this::entityManagerFactory, entityManager -> {
final Ticket ticket = new Ticket( 2, "data-t2" );
entityManager.persist( ticket );
} );
// Revision 3 | Update ticket with comments
TransactionUtil.doInJPA( this::entityManagerFactory, entityManager -> {
final Ticket ticket = entityManager.find( Ticket.class, 1 );
ticket.setData( "data-changed-t1" );
entityManager.merge( ticket );
} );
// Revision 4 | Update ticket without comments
TransactionUtil.doInJPA( this::entityManagerFactory, entityManager -> {
final Ticket ticket = entityManager.find( Ticket.class, 2 );
ticket.setData( "data-changed-t2" );
entityManager.merge( ticket );
} );
// Revision 5 | Update ticket and comment
TransactionUtil.doInJPA( this::entityManagerFactory, entityManager -> {
final Ticket ticket = entityManager.find( Ticket.class, 1 );
ticket.setData( "data-changed-twice" );
ticket.getComments().get( 0 ).setText( "comment-modified" );
ticket.getComments().forEach( entityManager::merge );
entityManager.merge( ticket );
} );
// Revision 6 | Update ticket and comment collection
TransactionUtil.doInJPA( this::entityManagerFactory, entityManager -> {
final Ticket ticket = entityManager.find( Ticket.class, 1 );
final Comment comment = new Comment( 2, "Comment2" );
ticket.addComment( comment );
entityManager.merge( comment );
entityManager.merge( ticket );
} );
}
@Test
public void testRevisionCounts() {
assertEquals( Arrays.asList( 1, 3, 5, 6 ), getAuditReader().getRevisions( Ticket.class, 1 ) );
assertEquals( Arrays.asList( 2, 4 ), getAuditReader().getRevisions( Ticket.class, 2 ) );
assertEquals( Arrays.asList( 1, 5 ), getAuditReader().getRevisions( Comment.class, 1 ) );
assertEquals( Arrays.asList( 6 ), getAuditReader().getRevisions( Comment.class, 2 ) );
}
@Test
public void testHasChanged() {
assertEquals( Arrays.asList( 1, 6 ), extractRevisionNumbers( queryForPropertyHasChanged( Ticket.class, 1, "comments" ) ) );
assertEquals( Arrays.asList( 2 ), extractRevisionNumbers( queryForPropertyHasChanged( Ticket.class, 2, "comments" ) ) );
}
@Test
public void testHasNotChanged() {
assertEquals( Arrays.asList( 3, 5 ), extractRevisionNumbers( queryForPropertyHasNotChanged( Ticket.class, 1, "comments" ) ) );
assertEquals( Arrays.asList( 4 ), extractRevisionNumbers( queryForPropertyHasNotChanged( Ticket.class, 2, "comments" ) ) );
}
@Entity(name = "Ticket")
@Audited(withModifiedFlag = true)
public static class Ticket {
@Id
private Integer id;
private String data;
@OneToMany(mappedBy = "ticket")
private List<Comment> comments = new ArrayList<>();
Ticket() {
}
public Ticket(Integer id, String data) {
this.id = id;
this.data = data;
}
public Integer getId() {
return id;
}
public String getData() {
return data;
}
public void setData(String data) {
this.data = data;
}
public List<Comment> getComments() {
return comments;
}
public void addComment(Comment comment) {
comment.setTicket( this );
comments.add( comment );
}
}
@Entity(name = "Comment")
@Audited(withModifiedFlag = true)
public static class Comment {
@Id
private Integer id;
@ManyToOne
private Ticket ticket;
private String text;
Comment() {
}
public Comment(Integer id, String text) {
this.id = id;
this.text = text;
}
public Integer getId() {
return id;
}
public Ticket getTicket() {
return ticket;
}
public void setTicket(Ticket ticket) {
this.ticket = ticket;
}
public String getText() {
return text;
}
public void setText(String text) {
this.text = text;
}
}
}