Fix sample test for ConsumerType via EntityManager.

4 files changed, 42 insertions, 1 deletions

diff --git a/proxy/code/src/META-INF/persistence.xml b/proxy/code/src/META-INF/persistence.xml
index 881b514..2ecd9fd 100644
--- a/proxy/code/src/META-INF/persistence.xml
+++ b/proxy/code/src/META-INF/persistence.xml
@@ -5,7 +5,6 @@
 
     <persistence-unit name="test" transaction-type="RESOURCE_LOCAL">
         <provider>org.hibernate.ejb.HibernatePersistence</provider>
-        <exclude-unlisted-classes>true</exclude-unlisted-classes>
         <properties>
             <property name="hibernate.connection.url" value="jdbc:hsqldb:mem:unit-testing-jpa"/>
             <property name="hibernate.connection.driver_class" value="org.hsqldb.jdbcDriver"/>
diff --git a/proxy/code/src/org/fedoraproject/candlepin/model/Consumer.java b/proxy/code/src/org/fedoraproject/candlepin/model/Consumer.java
index fe11886..9af8f4e 100644
--- a/proxy/code/src/org/fedoraproject/candlepin/model/Consumer.java
+++ b/proxy/code/src/org/fedoraproject/candlepin/model/Consumer.java
@@ -23,6 +23,8 @@ import javax.xml.bind.annotation.XmlAccessorType;
 import javax.xml.bind.annotation.XmlRootElement;
 import javax.xml.bind.annotation.XmlTransient;
 
+import javax.persistence.*;
+
 /**
  * Represents Consumer entity
  */
diff --git a/proxy/code/src/org/fedoraproject/candlepin/model/ConsumerType.java b/proxy/code/src/org/fedoraproject/candlepin/model/ConsumerType.java
index d441b27..9ead523 100644
--- a/proxy/code/src/org/fedoraproject/candlepin/model/ConsumerType.java
+++ b/proxy/code/src/org/fedoraproject/candlepin/model/ConsumerType.java
@@ -14,6 +14,8 @@
  */
 package org.fedoraproject.candlepin.model;
 
+import javax.persistence.Entity;
+import javax.persistence.Id;
 import javax.xml.bind.annotation.XmlAccessType;
 import javax.xml.bind.annotation.XmlAccessorType;
 import javax.xml.bind.annotation.XmlRootElement;
@@ -23,6 +25,7 @@ import javax.xml.bind.annotation.XmlRootElement;
  */
 @XmlRootElement
 @XmlAccessorType(XmlAccessType.PROPERTY)
+@Entity
 public class ConsumerType extends BaseModel {
 
     private String label;
@@ -46,6 +49,7 @@ public class ConsumerType extends BaseModel {
     /**
      * @return Returns the label.
      */
+    @Id
     public String getLabel() {
         return label;
     }
diff --git a/proxy/code/src/org/fedoraproject/candlepin/model/test/ConsumerTypeTest.java b/proxy/code/src/org/fedoraproject/candlepin/model/test/ConsumerTypeTest.java
new file mode 100644
index 0000000..92e810a
--- /dev/null
+++ b/proxy/code/src/org/fedoraproject/candlepin/model/test/ConsumerTypeTest.java
@@ -0,0 +1,36 @@
+package org.fedoraproject.candlepin.model.test;
+
+import java.util.List;
+
+import javax.persistence.EntityManager;
+import javax.persistence.EntityManagerFactory;
+import javax.persistence.EntityTransaction;
+import javax.persistence.Persistence;
+
+import org.fedoraproject.candlepin.model.ConsumerType;
+import org.junit.Test;
+import static org.junit.Assert.*;
+
+public class ConsumerTypeTest {
+
+    @Test
+    public void testSomething() {
+        EntityManagerFactory emf = 
+            Persistence.createEntityManagerFactory("test");
+//        HibernateUtil.getSession();
+        ConsumerType ct = new ConsumerType("standard-system");
+        EntityManager em = emf.createEntityManager();
+        EntityTransaction tx = null;
+        tx = em.getTransaction();
+        tx.begin();
+        
+        em.persist(ct);
+        tx.commit();
+        
+        List<EntityManager> results = em.createQuery("select ct from ConsumerType as ct")
+            .getResultList();
+        assertEquals(1, results.size());
+        
+    
+    }
+}