@Test
@Verifies(
value = "should return a list consisting of active, not retired, states",
method = "getPossibleNextStates")
public void getPossibleNextStates_shouldReturnNonRetiredStates() throws Exception {
executeDataSet(PROGRAM_NEXT_STATES_XML);
Integer patient = 11;
Integer workflow = 501;
Vector<ListItem> possibleNextStates;
/* retire a workflow state */
PatientProgram pp = Context.getProgramWorkflowService().getPatientProgram(patient);
ProgramWorkflow pw = pp.getProgram().getWorkflow(workflow);
Set<ProgramWorkflowState> pwss = pw.getStates();
for (ProgramWorkflowState pws : pwss) {
Concept cp = pws.getConcept();
ConceptName cn = cp.getName();
if (cn != null) {
String cnn = cn.getName();
if (cnn.equalsIgnoreCase("Test State 3")) {
pws.setRetired(true);
}
}
}
possibleNextStates = dwrProgramWorkflowService.getPossibleNextStates(patient, workflow);
assertFalse(possibleNextStates.isEmpty());
assertEquals(2, possibleNextStates.size());
}
/**
* Takes a "program_id:list" where program_id is the id of the program that this collection is for
* (or not present, if it's a new program) and list is a space-separated list of concept ids. This
* class is a bit of a hack, because I don't know a better way to do this. -DJ The purpose is to
* retire and un-retire workflows where possible rather than deleting and creating them.
*/
public void setAsText(String text) throws IllegalArgumentException {
if (StringUtils.hasText(text)) {
ConceptService cs = Context.getConceptService();
ProgramWorkflowService pws = Context.getProgramWorkflowService();
try {
int ind = text.indexOf(":");
String progIdStr = text.substring(0, ind);
text = text.substring(ind + 1);
if (program == null)
// if a program wasn't passed in, try to look it up now
program = pws.getProgram(Integer.valueOf(progIdStr));
} catch (Exception ex) {
}
String[] conceptIds = text.split(" ");
Set<ProgramWorkflow> oldSet =
program == null ? new HashSet<ProgramWorkflow>() : program.getAllWorkflows();
Set<Integer> newConceptIds = new HashSet<Integer>();
for (String id : conceptIds) {
if (id.trim().length() == 0) continue;
log.debug("trying " + id);
newConceptIds.add(Integer.valueOf(id.trim()));
}
// go through oldSet and see what we need to keep and what we need to unvoid
Set<Integer> alreadyDone = new HashSet<Integer>();
for (ProgramWorkflow pw : oldSet) {
if (!newConceptIds.contains(pw.getConcept().getConceptId())) {
pw.setRetired(true);
} else if (pw.isRetired()) { // && newConceptIds.contains(pw...)
pw.setRetired(false);
}
alreadyDone.add(pw.getConcept().getConceptId());
}
// now add any new ones
newConceptIds.removeAll(alreadyDone);
for (Integer conceptId : newConceptIds) {
ProgramWorkflow pw = new ProgramWorkflow();
pw.setProgram(program);
pw.setConcept(cs.getConcept(conceptId));
oldSet.add(pw);
}
setValue(oldSet);
} else {
setValue(null);
}
}
/**
* Convert this program's workflows into "id: wkflowid wkflowid wkflowid"
*
* @see java.beans.PropertyEditorSupport#getAsText()
*/
@SuppressWarnings("unchecked")
public String getAsText() {
Collection<ProgramWorkflow> pws = (Collection<ProgramWorkflow>) getValue();
if (pws == null || pws.size() == 0) {
return ":";
} else {
Integer progId = null;
for (ProgramWorkflow pw : pws) {
if (pw.getProgram() != null && pw.getProgram().getProgramId() != null) {
progId = pw.getProgram().getProgramId();
break;
}
}
StringBuilder ret = new StringBuilder();
if (progId != null) {
ret.append(progId);
}
ret.append(":");
for (ProgramWorkflow pw : pws) {
ret.append(pw.getConcept().getConceptId()).append(" ");
}
return ret.toString().trim();
}
}