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
|
// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.document.select;
import com.yahoo.document.BucketId;
import com.yahoo.document.BucketIdFactory;
import com.yahoo.document.select.BucketSelector;
import java.util.Set;
import java.util.TreeSet;
/**
* Date: Sep 6, 2007
*
* @author <a href="mailto:humbe@yahoo-inc.com">Håkon Humberset</a>
*/
public class BucketSelectorTestCase extends junit.framework.TestCase {
public BucketSelectorTestCase(String name) {
super(name);
}
public void testExpressions() throws Exception {
assertBucketCount("id = \"userdoc:ns:123:foobar\"", 1);
assertBucketCount("id = \"userdoc:ns:123:foo*\"", 0);
assertBucketCount("id == \"userdoc:ns:123:f?oo*\"", 1);
assertBucketCount("id =~ \"userdoc:ns:123:foo*\"", 0);
assertBucketCount("id =~ \"userdoc:ns:123:foo?\"", 0);
assertBucketCount("id.user = 123", 1);
assertBucketCount("id.user == 123", 1);
assertBucketCount("id.group = \"yahoo.com\"", 1);
assertBucketCount("id.group = \"yahoo.com\" or id.user=123", 2);
assertBucketCount("id.group = \"yahoo.com\" and id.user=123", 0);
assertBucketCount("id.group = \"yahoo.com\" and testdoctype1.hstringval=\"Doe\"", 1);
assertBucketCount("not id.group = \"yahoo.com\"", 0);
assertBucketCount("id.group != \"yahoo.com\"", 0);
assertBucketCount("id.group <= \"yahoo.com\"", 0);
assertBucketCount("id.bucket = 0x4000000000003018", 1); // Bucket 16:12312
assertBucketCount("id.bucket == 0x4000000000000258", 1); // Bucket 16:600
assertBucketCount("searchcolumn.3 == 1", 21845);
// Check that the correct buckets is found
assertBucket("id.bucket = 0x4000000000003018", new BucketId(16, 12312));
assertBucket("id.bucket == 0x4000000000000258", new BucketId(16, 600));
assertBucket("id = \"userdoc:ns:123:foobar\"", new BucketId(0xeafff5320000007bl));
assertBucket("id.user = 123", new BucketId(32, 123));
assertBucket("id.group = \"yahoo.com\"", new BucketId(32, 0x035837189a1acd50l));
// Check that overlapping works
Set<BucketId> expected = new TreeSet<BucketId>();
expected.add(new BucketId(32, 123));
expected.add(new BucketId(16, 123));
assertBuckets("id.user == 123 or id.bucket == 0x400000000000007b", expected);
}
public void assertBucketCount(String expr, int count) throws Exception {
BucketIdFactory factory = new BucketIdFactory();
BucketSelector selector = new BucketSelector(factory);
Set<BucketId> buckets = selector.getBucketList(expr);
assertEquals(count, buckets == null ? 0 : buckets.size());
}
public void assertBucket(String expr, BucketId bucket) throws Exception {
BucketIdFactory factory = new BucketIdFactory();
BucketSelector selector = new BucketSelector(factory);
Set<BucketId> buckets = selector.getBucketList(expr);
assertEquals(1, buckets == null ? 0 : buckets.size());
assertEquals(bucket, buckets.toArray()[0]);
}
public void assertBuckets(String expr, Set<BucketId> expected) throws Exception {
BucketIdFactory factory = new BucketIdFactory();
BucketSelector selector = new BucketSelector(factory);
Set<BucketId> actual = new TreeSet<BucketId>(selector.getBucketList(expr));
assertEquals(expected, actual);
}
}
|
