package pl.edu.icm.saos.search.indexing;

import static org.hamcrest.MatcherAssert.assertThat;
import static org.hamcrest.Matchers.containsInAnyOrder;

import java.util.List;

import org.junit.Test;
import org.junit.experimental.categories.Category;
import org.springframework.beans.factory.annotation.Autowired;

import pl.edu.icm.saos.common.json.JsonNormalizer;
import pl.edu.icm.saos.common.testcommon.category.SlowTest;
import pl.edu.icm.saos.persistence.common.TestInMemoryEnrichmentTagFactory;
import pl.edu.icm.saos.persistence.enrichment.EnrichmentTagRepository;
import pl.edu.icm.saos.persistence.enrichment.model.EnrichmentTag;
import pl.edu.icm.saos.persistence.enrichment.model.EnrichmentTagTypes;
import pl.edu.icm.saos.persistence.model.CommonCourtJudgment;
import pl.edu.icm.saos.persistence.model.CourtCase;
import pl.edu.icm.saos.persistence.model.Judgment;
import pl.edu.icm.saos.persistence.model.JudgmentSourceInfo;
import pl.edu.icm.saos.persistence.model.SourceCode;
import pl.edu.icm.saos.persistence.repository.JudgmentRepository;
import pl.edu.icm.saos.search.SearchTestSupport;

import com.google.common.collect.Lists;

/**
 * @author madryk
 */
@Category(SlowTest.class)
public class JudgmentIndexingItemFetcherTest extends SearchTestSupport {

    @Autowired
    private JudgmentIndexingItemFetcher judgmentIndexingItemFetcher;
    
    @Autowired
    private JudgmentRepository judgmentRepository;
    
    @Autowired
    private EnrichmentTagRepository enrichmentTagRepository;
    
    
    //------------------------ TESTS --------------------------
    
    @Test
    public void fetchJudgmentIndexingItems() {
        // given
        Judgment judgment1 = createCcJudgment(SourceCode.COMMON_COURT, "AA1", "ABC1");
        Judgment judgment2 = createCcJudgment(SourceCode.COMMON_COURT, "AA2", "ABC");
        Judgment judgment3 = createCcJudgment(SourceCode.COMMON_COURT, "AA3", "ABC");
        Judgment judgment4 = createCcJudgment(SourceCode.COMMON_COURT, "AA4", "ABC4");
        Judgment judgment5 = createCcJudgment(SourceCode.COMMON_COURT, "AA5", "ABC5");
        
        judgmentRepository.markAsIndexed(judgment3.getId());
        
        
        EnrichmentTag tag1 = TestInMemoryEnrichmentTagFactory.createEnrichmentTag(judgment1.getId(), EnrichmentTagTypes.REFERENCED_COURT_CASES,
                JsonNormalizer.normalizeJson("[{'caseNumber':'ABC','judgmentIds':[" + judgment2.getId() + "," + judgment3.getId() + "]},"
                        + "{'caseNumber':'ABC4','judgmentIds':[" + judgment4.getId() + "]}]"));
        
        EnrichmentTag tag2 = TestInMemoryEnrichmentTagFactory.createEnrichmentTag(judgment2.getId(), EnrichmentTagTypes.REFERENCED_COURT_CASES,
                JsonNormalizer.normalizeJson("[{'caseNumber':'ABC','judgmentIds':[" + judgment3.getId() + "]},"
                        + "{'caseNumber':'ABC4','judgmentIds':[" + judgment4.getId() + "]}]"));
        
        
        enrichmentTagRepository.save(Lists.newArrayList(tag1, tag2));
        
        // execute
        List<JudgmentIndexingItem> retJudgmentIndexingItems = 
                judgmentIndexingItemFetcher.fetchJudgmentIndexingItems();
        
        
        // assert
        JudgmentIndexingItem expectedIndexingItem1 = new JudgmentIndexingItem(judgment1.getId(), 0L);
        JudgmentIndexingItem expectedIndexingItem2 = new JudgmentIndexingItem(judgment2.getId(), 1L);
        JudgmentIndexingItem expectedIndexingItem4 = new JudgmentIndexingItem(judgment4.getId(), 2L);
        JudgmentIndexingItem expectedIndexingItem5 = new JudgmentIndexingItem(judgment5.getId(), 0L);
        
        assertThat(retJudgmentIndexingItems, containsInAnyOrder(
                expectedIndexingItem1, expectedIndexingItem2, expectedIndexingItem4, expectedIndexingItem5));
        
    }
    
    
    //------------------------ PRIVATE --------------------------
    
    private CommonCourtJudgment createCcJudgment(SourceCode sourceCode, String sourceJudgmentId, String caseNumber) {
        CommonCourtJudgment ccJudgment = new CommonCourtJudgment();
        JudgmentSourceInfo sourceInfo = new JudgmentSourceInfo();
        sourceInfo.setSourceCode(sourceCode);
        sourceInfo.setSourceJudgmentId(sourceJudgmentId);
        ccJudgment.setSourceInfo(sourceInfo);
        ccJudgment.addCourtCase(new CourtCase(caseNumber));
        judgmentRepository.save(ccJudgment);
        return ccJudgment;
    }
}