diff --git a/api/src/test/java/org/openmrs/module/patientflags/FlagServiceTest.java b/api/src/test/java/org/openmrs/module/patientflags/FlagServiceTest.java index 5f33abe1..a883c900 100644 --- a/api/src/test/java/org/openmrs/module/patientflags/FlagServiceTest.java +++ b/api/src/test/java/org/openmrs/module/patientflags/FlagServiceTest.java @@ -8,11 +8,13 @@ import static org.junit.Assert.assertFalse; import static org.junit.Assert.assertNotNull; import static org.junit.Assert.assertNull; +import static org.junit.Assert.assertTrue; import org.junit.Before; import org.junit.Test; import org.openmrs.Cohort; import org.openmrs.Patient; +import org.openmrs.api.APIException; import org.openmrs.api.PatientService; import org.openmrs.api.context.Context; @@ -112,7 +114,73 @@ public void savePriority_shouldSaveNewPriority() { assertNotNull(savedPriority); assertEquals(priority.getName(), savedPriority.getName()); } - + + @Test + public void getAllPriorities_shouldGetAllPriorities() { + List priorities = flagService.getAllPriorities(); + assertFalse(priorities.isEmpty()); + } + + @Test + public void getPriority_shouldGetPriorityById() { + Integer priorityId = 1; + Priority priority = flagService.getPriority(priorityId); + assertNotNull(priority); + assertEquals(priorityId, priority.getPriorityId()); + } + + @Test + public void getPriorityByUuid_shouldGetPriorityByUuid() { + String priorityUuid = "da7f524f-27ce-4bb2-86d6-6d1d05312bd5"; + Priority priority = flagService.getPriorityByUuid(priorityUuid); + assertNotNull(priority); + assertEquals(priorityUuid, priority.getUuid()); + } + + @Test + public void getPriorityByName_shouldGetPriorityByName() { + String priorityName = "pr 2"; + Priority priority = flagService.getPriorityByName(priorityName); + assertNotNull(priority); + assertEquals(priorityName, priority.getName()); + } + + @Test + public void purgePriority_shouldPurgePriority() { + Integer priorityId = 2; + flagService.purgePriority(priorityId); + Priority priority = flagService.getPriority(priorityId); + assertNull(priority); + } + + @Test(expected = APIException.class) + public void purgePriority_shouldTrowExceptionWhenPriorityAssociatedWithAFlag() { + Integer priorityId = 1; + + flagService.purgePriority(priorityId); + Priority purgedPriority = flagService.getPriority(priorityId); + assertNotNull(purgedPriority); + } + + @Test + public void retirePriority_shouldRetirePriorityWithReason() { + String reason = "reason"; + Priority priority = flagService.getPriority(2); + flagService.retirePriority(priority, reason); + Priority retiredpriority = flagService.getPriority(2); + assertTrue(retiredpriority.getRetired()); + assertEquals(reason, retiredpriority.getRetireReason()); + } + + @Test(expected = APIException.class) + public void retirePriority_shouldNotRetirePriorityWithoutReason() { + String reason = ""; + Priority priority = flagService.getPriority(2); + flagService.retirePriority(priority, reason); + Priority retiredpriority = flagService.getPriority(2); + assertFalse(retiredpriority.getRetired()); + } + /** * Tests of methods that save and retrieve flags */ diff --git a/api/src/test/resources/org/openmrs/module/patientflags/include/patientflagtest-dataset.xml b/api/src/test/resources/org/openmrs/module/patientflags/include/patientflagtest-dataset.xml index b1cd24b9..72cb49ab 100644 --- a/api/src/test/resources/org/openmrs/module/patientflags/include/patientflagtest-dataset.xml +++ b/api/src/test/resources/org/openmrs/module/patientflags/include/patientflagtest-dataset.xml @@ -27,7 +27,11 @@ - + + + + + \ No newline at end of file