aboutsummaryrefslogtreecommitdiffstats
path: root/config-model/src/test/java/com/yahoo/schema/RankingExpressionValidationTestCase.java
blob: e4b0b6439054dbba2b465d0f7658ee00ff135c49 (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
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.schema;

import com.yahoo.schema.derived.DerivedConfiguration;
import com.yahoo.schema.parser.ParseException;
import com.yahoo.yolean.Exceptions;
import org.junit.jupiter.api.Test;

import static org.junit.jupiter.api.Assertions.assertTrue;
import static org.junit.jupiter.api.Assertions.fail;

/**
 * @author bratseth
 */
public class RankingExpressionValidationTestCase extends AbstractSchemaTestCase {

    @Test
    void testInvalidExpressionProducesException() throws ParseException {
        assertFailsExpression("&/%(/%&");
        assertFailsExpression("if(a==b,b)");
    }

    private void assertFailsExpression(String expression) throws ParseException {
        try {
            RankProfileRegistry registry = new RankProfileRegistry();
            Schema schema = importWithExpression(expression, registry);
            new DerivedConfiguration(schema, registry); // cause rank profile parsing
            fail("No exception on incorrect ranking expression " + expression);
        } catch (IllegalArgumentException e) {
            // Success
            assertTrue(Exceptions.toMessageString(e).startsWith("Illegal first phase ranking function: Could not parse ranking expression '" + expression + "' in default, firstphase.:"));
        }
    }

    private Schema importWithExpression(String expression, RankProfileRegistry registry) throws ParseException {
        ApplicationBuilder builder = new ApplicationBuilder(registry);
        builder.addSchema("search test {" +
                          "    document test { " +
                          "        field a type string { " +
                          "            indexing: index " +
                          "        }" +
                          "    }" +
                          "    rank-profile default {" +
                          "        first-phase {" +
                          "            expression: " + expression +
                          "        }" +
                          "    }" +
                          "}");
        builder.build(true);
        return builder.getSchema();
    }

}