From 4880641cffb6f2c5179214e24a30857429790888 Mon Sep 17 00:00:00 2001
From: sureshms <suresh@getcollate.io>
Date: Tue, 26 Oct 2021 14:15:41 -0700
Subject: [PATCH] Update to chart tests to check patched entity

---
 .../catalog/resources/charts/ChartResourceTest.java         | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/catalog-rest-service/src/test/java/org/openmetadata/catalog/resources/charts/ChartResourceTest.java b/catalog-rest-service/src/test/java/org/openmetadata/catalog/resources/charts/ChartResourceTest.java
index 9982496a14a..baa69b0e090 100644
--- a/catalog-rest-service/src/test/java/org/openmetadata/catalog/resources/charts/ChartResourceTest.java
+++ b/catalog-rest-service/src/test/java/org/openmetadata/catalog/resources/charts/ChartResourceTest.java
@@ -659,7 +659,6 @@ public class ChartResourceTest extends EntityTestHelper<Chart> {
     CreateChart createRequest = (CreateChart) request;
     validateCommonEntityFields(getEntityInterface(chart), createRequest.getDescription(),
             TestUtils.getPrincipal(authHeaders), createRequest.getOwner());
-    // TODO add display name
     assertService(createRequest.getService(), chart.getService());
   }
 
@@ -669,7 +668,10 @@ public class ChartResourceTest extends EntityTestHelper<Chart> {
   }
 
   @Override
-  public void validatePatchedEntity(Chart expected, Chart updated, Map<String, String> authHeaders) {
+  public void validatePatchedEntity(Chart expected, Chart patched, Map<String, String> authHeaders) {
+    validateCommonEntityFields(getEntityInterface(patched), expected.getDescription(),
+            TestUtils.getPrincipal(authHeaders), expected.getOwner());
+    assertService(expected.getService(), patched.getService());
   }
 
   @Override