PatientServiceImplTest.java

package gov.samhsa.consent2share.service.patient;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;
import static org.mockito.Matchers.any;
import static org.mockito.Matchers.anyLong;
import static org.mockito.Matchers.anyString;
import static org.mockito.Matchers.argThat;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;
import gov.samhsa.consent2share.common.AuthenticatedUser;
import gov.samhsa.consent2share.common.UserContext;
import gov.samhsa.consent2share.domain.account.Users;
import gov.samhsa.consent2share.domain.account.UsersRepository;
import gov.samhsa.consent2share.domain.commondomainservices.EmailSender;
import gov.samhsa.consent2share.domain.patient.Patient;
import gov.samhsa.consent2share.domain.patient.PatientLegalRepresentativeAssociation;
import gov.samhsa.consent2share.domain.patient.PatientLegalRepresentativeAssociationPk;
import gov.samhsa.consent2share.domain.patient.PatientLegalRepresentativeAssociationRepository;
import gov.samhsa.consent2share.domain.patient.PatientRepository;
import gov.samhsa.consent2share.domain.provider.IndividualProvider;
import gov.samhsa.consent2share.domain.provider.OrganizationalProvider;
import gov.samhsa.consent2share.hl7.dto.PixPatientDto;
import gov.samhsa.consent2share.infrastructure.DtoToDomainEntityMapper;
import gov.samhsa.consent2share.infrastructure.PixService;
import gov.samhsa.consent2share.infrastructure.security.AuthenticationFailedException;
import gov.samhsa.consent2share.pixclient.util.PixManagerBean;
import gov.samhsa.consent2share.service.dto.AddConsentIndividualProviderDto;
import gov.samhsa.consent2share.service.dto.AddConsentOrganizationalProviderDto;
import gov.samhsa.consent2share.service.dto.LookupDto;
import gov.samhsa.consent2share.service.dto.PatientAdminDto;
import gov.samhsa.consent2share.service.dto.PatientProfileDto;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Date;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.ArgumentMatcher;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.runners.MockitoJUnitRunner;
import org.modelmapper.ModelMapper;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageRequest;
import org.springframework.security.crypto.password.PasswordEncoder;

@RunWith(MockitoJUnitRunner.class)
public class PatientServiceImplTest {

	@Mock
	private PatientRepository patientRepository;

	@Mock
	private PatientLegalRepresentativeAssociationRepository patientLegalRepresentativeAssociationRepository;

	@Mock
	private ModelMapper modelMapper;

	@Mock
	private UserContext userContext;

	@Mock
	private DtoToDomainEntityMapper<PatientProfileDto, Patient> patientProfileDtoToPatientMapper;

	@Mock
	private UsersRepository usersRepository;

	@Mock
	private PasswordEncoder passwordEncoder;

	@Mock
	private EmailSender emailSender;

	@Mock
	private PixService pixService;

	@InjectMocks
	private PatientServiceImpl sut;

	@Test
	public void testCountAllPatients() {
		// Arrange
		final Long expectedCount = new Long(100);
		when(patientRepository.count()).thenReturn(expectedCount);

		// Act
		Long result = sut.countAllPatients();

		// Assert
		assertEquals(expectedCount, result);
	}

	@Test
	public void testFindPatient() {
		// Arrange
		final Patient patient = mock(Patient.class);
		when(patientRepository.findOne(anyLong())).thenReturn(patient);
		final PatientProfileDto expectedPatientProfileDto = mock(PatientProfileDto.class);
		when(modelMapper.map(patient, PatientProfileDto.class)).thenReturn(
				expectedPatientProfileDto);

		// Act
		PatientProfileDto result = sut.findPatient(anyLong());

		// Assert
		assertEquals(expectedPatientProfileDto, result);
	}

	@Test
	public void testFindIdByUsername() {
		// Arrange
		final Patient patient = mock(Patient.class);
		when(patientRepository.findByUsername(anyString())).thenReturn(patient);
		final Long expectedId = new Long(100);
		when(patient.getId()).thenReturn(expectedId);

		// Act
		Long result = sut.findIdByUsername(anyString());

		// Assert
		assertEquals(expectedId, result);
	}

