1   /*
2    * Copyright 2009 The Apache Software Foundation
3    *
4    * Licensed to the Apache Software Foundation (ASF) under one
5    * or more contributor license agreements.  See the NOTICE file
6    * distributed with this work for additional information
7    * regarding copyright ownership.  The ASF licenses this file
8    * to you under the Apache License, Version 2.0 (the
9    * "License"); you may not use this file except in compliance
10   * with the License.  You may obtain a copy of the License at
11   *
12   *     http://www.apache.org/licenses/LICENSE-2.0
13   *
14   * Unless required by applicable law or agreed to in writing, software
15   * distributed under the License is distributed on an "AS IS" BASIS,
16   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
17   * See the License for the specific language governing permissions and
18   * limitations under the License.
19   */
20  
21  package org.apache.hadoop.hbase.regionserver;
22  
23  import java.util.ArrayList;
24  import java.util.List;
25  
26  import org.apache.hadoop.hbase.HBaseTestCase;
27  import org.apache.hadoop.hbase.regionserver.ScanQueryMatcher.MatchCode;
28  import org.apache.hadoop.hbase.util.Bytes;
29  
30  public class TestScanWildcardColumnTracker extends HBaseTestCase {
31  
32    final static int VERSIONS = 2;
33  
34    public void testCheckColumn_Ok() {
35      ScanWildcardColumnTracker tracker =
36        new ScanWildcardColumnTracker(VERSIONS);
37  
38      //Create list of qualifiers
39      List<byte[]> qualifiers = new ArrayList<byte[]>();
40      qualifiers.add(Bytes.toBytes("qualifer1"));
41      qualifiers.add(Bytes.toBytes("qualifer2"));
42      qualifiers.add(Bytes.toBytes("qualifer3"));
43      qualifiers.add(Bytes.toBytes("qualifer4"));
44  
45      //Setting up expected result
46      List<MatchCode> expected = new ArrayList<MatchCode>();
47      expected.add(ScanQueryMatcher.MatchCode.INCLUDE);
48      expected.add(ScanQueryMatcher.MatchCode.INCLUDE);
49      expected.add(ScanQueryMatcher.MatchCode.INCLUDE);
50      expected.add(ScanQueryMatcher.MatchCode.INCLUDE);
51  
52      List<ScanQueryMatcher.MatchCode> actual = new ArrayList<MatchCode>();
53  
54      for(byte [] qualifier : qualifiers) {
55        ScanQueryMatcher.MatchCode mc = tracker.checkColumn(qualifier, 0, qualifier.length);
56        actual.add(mc);
57      }
58  
59      //Compare actual with expected
60      for(int i=0; i<expected.size(); i++) {
61        assertEquals(expected.get(i), actual.get(i));
62      }
63    }
64  
65    public void testCheckColumn_EnforceVersions() {
66      ScanWildcardColumnTracker tracker =
67        new ScanWildcardColumnTracker(VERSIONS);
68  
69      //Create list of qualifiers
70      List<byte[]> qualifiers = new ArrayList<byte[]>();
71      qualifiers.add(Bytes.toBytes("qualifer1"));
72      qualifiers.add(Bytes.toBytes("qualifer1"));
73      qualifiers.add(Bytes.toBytes("qualifer1"));
74      qualifiers.add(Bytes.toBytes("qualifer2"));
75  
76      //Setting up expected result
77      List<ScanQueryMatcher.MatchCode> expected = new ArrayList<MatchCode>();
78      expected.add(ScanQueryMatcher.MatchCode.INCLUDE);
79      expected.add(ScanQueryMatcher.MatchCode.INCLUDE);
80      expected.add(ScanQueryMatcher.MatchCode.SKIP);
81      expected.add(ScanQueryMatcher.MatchCode.INCLUDE);
82  
83      List<MatchCode> actual = new ArrayList<ScanQueryMatcher.MatchCode>();
84  
85      for(byte [] qualifier : qualifiers) {
86        MatchCode mc = tracker.checkColumn(qualifier, 0, qualifier.length);
87        actual.add(mc);
88      }
89  
90      //Compare actual with expected
91      for(int i=0; i<expected.size(); i++) {
92        assertEquals(expected.get(i), actual.get(i));
93      }
94    }
95  
96    public void DisabledTestCheckColumn_WrongOrder() {
97      ScanWildcardColumnTracker tracker =
98        new ScanWildcardColumnTracker(VERSIONS);
99  
100     //Create list of qualifiers
101     List<byte[]> qualifiers = new ArrayList<byte[]>();
102     qualifiers.add(Bytes.toBytes("qualifer2"));
103     qualifiers.add(Bytes.toBytes("qualifer1"));
104 
105     boolean ok = false;
106 
107     try {
108       for(byte [] qualifier : qualifiers) {
109         tracker.checkColumn(qualifier, 0, qualifier.length);
110       }
111     } catch (Exception e) {
112       ok = true;
113     }
114 
115     assertEquals(true, ok);
116   }
117 
118 }