@BeforeClass public void setup() { troopA = new Troop(); troopA.setName("Angels"); troopA.setMoney(BigDecimal.ZERO); troopA.setMission("mise A"); troopB = new Troop(); troopB.setName("Bulls"); troopB.setMission("mise B"); troopB.setMoney(new BigDecimal("5")); }
@Test public void testFindTroopByName() { Troop troopE = new Troop(); troopE.setName("E"); troopE.setMoney(BigDecimal.ZERO); troopE.setMission("mise E"); troopDao.createTroop(troopE); Troop t1 = troopDao.findTroopById(troopE.getId()); Assert.assertEquals(t1.getName(), "E"); Assert.assertEquals(t1.getMission(), "mise E"); }
@Test public void testUpdateGoldForTroop() { EntityManager entityManager = emf.createEntityManager(); entityManager.getTransaction().begin(); Troop troopC = new Troop(); troopC.setName("Testers"); troopC.setMoney(new BigDecimal("30")); entityManager.persist(troopC); entityManager.getTransaction().commit(); entityManager.close(); troopDao.updateGoldForTroop(troopC.getId(), new BigDecimal("5")); Troop t1 = troopDao.findTroopById(troopC.getId()); Assert.assertEquals(t1.getMoney().compareTo(new BigDecimal("5")), 0); }
@Test public void testDeleteTroop() { EntityManager entityManager = emf.createEntityManager(); entityManager.getTransaction().begin(); Troop troopC = new Troop(); troopC.setName("Testers"); troopC.setMoney(new BigDecimal("30")); troopC.setMission("mise C"); entityManager.persist(troopC); entityManager.getTransaction().commit(); entityManager.close(); Troop t1 = troopDao.findTroopById(troopC.getId()); troopDao.deleteTroop(t1); Troop t2 = troopDao.findTroopById(troopC.getId()); Assert.assertEquals(t2, null); }