	@Test
	public void testFindPatientEmailByUsername() {
		// Arrange
		final Patient patient = mock(Patient.class);
		when(patientRepository.findByUsername(anyString())).thenReturn(patient);
		final String expectedEmail = "test@test.com";
		when(patient.getEmail()).thenReturn(expectedEmail);

		// Act
		String result = sut.findPatientEmailByUsername(anyString());

		// Assert
		assertEquals(expectedEmail, result);
	}

	@Test
	public void testFindPatientProfileByUsername() {
		// Arrange
		final Patient patient = mock(Patient.class);
		when(patientRepository.findByUsername(anyString())).thenReturn(patient);
		final PatientProfileDto expectedPatientProfileDto = mock(PatientProfileDto.class);
		when(modelMapper.map(patient, PatientProfileDto.class)).thenReturn(
				expectedPatientProfileDto);

		// Act
		PatientProfileDto result = sut
				.findPatientProfileByUsername(anyString());

		// Assert
		assertEquals(expectedPatientProfileDto, result);
	}

	@Test
	public void testFindPatientEntries() {
		// Arrange
		final int pageNumber = 1;
		final int pageSize = 9;
		final Patient patient1 = mock(Patient.class);
		final Patient patient2 = mock(Patient.class);
		List<Patient> patientList = new ArrayList<Patient>();
		patientList.add(patient1);
		patientList.add(patient2);

		int numberOfPatients = patientList.size();

		Page<Patient> patientPage = mock(Page.class);
		when(patientPage.getContent()).thenReturn(patientList);

		ArgumentMatcher<PageRequest> pageRequestArgumentMatcher = new ArgumentMatcher<PageRequest>() {

			@Override
			public boolean matches(Object argument) {
				PageRequest pageRequest = (PageRequest) argument;
				if (pageRequest == null) {
					return false;
				}

				if (pageRequest.getPageSize() != pageSize) {
					return false;
				}

				if (pageRequest.getPageNumber() != pageNumber) {
					return false;
				}

				return true;
			}
		};

		when(patientRepository.findAll(argThat(pageRequestArgumentMatcher)))
				.thenReturn(patientPage);

		final PatientProfileDto expectedPatientProfileDto1 = mock(PatientProfileDto.class);
		when(modelMapper.map(patient1, PatientProfileDto.class)).thenReturn(
				expectedPatientProfileDto1);

		final PatientProfileDto expectedPatientProfileDto2 = mock(PatientProfileDto.class);
		when(modelMapper.map(patient2, PatientProfileDto.class)).thenReturn(
				expectedPatientProfileDto2);

		// Act
		List<PatientProfileDto> result = sut.findPatientEntries(pageNumber,
				pageSize);

		// Assert
		assertEquals(numberOfPatients, result.size());
		for (PatientProfileDto patientProfileDto : result) {
			assertTrue(patientProfileDto == expectedPatientProfileDto1
					|| patientProfileDto == expectedPatientProfileDto2);
		}

		assertTrue(result.get(0) != result.get(1));
	}

	@Test
	public void testSavePatient() {
		// Arrange
		PatientProfileDto patientProfileDtoInput = mock(PatientProfileDto.class);
		Patient patient = mock(Patient.class);
		when(patientProfileDtoToPatientMapper.map(patientProfileDtoInput))
				.thenReturn(patient);
		PatientProfileDto expectedPatientProfileDto = mock(PatientProfileDto.class);
		when(modelMapper.map(patient, PatientProfileDto.class)).thenReturn(
				expectedPatientProfileDto);

		// Act
		PatientProfileDto result = sut.savePatient(patientProfileDtoInput);

		// Assert
		verify(patientRepository, times(1)).save(patient);
		assertEquals(expectedPatientProfileDto, result);
	}

