use of org.pmiops.workbench.db.model.ParticipantCohortStatus in project workbench by all-of-us.
the class ParticipantCohortStatusDaoTest method findAllSearchCriteriaIn.
@Test
public void findAllSearchCriteriaIn() throws Exception {
PageRequest pageRequest = new PageRequest(page, pageSize, SortOrder.ASC, ParticipantCohortStatusColumns.PARTICIPANTID.toString());
List<Filter> filters = new ArrayList<>();
filters.add(new Filter().property(ParticipantCohortStatusColumns.PARTICIPANTID).operator(Operator.IN).values(Arrays.asList("1", "2")));
filters.add(new Filter().property(ParticipantCohortStatusColumns.STATUS).operator(Operator.IN).values(Arrays.asList(CohortStatus.INCLUDED.toString(), CohortStatus.EXCLUDED.toString())));
filters.add(new Filter().property(ParticipantCohortStatusColumns.BIRTHDATE).operator(Operator.IN).values(Arrays.asList(new Date(System.currentTimeMillis()).toString())));
filters.add(new Filter().property(ParticipantCohortStatusColumns.GENDER).operator(Operator.IN).values(Arrays.asList("MALE", "FEMALE")));
filters.add(new Filter().property(ParticipantCohortStatusColumns.RACE).operator(Operator.IN).values(Arrays.asList("Asian", "White")));
filters.add(new Filter().property(ParticipantCohortStatusColumns.ETHNICITY).operator(Operator.IN).values(Arrays.asList("Not Hispanic", "Hispanic")));
List<ParticipantCohortStatus> results = participantCohortStatusDao.findAll(1L, filters, pageRequest);
assertEquals(2, results.size());
ParticipantCohortStatus expectedPCS1 = createExpectedPCSWithConceptValues(new ParticipantCohortStatusKey().cohortReviewId(COHORT_REVIEW_ID).participantId(1), CohortStatus.INCLUDED);
expectedPCS1.setBirthDate(results.get(0).getBirthDate());
ParticipantCohortStatus expectedPCS2 = createExpectedPCSWithConceptValues(new ParticipantCohortStatusKey().cohortReviewId(COHORT_REVIEW_ID).participantId(2), CohortStatus.EXCLUDED);
expectedPCS2.setBirthDate(results.get(0).getBirthDate());
assertEquals(expectedPCS1, results.get(0));
assertEquals(expectedPCS2, results.get(1));
}
use of org.pmiops.workbench.db.model.ParticipantCohortStatus in project workbench by all-of-us.
the class ParticipantCohortStatusDaoTest method findByParticipantKeyCohortReviewIdAndParticipantKeyParticipantId.
@Test
public void findByParticipantKeyCohortReviewIdAndParticipantKeyParticipantId() throws Exception {
ParticipantCohortStatus participant1 = createExpectedPCS(new ParticipantCohortStatusKey().cohortReviewId(COHORT_REVIEW_ID).participantId(1), CohortStatus.INCLUDED);
ParticipantCohortStatus actualParticipant = participantCohortStatusDao.findByParticipantKey_CohortReviewIdAndParticipantKey_ParticipantId(COHORT_REVIEW_ID, participant1.getParticipantKey().getParticipantId());
participant1.setBirthDate(actualParticipant.getBirthDate());
assertEquals(participant1, actualParticipant);
}
use of org.pmiops.workbench.db.model.ParticipantCohortStatus in project workbench by all-of-us.
the class CohortReviewControllerTest method createParticipantCohortStatus.
private ParticipantCohortStatus createParticipantCohortStatus(long cohortReviewId, long participantId, CohortStatus cohortStatus) {
ParticipantCohortStatusKey key = new ParticipantCohortStatusKey().cohortReviewId(cohortReviewId).participantId(participantId);
final Date dob = new Date(System.currentTimeMillis());
return new ParticipantCohortStatus().participantKey(key).status(cohortStatus).birthDate(dob).ethnicityConceptId(1L).genderConceptId(1L).raceConceptId(1L);
}
use of org.pmiops.workbench.db.model.ParticipantCohortStatus in project workbench by all-of-us.
the class CohortReviewControllerTest method assertCreateParticipantCohortAnnotation.
private void assertCreateParticipantCohortAnnotation(ParticipantCohortAnnotation request, AnnotationType annotationType) throws Exception {
ParticipantCohortStatusKey key = new ParticipantCohortStatusKey().cohortReviewId(cohortReviewId).participantId(participantId);
ParticipantCohortStatus participantCohortStatus = new ParticipantCohortStatus().participantKey(key);
CohortAnnotationDefinition cohortAnnotationDefinition = new CohortAnnotationDefinition().annotationType(annotationType);
if (request.getAnnotationValueEnum() != null) {
CohortAnnotationEnumValue cohortAnnotationEnumValue = new CohortAnnotationEnumValue().name(request.getAnnotationValueEnum());
cohortAnnotationDefinition.setEnumValues(new TreeSet(Arrays.asList(cohortAnnotationEnumValue)));
}
when(cohortReviewService.findCohort(cohortId)).thenReturn(createCohort(cohortId, workspaceId, null));
when(cohortReviewService.validateMatchingWorkspace(namespace, name, workspaceId, WorkspaceAccessLevel.WRITER)).thenReturn(new Workspace());
when(cohortReviewService.findCohortReview(cohortId, cdrVersionId)).thenReturn(createCohortReview(0, cohortId, cohortReviewId, cdrVersionId, null));
when(cohortReviewService.saveParticipantCohortAnnotation(isA(Long.class), isA(org.pmiops.workbench.db.model.ParticipantCohortAnnotation.class))).thenReturn(createParticipantCohortAnnotation(request));
ParticipantCohortAnnotation response = reviewController.createParticipantCohortAnnotation(namespace, name, cohortId, cdrVersionId, participantId, request).getBody();
assertEquals(request, response);
verify(cohortReviewService, atLeastOnce()).findCohort(cohortId);
verify(cohortReviewService, atLeastOnce()).validateMatchingWorkspace(namespace, name, workspaceId, WorkspaceAccessLevel.WRITER);
verify(cohortReviewService, atLeastOnce()).findCohortReview(cohortId, cdrVersionId);
verify(cohortReviewService, atLeastOnce()).saveParticipantCohortAnnotation(isA(Long.class), isA(org.pmiops.workbench.db.model.ParticipantCohortAnnotation.class));
verifyNoMoreMockInteractions();
}
use of org.pmiops.workbench.db.model.ParticipantCohortStatus in project workbench by all-of-us.
the class CohortReviewServiceImplTest method saveParticipantCohortStatus.
@Test
public void saveParticipantCohortStatus() throws Exception {
ParticipantCohortStatus pcs = new ParticipantCohortStatus();
when(participantCohortStatusDao.save(pcs)).thenReturn(pcs);
ParticipantCohortStatus actualPcs = cohortReviewService.saveParticipantCohortStatus(pcs);
assertEquals(pcs, actualPcs);
verify(participantCohortStatusDao).save(pcs);
verifyNoMoreMockInteractions();
}
Aggregations