@Test public void testClassificationTypeDefaultValue() { AtlasClassification defValue = classificationType.createDefaultValue(); assertNotNull(defValue); assertEquals(defValue.getTypeName(), classificationType.getTypeName()); }
public static AtlasClassification newClassification(AtlasClassificationDef classificationDef, AtlasTypeRegistry typesRegistry) { AtlasClassification ret = null; AtlasClassificationType classificationType = typesRegistry.getClassificationTypeByName(classificationDef.getName()); if (classificationType != null) { ret = classificationType.createDefaultValue(); } else { LOG.error("failed to get classification-type {}", classificationDef.getName()); } return ret; }
public static AtlasClassification newClassification(AtlasClassificationDef classificationDef, AtlasTypeRegistry typesRegistry) { AtlasClassification ret = null; AtlasClassificationType classificationType = typesRegistry.getClassificationTypeByName(classificationDef.getName()); if (classificationType != null) { ret = classificationType.createDefaultValue(); } else { LOG.error("failed to get classification-type {}", classificationDef.getName()); } return ret; }
@Test(dependsOnMethods = "atT0_ReturnsAllEntities") public void atT1_NewClassificationAttachedToTable_ReturnsChangedTable() throws AtlasBaseException { final int expectedEntityCount = 1; AtlasClassificationType ct = createNewClassification(); entityStore.addClassifications(TABLE_GUID, ImmutableList.of(ct.createDefaultValue())); AtlasExportRequest request = getIncrementalRequest(nextTimestamp); ZipSource source = runExportWithParameters(exportService, request); AtlasEntity.AtlasEntityWithExtInfo entities = getEntities(source, expectedEntityCount); AtlasEntity entity = null; for (Map.Entry<String, AtlasEntity> entry : entities.getReferredEntities().entrySet()) { entity = entry.getValue(); assertNotNull(entity); break; } nextTimestamp = updateTimesampForNextIncrementalExport(source); assertEquals(entity.getGuid(),TABLE_GUID); }
@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"); }
@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"); }
@Test public void testClassificationSerDeWithSuperTypes() throws AtlasBaseException { AtlasClassificationDef classificationDef = ModelTestUtil.getClassificationDefWithSuperTypes(); 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 superTypes"); } }
@Test public void testClassificationSerDeWithSuperTypes() throws AtlasBaseException { AtlasClassificationDef classificationDef = ModelTestUtil.getClassificationDefWithSuperTypes(); 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 superTypes"); } }
@Test(dependsOnMethods = "atT1_NewClassificationAttachedToTable_ReturnsChangedTable") public void atT2_NewClassificationAttachedToColumn_ReturnsChangedColumn() throws AtlasBaseException { final int expectedEntityCount = 1; AtlasEntity.AtlasEntityWithExtInfo tableEntity = entityStore.getById(TABLE_GUID); long preExportTableEntityTimestamp = tableEntity.getEntity().getUpdateTime().getTime(); entityStore.addClassifications(COLUMN_GUID_HIGH, ImmutableList.of(typeRegistry.getClassificationTypeByName("T1").createDefaultValue())); ZipSource source = runExportWithParameters(exportService, getIncrementalRequest(nextTimestamp)); AtlasEntity.AtlasEntityWithExtInfo entities = getEntities(source, expectedEntityCount); for (Map.Entry<String, AtlasEntity> entry : entities.getReferredEntities().entrySet()) { AtlasEntity entity = entry.getValue(); assertNotNull(entity.getGuid()); break; } long postUpdateTableEntityTimestamp = tableEntity.getEntity().getUpdateTime().getTime(); assertEquals(preExportTableEntityTimestamp, postUpdateTableEntityTimestamp); nextTimestamp = updateTimesampForNextIncrementalExport(source); }