From 2c27d10fd88addd32067024f027fb5f2bbfdf608 Mon Sep 17 00:00:00 2001
From: sureshms <suresh@getcollate.io>
Date: Tue, 26 Oct 2021 14:13:43 -0700
Subject: [PATCH] Update to databae tests to check patched entity

---
 .../catalog/resources/databases/DatabaseResourceTest.java  | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/catalog-rest-service/src/test/java/org/openmetadata/catalog/resources/databases/DatabaseResourceTest.java b/catalog-rest-service/src/test/java/org/openmetadata/catalog/resources/databases/DatabaseResourceTest.java
index f1d412681d1..a309f346126 100644
--- a/catalog-rest-service/src/test/java/org/openmetadata/catalog/resources/databases/DatabaseResourceTest.java
+++ b/catalog-rest-service/src/test/java/org/openmetadata/catalog/resources/databases/DatabaseResourceTest.java
@@ -601,7 +601,7 @@ public class DatabaseResourceTest extends EntityTestHelper<Database> {
             TestUtils.getPrincipal(authHeaders), createRequest.getOwner());
 
     // Validate service
-    assertService(createdEntity.getService(), createdEntity.getService());
+    assertService(createRequest.getService(), createdEntity.getService());
   }
 
   @Override
@@ -611,7 +611,10 @@ public class DatabaseResourceTest extends EntityTestHelper<Database> {
 
   @Override
   public void validatePatchedEntity(Database expected, Database updated, Map<String, String> authHeaders) {
-
+    validateCommonEntityFields(getEntityInterface(updated), expected.getDescription(),
+            TestUtils.getPrincipal(authHeaders), expected.getOwner());
+    // Validate service
+    assertService(expected.getService(), updated.getService());
   }
 
   @Override