svc.data.citations.datasources.mock.MockCitationDataSourceTest.java Source code

Introduction

Here is the source code for svc.data.citations.datasources.mock.MockCitationDataSourceTest.java

Source

package svc.data.citations.datasources.mock;

import static org.hamcrest.CoreMatchers.is;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertThat;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;
import static org.springframework.core.io.ResourceLoader.CLASSPATH_URL_PREFIX;

import com.google.common.collect.Lists;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.InjectMocks;
import org.mockito.Matchers;
import org.mockito.Mock;
import org.mockito.runners.MockitoJUnitRunner;
import org.springframework.core.io.Resource;
import org.springframework.core.io.ResourceLoader;
import org.springframework.jdbc.core.RowMapper;
import org.springframework.jdbc.core.namedparam.NamedParameterJdbcTemplate;

import svc.managers.ViolationManager;
import svc.models.Citation;
import svc.models.Violation;

import java.io.IOException;
import java.text.ParseException;
import java.time.LocalDate;
import java.time.format.DateTimeFormatter;
import java.util.List;

@RunWith(MockitoJUnitRunner.class)
public class MockCitationDataSourceTest {
    @InjectMocks
    MockCitationDataSource mockCitationDataSource;

    @Mock
    ViolationManager violationManagerMock;

    @Mock
    private ResourceLoader resourceLoader;

    @Mock
    NamedParameterJdbcTemplate jdbcTemplate;

    @SuppressWarnings("unchecked")
    @Test
    public void returnsCitationGivenCitationNumberAndDOB() throws ParseException {
        final Violation VIOLATION = new Violation();
        VIOLATION.id = 4;
        final List<Violation> VIOLATIONS = Lists.newArrayList(VIOLATION);

        final Citation CITATION = new Citation();
        CITATION.id = 3;
        final String CITATIONNUMBER = "F3453";
        String dateString = "08/05/1965";
        DateTimeFormatter formatter = DateTimeFormatter.ofPattern("MM/dd/yyyy");
        LocalDate date = LocalDate.parse(dateString, formatter);

        when(jdbcTemplate.query(Matchers.anyString(), Matchers.anyMap(), Matchers.<RowMapper<Citation>>any()))
                .thenReturn(Lists.newArrayList(CITATION));

        List<Citation> citations = mockCitationDataSource.getByCitationNumberAndDOB(CITATIONNUMBER, date);
        when(violationManagerMock.getViolationsByCitationNumber(Matchers.anyString())).thenReturn(VIOLATIONS);

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

    @SuppressWarnings("unchecked")
    @Test
    public void returnsCitationsGivenDOBAndLicense() throws ParseException {
        final Violation VIOLATION = new Violation();
        VIOLATION.id = 4;
        final List<Violation> VIOLATIONS = Lists.newArrayList(VIOLATION);

        final Citation CITATION = new Citation();
        CITATION.id = 3;

        final List<Citation> CITATIONS = Lists.newArrayList(CITATION);

        when(jdbcTemplate.query(Matchers.anyString(), Matchers.anyMap(), Matchers.<RowMapper<Citation>>any()))
                .thenReturn(CITATIONS);
        when(violationManagerMock.getViolationsByCitationNumber(Matchers.anyString())).thenReturn(VIOLATIONS);

        String dateString = "08/05/1965";
        DateTimeFormatter formatter = DateTimeFormatter.ofPattern("MM/dd/yyyy");
        LocalDate date = LocalDate.parse(dateString, formatter);

        List<Citation> citations = mockCitationDataSource.getByLicenseAndDOB("someLiscensNumber", "MO", date);

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

    @SuppressWarnings("unchecked")
    @Test
    public void returnsCitationsWhenGivenDOBLastNameAndMultipleMunicipalities() throws ParseException, IOException {
        final Violation VIOLATION = new Violation();
        VIOLATION.id = 4;
        final List<Violation> VIOLATIONS = Lists.newArrayList(VIOLATION);

        final Citation CITATION = new Citation();
        CITATION.id = 3;

        final List<Citation> CITATIONS = Lists.newArrayList(CITATION);

        Resource resource = mock(Resource.class);
        when(resource.getInputStream()).thenReturn(null);
        when(resourceLoader.getResource(CLASSPATH_URL_PREFIX + "sql/citation/get-by-location.sql"))
                .thenReturn(resource);

        when(jdbcTemplate.query(Matchers.anyString(), Matchers.anyMap(), Matchers.<RowMapper<Citation>>any()))
                .thenReturn(CITATIONS);

        String dateString = "08/05/1965";
        DateTimeFormatter formatter = DateTimeFormatter.ofPattern("MM/dd/yyyy");
        LocalDate date = LocalDate.parse(dateString, formatter);

        List<Citation> citations = mockCitationDataSource.getByNameAndMunicipalitiesAndDOB("someLastName",
                Lists.newArrayList(19L, 20L), date);
        when(violationManagerMock.getViolationsByCitationNumber(Matchers.anyString())).thenReturn(VIOLATIONS);

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