/**
* (c) Copyright 2012 WibiData, Inc.
*
* See the NOTICE file distributed with this work for additional
* information regarding copyright ownership.
*
* Licensed 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.kiji.schema;
import static org.junit.Assert.assertArrayEquals;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;
import java.io.IOException;
import java.util.Collections;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;
import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.hbase.HColumnDescriptor;
import org.apache.hadoop.hbase.HTableDescriptor;
import org.apache.hadoop.hbase.client.HBaseAdmin;
import org.apache.hadoop.hbase.client.HTableInterface;
import org.apache.hadoop.hbase.util.Bytes;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.kiji.schema.hbase.HBaseFactory;
import org.kiji.schema.impl.hbase.HBaseTableKeyValueDatabase;
/**
* This class tests that HBaseTableLayoutDatabase is correctly writing and reading from HBase
* when performing its operations.
*/
public class TestHBaseTableKeyValueDatabase extends KijiClientTest {
private static final String TABLE_NAME = "metaTable";
private static final Map<String, Map<String, byte[]>> TABLE_KV_MAP = createTableMap();
private static final String FAMILY_NAME = "meta";
private static Map<String, Map<String, byte[]>> createTableMap() {
Map<String, byte[]> innerMap1 = new HashMap<String, byte[]>();
innerMap1.put("config1", Bytes.toBytes("1one"));
innerMap1.put("config2", Bytes.toBytes("1two"));
Map<String, Map<String, byte[]>> result = new HashMap<String, Map<String, byte[]>>();
result.put("table1", innerMap1);
Map<String, byte[]> innerMap2 = new HashMap<String, byte[]>();
innerMap2.put("config1", Bytes.toBytes("2one"));
innerMap2.put("config2", Bytes.toBytes("2two"));
result.put("table2", innerMap2);
return Collections.unmodifiableMap(result);
}
private Configuration mConf;
private HBaseAdmin mHBaseAdmin;
private HTableInterface mTable;
private HBaseTableKeyValueDatabase mDb;
@Before
public final void setupHBaseTable() throws IOException {
final HBaseFactory factory = HBaseFactory.Provider.get();
mConf = super.getConf();
mHBaseAdmin = factory.getHBaseAdminFactory(getKiji().getURI()).create(mConf);
// Create an HBase table.
HTableDescriptor tableDescriptor = new HTableDescriptor(TABLE_NAME);
tableDescriptor.addFamily(new HColumnDescriptor(FAMILY_NAME));
mHBaseAdmin.createTable(tableDescriptor);
mTable = factory.getHTableInterfaceFactory(getKiji().getURI()).create(mConf, TABLE_NAME);
// Fill it with some data.
mDb = new HBaseTableKeyValueDatabase(mTable, FAMILY_NAME);
mDb.putValue("table1", "config1", Bytes.toBytes("1one"));
mDb.putValue("table1", "config2", Bytes.toBytes("1two"));
mDb.putValue("table2", "config1", Bytes.toBytes("2one"));
mDb.putValue("table2", "config2", Bytes.toBytes("2two"));
}
@After
public final void teardownHBaseTable() throws IOException {
mTable.close();
mHBaseAdmin.disableTable(TABLE_NAME);
mHBaseAdmin.deleteTable(TABLE_NAME);
mHBaseAdmin.close();
}
@Test
public void testGet() throws IOException {
byte[] result = mDb.getValue("table1", "config1");
assertArrayEquals(Bytes.toBytes("1one"), result);
}
@Test
public void testTableSet() throws IOException {
Set<String> tableNames = mDb.tableSet();
assertEquals(TABLE_KV_MAP.keySet(), tableNames);
}
@Test
public void testKeySet() throws IOException {
Set<String> keys = mDb.keySet("table1");
assertEquals(TABLE_KV_MAP.get("table1").keySet(), keys);
}
@Test
public void testRemoveValues() throws IOException {
mDb.removeValues("table1", "config1");
assertEquals(1, mDb.keySet("table1").size());
assertTrue("The key set should still contain config2", mDb.keySet("table1")
.contains("config2"));
}
@Test
public void testRemoveAllValues() throws IOException {
mDb.removeAllValues("table1");
try {
mDb.getValue("table1", "config1");
fail("An exception should have been thrown.");
} catch (IOException ioe) {
assertEquals("Could not find any values associated with table table1 and key config1",
ioe.getMessage());
}
}
}