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
|
// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.document.predicate;
import com.yahoo.slime.Inspector;
import com.yahoo.slime.Slime;
import org.junit.Test;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.fail;
/**
* @author <a href="mailto:simon@yahoo-inc.com">Simon Thoresen Hult</a>
*/
public class BinaryFormatTest {
@Test
public void requireThatEncodeNullThrows() {
try {
BinaryFormat.encode(null);
fail();
} catch (NullPointerException e) {
assertEquals("predicate", e.getMessage());
}
}
@Test
public void requireThatDecodeNullThrows() {
try {
BinaryFormat.decode(null);
fail();
} catch (NullPointerException e) {
assertEquals("buf", e.getMessage());
}
}
@Test
public void requireThatDecodeEmptyThrows() {
try {
BinaryFormat.decode(new byte[0]);
fail();
} catch (UnsupportedOperationException e) {
assertEquals("0", e.getMessage());
}
}
@Test
public void requireThatConjunctionCanBeSerialized() {
assertSerialize(new Conjunction(new FeatureSet("foo", "bar"), new FeatureSet("baz", "cox")));
}
@Test
public void requireThatDisjunctionCanBeSerialized() {
assertSerialize(new Disjunction(new FeatureSet("foo", "bar"), new FeatureSet("baz", "cox")));
}
@Test
public void requireThatFeatureRangeCanBeSerialized() {
assertSerialize(new FeatureRange("foo", null, null));
assertSerialize(new FeatureRange("foo", null, 9L));
assertSerialize(new FeatureRange("foo", 6L, null));
assertSerialize(new FeatureRange("foo", 6L, 9L));
}
@Test
public void requireThatPartitionedFeatureRangeCanBeSerialized() {
FeatureRange expected = new FeatureRange("foo", 8L, 20L);
FeatureRange f = new FeatureRange("foo", 8L, 20L);
f.addPartition(new RangeEdgePartition("foo=0", 0, 8, -1));
f.addPartition(new RangeEdgePartition("foo=20", 20, 0, 0));
f.addPartition(new RangePartition("foo", 10, 19, false));
assertSerializesTo(expected, f);
Slime slime = com.yahoo.slime.BinaryFormat.decode(BinaryFormat.encode(f));
assertEquals(BinaryFormat.TYPE_FEATURE_RANGE, slime.get().field(BinaryFormat.NODE_TYPE).asLong());
Inspector in1 = slime.get().field(BinaryFormat.HASHED_PARTITIONS);
assertEquals(1, in1.entries());
assertEquals(0xf2b6d1cc6322cb99L, in1.entry(0).asLong());
Inspector in2 = slime.get().field(BinaryFormat.HASHED_EDGE_PARTITIONS);
assertEquals(2, in2.entries());
Inspector obj1 = in2.entry(0);
assertEquals(0xb2b301e26efffdc2L, obj1.field(BinaryFormat.HASH).asLong());
assertEquals(0, obj1.field(BinaryFormat.VALUE).asLong());
assertEquals(0x80000008L, obj1.field(BinaryFormat.PAYLOAD).asLong());
Inspector obj2 = in2.entry(1);
assertEquals(0x22acb2ed72523c36L, obj2.field(BinaryFormat.HASH).asLong());
assertEquals(20, obj2.field(BinaryFormat.VALUE).asLong());
assertEquals(0x40000001L, obj2.field(BinaryFormat.PAYLOAD).asLong());
}
@Test
public void requireThatFeatureSetCanBeSerialized() {
assertSerialize(new FeatureSet("foo"));
assertSerialize(new FeatureSet("foo", "bar"));
assertSerialize(new FeatureSet("foo", "bar", "baz"));
}
@Test
public void requireThatNegationCanBeSerialized() {
assertSerialize(new Negation(new FeatureSet("foo", "bar")));
}
@Test
public void requireThatBooleanCanBeSerialized() {
assertSerialize(new BooleanPredicate(true));
assertSerialize(new BooleanPredicate(false));
}
@Test
public void requireThatUnknownNodeThrows() {
try {
BinaryFormat.encode(SimplePredicates.newString("foo"));
fail();
} catch (UnsupportedOperationException e) {
}
}
private static void assertSerializesTo(Predicate expected, Predicate predicate) {
assertEquals(expected, BinaryFormat.decode(BinaryFormat.encode(predicate)));
}
private static void assertSerialize(Predicate predicate) {
assertSerializesTo(predicate, predicate);
}
}
|