| /** |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with this |
| * work for additional information regarding copyright ownership. The ASF |
| * licenses this file to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT |
| * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the |
| * License for the specific language governing permissions and limitations under |
| * the License. |
| */ |
| package org.apache.pig.test; |
| |
| import org.apache.commons.cli.ParseException; |
| import org.apache.commons.logging.Log; |
| import org.apache.commons.logging.LogFactory; |
| import org.apache.hadoop.hbase.client.Scan; |
| import org.apache.pig.backend.hadoop.hbase.HBaseStorage; |
| import org.apache.pig.impl.util.UDFContext; |
| import org.junit.Assert; |
| import org.junit.BeforeClass; |
| import org.junit.Test; |
| |
| import java.io.IOException; |
| import java.lang.reflect.Field; |
| import java.util.Properties; |
| |
| public class TestHBaseStorageParams { |
| |
| private static final Log LOG = LogFactory.getLog(TestHBaseStorageParams.class); |
| |
| @BeforeClass |
| public static void setUp() throws Exception { |
| // This is needed by HBaseStorage |
| UDFContext.getUDFContext().setClientSystemProps(new Properties()); |
| } |
| |
| @Test |
| public void testColumnParsingWithSpaces1() throws IOException, ParseException { |
| HBaseStorage storage = new HBaseStorage("foo:a foo:b foo:c"); |
| doColumnParseTest(storage, "foo:a", "foo:b", "foo:c"); |
| } |
| |
| @Test |
| public void testColumnParsingWithSpaces2() throws IOException, ParseException { |
| HBaseStorage storage = new HBaseStorage("foo:a foo:b foo:c"); |
| doColumnParseTest(storage, "foo:a", "foo:b", "foo:c"); |
| } |
| |
| @Test |
| public void testColumnParsingWithCommas1() throws IOException, ParseException { |
| HBaseStorage storage = new HBaseStorage("foo:a,foo:b,foo:c"); |
| doColumnParseTest(storage, "foo:a", "foo:b", "foo:c"); |
| } |
| |
| @Test |
| public void testColumnParsingWithCommas2() throws IOException, ParseException { |
| HBaseStorage storage = new HBaseStorage("foo:a, foo:b , foo:c"); |
| doColumnParseTest(storage, "foo:a", "foo:b", "foo:c"); |
| } |
| |
| @Test |
| public void testColumnParsingWithDelim() throws IOException, ParseException { |
| HBaseStorage storage = new HBaseStorage("foo:a%foo:b % foo:c,d", "-delim %"); |
| doColumnParseTest(storage, "foo:a", "foo:b", "foo:c,d"); |
| } |
| |
| @Test |
| public void testColumnParsingWithDelimWithSpaces() |
| throws IOException, ParseException { |
| |
| HBaseStorage storage = |
| new HBaseStorage("foo:a%foo:b % foo:c,d", "-delim % -ignoreWhitespace false"); |
| doColumnParseTest(storage, "foo:a", "foo:b ", " foo:c,d"); |
| } |
| |
| /** |
| * Assert that -maxResultsPerColumnFamily actually gets set on Scan |
| */ |
| @Test |
| public void testSetsMaxResultsPerColumnFamily() throws Exception { |
| Field scanField = HBaseStorage.class.getDeclaredField("scan"); |
| scanField.setAccessible(true); |
| |
| HBaseStorage storageNoMax = new HBaseStorage("", ""); |
| Scan scan = (Scan)scanField.get(storageNoMax); |
| Assert.assertEquals(-1, scan.getMaxResultsPerColumnFamily()); |
| |
| HBaseStorage storageWithMax = new HBaseStorage("", "-maxResultsPerColumnFamily 123"); |
| scan = (Scan)scanField.get(storageWithMax); |
| Assert.assertEquals(123, scan.getMaxResultsPerColumnFamily()); |
| } |
| |
| private void doColumnParseTest(HBaseStorage storage, String... names) { |
| Assert.assertEquals("Wrong column count", |
| names.length, storage.getColumnInfoList().size()); |
| |
| for (String name : names) { |
| String[] cfAndName = name.split(":"); |
| HBaseStorage.ColumnInfo col = storage.getColumnInfoList().remove(0); |
| |
| Assert.assertEquals("Wrong CF", |
| cfAndName[0], new String(col.getColumnFamily())); |
| Assert.assertEquals("Wrong column descriptor", |
| cfAndName[1], new String(col.getColumnName())); |
| Assert.assertNull("Column prefix should be null", col.getColumnPrefix()); |
| } |
| } |
| } |