brightsolid.solr.plugins.TestTargetPositionQuerySynonyms.java Source code

Introduction

Here is the source code for brightsolid.solr.plugins.TestTargetPositionQuerySynonyms.java

Source

package brightsolid.solr.plugins;

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import java.io.IOException;
import java.io.Reader;
import java.io.StringReader;

import org.apache.lucene.analysis.Analyzer;
import org.apache.lucene.analysis.MockAnalyzer;
import org.apache.lucene.analysis.MockTokenizer;
import org.apache.lucene.analysis.Tokenizer;
import org.apache.lucene.analysis.synonym.SolrSynonymParser;
import org.apache.lucene.analysis.synonym.SynonymFilter;
import org.apache.lucene.analysis.synonym.SynonymMap;
import org.apache.lucene.document.Document;
import org.apache.lucene.document.Field;
import org.apache.lucene.document.FieldType;
import org.apache.lucene.index.IndexReader;
import org.apache.lucene.index.RandomIndexWriter;
import org.apache.lucene.index.Term;
import org.apache.lucene.search.IndexSearcher;
import org.apache.lucene.search.TopDocs;
import org.apache.lucene.search.spans.SpanQuery;
import org.apache.lucene.search.spans.SpanTermQuery;
import org.apache.lucene.store.Directory;
import org.apache.lucene.util.LuceneTestCase;

/**
 * Tests for {@link SpanTargetPositionQuery}.
 */
public class TestTargetPositionQuerySynonyms extends LuceneTestCase {
    private Directory directory;
    private IndexReader reader;
    private IndexSearcher searcher;

    @Override
    public void setUp() throws Exception {
        super.setUp();

        String testFile = "one, uno, un\n" + "two, dos, too\n" + "three, free, tres";

        SolrSynonymParser parser = new SolrSynonymParser(true, true, new MockAnalyzer(random()));
        parser.parse(new StringReader(testFile));

        final SynonymMap map = parser.build();
        Analyzer analyzer = new Analyzer() {
            @Override
            protected TokenStreamComponents createComponents(String fieldName, Reader reader) {
                Tokenizer tokenizer = new MockTokenizer(reader, MockTokenizer.WHITESPACE, true);
                return new TokenStreamComponents(tokenizer, new SynonymFilter(tokenizer, map, false));
            }
        };

        directory = newDirectory();
        RandomIndexWriter iw = new RandomIndexWriter(random(), directory, analyzer);
        Document doc = new Document();
        FieldType newType = new FieldType(org.apache.lucene.document.TextField.TYPE_STORED);
        newType.setOmitNorms(true);
        Field field = newField("field", "", newType);
        field.fieldType().setOmitNorms(true);

        doc.add(field);

        field.setStringValue("one two three");
        iw.addDocument(doc);
        field.setStringValue("two three one");
        iw.addDocument(doc);
        field.setStringValue("three one two");
        iw.addDocument(doc);

        reader = iw.getReader();
        iw.close();
        searcher = newSearcher(reader);
    }

    @Override
    public void tearDown() throws Exception {
        reader.close();
        directory.close();
        super.tearDown();
    }

    public void testTargetPosition2() throws Exception {
        SpanTermQuery stq = new SpanTermQuery(new Term("field", "uno"));
        SpanQuery tpq = new SpanTargetPositionQuery(stq, 2);
        TopDocs td = searcher.search(tpq, 10);

        assertEquals(fieldValue(td, 0), "two three one");
        assertEquals(3, td.totalHits);
    }

    public void testTargetPosition1() throws Exception {
        SpanTermQuery stq = new SpanTermQuery(new Term("field", "too"));
        SpanQuery tpq = new SpanTargetPositionQuery(stq, 1);
        TopDocs td = searcher.search(tpq, 10);

        assertEquals(fieldValue(td, 0), "one two three");
        assertEquals(3, td.totalHits);
    }

    private String fieldValue(TopDocs td, int i) throws IOException {
        return searcher.doc(td.scoreDocs[i].doc).getField("field").stringValue();
    }
}