MINOR - Data Contracts only execute tests without results

openmetadata-integration-tests/src/test/java/org/openmetadata/it/tests/TestCaseResourceIT.java

@@ -25,11 +25,13 @@
 import org.openmetadata.it.util.TestNamespace;
 import org.openmetadata.schema.api.classification.CreateClassification;
 import org.openmetadata.schema.api.classification.CreateTag;
+import org.openmetadata.schema.api.data.CreateDataContract;
 import org.openmetadata.schema.api.data.CreateTable;
 import org.openmetadata.schema.api.tests.CreateTestCase;
 import org.openmetadata.schema.api.tests.CreateTestSuite;
 import org.openmetadata.schema.entity.classification.Classification;
 import org.openmetadata.schema.entity.classification.Tag;
+import org.openmetadata.schema.entity.data.DataContract;
 import org.openmetadata.schema.entity.data.DatabaseSchema;
 import org.openmetadata.schema.entity.data.Table;
 import org.openmetadata.schema.entity.services.DatabaseService;
@@ -40,6 +42,7 @@
 import org.openmetadata.schema.type.Column;
 import org.openmetadata.schema.type.ColumnDataType;
 import org.openmetadata.schema.type.EntityHistory;
+import org.openmetadata.schema.type.EntityReference;
 import org.openmetadata.schema.type.TagLabel;
 import org.openmetadata.schema.type.csv.CsvImportResult;
 import org.openmetadata.schema.utils.JsonUtils;
@@ -4135,6 +4138,95 @@
     return value;
   }
 
+  // ===================================================================
+  // DATA CONTRACT REFERENCE TESTS
+  // ===================================================================
+
+  @Test
+  void test_createTestCase_withDataContractReference(TestNamespace ns) {
+    Table table = createTable(ns);
+
+    // Create a data contract for the table
+    CreateDataContract dcRequest =
+        new CreateDataContract()
+            .withName(ns.prefix("tc_dc"))
+            .withEntity(table.getEntityReference())
+            .withDescription("Contract for test case reference test");
+    DataContract contract = SdkClients.adminClient().dataContracts().create(dcRequest);
+
+    // Create a test case with dataContract FQN (the mapper resolves it to EntityReference)
+    CreateTestCase request =
+        TestCaseBuilder.create(SdkClients.adminClient())
+            .name(ns.prefix("tc_with_dc"))
+            .forTable(table)
+            .testDefinition("tableRowCountToEqual")
+            .parameter("value", "100")
+            .build();
+    request.setDataContract(contract.getFullyQualifiedName());
+
+    TestCase testCase = createEntity(request);
+
+    assertNotNull(testCase.getDataContract(), "Test case should have dataContract reference");
+    assertEquals(contract.getId(), testCase.getDataContract().getId());
+    assertEquals("dataContract", testCase.getDataContract().getType());
+  }
+
+  @Test
+  void test_createTestCase_withInvalidDataContractReference(TestNamespace ns) {
+    Table table = createTable(ns);
+
+    CreateTestCase request =
+        TestCaseBuilder.create(SdkClients.adminClient())
+            .name(ns.prefix("tc_bad_dc"))
+            .forTable(table)
+            .testDefinition("tableRowCountToEqual")
+            .parameter("value", "100")
+            .build();
+    request.setDataContract("non.existent.data.contract");
+
+    assertThrows(
+        Exception.class,
+        () -> createEntity(request),
+        "Should fail with nonexistent data contract FQN");
+  }
+
+  @Test
+  void test_testCase_dataContractLinkedViaContractCreate(TestNamespace ns) {
+    Table table = createTable(ns);
+
+    // Create test case first without dataContract
+    TestCase testCase =
+        TestCaseBuilder.create(SdkClients.adminClient())
+            .name(ns.prefix("tc_linked"))
+            .forTable(table)
+            .testDefinition("tableRowCountToEqual")
+            .parameter("value", "100")
+            .create();
+
+    // Verify no dataContract initially
+    TestCase beforeContract = SdkClients.adminClient().testCases().get(testCase.getId().toString());
+    assertTrue(
+        beforeContract.getDataContract() == null,
+        "Test case should have no dataContract before contract creation");
+
+    // Now create a data contract that references this test case
+    CreateDataContract dcRequest =
+        new CreateDataContract()
+            .withName(ns.prefix("dc_link"))
+            .withEntity(table.getEntityReference())
+            .withDescription("Contract linking to existing test case")
+            .withQualityExpectations(
+                List.of(new EntityReference().withId(testCase.getId()).withType("testCase")));
+    DataContract contract = SdkClients.adminClient().dataContracts().create(dcRequest);
+
+    // Verify test case now has dataContract reference (set by postCreate lifecycle hook)
+    TestCase afterContract = SdkClients.adminClient().testCases().get(testCase.getId().toString());
+    assertNotNull(
+        afterContract.getDataContract(),
+        "Test case should have dataContract after contract creation");
+    assertEquals(contract.getId(), afterContract.getDataContract().getId());
+  }
+
   // ===================================================================
   // COLUMN NAME FILTER TESTS (list and search/list endpoints)
   // ===================================================================

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

