1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20 package org.apache.hadoop.hbase.regionserver;
21
22 import java.io.IOException;
23 import java.util.ArrayList;
24 import java.util.List;
25
26 import org.apache.hadoop.hbase.*;
27 import org.apache.hadoop.hbase.regionserver.ScanQueryMatcher.MatchCode;
28 import org.apache.hadoop.hbase.util.Bytes;
29 import org.junit.experimental.categories.Category;
30
31 @Category(SmallTests.class)
32 public class TestScanWildcardColumnTracker extends HBaseTestCase {
33
34 final static int VERSIONS = 2;
35
36 public void testCheckColumn_Ok() throws IOException {
37 ScanWildcardColumnTracker tracker =
38 new ScanWildcardColumnTracker(0, VERSIONS, Long.MIN_VALUE);
39
40
41 List<byte[]> qualifiers = new ArrayList<byte[]>();
42 qualifiers.add(Bytes.toBytes("qualifer1"));
43 qualifiers.add(Bytes.toBytes("qualifer2"));
44 qualifiers.add(Bytes.toBytes("qualifer3"));
45 qualifiers.add(Bytes.toBytes("qualifer4"));
46
47
48 List<MatchCode> expected = new ArrayList<MatchCode>();
49 expected.add(ScanQueryMatcher.MatchCode.INCLUDE);
50 expected.add(ScanQueryMatcher.MatchCode.INCLUDE);
51 expected.add(ScanQueryMatcher.MatchCode.INCLUDE);
52 expected.add(ScanQueryMatcher.MatchCode.INCLUDE);
53
54 List<ScanQueryMatcher.MatchCode> actual = new ArrayList<MatchCode>();
55
56 for(byte [] qualifier : qualifiers) {
57 ScanQueryMatcher.MatchCode mc =
58 ScanQueryMatcher.checkColumn(tracker, qualifier, 0, qualifier.length, 1,
59 KeyValue.Type.Put.getCode(), false);
60 actual.add(mc);
61 }
62
63
64 for(int i=0; i<expected.size(); i++) {
65 assertEquals(expected.get(i), actual.get(i));
66 }
67 }
68
69 public void testCheckColumn_EnforceVersions() throws IOException {
70 ScanWildcardColumnTracker tracker =
71 new ScanWildcardColumnTracker(0, VERSIONS, Long.MIN_VALUE);
72
73
74 List<byte[]> qualifiers = new ArrayList<byte[]>();
75 qualifiers.add(Bytes.toBytes("qualifer1"));
76 qualifiers.add(Bytes.toBytes("qualifer1"));
77 qualifiers.add(Bytes.toBytes("qualifer1"));
78 qualifiers.add(Bytes.toBytes("qualifer2"));
79
80
81 List<ScanQueryMatcher.MatchCode> expected = new ArrayList<MatchCode>();
82 expected.add(ScanQueryMatcher.MatchCode.INCLUDE);
83 expected.add(ScanQueryMatcher.MatchCode.INCLUDE);
84 expected.add(ScanQueryMatcher.MatchCode.SEEK_NEXT_COL);
85 expected.add(ScanQueryMatcher.MatchCode.INCLUDE);
86
87 List<MatchCode> actual = new ArrayList<ScanQueryMatcher.MatchCode>();
88
89 long timestamp = 0;
90 for(byte [] qualifier : qualifiers) {
91 MatchCode mc =
92 ScanQueryMatcher.checkColumn(tracker, qualifier, 0, qualifier.length, ++timestamp,
93 KeyValue.Type.Put.getCode(), false);
94 actual.add(mc);
95 }
96
97
98 for(int i=0; i<expected.size(); i++) {
99 assertEquals(expected.get(i), actual.get(i));
100 }
101 }
102
103 public void DisabledTestCheckColumn_WrongOrder() {
104 ScanWildcardColumnTracker tracker =
105 new ScanWildcardColumnTracker(0, VERSIONS, Long.MIN_VALUE);
106
107
108 List<byte[]> qualifiers = new ArrayList<byte[]>();
109 qualifiers.add(Bytes.toBytes("qualifer2"));
110 qualifiers.add(Bytes.toBytes("qualifer1"));
111
112 boolean ok = false;
113
114 try {
115 for(byte [] qualifier : qualifiers) {
116 ScanQueryMatcher.checkColumn(tracker, qualifier, 0, qualifier.length, 1,
117 KeyValue.Type.Put.getCode(), false);
118 }
119 } catch (Exception e) {
120 ok = true;
121 }
122
123 assertEquals(true, ok);
124 }
125
126
127 }
128