lucene/analysis/common/src/test/org/apache/lucene/analysis/synonym/TestSolrSynonymParser.java

/*
 * 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.
 */
package org.apache.lucene.analysis.synonym;

import java.io.StringReader;
import java.text.ParseException;
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.en.EnglishAnalyzer;

/**
 * Tests parser for the Solr synonyms format
 *
 * @lucene.experimental
 */
@Deprecated
public class TestSolrSynonymParser extends BaseSynonymParserTestCase {

  /** Tests some simple examples from the solr wiki */
  public void testSimple() throws Exception {
    String testFile =
        "i-pod, ipod, ipoooood\n" + "foo => foo bar\n" + "foo => baz\n" + "this test, that testing";

    Analyzer analyzer = new MockAnalyzer(random());
    SolrSynonymParser parser = new SolrSynonymParser(true, true, analyzer);
    parser.parse(new StringReader(testFile));
    final SynonymMap map = parser.build();
    analyzer.close();

    analyzer =
        new Analyzer() {
          @Override
          protected TokenStreamComponents createComponents(String fieldName) {
            Tokenizer tokenizer = new MockTokenizer(MockTokenizer.WHITESPACE, true);
            return new TokenStreamComponents(tokenizer, new SynonymFilter(tokenizer, map, true));
          }
        };

    assertAnalyzesTo(analyzer, "ball", new String[] {"ball"}, new int[] {1});

    assertAnalyzesTo(
        analyzer, "i-pod", new String[] {"i-pod", "ipod", "ipoooood"}, new int[] {1, 0, 0});

    assertAnalyzesTo(analyzer, "foo", new String[] {"foo", "baz", "bar"}, new int[] {1, 0, 1});

    assertAnalyzesTo(
        analyzer,
        "this test",
        new String[] {"this", "that", "test", "testing"},
        new int[] {1, 0, 1, 0});
    analyzer.close();
  }

  /** parse a syn file with bad syntax */
  public void testInvalidDoubleMap() throws Exception {
    String testFile = "a => b => c";
    Analyzer analyzer = new MockAnalyzer(random());
    SolrSynonymParser parser = new SolrSynonymParser(true, true, analyzer);
    expectThrows(
        ParseException.class,
        () -> {
          parser.parse(new StringReader(testFile));
        });
    analyzer.close();
  }

  /** parse a syn file with bad syntax */
  public void testInvalidAnalyzesToNothingOutput() throws Exception {
    String testFile = "a => 1";
    Analyzer analyzer = new MockAnalyzer(random(), MockTokenizer.SIMPLE, false);
    SolrSynonymParser parser = new SolrSynonymParser(true, true, analyzer);
    expectThrows(
        ParseException.class,
        () -> {
          parser.parse(new StringReader(testFile));
        });
    analyzer.close();
  }

  /** parse a syn file with bad syntax */
  public void testInvalidAnalyzesToNothingInput() throws Exception {
    String testFile = "1 => a";
    Analyzer analyzer = new MockAnalyzer(random(), MockTokenizer.SIMPLE, false);
    SolrSynonymParser parser = new SolrSynonymParser(true, true, analyzer);
    expectThrows(
        ParseException.class,
        () -> {
          parser.parse(new StringReader(testFile));
        });
    analyzer.close();
  }

  /** parse a syn file with bad syntax */
  public void testInvalidPositionsInput() throws Exception {
    String testFile = "testola => the test";
    Analyzer analyzer = new EnglishAnalyzer();
    SolrSynonymParser parser = new SolrSynonymParser(true, true, analyzer);
    expectThrows(
        ParseException.class,
        () -> {
          parser.parse(new StringReader(testFile));
        });
    analyzer.close();
  }

  /** parse a syn file with bad syntax */
  public void testInvalidPositionsOutput() throws Exception {
    String testFile = "the test => testola";
    Analyzer analyzer = new EnglishAnalyzer();
    SolrSynonymParser parser = new SolrSynonymParser(true, true, analyzer);
    expectThrows(
        ParseException.class,
        () -> {
          parser.parse(new StringReader(testFile));
        });
    analyzer.close();
  }