	@Test
	public void testUpdatePatient_when_authentication_success()
			throws AuthenticationFailedException {
		final String username = "TheUsername";
		final String rawpassword = "rawpassword";
		final String firstname = "albert";
		final String lastname = "Smith";
		final String email = "albert.smith@consent2share.com";
		final String mrn = "mrn";
		final String eid = "eid";
		AuthenticatedUser authenticatedUser = mock(AuthenticatedUser.class);
		when(authenticatedUser.getUsername()).thenReturn(username);
		when(userContext.getCurrentUser()).thenReturn(authenticatedUser);

		Users user = mock(Users.class);
		when(user.getUsername()).thenReturn(username);
		when(user.getPassword()).thenReturn("hashedpassword");
		when(usersRepository.loadUserByUsername(anyString())).thenReturn(user);
		when(passwordEncoder.matches("rawpassword", "hashedpassword"))
				.thenReturn(true);

		// Arrange
		PatientProfileDto patientProfileDtoInput = new PatientProfileDto();
		Patient patient = mock(Patient.class);
		patientProfileDtoInput.setUsername(username);
		patientProfileDtoInput.setPassword(rawpassword);
		patientProfileDtoInput.setFirstName(firstname);
		patientProfileDtoInput.setLastName(lastname);
		patientProfileDtoInput.setEmail(email);
		patientProfileDtoInput.setBirthDate(new Date());
		LookupDto code = new LookupDto();
		code.setCode("M");
		patientProfileDtoInput.setMaritalStatusCode(code);

		when(patientProfileDtoToPatientMapper.map(patientProfileDtoInput))
				.thenReturn(patient);
		when(patientRepository.findByUsername(username)).thenReturn(patient);
		when(patient.getMedicalRecordNumber()).thenReturn(mrn);
		when(patient.getEnterpriseIdentifier()).thenReturn(eid);
		PixManagerBean pixManagerBean = new PixManagerBean();
		pixManagerBean.setSuccess(true);
		when(pixService.updatePatient(any(PixPatientDto.class))).thenReturn(
				pixManagerBean);
		when(pixService.getEid(mrn)).thenReturn(eid);

		// Act
		sut.updatePatient(patientProfileDtoInput);

		// Assert
		verify(patientRepository, times(1)).save(patient);
		verify(pixService, times(1)).updatePatient(any(PixPatientDto.class));
	}

	@Test
	public void testUpdatePatient_when_username_does_not_match_and_so_authentication_fails() {
		AuthenticatedUser authenticatedUser = mock(AuthenticatedUser.class);
		when(authenticatedUser.getUsername()).thenReturn("TheUsername");
		when(userContext.getCurrentUser()).thenReturn(authenticatedUser);
		Users user = mock(Users.class);
		when(user.getUsername()).thenReturn("AnotherUsername");

		// Arrange
		PatientProfileDto patientProfileDtoInput = mock(PatientProfileDto.class);
		when(patientProfileDtoInput.getUsername())
				.thenReturn("AnotherUsername");

		// Act
		try {
			sut.updatePatient(patientProfileDtoInput);
		} catch (AuthenticationFailedException e) {
			assertEquals(e.getMessage(),
					"Username does not match current active user.");
		}

		// Assert
		verify(patientRepository, times(0)).save(any(Patient.class));
	}

	@Test
	public void testUpdatePatient_when_username_matches_but_password_is_wrong() {
		final String username = "TheUsername";
		AuthenticatedUser authenticatedUser = mock(AuthenticatedUser.class);
		when(authenticatedUser.getUsername()).thenReturn(username);
		when(userContext.getCurrentUser()).thenReturn(authenticatedUser);

		Users user = mock(Users.class);
		when(user.getUsername()).thenReturn(username);
		when(user.getPassword()).thenReturn("hashedpassword");
		when(usersRepository.loadUserByUsername(anyString())).thenReturn(user);
		when(passwordEncoder.matches("rawpassword", "hashedpassword"))
				.thenReturn(false);

		// Arrange
		PatientProfileDto patientProfileDtoInput = mock(PatientProfileDto.class);
		Patient patient = mock(Patient.class);
		when(patientProfileDtoInput.getUsername()).thenReturn(username);
		when(patientProfileDtoInput.getPassword()).thenReturn("rawpassword");
		when(patientProfileDtoInput.getFirstName()).thenReturn("albert");
		when(patientProfileDtoInput.getLastName()).thenReturn("Smith");
		when(patientProfileDtoInput.getEmail()).thenReturn(
				"albert.smith@consent2share.com");
		when(patientProfileDtoToPatientMapper.map(patientProfileDtoInput))
				.thenReturn(patient);

		// Act
		try {
			sut.updatePatient(patientProfileDtoInput);
		} catch (AuthenticationFailedException e) {
			assertEquals(e.getMessage(), "Password is incorrect.");
		}
		// Assert
		verify(patientRepository, times(0)).save(patient);
	}

