grants-writer/src/test/java/scot/carricksoftware/grants/bootstrap/DataLoadCensusTest.java

package scot.carricksoftware.grants.bootstrap;

import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.ArgumentCaptor;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;
import scot.carricksoftware.grants.commands.census.CensusCommand;
import scot.carricksoftware.grants.commands.census.CensusEntryCommand;
import scot.carricksoftware.grants.domains.census.Census;
import scot.carricksoftware.grants.domains.people.Person;
import scot.carricksoftware.grants.domains.places.Place;
import scot.carricksoftware.grants.enums.censusentry.CensusEntryRelationship;
import scot.carricksoftware.grants.services.census.CensusEntryService;
import scot.carricksoftware.grants.services.census.CensusService;
import scot.carricksoftware.grants.services.people.PersonService;
import scot.carricksoftware.grants.services.places.places.PlaceService;


import java.time.LocalDate;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;
import static scot.carricksoftware.grants.GenerateRandomCensusValues.GetRandomCensus;
import static scot.carricksoftware.grants.GenerateRandomPeopleValues.GetRandomPerson;
import static scot.carricksoftware.grants.GenerateRandomPlaceValues.GetRandomPlace;


@ExtendWith(MockitoExtension.class)
public class DataLoadCensusTest {

    private DataLoadCensus dataLoadCensus;

    @Mock
    private CensusEntryService censusEntryServiceMock;

    @Mock
    private CensusService censusServiceMock;

    @Mock
    private PlaceService placeServiceMock;

    @Mock
    private PersonService personServiceMock;


    @BeforeEach
    public void setUp() {
        dataLoadCensus = new DataLoadCensus(censusServiceMock,
                censusEntryServiceMock,
                placeServiceMock,
                personServiceMock);
    }

    @Test
    public void censusIsCreatedTest() {
        Place place = GetRandomPlace();
        ArgumentCaptor<CensusCommand> captor = ArgumentCaptor.forClass(CensusCommand.class);
        when(placeServiceMock.findById(1L)).thenReturn(place);
        dataLoadCensus.load();
        verify(censusServiceMock).saveCensusCommand(captor.capture());

        assertEquals(captor.getValue().getDate(), LocalDate.now());
        assertEquals(captor.getValue().getPlace(), place);
    }

    @Test
    public void censusEntryIsCreatedTest() {
        ArgumentCaptor<CensusEntryCommand> captor = ArgumentCaptor.forClass(CensusEntryCommand.class);
        Census census = GetRandomCensus();
        Person person = GetRandomPerson();
        CensusEntryRelationship relationship = CensusEntryRelationship.COUSIN;
        when(censusServiceMock.findById(1L)).thenReturn(census);
        when(personServiceMock.findById(1L)).thenReturn(person);

        dataLoadCensus.load();

        verify(censusEntryServiceMock).saveCensusEntryCommand(captor.capture());
        assertEquals("Archie Grant", captor.getValue().getName());
        assertEquals(census, captor.getValue().getCensus());
        assertEquals(person, captor.getValue().getPerson());
        assertEquals(relationship, captor.getValue().getRelationship());
    }


}