GEN 1507 - Implement Origin Entity FQN parms to incident manager listing

openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/CollectionDAO.java

@@ -4408,10 +4408,13 @@ default List<String> listWithOffset(
  outerFilter.addQueryParam("testCaseResolutionStatusType", testCaseResolutionStatusType);
  outerFilter.addQueryParam("assignee", assignee);
 
+ String condition = filter.getCondition();
+ condition = TestCaseResolutionStatusRepository.addOriginEntityFQNJoin(filter, condition);
+
  return listWithOffset(
  getTimeSeriesTableName(),
  filter.getQueryParams(),
- filter.getCondition(),
+ condition,
  getPartitionFieldName(),
  limit,
  offset,
@@ -4420,15 +4423,32 @@ default List<String> listWithOffset(
  filter.getQueryParams(),
  outerFilter.getCondition());
  }
+ String condition = filter.getCondition();
+ condition = TestCaseResolutionStatusRepository.addOriginEntityFQNJoin(filter, condition);
  return listWithOffset(
  getTimeSeriesTableName(),
  filter.getQueryParams(),
- filter.getCondition(),
+ condition,
  limit,
  offset,
  startTs,
  endTs);
  }
+
+ @Override
+ default int listCount(ListFilter filter, Long startTs, Long endTs, boolean latest) {
+ String condition = filter.getCondition();
+ condition = TestCaseResolutionStatusRepository.addOriginEntityFQNJoin(filter, condition);
+ return latest
+ ? listCount(
+ getTimeSeriesTableName(),
+ getPartitionFieldName(),
+ filter.getQueryParams(),
+ condition,
+ startTs,
+ endTs)
+ : listCount(getTimeSeriesTableName(), filter.getQueryParams(), condition, startTs, endTs);
+ }
  }
 
  class EntitiesCountRowMapper implements RowMapper<EntitiesCount> {

openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/ListFilter.java

@@ -36,6 +36,7 @@ public String getCondition(String tableName) {
  conditions.add(getWebhookCondition(tableName));
  conditions.add(getWebhookTypeCondition(tableName));
  conditions.add(getTestCaseCondition());
+ conditions.add(getTestCaseIncidentCondition());
  conditions.add(getTestSuiteTypeCondition(tableName));
  conditions.add(getTestSuiteFQNCondition());
  conditions.add(getDomainCondition(tableName));
@@ -231,6 +232,18 @@ private String getTestCaseCondition() {
  return addCondition(conditions);
  }
 
+ private String getTestCaseIncidentCondition() {
+ String originEntityFQN = getQueryParam("originEntityFQN");
+ if (originEntityFQN != null) {
+ queryParams.put(
+ "originEntityFQNLike",
+ originEntityFQN + ".%"); // Add wildcard to get all column test cases under the entity
+ return "(testCaseEntityFQN = :originEntityFQN\n"
+ + " OR testCaseEntityFQN LIKE :originEntityFQNLike)";
+ }
+ return "";
+ }
+
  private String getTestSuiteTypeCondition(String tableName) {
  String testSuiteType = getQueryParam("testSuiteType");
 

openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/TestCaseResolutionStatusRepository.java

@@ -375,4 +375,17 @@ public void deleteTestCaseFailedSamples(TestCaseResolutionStatus entity) {
  (TestCaseRepository) Entity.getEntityRepository(Entity.TEST_CASE);
  testCaseRepository.deleteTestCaseFailedRowsSample(entity.getTestCaseReference().getId());
  }
+
+ public static String addOriginEntityFQNJoin(ListFilter filter, String condition) {
+ // if originEntityFQN is present, we need to join with test_case table
+ if (filter.getQueryParam("originEntityFQN") != null) {
+ condition =
+ """
+ INNER JOIN (SELECT entityFQN AS testCaseEntityFQN,fqnHash AS testCaseHash FROM test_case) \
+ ON entityFQNHash = testCaseHash
+ """
+ + condition;
+ }
+ return condition;
+ }
 }

openmetadata-service/src/main/java/org/openmetadata/service/resources/dqtests/TestCaseResolutionStatusResource.java

@@ -48,6 +48,7 @@
 import org.openmetadata.service.security.policyevaluator.OperationContext;
 import org.openmetadata.service.security.policyevaluator.ReportDataContext;
 import org.openmetadata.service.security.policyevaluator.ResourceContextInterface;
+import org.openmetadata.service.util.FullyQualifiedName;
 import org.openmetadata.service.util.RestUtil;
 import org.openmetadata.service.util.ResultList;
 
@@ -134,7 +135,12 @@ public ResultList<TestCaseResolutionStatus> list(
  String assignee,
  @Parameter(description = "Test case fully qualified name", schema = @Schema(type = "String"))
  @QueryParam("testCaseFQN")
- String testCaseFQN) {
+ String testCaseFQN,
+ @Parameter(
+ description = "Origin entity for which the incident was opened for",
+ schema = @Schema(type = "String"))
+ @QueryParam("originEntityFQN")
+ String originEntityFQN) {
  OperationContext operationContext =
  new OperationContext(Entity.TEST_CASE, MetadataOperation.VIEW_ALL);
  ResourceContextInterface resourceContext = ReportDataContext.builder().build();
@@ -143,7 +149,8 @@ public ResultList<TestCaseResolutionStatus> list(
  ListFilter filter = new ListFilter(null);
  filter.addQueryParam("testCaseResolutionStatusType", testCaseResolutionStatusType);
  filter.addQueryParam("assignee", assignee);
- filter.addQueryParam("entityFQNHash", testCaseFQN);
+ filter.addQueryParam("entityFQNHash", FullyQualifiedName.buildHash(testCaseFQN));
+ filter.addQueryParam("originEntityFQN", originEntityFQN);
 
  return repository.list(offset, startTs, endTs, limitParam, filter, latest);
  }

openmetadata-service/src/test/java/org/openmetadata/service/resources/dqtests/TestCaseResourceTest.java

@@ -95,6 +95,7 @@
 import org.openmetadata.schema.type.ChangeDescription;
 import org.openmetadata.schema.type.Column;
 import org.openmetadata.schema.type.ColumnDataType;
+import org.openmetadata.schema.type.EntityReference;
 import org.openmetadata.schema.type.TableData;
 import org.openmetadata.schema.type.TagLabel;
 import org.openmetadata.schema.type.TaskStatus;
@@ -1569,6 +1570,34 @@
  TestCaseResolutionStatusTypes.Ack,
  storedTestCaseResolutions.getData().get(0).getTestCaseResolutionStatusType());
 
+ // Get the test case resolution by FQN
+ Map<String, String> queryParams = new HashMap<>();
+ queryParams.put("testCaseFQN", TEST_TABLE1.getFullyQualifiedName());
+ storedTestCaseResolutions = getTestCaseFailureStatus(startTs, endTs, null, null, queryParams);
+ assertTrue(
+ storedTestCaseResolutions.getData().stream()
+ .allMatch(
+ t ->
+ t.getTestCaseReference()
+ .getFullyQualifiedName()
+ .equals(testCaseEntity1.getFullyQualifiedName())));
+
+ // Get the test case resolution by origin entity FQN
+ queryParams.clear();
+ queryParams.put("originEntityFQN", TEST_TABLE1.getFullyQualifiedName());
+ storedTestCaseResolutions = getTestCaseFailureStatus(startTs, endTs, null, null, queryParams);
+ for (TestCaseResolutionStatus testCaseResolution : storedTestCaseResolutions.getData()) {
+ EntityReference testCaseReference = testCaseResolution.getTestCaseReference();
+ TestCase testCase = getEntity(testCaseReference.getId(), ADMIN_AUTH_HEADERS);
+ MessageParser.EntityLink entityLink =
+ MessageParser.EntityLink.parse(testCase.getEntityLink());
+ assertEquals(entityLink.getEntityFQN(), TEST_TABLE1.getFullyQualifiedName());
+ }
+
+ queryParams.put("originEntityFQN", "IDONOTEXIST123");
+ storedTestCaseResolutions = getTestCaseFailureStatus(startTs, endTs, null, null, queryParams);
+ assertEquals(0, storedTestCaseResolutions.getData().size());
+
  // Delete test case recursively and check that the test case resolution status is also deleted
  // 1. soft delete - should not delete the test case resolution status
  // 2. hard delete - should delete the test case resolution status
@@ -2385,9 +2414,13 @@
  Long startTs,
  Long endTs,
  String assignee,
- TestCaseResolutionStatusTypes testCaseResolutionStatusType)
+ TestCaseResolutionStatusTypes testCaseResolutionStatusType,
+ Map<String, String> fields)
  throws HttpResponseException {
  WebTarget target = getCollection().path("/testCaseIncidentStatus");
+ for (Map.Entry<String, String> entry : fields.entrySet()) {
+ target = target.queryParam(entry.getKey(), entry.getValue());
+ }
  target = target.queryParam("startTs", startTs);
  target = target.queryParam("endTs", endTs);
  target = assignee != null ? target.queryParam("assignee", assignee) : target;
@@ -2401,6 +2434,16 @@
  ADMIN_AUTH_HEADERS);
  }
 
+ public ResultList<TestCaseResolutionStatus> getTestCaseFailureStatus(
+ Long startTs,
+ Long endTs,
+ String assignee,
+ TestCaseResolutionStatusTypes testCaseResolutionStatusType)
+ throws HttpResponseException {
+ return getTestCaseFailureStatus(
+ startTs, endTs, assignee, testCaseResolutionStatusType, new HashMap<>());
+ }
+
  private TestCaseResolutionStatus getTestCaseFailureStatusById(UUID id)
  throws HttpResponseException {
  String pathUrl = "/testCaseIncidentStatus/" + id;