svc.data.citations.datasources.tyler.TylerCitationDataSourceTest.java Source code

Introduction

Here is the source code for svc.data.citations.datasources.tyler.TylerCitationDataSourceTest.java

Source

package svc.data.citations.datasources.tyler;

import static org.hamcrest.CoreMatchers.is;
import static org.junit.Assert.assertThat;
import static org.mockito.Matchers.any;
import static org.mockito.Matchers.eq;
import static org.mockito.Mockito.when;

import java.net.URI;
import java.time.LocalDate;
import java.util.List;

import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.Mockito;
import org.mockito.Spy;
import org.mockito.runners.MockitoJUnitRunner;

import com.google.common.collect.Lists;

import org.springframework.web.client.RestClientException;
import org.springframework.web.client.RestTemplate;
import org.springframework.web.util.UriComponentsBuilder;
import org.springframework.core.ParameterizedTypeReference;
import org.springframework.http.HttpEntity;
import org.springframework.http.HttpMethod;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;

import svc.data.citations.datasources.tyler.models.TylerCitation;
import svc.data.citations.datasources.tyler.transformers.CitationTransformer;
import svc.data.citations.filters.CitationFilter;
import svc.models.Citation;

@RunWith(MockitoJUnitRunner.class)
public class TylerCitationDataSourceTest {
    @InjectMocks
    TylerCitationDataSource mockTylerCitationDataSource;

    @Mock
    TylerConfiguration mockTylerConfiguration;
    @Mock
    UriComponentsBuilder mockUriComponentsBuilder;
    @Mock
    RestTemplate restTemplate;
    @Mock
    CitationTransformer mockCitationTransformer;
    @Mock
    CitationFilter mockCitationFilter;

    @Spy
    ResponseEntity<List<TylerCitation>> tylerCitationsResponseSpy = new ResponseEntity<List<TylerCitation>>(
            HttpStatus.ACCEPTED);

    @SuppressWarnings("unchecked")
    @Test
    public void returnsCitationsGivenCitationNumberAndDOB() {
        final String CITATIONNUMBER = "F3453";
        final LocalDate DOB = LocalDate.parse("2000-06-01");
        mockTylerConfiguration.rootUrl = "http://myURL.com";
        mockTylerConfiguration.apiKey = "1234";
        final Citation CITATION = new Citation();
        CITATION.id = 3;
        final List<Citation> CITATIONS = Lists.newArrayList(CITATION);
        final List<TylerCitation> tylerCitations = Lists.newArrayList();
        Mockito.doReturn(tylerCitations).when(tylerCitationsResponseSpy).getBody();

        when(restTemplate.exchange(any(URI.class), eq(HttpMethod.GET), any(HttpEntity.class),
                any(ParameterizedTypeReference.class))).thenReturn(tylerCitationsResponseSpy);

        when(mockCitationTransformer.fromTylerCitations(tylerCitations)).thenReturn(CITATIONS);
        when(mockCitationFilter.RemoveCitationsWithExpiredDates(CITATIONS)).thenReturn(CITATIONS);

        List<Citation> citations = mockTylerCitationDataSource.getByCitationNumberAndDOB(CITATIONNUMBER, DOB);

        assertThat(citations.get(0).id, is(3));
    }

    @SuppressWarnings("unchecked")
    @Test
    public void returnsCitationsGivenLicenseAndDOB() {
        final String DRIVERSLICENSENUMBER = "ABCDE";
        final String DRIVERSLICENSESTATE = "AZ";
        final LocalDate DOB = LocalDate.parse("2000-06-01");
        mockTylerConfiguration.rootUrl = "http://myURL.com";
        mockTylerConfiguration.apiKey = "1234";
        final Citation CITATION = new Citation();
        CITATION.id = 3;
        final List<Citation> CITATIONS = Lists.newArrayList(CITATION);
        final List<TylerCitation> tylerCitations = Lists.newArrayList();
        Mockito.doReturn(tylerCitations).when(tylerCitationsResponseSpy).getBody();

        when(restTemplate.exchange(any(URI.class), eq(HttpMethod.GET), any(HttpEntity.class),
                any(ParameterizedTypeReference.class))).thenReturn(tylerCitationsResponseSpy);

        when(mockCitationTransformer.fromTylerCitations(tylerCitations)).thenReturn(CITATIONS);
        when(mockCitationFilter.RemoveCitationsWithExpiredDates(CITATIONS)).thenReturn(CITATIONS);

        List<Citation> citations = mockTylerCitationDataSource.getByLicenseAndDOB(DRIVERSLICENSENUMBER,
                DRIVERSLICENSESTATE, DOB);

        assertThat(citations.get(0).id, is(3));
    }

    @SuppressWarnings("unchecked")
    @Test
    public void returnsCitationsGivenNameAndMunicipalitiesAndDOB() {
        final String NAME = "Smith";
        final List<Long> MUNICIPALITIES = Lists.newArrayList(5L);
        final LocalDate DOB = LocalDate.parse("2000-06-01");
        mockTylerConfiguration.rootUrl = "http://myURL.com";
        mockTylerConfiguration.apiKey = "1234";
        final Citation CITATION = new Citation();
        CITATION.id = 3;
        final List<Citation> CITATIONS = Lists.newArrayList(CITATION);
        final List<TylerCitation> tylerCitations = Lists.newArrayList();
        Mockito.doReturn(tylerCitations).when(tylerCitationsResponseSpy).getBody();

        when(restTemplate.exchange(any(URI.class), eq(HttpMethod.GET), any(HttpEntity.class),
                any(ParameterizedTypeReference.class))).thenReturn(tylerCitationsResponseSpy);

        when(mockCitationTransformer.fromTylerCitations(tylerCitations)).thenReturn(CITATIONS);
        when(mockCitationFilter.RemoveCitationsWithExpiredDates(CITATIONS)).thenReturn(CITATIONS);

        List<Citation> citations = mockTylerCitationDataSource.getByNameAndMunicipalitiesAndDOB(NAME,
                MUNICIPALITIES, DOB);

        assertThat(citations.get(0).id, is(3));
    }

    @SuppressWarnings("unchecked")
    @Test
    public void returnsEmptyCitationsOnRestTemplateError() {
        final String CITATIONNUMBER = "F3453";
        final LocalDate DOB = LocalDate.parse("2000-06-01");
        mockTylerConfiguration.rootUrl = "http://myURL.com";
        mockTylerConfiguration.apiKey = "1234";
        final Citation CITATION = new Citation();
        CITATION.id = 3;

        when(restTemplate.exchange(any(URI.class), eq(HttpMethod.GET), any(HttpEntity.class),
                any(ParameterizedTypeReference.class))).thenThrow(RestClientException.class);

        List<Citation> citations = mockTylerCitationDataSource.getByCitationNumberAndDOB(CITATIONNUMBER, DOB);

        assertThat(citations.size(), is(0));
    }
}