aboutsummaryrefslogtreecommitdiffstats
path: root/container-search/src/test/java/com/yahoo/prelude/query/test/SameElementItemTestCase.java
blob: 198ad2b46f9881a2951060d67b5ff795911e3bfa (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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.prelude.query.test;

import com.yahoo.prelude.query.AndItem;
import com.yahoo.prelude.query.IntItem;
import com.yahoo.prelude.query.Item;
import com.yahoo.prelude.query.SameElementItem;
import com.yahoo.prelude.query.Substring;
import com.yahoo.prelude.query.TermItem;
import com.yahoo.prelude.query.WordAlternativesItem;
import com.yahoo.prelude.query.WordItem;
import org.junit.jupiter.api.Test;

import java.util.Collections;
import java.util.List;
import java.util.Optional;

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

public class SameElementItemTestCase {

    @Test
    void testAddItem() {
        SameElementItem s = new SameElementItem("structa");
        s.addItem(new WordItem("b", "f1"));
        s.addItem(new WordItem("c", "f2"));
        s.addItem(new WordItem("d", "f3"));
        assertEquals("structa:{f1:b f2:c f3:d}", s.toString());
    }

    @Test
    void testClone() {
        SameElementItem s = new SameElementItem("structa");
        s.addItem(new WordItem("b", "f1"));
        s.addItem(new WordItem("c", "f2"));
        s.addItem(new WordItem("d", "f3"));
        assertEquals("structa:{f1:b f2:c f3:d}", s.toString());
        SameElementItem c = (SameElementItem) s.clone();
        assertEquals("structa:{f1:b f2:c f3:d}", c.toString());
    }

    @Test
    void requireAllChildrenHaveStructMemberNameSet() {
        try {
            SameElementItem s = new SameElementItem("structa");
            s.addItem(new WordItem("b", "f1"));
            s.addItem(new WordItem("c"));
            fail("Expected exception");
        }
        catch (IllegalArgumentException e) { // success
            assertEquals("Struct fieldname can not be empty", e.getMessage());
        }
    }

    @Test
    void requireAllowCommonPrefix() {
        SameElementItem s = new SameElementItem("structa");
        s.addItem(new WordItem("b", "f1"));
        s.addItem(new WordItem("c", "structaf2"));
        assertEquals("structa:{f1:b structaf2:c}", s.toString());
    }

    @Test
    void requireChildrenCanHavePrefixCommonWithParent() {
        SameElementItem s = new SameElementItem("structa");
        s.addItem(new WordItem("b", "f1"));
        s.addItem(new WordItem("c", "structa.f2"));
        assertEquals("structa:{f1:b structa.f2:c}", s.toString());
    }

    @Test
    void requireAllChildrenHaveNonEmptyTerm() {
        try {
            SameElementItem s = new SameElementItem("structa");
            s.addItem(new WordItem("", "f2"));
            fail("Expected exception");
        }
        catch (IllegalArgumentException e) { // Success
            assertEquals("The word of a word item can not be empty", e.getMessage());
        }
    }

    @Test
    void requireNoChildrenAreWordAlternatives() {
        try {
            SameElementItem s = new SameElementItem("structa");
            s.addItem(new AndItem());
            fail("Expected exception");
        }
        catch (IllegalArgumentException e) { // Success
            assertEquals("Child item (AND ) should be an instance of class com.yahoo.prelude.query.TermItem but is class com.yahoo.prelude.query.AndItem",
                    e.getMessage());
        }
    }

    @Test
    void requireAllChildrenAreTermItems() {
        try {
            SameElementItem s = new SameElementItem("structa");
            s.addItem(new WordAlternativesItem("test", true, new Substring("origin"), List.of(new WordAlternativesItem.Alternative("a", 0.3))));
            fail("Expected exception");
        }
        catch (IllegalArgumentException e) { // Success
            assertEquals("Child item WORD_ALTERNATIVES test:[ a(0.3) ] should NOT be an instance of class com.yahoo.prelude.query.WordAlternativesItem but is class com.yahoo.prelude.query.WordAlternativesItem",
                    e.getMessage());
        }
    }

    private void verifyExtractSingle(TermItem term) {
        String subFieldName = term.getIndexName();
        SameElementItem s = new SameElementItem("structa");
        s.addItem(term);
        Optional<Item> single =s.extractSingleChild();
        assertTrue(single.isPresent());
        assertEquals(((TermItem)single.get()).getIndexName(), s.getFieldName() + "." + subFieldName);
    }

    @Test
    void requireExtractSingleItemToExtractSingles() {
        verifyExtractSingle(new WordItem("b", "f1"));
        verifyExtractSingle(new IntItem("7", "f1"));
    }

    @Test
    void requireExtractSingleItemToExtractSinglesOnly() {
        SameElementItem s = new SameElementItem("structa");
        s.addItem(new WordItem("b", "f1"));
        s.addItem(new WordItem("c", "f2"));
        assertTrue(s.extractSingleChild().isEmpty());
    }

}