aboutsummaryrefslogtreecommitdiffstats
path: root/container-search/src/test/java/com/yahoo/search/result/CoverageTestCase.java
blob: cadbc6adf5cf5aa8a19364f73e24e5ddd858ed4d (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
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.search.result;

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

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertFalse;
import static org.junit.jupiter.api.Assertions.assertNull;
import static org.junit.jupiter.api.Assertions.assertTrue;

/**
 * @author Steinar Knutsen
 */
public class CoverageTestCase {

    @Test
    void testZeroCoverage() {
        Coverage c = new Coverage(0L, 0, 0, 0);
        assertEquals(0, c.getResultPercentage());
        assertEquals(0, c.getResultSets());
    }

    @Test
    void testActiveCoverage() {
        Coverage c = new Coverage(6, 5, 1);
        assertEquals(5, c.getActive());
        assertEquals(6, c.getDocs());

        Coverage d = new Coverage(7, 6, 1);
        c.merge(d);
        assertEquals(11, c.getActive());
        assertEquals(13, c.getDocs());
    }

    @Test
    void testCoverageBasedOnTargetActive() {
        var c = new Coverage(8, 10, 1).setTargetActive(16);
        assertEquals(50, c.getResultPercentage());
    }

    @Test
    void testDefaultCoverage() {
        boolean create = true;

        Result r1 = new Result(new Query());
        assertEquals(0, r1.getCoverage(create).getResultSets());
        Result r2 = new Result(new Query());

        r1.mergeWith(r2);
        assertEquals(0, r1.getCoverage(create).getResultSets());
    }

    @Test
    void testDefaultSearchScenario() {
        boolean create = true;

        Result federationSearcherResult = new Result(new Query());
        Result singleSourceResult = new Result(new Query());
        federationSearcherResult.mergeWith(singleSourceResult);
        assertNull(federationSearcherResult.getCoverage(!create));
        assertEquals(0, federationSearcherResult.getCoverage(create).getResultSets());
    }

    @Test
    void testRequestingCoverageSearchScenario() {
        boolean create = true;

        Result federationSearcherResult = new Result(new Query());
        Result singleSourceResult = new Result(new Query());
        singleSourceResult.setCoverage(new Coverage(10, 1, 1));
        federationSearcherResult.mergeWith(singleSourceResult);
        assertEquals(1, federationSearcherResult.getCoverage(create).getResultSets());
    }

    void verifyCoverageConversion(com.yahoo.container.handler.Coverage c) {
        com.yahoo.container.logging.Coverage lc = c.toLoggingCoverage();
        assertEquals(lc.getDocs(), c.getDocs());
        assertEquals(lc.getActive(), c.getActive());
        assertEquals(lc.getTargetActive(), c.getTargetActive());
        assertEquals(lc.getResultPercentage(), c.getResultPercentage());
        assertEquals(lc.isDegraded(), c.isDegraded());
        assertEquals(lc.isDegradedByNonIdealState(), c.isDegradedByNonIdealState());
        assertEquals(lc.isDegradedByAdapativeTimeout(), c.isDegradedByAdapativeTimeout());
        assertEquals(lc.isDegradedByMatchPhase(), c.isDegradedByMatchPhase());
        assertEquals(lc.isDegradedByTimeout(), c.isDegradedByTimeout());
    }

    @Test
    void testCoverageConversion() {
        verifyCoverageConversion(new Coverage(6, 10, 1).setDegradedReason(7).setTargetActive(12));
    }

    @Test
    void testCoverageWorksForCorpusAboveZero() {
        Coverage zero = new Coverage(0, 0, 1);
        assertEquals(1, zero.getNodes());
        assertEquals(1, zero.getFullResultSets());
        for (int i =1; i < 10; i++) {
            Coverage coverage = new Coverage(i, i, 1);
            assertEquals(1, coverage.getNodes());
            assertEquals(1, coverage.getFullResultSets());
            verifyCoverageConversion(coverage);
        }
    }

    private void verifyNoCoverage(Coverage zero) {
        assertFalse(zero.isDegraded());
        assertEquals(100, zero.getResultPercentage());
        assertTrue(zero.getFull());
        zero.setDegradedReason(com.yahoo.container.handler.Coverage.DEGRADED_BY_TIMEOUT);
        assertTrue(zero.isDegraded());
        assertEquals(0, zero.getResultPercentage());
        assertFalse(zero.getFull());
    }
    @Test
    void testCoverageWithNoResponseFromSearchNodesAndTimeout() {
        verifyNoCoverage(new Coverage(0, 0, 0));
    }
    @Test
    void testCoverageWithResponseFromSearchNodesAndTimeout() {
        verifyNoCoverage(new Coverage(0, 0, 1));
    }

}