aboutsummaryrefslogtreecommitdiffstats
path: root/container-search/src/test/java/com/yahoo/prelude/semantics/test/ParameterTestCase.java
blob: 4474df072e05dfcf17991cb90d15001be421050c (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.prelude.semantics.test;

import com.yahoo.search.Query;
import org.junit.jupiter.api.Test;

import static org.junit.jupiter.api.Assertions.assertEquals;

/**
 * Tests parameter matching and production
 *
 * @author bratseth
 */
public class ParameterTestCase extends RuleBaseAbstractTestCase {

    public ParameterTestCase() {
        super("parameter.sr");
    }

    /** Tests parameter literal matching */
    @Test
    void testLiteralEquals() {
        assertSemantics("a", "a");
        assertSemantics("RANK a foo:a", "a&ranking=category");
        assertSemantics("a", "a&ranking=somethingelse");
        assertSemantics("a", "a&otherparam=category");
    }

    /** Tests parameter matching of larger */
    @Test
    void testLarger() {
        assertSemantics("a", "a");
        assertSemantics("AND a largepage", "a&hits=11");
        assertSemantics("AND a largepage", "a&hits=12");
    }

    /** Tests parameter containment matching */
    @Test
    void testContainsAsList() {
        assertSemantics("a", "a");
        assertSemantics("AND a intent:music", "a&search=music");
        assertSemantics("AND a intent:music", "a&search=music,books");
        assertSemantics("AND a intent:music", "a&search=kanoos,music,books");
    }

    /** Tests parameter production */
    @Test
    void testParameterProduction() {
        assertParameterSemantics("AND a b c", "a b c", "search", "[letters, alphabet]");
        assertParameterSemantics("AND a c d", "a c d", "search", "[letters, someletters]");
        assertParameterSemantics("+(AND a d e) -letter:c", "a d e", "search", "[someletters]");
        assertParameterSemantics("AND a d f", "a d f", "rank-profile", "foo");
        assertParameterSemantics("AND a f g", "a f g", "grouping.nolearning", "true");
    }

    @Test
    void testMultipleAlternativeParameterValuesInCondition() {
        assertInputRankParameterSemantics("WEAKAND(100) one", "foo", "cat");
        assertInputRankParameterSemantics("WEAKAND(100) one", "foo", "cat0");
        assertInputRankParameterSemantics("WEAKAND(100) one", "bar", "cat");
        assertInputRankParameterSemantics("WEAKAND(100) one", "bar", "cat0");
        assertInputRankParameterSemantics("WEAKAND(100) one one", "foo+bar", "cat0");
        assertInputRankParameterSemantics("WEAKAND(100) fuki sushi", "fuki+sushi", "cat0");
    }

    private void assertInputRankParameterSemantics(String producedQuery,String inputQuery, String rankParameterValue) {
        assertInputRankParameterSemantics(producedQuery, inputQuery, rankParameterValue, 0);
    }

    private void assertInputRankParameterSemantics(String producedQuery, String inputQuery,
                                                   String rankParameterValue, int tracelevel) {
        Query query = new Query("?query=" + inputQuery + "&tracelevel=0&tracelevel.rules=" + tracelevel);
        query.getRanking().setProfile(rankParameterValue);
        query.properties().set("tracelevel.rules", tracelevel);
        assertSemantics(producedQuery, query);
    }

    private void assertParameterSemantics(String producedQuery, String inputQuery,
                                          String producedParameterName, String producedParameterValue) {
        Query query = assertSemantics(producedQuery, inputQuery);
        assertEquals(producedParameterValue, query.properties().getString(producedParameterName));
    }

}