container-search/src/test/java/com/yahoo/search/result/test/FillingTestCase.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
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

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

import com.yahoo.search.result.Hit;
import com.yahoo.search.result.HitGroup;
import org.junit.jupiter.api.Test;

import java.util.Collections;

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

/**
 * @author bratseth
 */
public class FillingTestCase {

    @Test
    void testFillingAPIConsistency() {
        HitGroup group = new HitGroup();
        group.add(new Hit("hit:1"));
        group.add(new Hit("hit:2"));
        assertTrue(group.isFilled("summary"));
    }

    @Test
    void testFillingAPIConsistencyTwoPhase() {
        HitGroup group = new HitGroup();
        group.add(createNonFilled("hit:1"));
        group.add(createNonFilled("hit:2"));
        assertFalse(group.isFilled("summary"));
        fillHitsIn(group, "summary");
        group.analyze();
        assertTrue(group.isFilled("summary"));  // consistent again
    }

    @Test
    void testFillingAPIConsistencyThreePhase() {
        HitGroup group = new HitGroup();
        group.add(createNonFilled("hit:1"));
        group.add(createNonFilled("hit:2"));
        assertFalse(group.isFilled("summary"));
        assertFalse(group.isFilled("otherSummary"));
        fillHitsIn(group, "otherSummary");
        group.analyze();
        assertFalse(group.isFilled("summary"));
        assertTrue(group.isFilled("otherSummary"));
        fillHitsIn(group, "summary");
        assertTrue(group.isFilled("otherSummary"));
        group.analyze();
        assertTrue(group.isFilled("summary"));  // consistent again
        assertTrue(group.isFilled("otherSummary"));
    }

    @Test
    void testPartiallyFilledWith2Hits() {
        Hit hit1 = new Hit("id1");
        Hit hit2 = new Hit("id2");

        hit1.setFilled("summary");
        hit2.setFillable();

        HitGroup hits = new HitGroup();
        hits.add(hit1);
        hits.add(hit2);

        assertEquals(Collections.emptySet(), hits.getFilled());
    }

    @Test
    void testPartiallyFilledDiverse() {
        Hit hit1 = new Hit("id1");
        Hit hit2 = new Hit("id2");
        Hit hit3 = new Hit("id3");

        hit1.setFilled("summary1");
        hit1.setFilled("summary2");
        hit2.setFilled("summary1");
        hit3.setFilled("summary1");

        HitGroup hits = new HitGroup();
        hits.add(hit1);
        hits.add(hit2);
        hits.add(hit3);

        assertEquals(Collections.singleton("summary1"), hits.getFilled());
    }

    private Hit createNonFilled(String id) {
        Hit hit=new Hit(id);
        hit.setFillable();
        return hit;
    }

    private void fillHitsIn(HitGroup group,String summary) {
        for (Hit hit : group.asList()) {
            if (hit.isMeta()) continue;
            hit.setFilled(summary);
        }
    }

}