@Override
public void setUp() {
super.setUp();
generator =
new DcpSponsorTasksGenerator(templateService, mockRegistrationService, mockSponsorService);
SponsorRole sponsorRole = sponsor.addSponsorRepresentativeRole(sponsorOrganization);
groupNames.add(sponsorRole.getVerifiedSponsorGroupName());
List<AbstractProtocolRegistration> emptyList = Collections.emptyList();
when(mockRegistrationService.getByStatusForUser(
any(RegistrationStatus.class), eq(sponsor), anySetOf(String.class)))
.thenReturn(emptyList);
}
@Test
public void testGetSubinvestigatorRegistrationsJson() throws JSONException {
SubInvestigatorRegistration inProgressReg =
createSubInvestigatorRegistration(RegistrationStatus.IN_PROGRESS);
SubInvestigatorRegistration returnedReg =
createSubInvestigatorRegistration(RegistrationStatus.RETURNED);
SubInvestigatorRegistration submittedReg =
createSubInvestigatorRegistration(RegistrationStatus.SUBMITTED);
when(mockRegistrationService.getSubinvestigatorRegistrations(profile, person))
.thenReturn(Lists.newArrayList(inProgressReg, returnedReg, submittedReg));
String json = action.getSubinvestigatorRegistrationsJson();
checkRegistrationInJson(inProgressReg, json);
checkRegistrationInJson(returnedReg, json);
checkRegistrationInJson(submittedReg, json);
}
private void setUpMockRegistrationServiceToReturnRegistration(
AbstractProtocolRegistration registration) {
when(mockRegistrationService.getByStatusForUser(registration.getStatus(), sponsor, groupNames))
.thenReturn(Lists.newArrayList(registration));
}