ATLAS-4807: Fix IT test related to Labels

Signed-off-by: Pinal Shah <pinal.shah@freestoneinfotech.com>
This commit is contained in:
Pinal Shah 2023-11-01 13:34:04 +07:00
parent 35ce839f0a
commit 55d79065b3
1 changed files with 2 additions and 2 deletions

View File

@ -967,7 +967,7 @@ public class EntityV2JerseyResourceIT extends BaseResourceIT {
createdDBName = (String) createHiveDB().getAttribute(NAME);
Set<String> labels = new HashSet<>();
labels.add("labelByTypeName");
atlasClientV2.addLabels(createHiveDB().getTypeName(), toMap(NAME, createdDBName), labels);
atlasClientV2.setLabels(createHiveDB().getTypeName(), toMap(NAME, createdDBName), labels);
AtlasEntityWithExtInfo info = atlasClientV2.getEntityByGuid(createHiveDB().getGuid(), false, true);
assertNotNull(info);
assertNotNull(info.getEntity().getLabels());
@ -979,7 +979,7 @@ public class EntityV2JerseyResourceIT extends BaseResourceIT {
createdDBName = (String) createHiveDB().getAttribute(NAME);
Set<String> labels = new HashSet<>();
labels.add("labelByTypeNameNext");
atlasClientV2.setLabels(createHiveDB().getTypeName(), toMap(NAME, createdDBName), labels);
atlasClientV2.addLabels(createHiveDB().getTypeName(), toMap(NAME, createdDBName), labels);
AtlasEntityWithExtInfo infoForSet = atlasClientV2.getEntityByGuid(createHiveDB().getGuid(), false, true);
assertNotNull(infoForSet);
assertNotNull(infoForSet.getEntity().getLabels());