Start testing Consumer and ConsumerType.

1 files changed, 19 insertions, 4 deletions

diff --git a/proxy/code/src/org/fedoraproject/candlepin/model/test/ConsumerTest.java b/proxy/code/src/org/fedoraproject/candlepin/model/test/ConsumerTest.java
index bdc0912..52de9b6 100644
--- a/proxy/code/src/org/fedoraproject/candlepin/model/test/ConsumerTest.java
+++ b/proxy/code/src/org/fedoraproject/candlepin/model/test/ConsumerTest.java
@@ -15,6 +15,7 @@
 package org.fedoraproject.candlepin.model.test;
 
 import org.fedoraproject.candlepin.model.Consumer;
+import org.fedoraproject.candlepin.model.ConsumerType;
 import org.fedoraproject.candlepin.model.Owner;
 import org.fedoraproject.candlepin.model.Product;
 import org.fedoraproject.candlepin.test.DatabaseTestFixture;
@@ -29,6 +30,10 @@ public class ConsumerTest extends DatabaseTestFixture {
     
     private Owner owner;
     private Product rhel;
+    private Consumer consumer;
+    private ConsumerType consumerType;
+    private static final String CONSUMER_TYPE_NAME = "test-consumer-type";
+    private static final String CONSUMER_NAME = "Test Consumer";
     
     @Before
     public void setUpTestObjects() {
@@ -40,17 +45,27 @@ public class ConsumerTest extends DatabaseTestFixture {
         em.persist(owner);
         em.persist(rhel);
         
+        consumerType = new ConsumerType(CONSUMER_TYPE_NAME);
+        em.persist(consumerType);
+        consumer = new Consumer(CONSUMER_NAME, owner, consumerType);
+        em.persist(consumer);
+        
         em.getTransaction().commit();
     }
 
     @Test
-    public void testConsumedProduct() throws Exception {
+    public void testLookup() throws Exception {
+        
+        Consumer lookedUp = (Consumer)em.createQuery(
+                "from Consumer c where c.name = :name").
+                setParameter("name", CONSUMER_NAME).
+                getSingleResult();
+        assertEquals(consumer.getId(), lookedUp.getId());
+        assertEquals(consumer.getName(), lookedUp.getName());
+        assertEquals(consumer.getType().getLabel(), lookedUp.getType().getLabel());
 //        
 //        Consumer c = TestUtil.createConsumer(o);
 //        c.addConsumedProduct(rhel);
-        
-        
-        
     }
     
 //    @Test