Search in sources :

Example 6 with AssignedModules

use of org.motechproject.mots.domain.AssignedModules in project mots by motech-implementations.

the class ModuleAssignmentServiceTest method assignModulesToDistrictShouldThrowIfIvrIdIsNotSet.

@Test(expected = ModuleAssignmentException.class)
public void assignModulesToDistrictShouldThrowIfIvrIdIsNotSet() {
    CommunityHealthWorker chw = new CommunityHealthWorkerDataBuilder().withIvrId(null).build();
    AssignedModules assignedModules = new AssignedModulesDataBuilder().withChw(chw).build();
    when(communityHealthWorkerRepository.findByCommunityFacilityChiefdomDistrictIdAndSelected(any(), any())).thenReturn(Collections.singletonList(chw));
    when(assignedModulesRepository.findByHealthWorkerId(eq(chw.getId()))).thenReturn(Optional.of(assignedModules));
    moduleAssignmentService.assignModulesToDistrict(districtAssignmentDto);
}
Also used : AssignedModules(org.motechproject.mots.domain.AssignedModules) AssignedModulesDataBuilder(org.motechproject.mots.testbuilder.AssignedModulesDataBuilder) CommunityHealthWorker(org.motechproject.mots.domain.CommunityHealthWorker) CommunityHealthWorkerDataBuilder(org.motechproject.mots.testbuilder.CommunityHealthWorkerDataBuilder) PrepareForTest(org.powermock.core.classloader.annotations.PrepareForTest) Test(org.junit.Test)

Example 7 with AssignedModules

use of org.motechproject.mots.domain.AssignedModules in project mots by motech-implementations.

the class ModuleAssignmentServiceTest method shouldAssignModulesToDistrict.

