Skip to content

Commit

Permalink
HHH-7949 - Added test case.
Browse files Browse the repository at this point in the history
  • Loading branch information
Naros committed Dec 9, 2016
1 parent b0f6f77 commit 1081c02
Showing 1 changed file with 186 additions and 0 deletions.
@@ -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;
}
}
}

0 comments on commit 1081c02

Please sign in to comment.