Add get tableProfilerConfig test (#9324)

This commit is contained in:
Sriharsha Chintalapani 2023-06-16 03:32:38 -07:00 committed by GitHub
parent 41c49b9de8
commit 817c5f3d4a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1055,11 +1055,11 @@ public class TableResourceTest extends EntityResourceTest<Table, CreateTable> {
.withIncludeColumns(columnProfilerConfigs); .withIncludeColumns(columnProfilerConfigs);
table = putTableProfilerConfig(table.getId(), tableProfilerConfig, authHeaders); table = putTableProfilerConfig(table.getId(), tableProfilerConfig, authHeaders);
assertEquals(tableProfilerConfig, table.getTableProfilerConfig()); assertEquals(tableProfilerConfig, table.getTableProfilerConfig());
storedTable = getEntity(table.getId(), "tableProfilerConfig", authHeaders); Table tableWithConfig = getTableProfileConfig(table.getId(), authHeaders);
assertEquals(tableProfilerConfig, storedTable.getTableProfilerConfig()); assertEquals(tableProfilerConfig, tableWithConfig.getTableProfilerConfig());
table = deleteTableProfilerConfig(table.getId(), authHeaders); table = deleteTableProfilerConfig(table.getId(), authHeaders);
assertNull(table.getTableProfilerConfig()); assertNull(table.getTableProfilerConfig());
storedTable = getEntity(table.getId(), "tableProfilerConfig", authHeaders); storedTable = getTableProfileConfig(table.getId(), authHeaders);
assertNull(storedTable.getTableProfilerConfig()); assertNull(storedTable.getTableProfilerConfig());
tableProfilerConfig = new TableProfilerConfig().withProfileSample(80.0); tableProfilerConfig = new TableProfilerConfig().withProfileSample(80.0);
table = putTableProfilerConfig(table.getId(), tableProfilerConfig, authHeaders); table = putTableProfilerConfig(table.getId(), tableProfilerConfig, authHeaders);
@ -2026,6 +2026,11 @@ public class TableResourceTest extends EntityResourceTest<Table, CreateTable> {
return TestUtils.put(target, data, Table.class, OK, authHeaders); return TestUtils.put(target, data, Table.class, OK, authHeaders);
} }
public Table getTableProfileConfig(UUID tableId, Map<String, String> authHeaders) throws HttpResponseException {
WebTarget target = getResource(tableId).path("/tableProfilerConfig");
return TestUtils.get(target, Table.class, authHeaders);
}
public Table deleteTableProfilerConfig(UUID tableId, Map<String, String> authHeaders) throws HttpResponseException { public Table deleteTableProfilerConfig(UUID tableId, Map<String, String> authHeaders) throws HttpResponseException {
WebTarget target = getResource(tableId).path("/tableProfilerConfig"); WebTarget target = getResource(tableId).path("/tableProfilerConfig");
return TestUtils.delete(target, Table.class, authHeaders); return TestUtils.delete(target, Table.class, authHeaders);