use of org.apache.hadoop.hive.metastore.api.SQLDefaultConstraint in project hive by apache.
the class TestDefaultConstraint method createGetDrop.
@Test
public void createGetDrop() throws TException {
Table table = testTables[0];
// Make sure get on a table with no key returns empty list
DefaultConstraintsRequest rqst = new DefaultConstraintsRequest(table.getCatName(), table.getDbName(), table.getTableName());
List<SQLDefaultConstraint> fetched = client.getDefaultConstraints(rqst);
Assert.assertTrue(fetched.isEmpty());
// Single column unnamed primary key in default catalog and database
List<SQLDefaultConstraint> dv = new SQLDefaultConstraintBuilder().onTable(table).addColumn("col1").setDefaultVal(0).build(metaStore.getConf());
client.addDefaultConstraint(dv);
rqst = new DefaultConstraintsRequest(table.getCatName(), table.getDbName(), table.getTableName());
fetched = client.getDefaultConstraints(rqst);
dv.get(0).setDc_name(fetched.get(0).getDc_name());
Assert.assertEquals(dv, fetched);
// Drop a primary key
client.dropConstraint(table.getCatName(), table.getDbName(), table.getTableName(), dv.get(0).getDc_name());
rqst = new DefaultConstraintsRequest(table.getCatName(), table.getDbName(), table.getTableName());
fetched = client.getDefaultConstraints(rqst);
Assert.assertTrue(fetched.isEmpty());
// Make sure I can add it back
client.addDefaultConstraint(dv);
}
Aggregations