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 {
}
public void deleteClassifications(String guid) throws AtlasBaseException {
AtlasVertex instanceVertex = AtlasGraphUtilsV1.findByGuid(guid);
if (instanceVertex == null) {
throw new AtlasBaseException(AtlasErrorCode.INSTANCE_GUID_NOT_FOUND, guid);
}
List<String> traitNames = getTraitNames(instanceVertex);
deleteClassifications(guid, traitNames);
if (CollectionUtils.isNotEmpty(traitNames)) {
deleteClassifications(guid, 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