// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.searchdefinition.processing;

import com.yahoo.searchdefinition.Search;
import com.yahoo.searchdefinition.SearchBuilder;
import com.yahoo.searchdefinition.SearchDefinitionTestCase;
import com.yahoo.searchdefinition.document.Matching;
import com.yahoo.searchdefinition.document.SDField;
import com.yahoo.searchdefinition.document.Stemming;
import com.yahoo.searchdefinition.parser.ParseException;
import org.junit.Test;

import java.io.IOException;
import java.util.List;

import static org.junit.Assert.*;
/**
 * @author bratseth
 */
public class NGramTestCase extends SearchDefinitionTestCase {

    @Test
    public void testNGram() throws IOException, ParseException {
        Search search = SearchBuilder.buildFromFile("src/test/examples/ngram.sd");
        assertNotNull(search);

        SDField gram1=search.getConcreteField("gram_1");
        assertEquals(Matching.Type.GRAM,gram1.getMatching().getType());
        assertEquals(1,gram1.getMatching().getGramSize());

        SDField gram2=search.getConcreteField("gram_2");
        assertEquals(Matching.Type.GRAM,gram2.getMatching().getType());
        assertEquals(-1,gram2.getMatching().getGramSize()); // Not set explicitly

        SDField gram3=search.getConcreteField("gram_3");
        assertEquals(Matching.Type.GRAM,gram3.getMatching().getType());
        assertEquals(3,gram3.getMatching().getGramSize());

        assertEquals("input gram_1 | ngram 1 | index gram_1 | summary gram_1",gram1.getIndexingScript().iterator().next().toString());
        assertEquals("input gram_2 | ngram 2 | index gram_2",gram2.getIndexingScript().iterator().next().toString());
        assertEquals("input gram_3 | ngram 3 | index gram_3",gram3.getIndexingScript().iterator().next().toString());

        assertFalse(gram1.getNormalizing().doRemoveAccents());
        assertEquals(Stemming.NONE,gram1.getStemming());

        List<String> queryCommands=gram1.getQueryCommands();
        assertEquals(1,queryCommands.size());
        assertEquals("ngram 1",queryCommands.get(0));
    }

    @Test
    public void testInvalidNGramSetting1() throws IOException, ParseException {
        try {
            Search search = SearchBuilder.buildFromFile("src/test/examples/invalidngram1.sd");
            fail("Should cause an exception");
        }
        catch (IllegalArgumentException e) {
            assertEquals("gram-size can only be set when the matching mode is 'gram'",e.getMessage());
        }
    }

    @Test
    public void testInvalidNGramSetting2() throws IOException, ParseException {
        try {
            Search search = SearchBuilder.buildFromFile("src/test/examples/invalidngram2.sd");
            fail("Should cause an exception");
        }
        catch (IllegalArgumentException e) {
            assertEquals("gram-size can only be set when the matching mode is 'gram'",e.getMessage());
        }
    }

    @Test
    public void testInvalidNGramSetting3() throws IOException, ParseException {
        try {
            Search search = SearchBuilder.buildFromFile("src/test/examples/invalidngram3.sd");
            fail("Should cause an exception");
        }
        catch (IllegalArgumentException e) {
            assertEquals("gram matching is not supported with attributes, use 'index' not 'attribute' in indexing",e.getMessage());
        }
    }

}