Map Product hierarchy, improve unit test db cleanup.

1 files changed, 78 insertions, 1 deletions

diff --git a/proxy/code/src/org/fedoraproject/candlepin/model/test/ProductTest.java b/proxy/code/src/org/fedoraproject/candlepin/model/test/ProductTest.java
index a492741..c139ce5 100644
--- a/proxy/code/src/org/fedoraproject/candlepin/model/test/ProductTest.java
+++ b/proxy/code/src/org/fedoraproject/candlepin/model/test/ProductTest.java
@@ -4,9 +4,11 @@ import static org.junit.Assert.assertEquals;
 
 import java.util.List;
 
+import javax.persistence.EntityManager;
 import javax.persistence.PersistenceException;
 
 import org.fedoraproject.candlepin.model.Product;
+import org.fedoraproject.candlepin.util.EntityManagerUtil;
 import org.junit.Test;
 
 public class ProductTest extends ModelTestFixture {
@@ -15,12 +17,16 @@ public class ProductTest extends ModelTestFixture {
     @Test
     public void normalCreate() { 
     
-        Product prod = new Product("myproductlabel", "My Product");
+        Product prod = new Product("cptest-label", "My Product");
         persistAndCommit(prod);
         
         List<Product> results = em.createQuery("select p from Product as p")
             .getResultList();
         assertEquals(1, results.size());
+        
+        em.getTransaction().begin();
+        em.remove(prod);
+        em.getTransaction().commit();
     }
     
     @Test(expected = PersistenceException.class)
@@ -38,7 +44,78 @@ public class ProductTest extends ModelTestFixture {
         Product prod = new Product();
         prod.setName("My Product Name");
         persistAndCommit(prod);
+    }
+    
+    @Test(expected = PersistenceException.class)
+    public void nameUnique() { 
+    
+        Product prod = new Product("label1", "name");
+        persistAndCommit(prod);
+        
+        Product prod2 = new Product("label2", "name");
+        persistAndCommit(prod2);
+    }
+    
+    @Test(expected = PersistenceException.class)
+    public void labelUnique() { 
+
+        Product prod = new Product("label1", "name");
+        Product prod2 = new Product("label1", "name2");
+        persistAndCommit(prod);
         
+        try {
+            persistAndCommit(prod2);
+        }
+        catch (PersistenceException e) {
+            em.remove(prod);
+            em.getTransaction().commit();
+            throw e;
+        }
     }
     
+    @Test
+    public void addChildProducts() {
+        em.getTransaction().begin();
+        Product parent = new Product("parent-product", "Parent Product");
+        Product child1 = new Product("child-product1", "Child Product 1");
+        Product child2 = new Product("child-product2", "Child Product 2");
+        
+        parent.addChildProduct(child1);
+        parent.addChildProduct(child2);
+        em.persist(child1);
+        em.persist(child2);
+        em.persist(parent);
+        em.getTransaction().commit();
+        
+        EntityManager em2 = EntityManagerUtil.createEntityManager();
+        Product result = (Product)em2.createQuery(
+                "select p from Product as p where name = :name")
+                .setParameter("name", parent.getName())
+                .getSingleResult();
+        assertEquals(2, result.getChildProducts().size());
+        
+        em.getTransaction().begin();
+        em.remove(parent);
+        em.remove(child1);
+        em.remove(child2);
+        em.getTransaction().commit();
+    }
+
+    @Test
+    public void childHasSingleParentOnly() {
+        em.getTransaction().begin();
+        
+        Product parent1 = new Product("parent-product1", "Parent Product 1");
+        Product child1 = new Product("child-product1", "Child Product 1");
+        Product parent2 = new Product("parent-product2", "Parent Product 2");
+        
+        parent1.addChildProduct(child1);
+//        parent2.addChildProduct(child1);
+        
+        em.persist(child1);
+        em.persist(parent1);
+        em.persist(parent2);
+        em.getTransaction().commit();
+    }
+
 }