HHH-6661 - Test case

hibernate-envers/src/matrix/java/org/hibernate/envers/test/integration/onetomany/hierarchy/HierarchyTest.java

@@ -0,0 +1,106 @@
+package org.hibernate.envers.test.integration.onetomany.hierarchy;
+
+import org.hibernate.ejb.Ejb3Configuration;
+import org.hibernate.envers.test.AbstractEntityTest;
+import org.hibernate.envers.test.Priority;
+import org.hibernate.envers.test.tools.TestTools;
+import org.hibernate.testing.TestForIssue;
+import org.junit.Assert;
+import org.junit.Test;
+
+import javax.persistence.EntityManager;
+import java.util.Arrays;
+
+/**
+ * @author Lukasz Antoniak (lukasz dot antoniak at gmail dot com)
+ */
+@TestForIssue(jiraKey = "HHH-6661")
+public class HierarchyTest extends AbstractEntityTest {
+    private Long parentId = null;
+    private Long child1Id = null;
+    private Long child2Id = null;
+
+    public void configure(Ejb3Configuration cfg) {
+        cfg.addAnnotatedClass(Node.class);
+    }
+
+    @Test
+    @Priority(10)
+    public void initData() {
+        EntityManager em = getEntityManager();
+
+        // Revision 1
+        em.getTransaction().begin();
+        Node parent = new Node("parent", (Node)null);
+        Node child1 = new Node("child1", parent);
+        Node child2 = new Node("child2", parent);
+        parent.getChildren().add(child1); 
+        parent.getChildren().add(child2);
+        em.persist(parent);
+        em.persist(child1);
+        em.persist(child2);
+        em.getTransaction().commit();
+
+        parentId = parent.getId();
+        child1Id = child1.getId();
+        child2Id = child2.getId();
+
+        // Revision 2
+        em.getTransaction().begin();
+        parent = em.find(Node.class, parent.getId());
+        parent.getChildren().get(0).setData("child1 modified");
+        em.getTransaction().commit();
+
+        // Revision 3
+        em.getTransaction().begin();
+        child2 = em.find(Node.class, child2.getId());
+        em.remove(child2);
+        em.getTransaction().commit();
+    }
+
+    @Test
+    public void testRevisionsCounts() {
+        Assert.assertEquals(Arrays.asList(1, 3), getAuditReader().getRevisions(Node.class, parentId));
+        Assert.assertEquals(Arrays.asList(1, 2), getAuditReader().getRevisions(Node.class, child1Id));
+        Assert.assertEquals(Arrays.asList(1, 3), getAuditReader().getRevisions(Node.class, child2Id));
+    }
+
+    @Test
+    public void testHistoryOfParentNode() {
+        Node parent = new Node("parent", parentId);
+        Node child1 = new Node("child1", child1Id);
+        Node child2 = new Node("child2", child2Id);
+
+        Node ver1 = getAuditReader().find(Node.class, parentId, 1);
+        Assert.assertEquals(parent, ver1);
+        Assert.assertTrue(TestTools.checkList(ver1.getChildren(), child1, child2));
+
+        child1.setData("child1 modified");
+
+        Node ver2 = getAuditReader().find(Node.class, parentId, 2);
+        Assert.assertEquals(parent, ver2);
+        Assert.assertTrue(TestTools.checkList(ver2.getChildren(), child1, child2));
+
+        Node ver3 = getAuditReader().find(Node.class, parentId, 3);
+        Assert.assertEquals(parent, ver3);
+        Assert.assertTrue(TestTools.checkList(ver3.getChildren(), child1));
+    }
+
+    @Test
+    public void testHistoryOfChild1Node() {
+        Node parent = new Node("parent", parentId);
+        Node child1 = new Node("child1", child1Id);
+
+        Node ver1 = getAuditReader().find(Node.class, child1Id, 1);
+        Assert.assertEquals(child1, ver1);
+        Assert.assertEquals(parent.getId(), ver1.getParent().getId());
+        Assert.assertEquals(parent.getData(), ver1.getParent().getData());
+
+        child1.setData("child1 modified");
+
+        Node ver2 = getAuditReader().find(Node.class, child1Id, 2);
+        Assert.assertEquals(child1, ver2);
+        Assert.assertEquals(parent.getId(), ver2.getParent().getId());
+        Assert.assertEquals(parent.getData(), ver2.getParent().getData());
+    }
+}

hibernate-envers/src/matrix/java/org/hibernate/envers/test/integration/onetomany/hierarchy/Node.java

@@ -0,0 +1,104 @@
+package org.hibernate.envers.test.integration.onetomany.hierarchy;
+
+import org.hibernate.envers.AuditJoinTable;
+import org.hibernate.envers.Audited;
+
+import javax.persistence.*;
+import java.io.Serializable;
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ * @author Lukasz Antoniak (lukasz dot antoniak at gmail dot com)
+ */
+@Entity
+@Table(name = "NODES")
+@Audited
+public class Node implements Serializable {
+    @Id
+    @GeneratedValue
+    private Long id;
+
+    private String data;
+
+    @ManyToOne(fetch = FetchType.LAZY, optional = true)
+    @AuditJoinTable(name = "NODES_JOIN_TABLE_AUD",
+                    inverseJoinColumns = {@JoinColumn(name = "PARENT_ID", nullable = true, updatable = false)})
+    @JoinTable(name = "NODES_JOIN_TABLE",
+               joinColumns = {@JoinColumn(name = "CHILD_ID", nullable = true, updatable = false)},
+               inverseJoinColumns = {@JoinColumn(name = "PARENT_ID", nullable = true, updatable = false)})
+    private Node parent;
+
+    @OneToMany(mappedBy = "parent")
+    private List<Node> children = new ArrayList<Node>();
+
+    public Node() {
+    }
+
+    public Node(String data, Node parent) {
+        this.data = data;
+        this.parent = parent;
+    }
+
+    public Node(String data, Long id) {
+        this.id = id;
+        this.data = data;
+    }
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (!(o instanceof Node)) return false;
+
+        Node node = (Node) o;
+
+        if (data != null ? !data.equals(node.data) : node.data != null) return false;
+        if (id != null ? !id.equals(node.id) : node.id != null) return false;
+
+        return true;
+    }
+
+    @Override
+    public int hashCode() {
+        int result = id != null ? id.hashCode() : 0;
+        result = 31 * result + (data != null ? data.hashCode() : 0);
+        return result;
+    }
+
+    @Override
+    public String toString() {
+        return "Node(id = " + id + ", data = " + data + ")";
+    }
+
+    public Long getId() {
+        return id;
+    }
+
+    public void setId(Long id) {
+        this.id = id;
+    }
+
+    public String getData() {
+        return data;
+    }
+
+    public void setData(String data) {
+        this.data = data;
+    }
+
+    public Node getParent() {
+        return parent;
+    }
+
+    public void setParent(Node parent) {
+        this.parent = parent;
+    }
+
+    public List<Node> getChildren() {
+        return children;
+    }
+
+    public void setChildren(List<Node> children) {
+        this.children = children;
+    }
+}