1 22 package org.jboss.test.cmp2.enums.test; 23 24 import java.util.List ; 25 import net.sourceforge.junitejb.EJBTestCase; 26 import junit.framework.Test; 27 import org.jboss.test.JBossTestCase; 28 import org.jboss.test.cmp2.enums.ejb.Facade; 29 import org.jboss.test.cmp2.enums.ejb.FacadeUtil; 30 import org.jboss.test.cmp2.enums.ejb.ColorEnum; 31 import org.jboss.test.cmp2.enums.ejb.AnimalEnum; 32 import org.jboss.test.cmp2.enums.ejb.IDClass; 33 34 39 public class EnumUnitTestCase 40 extends EJBTestCase 41 { 42 public static Test suite() throws Exception 43 { 44 return JBossTestCase.getDeploySetup(EnumUnitTestCase.class, "cmp2-enum.jar"); 45 } 46 47 public EnumUnitTestCase(String s) 48 { 49 super(s); 50 } 51 52 54 public void testColorEnum() 55 throws Exception 56 { 57 Facade facade = FacadeUtil.getHome().create(); 58 IDClass childId = new IDClass(1); 59 facade.createChild(childId); 60 assertTrue(ColorEnum.RED == facade.getColorForId(childId)); 61 facade.setColor(childId, ColorEnum.GREEN); 62 assertTrue(ColorEnum.GREEN == facade.getColorForId(childId)); 63 facade.setColor(childId, ColorEnum.BLUE); 64 assertTrue(ColorEnum.BLUE == facade.getColorForId(childId)); 65 facade.removeChild(childId); 66 } 67 68 public void testAnimalEnum() 69 throws Exception 70 { 71 Facade facade = FacadeUtil.getHome().create(); 72 IDClass childId = new IDClass(2); 73 facade.createChild(childId); 74 assertTrue(AnimalEnum.PENGUIN == facade.getAnimalForId(childId)); 75 facade.setAnimal(childId, AnimalEnum.DOG); 76 assertTrue(AnimalEnum.DOG == facade.getAnimalForId(childId)); 77 facade.setAnimal(childId, AnimalEnum.CAT); 78 assertTrue(AnimalEnum.CAT == facade.getAnimalForId(childId)); 79 facade.removeChild(childId); 80 } 81 82 public void testFindByColor() 83 throws Exception 84 { 85 Facade facade = FacadeUtil.getHome().create(); 86 IDClass childId = new IDClass(3); 87 facade.createChild(childId); 88 try 89 { 90 facade.setColor(childId, ColorEnum.BLUE); 91 IDClass id = facade.findByColor(ColorEnum.BLUE); 92 assertEquals(childId, id); 93 } 94 finally 95 { 96 facade.removeChild(childId); 97 } 98 } 99 100 public void testFindAndOrderByColor() 101 throws Exception 102 { 103 Facade facade = FacadeUtil.getHome().create(); 104 IDClass childId = new IDClass(3); 105 facade.createChild(childId); 106 try 107 { 108 facade.setColor(childId, ColorEnum.BLUE); 109 IDClass id = facade.findAndOrderByColor(ColorEnum.BLUE); 110 assertEquals(childId, id); 111 } 112 finally 113 { 114 facade.removeChild(childId); 115 } 116 } 117 118 public void testFindByColorDeclaredSql() 119 throws Exception 120 { 121 Facade facade = FacadeUtil.getHome().create(); 122 IDClass childId = new IDClass(4); 123 facade.createChild(childId); 124 try 125 { 126 facade.setColor(childId, ColorEnum.BLUE); 127 IDClass id = facade.findByColorDeclaredSql(ColorEnum.BLUE); 128 assertEquals(childId, id); 129 } 130 finally 131 { 132 facade.removeChild(childId); 133 } 134 } 135 136 public void testLowColor() 137 throws Exception 138 { 139 Facade facade = FacadeUtil.getHome().create(); 140 IDClass childId = new IDClass(3); 141 facade.createChild(childId); 142 try 143 { 144 facade.setColor(childId, ColorEnum.RED); 145 List ids = facade.findLowColor(ColorEnum.BLUE); 146 assertEquals(1, ids.size()); 147 assertEquals(childId, ids.get(0)); 148 } 149 finally 150 { 151 facade.removeChild(childId); 152 } 153 } 154 } 155 | Popular Tags |