@Test
public void shouldAssignModulesToDistrict() throws Exception {
    when(userService.getUserByUserName(eq(user.getUsername()))).thenReturn(user);
    when(communityHealthWorkerRepository.findByCommunityFacilityChiefdomDistrictIdAndSelected(any(), any())).thenReturn(Collections.singletonList(CHW));
    moduleAssignmentService.assignModulesToDistrict(districtAssignmentDto);
    ArgumentCaptor<DistrictAssignmentLog> districtAssignmentLogCaptor = ArgumentCaptor.forClass(DistrictAssignmentLog.class);
    verify(districtAssignmentLogRepository, times(2)).save(districtAssignmentLogCaptor.capture());
    List<DistrictAssignmentLog> districtAssignmentLogs = districtAssignmentLogCaptor.getAllValues();
    final Set<Module> passedModules = new HashSet<>(Arrays.asList(MODULE_2, MODULE_3));
    assertTrue(districtAssignmentLogs.stream().allMatch(l -> passedModules.contains(l.getModule())));
    for (DistrictAssignmentLog log : districtAssignmentLogs) {
        assertEquals(DISTRICT, log.getDistrict());
        assertEquals(districtAssignmentDto.getStartDate(), log.getStartDate().toString());
        assertEquals(districtAssignmentDto.getEndDate(), log.getEndDate().toString());
        assertEquals(user, log.getOwner());
    }
    ArgumentCaptor<AssignedModules> assignedModulesCaptor = ArgumentCaptor.forClass(AssignedModules.class);
    verify(assignedModulesRepository).save(assignedModulesCaptor.capture());
    final Set<Module> allModules = new HashSet<>(Arrays.asList(MODULE_1, MODULE_2, MODULE_3));
    assertEquals(allModules, assignedModulesCaptor.getValue().getModules());
    verify(ivrService).addSubscriberToGroups(eq(CHW.getIvrId()), eq(Collections.singletonList(IVR_GROUP)));
    verify(moduleProgressService).createModuleProgresses(any(), eq(Collections.singleton(MODULE_3)));
}
Also used : Arrays(java.util.Arrays) DistrictAssignmentLogRepository(org.motechproject.mots.repository.DistrictAssignmentLogRepository) ModuleProgress(org.motechproject.mots.domain.ModuleProgress) CommunityHealthWorkerRepository(org.motechproject.mots.repository.CommunityHealthWorkerRepository) Mockito.doThrow(org.mockito.Mockito.doThrow) ModuleAssignmentException(org.motechproject.mots.exception.ModuleAssignmentException) AssignedModules(org.motechproject.mots.domain.AssignedModules) Matchers.eq(org.mockito.Matchers.eq) DistrictAssignmentLog(org.motechproject.mots.domain.DistrictAssignmentLog) DistrictAssignmentDtoDataBuilder(org.motechproject.mots.testbuilder.DistrictAssignmentDtoDataBuilder) EntityNotFoundException(org.motechproject.mots.exception.EntityNotFoundException) DistrictAssignmentDto(org.motechproject.mots.dto.DistrictAssignmentDto) SecurityContextHolder(org.springframework.security.core.context.SecurityContextHolder) AssignedModulesDataBuilder(org.motechproject.mots.testbuilder.AssignedModulesDataBuilder) MotsException(org.motechproject.mots.exception.MotsException) Set(java.util.Set) UUID(java.util.UUID) Matchers.any(org.mockito.Matchers.any) List(java.util.List) District(org.motechproject.mots.domain.District) Optional(java.util.Optional) User(org.motechproject.mots.domain.security.User) ModuleDataBuilder(org.motechproject.mots.testbuilder.ModuleDataBuilder) Mock(org.mockito.Mock) RunWith(org.junit.runner.RunWith) ProgressStatus(org.motechproject.mots.domain.enums.ProgressStatus) CommunityHealthWorkerDataBuilder(org.motechproject.mots.testbuilder.CommunityHealthWorkerDataBuilder) CommunityHealthWorker(org.motechproject.mots.domain.CommunityHealthWorker) HashSet(java.util.HashSet) IvrException(org.motechproject.mots.exception.IvrException) ArgumentCaptor(org.mockito.ArgumentCaptor) PrepareForTest(org.powermock.core.classloader.annotations.PrepareForTest) Module(org.motechproject.mots.domain.Module) ModuleProgressDataBuilder(org.motechproject.mots.testbuilder.ModuleProgressDataBuilder) PowerMockRunner(org.powermock.modules.junit4.PowerMockRunner) Before(org.junit.Before) InjectMocks(org.mockito.InjectMocks) ModuleRepository(org.motechproject.mots.repository.ModuleRepository) DistrictDataBuilder(org.motechproject.mots.testbuilder.DistrictDataBuilder) Assert.assertTrue(org.junit.Assert.assertTrue) Mockito.times(org.mockito.Mockito.times) Test(org.junit.Test) Mockito.when(org.mockito.Mockito.when) EntityManager(javax.persistence.EntityManager) Mockito.verify(org.mockito.Mockito.verify) AssignedModulesRepository(org.motechproject.mots.repository.AssignedModulesRepository) Collections(java.util.Collections) DistrictRepository(org.motechproject.mots.repository.DistrictRepository) Assert.assertEquals(org.junit.Assert.assertEquals) TestUtils(org.motechproject.mots.utils.TestUtils) AssignedModules(org.motechproject.mots.domain.AssignedModules) DistrictAssignmentLog(org.motechproject.mots.domain.DistrictAssignmentLog) Module(org.motechproject.mots.domain.Module) HashSet(java.util.HashSet) PrepareForTest(org.powermock.core.classloader.annotations.PrepareForTest) Test(org.junit.Test)

Example 8 with AssignedModules

use of org.motechproject.mots.domain.AssignedModules in project mots by motech-implementations.

the class ModuleAssignmentService method assignModules.

/**
 * Assign modules to CHW.
 * @param assignedModules modules assigned for CHW
 */
@SuppressWarnings("checkstyle:variabledeclarationusagedistance")
@Transactional
@PreAuthorize(RoleNames.HAS_ASSIGN_MODULES_ROLE)
public void assignModules(AssignedModules assignedModules) {
    AssignedModules existingAssignedModules = getAssignedModules(assignedModules.getHealthWorker().getId());
    CommunityHealthWorker assignedModulesChw = existingAssignedModules.getHealthWorker();
    Set<Module> oldModules = new HashSet<>(existingAssignedModules.getModules());
    existingAssignedModules.setModules(assignedModules.getModules());
    repository.save(existingAssignedModules);
    entityManager.flush();
    entityManager.refresh(existingAssignedModules);
    Set<Module> newModules = new HashSet<>(existingAssignedModules.getModules());
    Set<Module> modulesToAdd = getModulesToAdd(oldModules, newModules);
    Set<Module> modulesToRemove = getModulesToRemove(oldModules, newModules);
    validateModulesToUnassign(assignedModulesChw, modulesToRemove);
    String ivrId = assignedModulesChw.getIvrId();
    if (ivrId == null) {
        throw new ModuleAssignmentException("Could not assign module to CHW, because CHW has empty IVR id");
    }
    try {
        ivrService.addSubscriberToGroups(ivrId, getIvrGroupsFromModules(modulesToAdd));
        ivrService.removeSubscriberFromGroups(ivrId, getIvrGroupsFromModules(modulesToRemove));
    } catch (IvrException ex) {
        String message = "Could not assign or delete module for CHW, " + "because of IVR module assignment error.\n\n" + ex.getClearVotoInfo();
        throw new ModuleAssignmentException(message, ex);
    }
    moduleProgressService.removeModuleProgresses(assignedModulesChw, modulesToRemove);
    moduleProgressService.createModuleProgresses(assignedModulesChw, modulesToAdd);
}
Also used : AssignedModules(org.motechproject.mots.domain.AssignedModules) IvrException(org.motechproject.mots.exception.IvrException) ModuleAssignmentException(org.motechproject.mots.exception.ModuleAssignmentException) CommunityHealthWorker(org.motechproject.mots.domain.CommunityHealthWorker) CourseModule(org.motechproject.mots.domain.CourseModule) Module(org.motechproject.mots.domain.Module) HashSet(java.util.HashSet) PreAuthorize(org.springframework.security.access.prepost.PreAuthorize) Transactional(org.springframework.transaction.annotation.Transactional)

Example 9 with AssignedModules

use of org.motechproject.mots.domain.AssignedModules in project mots by motech-implementations.

the class ModuleAssignmentService method unassignOldModuleVersion.

private void unassignOldModuleVersion(Module oldModule) {
    List<AssignedModules> assignedModulesList = repository.findByModules_id(oldModule.getId());
    if (assignedModulesList != null && !assignedModulesList.isEmpty()) {
        assignedModulesList.forEach(assignedModules -> {
            assignedModules.unassignModule(oldModule);
            String ivrGroup = oldModule.getIvrGroup();
            String chwIvrId = assignedModules.getHealthWorker().getIvrId();
            try {
                ivrService.removeSubscriberFromGroups(chwIvrId, Collections.singletonList(ivrGroup));
            } catch (IvrException ex) {
                throw new ModuleAssignmentException("Could not unassign old module version, " + "because of IVR module assignment error.\n\n" + ex.getClearVotoInfo(), ex);
            }
        });
        repository.save(assignedModulesList);
    }
}
Also used : AssignedModules(org.motechproject.mots.domain.AssignedModules) IvrException(org.motechproject.mots.exception.IvrException) ModuleAssignmentException(org.motechproject.mots.exception.ModuleAssignmentException)

Example 10 with AssignedModules

use of org.motechproject.mots.domain.AssignedModules in project mots by motech-implementations.

the class ModuleAssignmentService method assignModulesToDistrict.

/**
 * Creates DistrictAssignmentLog for district assignment,
 * and assigns modules to each CHW from a district.
 * @param assignmentDto dto with district id, list of modules assigned to it
 *     and start and end dates
 */
@Transactional
@PreAuthorize(RoleNames.HAS_ASSIGN_MODULES_ROLE)
public void assignModulesToDistrict(DistrictAssignmentDto assignmentDto) {
    List<CommunityHealthWorker> communityHealthWorkers = communityHealthWorkerRepository.findByCommunityFacilityChiefdomDistrictIdAndSelected(UUID.fromString(assignmentDto.getDistrictId()), true);
    Set<Module> newChwModules = new HashSet<>();
    String userName = (String) SecurityContextHolder.getContext().getAuthentication().getPrincipal();
    User currentUser = userService.getUserByUserName(userName);
    for (String moduleId : assignmentDto.getModules()) {
        Module moduleToAssign = moduleRepository.findById(UUID.fromString(moduleId)).orElseThrow(() -> new EntityNotFoundException("No module found for Id: {0}", moduleId));
        newChwModules.add(moduleToAssign);
        assignmentLogRepository.save(new DistrictAssignmentLog(districtRepository.findOne(UUID.fromString(assignmentDto.getDistrictId())), LocalDate.parse(assignmentDto.getStartDate()), LocalDate.parse(assignmentDto.getEndDate()), moduleToAssign, currentUser));
    }
    for (CommunityHealthWorker chw : communityHealthWorkers) {
        AssignedModules existingAssignedModules = getAssignedModules(chw.getId());
        Set<Module> oldModules = existingAssignedModules.getModules();
        Set<Module> modulesToAdd = getModulesToAdd(oldModules, newChwModules);
        existingAssignedModules.getModules().addAll(modulesToAdd);
        repository.save(existingAssignedModules);
        entityManager.flush();
        entityManager.refresh(existingAssignedModules);
        String ivrId = existingAssignedModules.getHealthWorker().getIvrId();
        if (ivrId == null) {
            throw new ModuleAssignmentException("Could not assign module to CHW, because CHW has empty IVR id");
        }
        try {
            ivrService.addSubscriberToGroups(ivrId, getIvrGroupsFromModules(modulesToAdd));
        } catch (IvrException ex) {
            String message = "Could not assign or delete module for CHW, " + "because of IVR module assignment error.\n\n" + ex.getClearVotoInfo();
            throw new ModuleAssignmentException(message, ex);
        }
        moduleProgressService.createModuleProgresses(chw, modulesToAdd);
    }
}
Also used : AssignedModules(org.motechproject.mots.domain.AssignedModules) IvrException(org.motechproject.mots.exception.IvrException) User(org.motechproject.mots.domain.security.User) ModuleAssignmentException(org.motechproject.mots.exception.ModuleAssignmentException) CommunityHealthWorker(org.motechproject.mots.domain.CommunityHealthWorker) EntityNotFoundException(org.motechproject.mots.exception.EntityNotFoundException) CourseModule(org.motechproject.mots.domain.CourseModule) Module(org.motechproject.mots.domain.Module) DistrictAssignmentLog(org.motechproject.mots.domain.DistrictAssignmentLog) HashSet(java.util.HashSet) PreAuthorize(org.springframework.security.access.prepost.PreAuthorize) Transactional(org.springframework.transaction.annotation.Transactional)

Aggregations

AssignedModules (org.motechproject.mots.domain.AssignedModules)11 CommunityHealthWorker (org.motechproject.mots.domain.CommunityHealthWorker)5 IvrException (org.motechproject.mots.exception.IvrException)5 Test (org.junit.Test)4 ModuleAssignmentException (org.motechproject.mots.exception.ModuleAssignmentException)4 PrepareForTest (org.powermock.core.classloader.annotations.PrepareForTest)4 HashSet (java.util.HashSet)3 Module (org.motechproject.mots.domain.Module)3 AssignedModulesDataBuilder (org.motechproject.mots.testbuilder.AssignedModulesDataBuilder)3 CommunityHealthWorkerDataBuilder (org.motechproject.mots.testbuilder.CommunityHealthWorkerDataBuilder)3 PreAuthorize (org.springframework.security.access.prepost.PreAuthorize)3 CourseModule (org.motechproject.mots.domain.CourseModule)2 DistrictAssignmentLog (org.motechproject.mots.domain.DistrictAssignmentLog)2 User (org.motechproject.mots.domain.security.User)2 EntityNotFoundException (org.motechproject.mots.exception.EntityNotFoundException)2 Transactional (org.springframework.transaction.annotation.Transactional)2 Arrays (java.util.Arrays)1 Collections (java.util.Collections)1 List (java.util.List)1 Optional (java.util.Optional)1