aboutsummaryrefslogtreecommitdiffstats
path: root/container-search/src/test/java/com/yahoo/search/grouping/vespa/CompositeContinuationTestCase.java
blob: ee177709b59a8517757085f3bdabf110c9bcfee9 (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
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.search.grouping.vespa;

import com.yahoo.search.grouping.Continuation;
import org.junit.jupiter.api.Test;

import java.util.Iterator;

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

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

    @Test
    void requireThatAccessorsWork() {
        CompositeContinuation cnt = new CompositeContinuation();
        Iterator<EncodableContinuation> it = cnt.iterator();
        assertFalse(it.hasNext());

        EncodableContinuation foo = new MyContinuation();
        cnt.add(foo);
        it = cnt.iterator();
        assertTrue(it.hasNext());
        assertSame(foo, it.next());
        assertFalse(it.hasNext());

        EncodableContinuation bar = new MyContinuation();
        cnt.add(bar);
        it = cnt.iterator();
        assertTrue(it.hasNext());
        assertSame(foo, it.next());
        assertTrue(it.hasNext());
        assertSame(bar, it.next());
        assertFalse(it.hasNext());
    }

    @Test
    void requireThatCompositeContinuationsAreFlattened() {
        assertEncode("BCBCBCBEBGBCBKCBACBKCCK",
                newComposite(newOffset(1, 1, 2, 3), newOffset(5, 8, 13, 21)));
        assertEncode("BCBBBBBDBFBCBJBPCBJCCJ",
                newComposite(newComposite(newOffset(-1, -1, -2, -3)), newComposite(newOffset(-5, -8, -13, -21))));
    }

    @Test
    void requireThatEmptyStringCanBeDecoded() {
        assertDecode("", new CompositeContinuation());
    }

    @Test
    void requireThatCompositeContinuationsCanBeDecoded() {
        assertDecode("BCBCBCBEBGBCBKCBACBKCCK",
                newComposite(newOffset(1, 1, 2, 3), newOffset(5, 8, 13, 21)));
        assertDecode("BCBBBBBDBFBCBJBPCBJCCJ",
                newComposite(newOffset(-1, -1, -2, -3), newOffset(-5, -8, -13, -21)));
    }

    @Test
    void requireThatHashCodeIsImplemented() {
        assertEquals(newComposite().hashCode(), newComposite().hashCode());
    }

    @Test
    void requireThatEqualsIsImplemented() {
        CompositeContinuation cnt = newComposite();
        assertNotEquals(cnt, new Object());
        assertEquals(cnt, newComposite());
        assertNotEquals(cnt, newComposite(newOffset(1, 1, 2, 3)));
        assertNotEquals(cnt, newComposite(newOffset(1, 1, 2, 3), newOffset(5, 8, 13, 21)));
        assertNotEquals(cnt, newComposite(newOffset(5, 8, 13, 21)));

        cnt = newComposite(newOffset(1, 1, 2, 3));
        assertNotEquals(cnt, new Object());
        assertEquals(cnt, newComposite(newOffset(1, 1, 2, 3)));
        assertNotEquals(cnt, newComposite(newOffset(1, 1, 2, 3), newOffset(5, 8, 13, 21)));
        assertNotEquals(cnt, newComposite(newOffset(5, 8, 13, 21)));

        cnt = newComposite(newOffset(1, 1, 2, 3), newOffset(5, 8, 13, 21));
        assertNotEquals(cnt, new Object());
        assertNotEquals(cnt, newComposite(newOffset(1, 1, 2, 3)));
        assertEquals(cnt, newComposite(newOffset(1, 1, 2, 3), newOffset(5, 8, 13, 21)));
        assertNotEquals(cnt, newComposite(newOffset(5, 8, 13, 21)));
    }

    private static CompositeContinuation newComposite(EncodableContinuation... children) {
        CompositeContinuation ret = new CompositeContinuation();
        for (EncodableContinuation child : children) {
            ret.add(child);
        }
        return ret;
    }

    private static OffsetContinuation newOffset(int resultId, int tag, int offset, int flags) {
        return new OffsetContinuation(ResultId.valueOf(resultId), tag, offset, flags);
    }

    private static void assertEncode(String expected, EncodableContinuation toEncode) {
        IntegerEncoder actual = new IntegerEncoder();
        toEncode.encode(actual);
        assertEquals(expected, actual.toString());
    }

    private static void assertDecode(String toDecode, Continuation expected) {
        assertEquals(expected, ContinuationDecoder.decode(toDecode));
    }

    private static class MyContinuation extends EncodableContinuation {

        @Override
        public EncodableContinuation copy() {
            return null;
        }

        @Override
        public void encode(IntegerEncoder out) {

        }
    }
}