From 817c5f3d4a8fffe0c3af172b1feaf41d6de62a9f Mon Sep 17 00:00:00 2001 From: Sriharsha Chintalapani Date: Fri, 16 Jun 2023 03:32:38 -0700 Subject: [PATCH] Add get tableProfilerConfig test (#9324) --- .../resources/databases/TableResourceTest.java | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/openmetadata-service/src/test/java/org/openmetadata/service/resources/databases/TableResourceTest.java b/openmetadata-service/src/test/java/org/openmetadata/service/resources/databases/TableResourceTest.java index 2124b899bc7..e91daaf615b 100644 --- a/openmetadata-service/src/test/java/org/openmetadata/service/resources/databases/TableResourceTest.java +++ b/openmetadata-service/src/test/java/org/openmetadata/service/resources/databases/TableResourceTest.java @@ -1055,11 +1055,11 @@ public class TableResourceTest extends EntityResourceTest { .withIncludeColumns(columnProfilerConfigs); table = putTableProfilerConfig(table.getId(), tableProfilerConfig, authHeaders); assertEquals(tableProfilerConfig, table.getTableProfilerConfig()); - storedTable = getEntity(table.getId(), "tableProfilerConfig", authHeaders); - assertEquals(tableProfilerConfig, storedTable.getTableProfilerConfig()); + Table tableWithConfig = getTableProfileConfig(table.getId(), authHeaders); + assertEquals(tableProfilerConfig, tableWithConfig.getTableProfilerConfig()); table = deleteTableProfilerConfig(table.getId(), authHeaders); assertNull(table.getTableProfilerConfig()); - storedTable = getEntity(table.getId(), "tableProfilerConfig", authHeaders); + storedTable = getTableProfileConfig(table.getId(), authHeaders); assertNull(storedTable.getTableProfilerConfig()); tableProfilerConfig = new TableProfilerConfig().withProfileSample(80.0); table = putTableProfilerConfig(table.getId(), tableProfilerConfig, authHeaders); @@ -2026,6 +2026,11 @@ public class TableResourceTest extends EntityResourceTest { return TestUtils.put(target, data, Table.class, OK, authHeaders); } + public Table getTableProfileConfig(UUID tableId, Map authHeaders) throws HttpResponseException { + WebTarget target = getResource(tableId).path("/tableProfilerConfig"); + return TestUtils.get(target, Table.class, authHeaders); + } + public Table deleteTableProfilerConfig(UUID tableId, Map authHeaders) throws HttpResponseException { WebTarget target = getResource(tableId).path("/tableProfilerConfig"); return TestUtils.delete(target, Table.class, authHeaders);