  /** parse a syn file with some escaped syntax chars */
  public void testEscapedStuff() throws Exception {
    String testFile = "a\\=>a => b\\=>b\n" + "a\\,a => b\\,b";
    Analyzer analyzer = new MockAnalyzer(random(), MockTokenizer.KEYWORD, false);
    SolrSynonymParser parser = new SolrSynonymParser(true, true, analyzer);
    parser.parse(new StringReader(testFile));
    final SynonymMap map = parser.build();
    analyzer.close();
    analyzer =
        new Analyzer() {
          @Override
          protected TokenStreamComponents createComponents(String fieldName) {
            Tokenizer tokenizer = new MockTokenizer(MockTokenizer.KEYWORD, false);
            return new TokenStreamComponents(tokenizer, new SynonymFilter(tokenizer, map, false));
          }
        };

    assertAnalyzesTo(analyzer, "ball", new String[] {"ball"}, new int[] {1});

    assertAnalyzesTo(analyzer, "a=>a", new String[] {"b=>b"}, new int[] {1});

    assertAnalyzesTo(analyzer, "a,a", new String[] {"b,b"}, new int[] {1});
    analyzer.close();
  }

  /** Verify type of token and positionLength after analyzer. */
  public void testPositionLengthAndTypeSimple() throws Exception {
    String testFile = "spider man, spiderman";

    Analyzer analyzer = new MockAnalyzer(random());
    SolrSynonymParser parser = new SolrSynonymParser(true, true, analyzer);
    parser.parse(new StringReader(testFile));
    final SynonymMap map = parser.build();
    analyzer.close();

    analyzer =
        new Analyzer() {
          @Override
          protected TokenStreamComponents createComponents(String fieldName) {
            Tokenizer tokenizer = new MockTokenizer(MockTokenizer.WHITESPACE, true);
            return new TokenStreamComponents(tokenizer, new SynonymFilter(tokenizer, map, true));
          }
        };

    assertAnalyzesToPositions(
        analyzer,
        "spider man",
        new String[] {"spider", "spiderman", "man"},
        new String[] {"word", "SYNONYM", "word"},
        new int[] {1, 0, 1},
        new int[] {1, 2, 1});
  }

  /** Verify type of original token is "word", others are Synonym. */
  public void testTypes() throws Exception {
    String testFile = "woods, wood, forest";

    Analyzer analyzer = new MockAnalyzer(random());
    SolrSynonymParser parser = new SolrSynonymParser(true, true, analyzer);
    parser.parse(new StringReader(testFile));
    final SynonymMap map = parser.build();
    analyzer.close();

    analyzer =
        new Analyzer() {
          @Override
          protected TokenStreamComponents createComponents(String fieldName) {
            Tokenizer tokenizer = new MockTokenizer(MockTokenizer.WHITESPACE, true);
            return new TokenStreamComponents(tokenizer, new SynonymFilter(tokenizer, map, true));
          }
        };

    assertAnalyzesTo(
        analyzer,
        "lost in the forest",
        new String[] {"lost", "in", "the", "forest", "woods", "wood"},
        new String[] {"word", "word", "word", "word", "SYNONYM", "SYNONYM"});
  }

  /** Test parsing of simple examples. */
  public void testParseSimple() throws Exception {
    String testFile =
        "spider man, spiderman\n"
            + "usa,united states,u s a,united states of america\n"
            + "mystyped, mistyped => mistyped\n"
            + "foo => foo bar\n"
            + "foo => baz";

    Analyzer analyzer = new MockAnalyzer(random());
    SolrSynonymParser parser = new SolrSynonymParser(true, true, analyzer);
    parser.parse(new StringReader(testFile));
    final SynonymMap map = parser.build();
    analyzer.close();

    assertEntryEquals(map, "spiderman", true, "spider man");
    assertEntryEquals(map, "spider man", true, "spiderman");

    assertEntryEquals(
        map, "usa", true, new String[] {"united states", "u s a", "united states of america"});
    assertEntryEquals(
        map, "united states", true, new String[] {"usa", "u s a", "united states of america"});
    assertEntryEquals(
        map, "u s a", true, new String[] {"usa", "united states", "united states of america"});
    assertEntryEquals(
        map, "united states of america", true, new String[] {"usa", "u s a", "united states"});

    assertEntryEquals(map, "mistyped", false, "mistyped");
    assertEntryEquals(map, "mystyped", false, "mistyped");

    assertEntryEquals(map, "foo", false, new String[] {"foo bar", "baz"});
    assertEntryAbsent(map, "baz");
    assertEntryAbsent(map, "bar");
  }
}