Commit 39be2ccf by nixonrodrigues

ATLAS-2376: Fix for unitTest case in ExportServiceTest due to NPE

parent bd87c3f6
...@@ -457,13 +457,17 @@ public class ExportService { ...@@ -457,13 +457,17 @@ public class ExportService {
} }
private void removeRelationshipAttributes(AtlasEntityWithExtInfo entity) { private void removeRelationshipAttributes(AtlasEntityWithExtInfo entity) {
if (entity.getEntity().getRelationshipAttributes() != null) {
entity.getEntity().getRelationshipAttributes().clear(); entity.getEntity().getRelationshipAttributes().clear();
if(entity.getReferredEntities() != null) { }
for (AtlasEntity e: entity.getReferredEntities().values()) { if (entity.getReferredEntities() != null) {
for (AtlasEntity e : entity.getReferredEntities().values()) {
if (e.getRelationshipAttributes() != null) {
e.getRelationshipAttributes().clear(); e.getRelationshipAttributes().clear();
} }
} }
} }
}
private void addTypes(AtlasEntity entity, ExportContext context) { private void addTypes(AtlasEntity entity, ExportContext context) {
addEntityType(entity.getTypeName(), context); addEntityType(entity.getTypeName(), context);
......
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