use of org.apache.hadoop.hive.metastore.api.SQLForeignKey in project hive by apache.
the class NonCatCallsWithCatalog method primaryKeyAndForeignKey.
@Test
public void primaryKeyAndForeignKey() throws TException {
Table parentTable = testTables[2];
Table table = testTables[3];
String constraintName = "othercatfk";
// Single column unnamed primary key in default catalog and database
List<SQLPrimaryKey> pk = new SQLPrimaryKeyBuilder().onTable(parentTable).addColumn("test_col1").build(conf);
client.addPrimaryKey(pk);
List<SQLForeignKey> fk = new SQLForeignKeyBuilder().fromPrimaryKey(pk).onTable(table).addColumn("test_col1").setConstraintName(constraintName).build(conf);
client.addForeignKey(fk);
PrimaryKeysRequest pkRqst = new PrimaryKeysRequest(parentTable.getDbName(), parentTable.getTableName());
pkRqst.setCatName(parentTable.getCatName());
List<SQLPrimaryKey> pkFetched = client.getPrimaryKeys(pkRqst);
Assert.assertEquals(1, pkFetched.size());
Assert.assertEquals(expectedCatalog(), pkFetched.get(0).getCatName());
Assert.assertEquals(parentTable.getDbName(), pkFetched.get(0).getTable_db());
Assert.assertEquals(parentTable.getTableName(), pkFetched.get(0).getTable_name());
Assert.assertEquals("test_col1", pkFetched.get(0).getColumn_name());
Assert.assertEquals(1, pkFetched.get(0).getKey_seq());
Assert.assertTrue(pkFetched.get(0).isEnable_cstr());
Assert.assertFalse(pkFetched.get(0).isValidate_cstr());
Assert.assertFalse(pkFetched.get(0).isRely_cstr());
Assert.assertEquals(parentTable.getCatName(), pkFetched.get(0).getCatName());
ForeignKeysRequest rqst = new ForeignKeysRequest(parentTable.getDbName(), parentTable.getTableName(), table.getDbName(), table.getTableName());
rqst.setCatName(table.getCatName());
List<SQLForeignKey> fetched = client.getForeignKeys(rqst);
Assert.assertEquals(1, fetched.size());
Assert.assertEquals(table.getDbName(), fetched.get(0).getFktable_db());
Assert.assertEquals(table.getTableName(), fetched.get(0).getFktable_name());
Assert.assertEquals(expectedCatalog(), fetched.get(0).getCatName());
Assert.assertEquals("test_col1", fetched.get(0).getFkcolumn_name());
Assert.assertEquals(parentTable.getDbName(), fetched.get(0).getPktable_db());
Assert.assertEquals(parentTable.getTableName(), fetched.get(0).getPktable_name());
Assert.assertEquals("test_col1", fetched.get(0).getFkcolumn_name());
Assert.assertEquals(1, fetched.get(0).getKey_seq());
Assert.assertEquals(parentTable.getTableName() + "_primary_key", fetched.get(0).getPk_name());
Assert.assertEquals(constraintName, fetched.get(0).getFk_name());
String table0FkName = fetched.get(0).getFk_name();
Assert.assertTrue(fetched.get(0).isEnable_cstr());
Assert.assertFalse(fetched.get(0).isValidate_cstr());
Assert.assertFalse(fetched.get(0).isRely_cstr());
Assert.assertEquals(table.getCatName(), fetched.get(0).getCatName());
// Drop a foreign key
client.dropConstraint(table.getDbName(), table.getTableName(), table0FkName);
rqst = new ForeignKeysRequest(parentTable.getDbName(), parentTable.getTableName(), table.getDbName(), table.getTableName());
rqst.setCatName(table.getCatName());
fetched = client.getForeignKeys(rqst);
Assert.assertTrue(fetched.isEmpty());
}
use of org.apache.hadoop.hive.metastore.api.SQLForeignKey in project hive by apache.
the class TestForeignKey method createTableWithConstraintsInOtherCatalog.
@Test
public void createTableWithConstraintsInOtherCatalog() throws TException {
String constraintName = "ctwcocfk";
Table parentTable = testTables[2];
Table table = new TableBuilder().setTableName("table_with_constraints").inDb(inOtherCatalog).addCol("col1", "int").addCol("col2", "varchar(32)").build(metaStore.getConf());
List<SQLPrimaryKey> pk = new SQLPrimaryKeyBuilder().onTable(parentTable).addColumn("col1").build(metaStore.getConf());
client.addPrimaryKey(pk);
List<SQLForeignKey> fk = new SQLForeignKeyBuilder().fromPrimaryKey(pk).onTable(table).addColumn("col1").setConstraintName(constraintName).build(metaStore.getConf());
client.createTableWithConstraints(table, null, fk, null, null, null, null);
ForeignKeysRequest rqst = new ForeignKeysRequest(parentTable.getDbName(), parentTable.getTableName(), table.getDbName(), table.getTableName());
rqst.setCatName(table.getCatName());
List<SQLForeignKey> fetched = client.getForeignKeys(rqst);
Assert.assertEquals(fk, fetched);
}
use of org.apache.hadoop.hive.metastore.api.SQLForeignKey in project hive by apache.
the class TestForeignKey method addNoSuchDb.
@Test
public void addNoSuchDb() throws TException {
Table parentTable = testTables[0];
List<SQLPrimaryKey> pk = new SQLPrimaryKeyBuilder().onTable(parentTable).addColumn("col1").build(metaStore.getConf());
client.addPrimaryKey(pk);
try {
List<SQLForeignKey> fk = new SQLForeignKeyBuilder().setTableName(testTables[0].getTableName()).setDbName("nosuch").fromPrimaryKey(pk).addColumn("col2").build(metaStore.getConf());
client.addForeignKey(fk);
Assert.fail();
} catch (InvalidObjectException | TApplicationException e) {
// NOP
}
}
use of org.apache.hadoop.hive.metastore.api.SQLForeignKey in project hive by apache.
the class TestForeignKey method addNoSuchTable.
@Test
public void addNoSuchTable() throws TException {
Table parentTable = testTables[0];
List<SQLPrimaryKey> pk = new SQLPrimaryKeyBuilder().onTable(parentTable).addColumn("col1").build(metaStore.getConf());
client.addPrimaryKey(pk);
try {
List<SQLForeignKey> fk = new SQLForeignKeyBuilder().setTableName("nosuch").fromPrimaryKey(pk).addColumn("col2").build(metaStore.getConf());
client.addForeignKey(fk);
Assert.fail();
} catch (InvalidObjectException | TApplicationException e) {
// NOP
}
}
use of org.apache.hadoop.hive.metastore.api.SQLForeignKey in project hive by apache.
the class TestGetAllTableConstraints method allConstraintsPresent.
/**
* Test to verify all constraint are present in table
* @throws TException
*/
@Test
public void allConstraintsPresent() throws TException {
Table table = testTables[0];
Table parentTable = testTables[1];
SQLAllTableConstraints expected = new SQLAllTableConstraints();
// Set col1 as primary key Constraint in default catalog and database
List<SQLPrimaryKey> pk = new SQLPrimaryKeyBuilder().onTable(table).addColumn("col1").setConstraintName("col1_pk").build(metaStore.getConf());
client.addPrimaryKey(pk);
expected.setPrimaryKeys(pk);
// Set col2 with Unique Constraint in default catalog and database
String uniqueConstraintName = "col2_unique";
List<SQLUniqueConstraint> uc = new SQLUniqueConstraintBuilder().onTable(table).addColumn("col2").setConstraintName("col2_unique").build(metaStore.getConf());
client.addUniqueConstraint(uc);
expected.setUniqueConstraints(uc);
// Set col3 with default Constraint in default catalog and database
List<SQLDefaultConstraint> dv = new SQLDefaultConstraintBuilder().onTable(table).addColumn("col3").setConstraintName("col3_default").setDefaultVal(false).build(metaStore.getConf());
client.addDefaultConstraint(dv);
expected.setDefaultConstraints(dv);
// Set col3 with not null constraint in default catalog and database;
List<SQLNotNullConstraint> nn = new SQLNotNullConstraintBuilder().onTable(table).addColumn("col3").setConstraintName("col3_not_null").build(metaStore.getConf());
client.addNotNullConstraint(nn);
expected.setNotNullConstraints(nn);
// Set col2 with not check constraint in default catalog and database;
List<SQLCheckConstraint> cc = new SQLCheckConstraintBuilder().onTable(table).addColumn("col2").setConstraintName("col2_check").setCheckExpression("= 5").build(metaStore.getConf());
client.addCheckConstraint(cc);
expected.setCheckConstraints(cc);
// Set col1 of parent table to PK and Set Col4 of table to FK
List<SQLPrimaryKey> parentPk = new SQLPrimaryKeyBuilder().onTable(parentTable).addColumn("col1").setConstraintName("parentpk").build(metaStore.getConf());
client.addPrimaryKey(parentPk);
String fkConstraintName = "fk";
List<SQLForeignKey> fk = new SQLForeignKeyBuilder().fromPrimaryKey(parentPk).onTable(table).setConstraintName(fkConstraintName).addColumn("col4").build(metaStore.getConf());
client.addForeignKey(fk);
expected.setForeignKeys(fk);
// Fetch all constraints for the table in default catalog and database
AllTableConstraintsRequest request = new AllTableConstraintsRequest(table.getDbName(), table.getTableName(), table.getCatName());
SQLAllTableConstraints actual = client.getAllTableConstraints(request);
Assert.assertEquals(expected, actual);
}
Aggregations