diff --git a/YACOSTest/src/org/yacos/tests/core/UserManagerTest.java b/YACOSTest/src/org/yacos/tests/core/UserManagerTest.java index 45afb37..4a747bf 100644 --- a/YACOSTest/src/org/yacos/tests/core/UserManagerTest.java +++ b/YACOSTest/src/org/yacos/tests/core/UserManagerTest.java @@ -3,6 +3,7 @@ package org.yacos.tests.core; import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertFalse; import static org.junit.Assert.assertNotNull; +import static org.junit.Assert.assertTrue; import java.util.ArrayList; import java.util.Date; @@ -83,7 +84,7 @@ public class UserManagerTest { um.addSpeciality("xml"); list = um.getSpecialities(); assertEquals(size, list.size()); - um.removeSpeciality(list.get(list.size()-1).getId()); + um.removeSpeciality(spe.getId()); } @Test @@ -94,11 +95,17 @@ public class UserManagerTest { um.addSpecialityToUser(spe.getId(), "truc"); list = um.getSpecialitiesFromUser("truc"); assertFalse(list.isEmpty()); + um.removeSpeciality(spe.getId()); } @Test - public void getSpecialitiesFromUser(){ - + public void removeSpecialitiesFromUser(){ + um.addSpeciality("xml"); + List list = um.getSpecialities(); + Speciality spe = list.get(list.size()-1); + um.addSpecialityToUser(spe.getId(), "truc"); + um.removeSpecialityFromUser(spe.getId(), "truc"); + assertTrue(list.isEmpty()); } @Test