indexinglanguage/src/test/java/com/yahoo/vespa/indexinglanguage/TypedExpressionConverterTestCase.java


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

// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.indexinglanguage;

import com.yahoo.vespa.indexinglanguage.expressions.*;
import org.junit.Test;

import static org.junit.Assert.assertEquals;

/**
 * @author Simon Thoresen Hult
 */
public class TypedExpressionConverterTestCase {

    @Test
    public void requireThatOnlyExpressionsOfGivenTypeIsConverted() {
        assertConvert(new AttributeExpression("foo"),
                      new IndexExpression("foo"));
        assertConvert(new StatementExpression(new AttributeExpression("foo")),
                      new StatementExpression(new IndexExpression("foo")));
        assertConvert(new SummaryExpression("foo"),
                      new SummaryExpression("foo"));
        assertConvert(new StatementExpression(new SummaryExpression("foo")),
                      new StatementExpression(new SummaryExpression("foo")));
    }

    private static void assertConvert(Expression before, Expression expectedAfter) {
        assertEquals(expectedAfter, new MyConverter().convert(before));
    }

    private static class MyConverter extends TypedExpressionConverter<AttributeExpression> {

        public MyConverter() {
            super(AttributeExpression.class);
        }

        @Override
        protected Expression typedConvert(AttributeExpression exp) {
            return new IndexExpression(exp.getFieldName());
        }
    }
}