private Map getMapFromValue(Object val) { final Map ret; if (val instanceof Map) { ret = (Map) val; } else if (val instanceof String) { ret = AtlasType.fromJson(val.toString(), Map.class); } else { ret = null; } return ret; } }
private boolean isEmptyMapValue(Object val) { if (val == null) { return true; } else if (val instanceof Map) { return ((Map) val).isEmpty(); } else if (val instanceof String) { Map map = AtlasType.fromJson(val.toString(), Map.class); return map == null || map.isEmpty(); } return false; }
@Override public AtlasTypesDef getScrubbedTypesDef(String jsonStr) { AtlasTypesDef typesDef = AtlasType.fromJson(jsonStr, AtlasTypesDef.class); return typesDefStrubberForMigrationImport.scrub(typesDef); }
private static boolean getSoftRef(JSONObject json) throws JSONException { final String SOFT_REF_KEY = "isSoftReference"; boolean isSoftRef; Map map = AtlasType.fromJson(json.getString("options"), Map.class); isSoftRef = (map != null && map.containsKey(SOFT_REF_KEY)) ? Boolean.parseBoolean((String) map.get(SOFT_REF_KEY)) : false; return isSoftRef; } }
public static <T> T readObjectFromJson(String subDir, String filename, Class<T> objectClass) throws IOException { final String userDir = System.getProperty("user.dir"); String filePath = getTestJsonPath(userDir, subDir, filename); String json = FileUtils.readFileToString(new File(filePath)); return AtlasType.fromJson(json, objectClass); }
@Test public void testEntityDefSerDeEmpty() { AtlasEntityDef entityDef = new AtlasEntityDef("emptyEntityDef"); String jsonString = AtlasType.toJson(entityDef); AtlasEntityDef entityDef2 = AtlasType.fromJson(jsonString, AtlasEntityDef.class); assertEquals(entityDef2, entityDef, "Incorrect serialization/deserialization of AtlasEntityDef"); }
@Test public void testClassificationDefSerDeWithSuperTypes() { AtlasClassificationDef classificationDef = ModelTestUtil.getClassificationDefWithSuperTypes(); String jsonString = AtlasType.toJson(classificationDef); AtlasClassificationDef classificationDef2 = AtlasType.fromJson(jsonString, AtlasClassificationDef.class); assertEquals(classificationDef2, classificationDef, "Incorrect serialization/deserialization of AtlasClassificationDef with superTypes"); }
@Test public void testClassificationDefSerDe() { AtlasClassificationDef classificationDef = ModelTestUtil.getClassificationDef(); String jsonString = AtlasType.toJson(classificationDef); AtlasClassificationDef classificationDef2 = AtlasType.fromJson(jsonString, AtlasClassificationDef.class); assertEquals(classificationDef2, classificationDef, "Incorrect serialization/deserialization of AtlasClassificationDef"); }
@Test public void testStructDefSerDe() { AtlasStructDef structDef = ModelTestUtil.getStructDef(); String jsonString = AtlasType.toJson(structDef); AtlasStructDef structDef2 = AtlasType.fromJson(jsonString, AtlasStructDef.class); assertEquals(structDef2, structDef, "Incorrect serialization/deserialization of AtlasStructDef"); }
@Test public void testEntityDefSerDeWithSuperType() { AtlasEntityDef entityDef = ModelTestUtil.getEntityDefWithSuperType(); String jsonString = AtlasType.toJson(entityDef); AtlasEntityDef entityDef2 = AtlasType.fromJson(jsonString, AtlasEntityDef.class); assertEquals(entityDef2, entityDef, "Incorrect serialization/deserialization of AtlasEntityDef with superType"); }
@Test public void testEntityDefSerDeWithSuperTypes() { AtlasEntityDef entityDef = ModelTestUtil.getEntityDefWithSuperTypes(); String jsonString = AtlasType.toJson(entityDef); AtlasEntityDef entityDef2 = AtlasType.fromJson(jsonString, AtlasEntityDef.class); assertEquals(entityDef2, entityDef, "Incorrect serialization/deserialization of AtlasEntityDef with superTypes"); }
@Test public void testClassificationDefSerDeEmpty() { AtlasClassificationDef classificationDef = new AtlasClassificationDef("emptyClassificationDef"); String jsonString = AtlasType.toJson(classificationDef); AtlasClassificationDef classificationDef2 = AtlasType.fromJson(jsonString, AtlasClassificationDef.class); assertEquals(classificationDef2, classificationDef, "Incorrect serialization/deserialization of AtlasClassificationDef"); }
@Test public void testClassificationDefSerDeWithSuperType() { AtlasClassificationDef classificationDef = ModelTestUtil.getClassificationDefWithSuperType(); String jsonString = AtlasType.toJson(classificationDef); AtlasClassificationDef classificationDef2 = AtlasType.fromJson(jsonString, AtlasClassificationDef.class); assertEquals(classificationDef2, classificationDef, "Incorrect serialization/deserialization of AtlasClassificationDef with superType"); }
@Test public void testEnumDefSerDeEmpty() { AtlasEnumDef enumDef1 = new AtlasEnumDef(); String jsonString = AtlasType.toJson(enumDef1); AtlasEnumDef enumDef2 = AtlasType.fromJson(jsonString, AtlasEnumDef.class); assertEquals(enumDef1, enumDef2, "Incorrect serialization/deserialization of AtlasEnumDef"); }
@Test public void testEnumDefSerDe() { AtlasEnumDef enumDef = ModelTestUtil.getEnumDef(); String jsonString = AtlasType.toJson(enumDef); AtlasEnumDef enumDef2 = AtlasType.fromJson(jsonString, AtlasEnumDef.class); assertEquals(enumDef, enumDef2, "Incorrect serialization/deserialization of AtlasEnumDef"); }
@Test public void testStructDefSerDeEmpty() { AtlasStructDef structDef = new AtlasStructDef("emptyStructDef"); String jsonString = AtlasType.toJson(structDef); AtlasStructDef structDef2 = AtlasType.fromJson(jsonString, AtlasStructDef.class); assertEquals(structDef2, structDef, "Incorrect serialization/deserialization of AtlasStructDef"); }
@Test public void serializeAtlasImportRequstFromJsonWithEmptyOptions() { String jsonData = "{ \"options\": {} }"; AtlasImportRequest request = AtlasType.fromJson(jsonData, AtlasImportRequest.class); assertNotNull(request); assertNotNull(request.getOptions()); assertNull(request.getOptions().get(AtlasImportRequest.TRANSFORMS_KEY)); ImportTransforms tr = ImportTransforms.fromJson(request.getOptions().get(AtlasImportRequest.TRANSFORMS_KEY)); assertNull(tr); }
@Test public void testEntitySerDe() throws AtlasBaseException { AtlasEntityDef entityDef = ModelTestUtil.getEntityDef(); AtlasTypeRegistry typeRegistry = ModelTestUtil.getTypesRegistry(); AtlasEntityType entityType = typeRegistry.getEntityTypeByName(entityDef.getName()); assertNotNull(entityType); AtlasEntity ent1 = entityType.createDefaultValue(); String jsonString = AtlasType.toJson(ent1); AtlasEntity ent2 = AtlasType.fromJson(jsonString, AtlasEntity.class); entityType.normalizeAttributeValues(ent2); assertEquals(ent2, ent1, "Incorrect serialization/deserialization of AtlasEntity"); }
@Test public void testClassificationSerDe() throws AtlasBaseException { AtlasClassificationDef classificationDef = ModelTestUtil.getClassificationDef(); AtlasTypeRegistry typeRegistry = ModelTestUtil.getTypesRegistry(); AtlasClassificationType classificationType = typeRegistry.getClassificationTypeByName(classificationDef.getName()); assertNotNull(classificationType); AtlasClassification ent1 = ModelTestUtil.newClassification(classificationDef, typeRegistry); String jsonString = AtlasType.toJson(ent1); AtlasClassification ent2 = AtlasType.fromJson(jsonString, AtlasClassification.class); classificationType.normalizeAttributeValues(ent2); assertEquals(ent2, ent1, "Incorrect serialization/deserialization of AtlasClassification"); }
@Test public void testClassificationSerDeWithSuperType() throws AtlasBaseException { AtlasClassificationDef classificationDef = ModelTestUtil.getClassificationDefWithSuperType(); AtlasTypeRegistry typeRegistry = ModelTestUtil.getTypesRegistry(); AtlasClassificationType classificationType = typeRegistry.getClassificationTypeByName(classificationDef.getName()); assertNotNull(classificationType); AtlasClassification ent1 = classificationType.createDefaultValue(); String jsonString = AtlasType.toJson(ent1); AtlasClassification ent2 = AtlasType.fromJson(jsonString, AtlasClassification.class); classificationType.normalizeAttributeValues(ent2); assertEquals(ent2, ent1, "Incorrect serialization/deserialization of AtlasClassification with superType"); }