	@Test
	public void testFindAddConsentIndividualProviderDtoByUsername() {
		// Arrange
		final String username = "TheUsername";
		Patient patient = mock(Patient.class);
		when(patientRepository.findByUsername(username)).thenReturn(patient);

		IndividualProvider provider1 = mock(IndividualProvider.class);
		IndividualProvider provider2 = mock(IndividualProvider.class);

		Set<IndividualProvider> providerSet = new HashSet();
		providerSet.add(provider1);
		providerSet.add(provider2);

		when(patient.getIndividualProviders()).thenReturn(providerSet);

		AddConsentIndividualProviderDto providerDto1 = mock(AddConsentIndividualProviderDto.class);
		AddConsentIndividualProviderDto providerDto2 = mock(AddConsentIndividualProviderDto.class);

		when(modelMapper.map(provider1, AddConsentIndividualProviderDto.class))
				.thenReturn(providerDto1);
		when(modelMapper.map(provider2, AddConsentIndividualProviderDto.class))
				.thenReturn(providerDto2);

		// Act
		List<AddConsentIndividualProviderDto> result = sut
				.findAddConsentIndividualProviderDtoByUsername(username);

		// Assert
		// Assert
		assertEquals(2, result.size());
		for (AddConsentIndividualProviderDto providerDto : result) {
			assertTrue(providerDto == providerDto1
					|| providerDto == providerDto2);
		}

		assertTrue(result.get(0) != result.get(1));
	}

	@Test
	public void testFindAddConsentOrganizationalProviderDtoByUsername() {
		// Arrange
		final String username = "TheUsername";
		Patient patient = mock(Patient.class);
		when(patientRepository.findByUsername(username)).thenReturn(patient);

		OrganizationalProvider provider1 = mock(OrganizationalProvider.class);
		OrganizationalProvider provider2 = mock(OrganizationalProvider.class);

		Set<OrganizationalProvider> providerSet = new HashSet();
		providerSet.add(provider1);
		providerSet.add(provider2);

		when(patient.getOrganizationalProviders()).thenReturn(providerSet);

		AddConsentOrganizationalProviderDto providerDto1 = mock(AddConsentOrganizationalProviderDto.class);
		AddConsentOrganizationalProviderDto providerDto2 = mock(AddConsentOrganizationalProviderDto.class);

		when(
				modelMapper.map(provider1,
						AddConsentOrganizationalProviderDto.class)).thenReturn(
				providerDto1);
		when(
				modelMapper.map(provider2,
						AddConsentOrganizationalProviderDto.class)).thenReturn(
				providerDto2);

		// Act
		List<AddConsentOrganizationalProviderDto> result = sut
				.findAddConsentOrganizationalProviderDtoByUsername(username);

		// Assert
		// Assert
		assertEquals(2, result.size());
		for (AddConsentOrganizationalProviderDto providerDto : result) {
			assertTrue(providerDto == providerDto1
					|| providerDto == providerDto2);
		}

		assertTrue(result.get(0) != result.get(1));
	}

	@Test
	public void testIsLegalRepForCurrentUser_Given_LegalRep_Not_For_Current_User() {
		// Arrange
		final Long patientOneId = new Long(1);
		final Long patientTwoId = new Long(2);

		final Long legalRepIdInput = new Long(100);

		final String username = "TheUsername";
		AuthenticatedUser authenticatedUser = mock(AuthenticatedUser.class);
		when(authenticatedUser.getUsername()).thenReturn(username);
		when(userContext.getCurrentUser()).thenReturn(authenticatedUser);

		Patient patient1 = mock(Patient.class);
		when(patient1.getId()).thenReturn(patientOneId);
		when(patientRepository.findByUsername(username)).thenReturn(patient1);

		PatientLegalRepresentativeAssociation association = mock(PatientLegalRepresentativeAssociation.class);
		List<PatientLegalRepresentativeAssociation> associations = new ArrayList<PatientLegalRepresentativeAssociation>();
		associations.add(association);
		when(
				patientLegalRepresentativeAssociationRepository
						.findByPatientLegalRepresentativeAssociationPkLegalRepresentativeId(legalRepIdInput))
				.thenReturn(associations);

		PatientLegalRepresentativeAssociationPk associationPk = mock(PatientLegalRepresentativeAssociationPk.class);
		when(association.getPatientLegalRepresentativeAssociationPk())
				.thenReturn(associationPk);
		Patient patient2 = mock(Patient.class);
		when(patient2.getId()).thenReturn(patientTwoId);
		when(associationPk.getPatient()).thenReturn(patient2);

		// Act
		boolean result = sut.isLegalRepForCurrentUser(legalRepIdInput);

		// Assert
		assertFalse(result);
	}

	@Test
	public void testIsLegalRepForCurrentUser_Given_LegalRep_For_Current_User() {
		// Arrange
		final Long patientOneId = new Long(1);
		final Long patientTwoId = new Long(1);

		final Long legalRepIdInput = new Long(100);

		final String username = "TheUsername";
		AuthenticatedUser authenticatedUser = mock(AuthenticatedUser.class);
		when(authenticatedUser.getUsername()).thenReturn(username);
		when(userContext.getCurrentUser()).thenReturn(authenticatedUser);

		Patient patient1 = mock(Patient.class);
		when(patient1.getId()).thenReturn(patientOneId);
		when(patientRepository.findByUsername(username)).thenReturn(patient1);

		PatientLegalRepresentativeAssociation association = mock(PatientLegalRepresentativeAssociation.class);
		List<PatientLegalRepresentativeAssociation> associations = new ArrayList<PatientLegalRepresentativeAssociation>();
		associations.add(association);
		when(
				patientLegalRepresentativeAssociationRepository
						.findByPatientLegalRepresentativeAssociationPkLegalRepresentativeId(legalRepIdInput))
				.thenReturn(associations);

		PatientLegalRepresentativeAssociationPk associationPk = mock(PatientLegalRepresentativeAssociationPk.class);
		when(association.getPatientLegalRepresentativeAssociationPk())
				.thenReturn(associationPk);
		Patient patient2 = mock(Patient.class);
		when(patient2.getId()).thenReturn(patientTwoId);
		when(associationPk.getPatient()).thenReturn(patient2);

		// Act
		boolean result = sut.isLegalRepForCurrentUser(legalRepIdInput);

		// Assert
		assertTrue(result);
	}

	@Test
	public void testFindAllPatientByFirstNameAndLastName() {
		List<Patient> patients = new ArrayList<Patient>();
		Patient patient = mock(Patient.class);
		Patient patient2 = mock(Patient.class);
		patients.add(patient);
		patients.add(patient2);
		when(
				patientRepository.findAllByFirstNameLikesAndLastNameLikes(
						"%john%", "%smith%")).thenReturn(patients);
		PatientAdminDto patientAdminDto = mock(PatientAdminDto.class);
		PatientAdminDto patientAdminDto2 = mock(PatientAdminDto.class);
		when(modelMapper.map(patient, PatientAdminDto.class)).thenReturn(
				patientAdminDto);
		when(modelMapper.map(patient2, PatientAdminDto.class)).thenReturn(
				patientAdminDto2);
		assertEquals(
				Arrays.asList(patientAdminDto, patientAdminDto2),
				sut.findAllPatientByFirstNameAndLastName(new String[] { "john",
						"smith" }));
	}

	@Test
	public void testFindPatientConnectionByUsername() {
		Patient patient = mock(Patient.class);
		when(patientRepository.findByUsername(anyString())).thenReturn(patient);
	}

	@Test
	public void testFindPatientConnectionById() {
		Patient patient = mock(Patient.class);
		when(patientRepository.findOne(anyLong())).thenReturn(patient);
	}
}