@@ -8840,6 +8840,46 @@ public TestCaseRecord map(ResultSet rs, StatementContext ctx) throws SQLExceptio
         return new TestCaseRecord(rs.getString("json"), rs.getInt("ranked"));
       }
     }
+
+    @ConnectionAwareSqlUpdate(
+        value =
+            "UPDATE test_case SET json = JSON_SET(json, '$.dataContract', CAST(:dataContractJson AS JSON)) WHERE id = :id",
+        connectionType = MYSQL)
+    @ConnectionAwareSqlUpdate(
+        value =
+            "UPDATE test_case SET json = jsonb_set(json, '{dataContract}', :dataContractJson::jsonb, true) WHERE id = :id",
+        connectionType = POSTGRES)
+    void updateTestCaseDataContract(
+        @Bind("id") String id, @Bind("dataContractJson") String dataContractJson);
+
+    @ConnectionAwareSqlUpdate(
+        value = "UPDATE test_case SET json = JSON_REMOVE(json, '$.dataContract') WHERE id = :id",
+        connectionType = MYSQL)
+    @ConnectionAwareSqlUpdate(
+        value = "UPDATE test_case SET json = json - 'dataContract' WHERE id = :id",
+        connectionType = POSTGRES)
+    void removeTestCaseDataContract(@Bind("id") String id);
+
+    @ConnectionAwareSqlUpdate(
+        value =
+            "UPDATE test_case SET json = JSON_REMOVE(json, '$.dataContract') WHERE id = :id AND JSON_UNQUOTE(JSON_EXTRACT(json, '$.dataContract.id')) = :dataContractId",
+        connectionType = MYSQL)
+    @ConnectionAwareSqlUpdate(
+        value =
+            "UPDATE test_case SET json = json - 'dataContract' WHERE id = :id AND json->'dataContract'->>'id' = :dataContractId",
+        connectionType = POSTGRES)
+    void removeTestCaseDataContractForSpecificContract(
+        @Bind("id") String id, @Bind("dataContractId") String dataContractId);
+
+    @ConnectionAwareSqlUpdate(
+        value =
+            "UPDATE test_case SET json = JSON_REMOVE(json, '$.dataContract') WHERE JSON_UNQUOTE(JSON_EXTRACT(json, '$.dataContract.id')) = :dataContractId",
+        connectionType = MYSQL)
+    @ConnectionAwareSqlUpdate(
+        value =
+            "UPDATE test_case SET json = json - 'dataContract' WHERE json->'dataContract'->>'id' = :dataContractId",
+        connectionType = POSTGRES)
+    void removeAllTestCaseDataContractReferences(@Bind("dataContractId") String dataContractId);
   }
 
   interface WebAnalyticEventDAO extends EntityDAO<WebAnalyticEvent> {