CcRemovedJudgmentsFinderTest.java example

Explorer
saos-master
package pl.edu.icm.saos.importer.commoncourt.judgment.remove;

import static org.hamcrest.MatcherAssert.assertThat;
import static org.hamcrest.Matchers.containsInAnyOrder;
import static org.mockito.Mockito.when;

import java.util.List;

import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.runners.MockitoJUnitRunner;
import org.powermock.reflect.Whitebox;

import com.google.common.collect.Lists;

import pl.edu.icm.saos.importer.commoncourt.judgment.SourceCcjExternalRepository;
import pl.edu.icm.saos.persistence.model.SourceCode;
import pl.edu.icm.saos.persistence.repository.JudgmentRepository;

/**
 * @author madryk
 */
@RunWith(MockitoJUnitRunner.class)
public class CcRemovedJudgmentsFinderTest {

    @InjectMocks
    private CcRemovedJudgmentsFinder ccRemovedJudgmentsFinder = new CcRemovedJudgmentsFinder();
    
    @Mock
    private SourceCcjExternalRepository sourceCcjExternalRepository;
    
    @Mock
    private JudgmentRepository judgmentRepository;
    
    
    @Before
    public void setup() {
        ccRemovedJudgmentsFinder.setExternalRepositoryPageSize(3);
        Whitebox.setInternalState(ccRemovedJudgmentsFinder, "judgmentRepositoryPageSize", 2);
    }
    
    //------------------------ TESTS --------------------------
    
    @Test
    public void findRemovedJudgments() {
        
        // given
        
        List<String> sourceIdsPage1 = Lists.newArrayList("sourceId_1", "sourceId_2", "sourceId_3");
        List<String> sourceIdsPage2 = Lists.newArrayList("sourceId_4", "sourceId_5");
        
        when(judgmentRepository.findAllIdsBySourceCode(SourceCode.COMMON_COURT)).thenReturn(Lists.newArrayList(1L, 3L, 4L, 6L, 7L));
        
        when(sourceCcjExternalRepository.findJudgmentIds(0, 3, null)).thenReturn(sourceIdsPage1);
        when(sourceCcjExternalRepository.findJudgmentIds(1, 3, null)).thenReturn(sourceIdsPage2);
        when(sourceCcjExternalRepository.findJudgmentIds(2, 3, null)).thenReturn(Lists.newArrayList());
        
        when(judgmentRepository.findAllIdsBySourceCodeAndSourceJudgmentIds(SourceCode.COMMON_COURT, Lists.newArrayList("sourceId_1", "sourceId_2")))
                .thenReturn(Lists.newArrayList(1L));
        when(judgmentRepository.findAllIdsBySourceCodeAndSourceJudgmentIds(SourceCode.COMMON_COURT, Lists.newArrayList("sourceId_3")))
                .thenReturn(Lists.newArrayList(3L));
        when(judgmentRepository.findAllIdsBySourceCodeAndSourceJudgmentIds(SourceCode.COMMON_COURT, sourceIdsPage2))
                .thenReturn(Lists.newArrayList(4L));
        
        // execute
        
        List<Long> removedSourceIds = ccRemovedJudgmentsFinder.findCcRemovedJudgmentSourceIds();
        
        
        // assert
        
        assertThat(removedSourceIds, containsInAnyOrder(6L, 7L));
    }
    
}