Commit 295595db by Sarath Subramanian

ATLAS-2510: Add support to disable/enable propagated classification in entity - fix UT failure

parent 0f689faa
...@@ -1739,15 +1739,17 @@ public class EntityGraphMapper { ...@@ -1739,15 +1739,17 @@ public class EntityGraphMapper {
} }
public void deleteClassifications(String guid) throws AtlasBaseException { public void deleteClassifications(String guid) throws AtlasBaseException {
AtlasVertex instanceVertex = AtlasGraphUtilsV1.findByGuid(guid); AtlasVertex instanceVertex = AtlasGraphUtilsV1.findByGuid(guid);
if (instanceVertex == null) { if (instanceVertex == null) {
throw new AtlasBaseException(AtlasErrorCode.INSTANCE_GUID_NOT_FOUND, guid); throw new AtlasBaseException(AtlasErrorCode.INSTANCE_GUID_NOT_FOUND, guid);
} }
List<String> traitNames = getTraitNames(instanceVertex); List<String> traitNames = getTraitNames(instanceVertex);
deleteClassifications(guid, traitNames); if (CollectionUtils.isNotEmpty(traitNames)) {
deleteClassifications(guid, traitNames);
}
} }
private void updateTraitNamesProperty(AtlasVertex entityVertex, List<String> traitNames) { private void updateTraitNamesProperty(AtlasVertex entityVertex, List<String> traitNames) {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment