diff --git a/catalog-rest-service/src/main/java/org/openmetadata/catalog/jdbi3/TestSuiteRepository.java b/catalog-rest-service/src/main/java/org/openmetadata/catalog/jdbi3/TestSuiteRepository.java index 581e7d388db..4ff6a5bf3d8 100644 --- a/catalog-rest-service/src/main/java/org/openmetadata/catalog/jdbi3/TestSuiteRepository.java +++ b/catalog-rest-service/src/main/java/org/openmetadata/catalog/jdbi3/TestSuiteRepository.java @@ -3,9 +3,11 @@ package org.openmetadata.catalog.jdbi3; import static org.openmetadata.catalog.Entity.TEST_SUITE; import java.io.IOException; +import org.openmetadata.catalog.Entity; import org.openmetadata.catalog.resources.dqtests.TestSuiteResource; import org.openmetadata.catalog.tests.TestSuite; import org.openmetadata.catalog.type.EntityReference; +import org.openmetadata.catalog.type.Relationship; import org.openmetadata.catalog.util.EntityUtil; public class TestSuiteRepository extends EntityRepository { @@ -26,6 +28,7 @@ public class TestSuiteRepository extends EntityRepository { @Override public TestSuite setFields(TestSuite entity, EntityUtil.Fields fields) throws IOException { entity.setOwner(fields.contains("owner") ? getOwner(entity) : null); + entity.setPipeline(fields.contains("pipelines") ? getIngestionPipeline(entity) : null); return entity; } @@ -50,6 +53,10 @@ public class TestSuiteRepository extends EntityRepository { storeOwner(entity, entity.getOwner()); } + private EntityReference getIngestionPipeline(TestSuite testSuite) throws IOException { + return getToEntityRef(testSuite.getId(), Relationship.CONTAINS, Entity.INGESTION_PIPELINE, false); + } + public class TestSuiteUpdater extends EntityUpdater { public TestSuiteUpdater(TestSuite original, TestSuite updated, Operation operation) { super(original, updated, operation); diff --git a/catalog-rest-service/src/main/java/org/openmetadata/catalog/resources/dqtests/TestSuiteResource.java b/catalog-rest-service/src/main/java/org/openmetadata/catalog/resources/dqtests/TestSuiteResource.java index 3b75c65ddf2..ae6df0af973 100644 --- a/catalog-rest-service/src/main/java/org/openmetadata/catalog/resources/dqtests/TestSuiteResource.java +++ b/catalog-rest-service/src/main/java/org/openmetadata/catalog/resources/dqtests/TestSuiteResource.java @@ -337,7 +337,6 @@ public class TestSuiteResource extends EntityResource