void defineEnums(TypeSystem ts) throws AtlasException { ts.defineEnumType("HiveObjectType", new EnumValue("GLOBAL", 1), new EnumValue("DATABASE", 2), new EnumValue("TABLE", 3), new EnumValue("PARTITION", 4), new EnumValue("COLUMN", 5)); ts.defineEnumType("PrincipalType", new EnumValue("USER", 1), new EnumValue("ROLE", 2), new EnumValue("GROUP", 3)); ts.defineEnumType("TxnState", new EnumValue("COMMITTED", 1), new EnumValue("ABORTED", 2), new EnumValue("OPEN", 3)); ts.defineEnumType("LockLevel", new EnumValue("DB", 1), new EnumValue("TABLE", 2), new EnumValue("PARTITION", 3)); }
void defineEnums(TypeSystem ts) throws AtlasException { ts.defineEnumType("HiveObjectType", new EnumValue("GLOBAL", 1), new EnumValue("DATABASE", 2), new EnumValue("TABLE", 3), new EnumValue("PARTITION", 4), new EnumValue("COLUMN", 5)); ts.defineEnumType("PrincipalType", new EnumValue("USER", 1), new EnumValue("ROLE", 2), new EnumValue("GROUP", 3)); ts.defineEnumType("TxnState", new EnumValue("COMMITTED", 1), new EnumValue("ABORTED", 2), new EnumValue("OPEN", 3)); ts.defineEnumType("LockLevel", new EnumValue("DB", 1), new EnumValue("TABLE", 2), new EnumValue("PARTITION", 3)); }
@DataProvider(name = "enumTypeData") private Object[][] createEnumTypeData() { EnumValue value = new EnumValue("name", 1); return new Object[][]{{null, value}, {"", value}, {"name"}}; }
@DataProvider(name = "enumTypeData") private Object[][] createEnumTypeData() { EnumValue value = new EnumValue("name", 1); return new Object[][]{{null, value}, {"", value}, {"name"}}; }
@Test(dataProvider = "enumValueData", expectedExceptions = {IllegalArgumentException.class}) public void testEnumValue(String name) { new EnumValue(name, 1); }
@Test(dataProvider = "enumValueData", expectedExceptions = {IllegalArgumentException.class}) public void testEnumValue(String name) { new EnumValue(name, 1); }
@Test public void testTypeUpdate() throws Exception { TypeSystem ts = getTypeSystem(); EnumTypeDefinition etd = new EnumTypeDefinition(newName(), new EnumValue("A", 1)); TypesDef typesDef = getTypesDef(etd); ts.defineTypes(typesDef); etd = new EnumTypeDefinition(etd.name, new EnumValue("A", 1), new EnumValue("B", 2)); typesDef = getTypesDef(etd); ts.updateTypes(typesDef); etd = new EnumTypeDefinition(etd.name, new EnumValue("A", 1)); typesDef = getTypesDef(etd); try { etd = new EnumTypeDefinition(etd.name, new EnumValue("A", 2)); typesDef = getTypesDef(etd); try {
@Test public void testTypeUpdate() throws Exception { TypeSystem ts = getTypeSystem(); EnumTypeDefinition etd = new EnumTypeDefinition(newName(), new EnumValue("A", 1)); TypesDef typesDef = getTypesDef(etd); ts.defineTypes(typesDef); etd = new EnumTypeDefinition(etd.name, new EnumValue("A", 1), new EnumValue("B", 2)); typesDef = getTypesDef(etd); ts.updateTypes(typesDef); etd = new EnumTypeDefinition(etd.name, new EnumValue("A", 1)); typesDef = getTypesDef(etd); try { etd = new EnumTypeDefinition(etd.name, new EnumValue("A", 2)); typesDef = getTypesDef(etd); try {
@Test public void testUTFNames() throws Exception { TypeSystem ts = getTypeSystem(); String enumType = random(); EnumTypeDefinition orgLevelEnum = new EnumTypeDefinition(enumType, new EnumValue(random(), 1), new EnumValue(random(), 2)); String structName = random(); String attrType = random(); StructTypeDefinition structType = createStructTypeDef(structName, createRequiredAttrDef(attrType, DataTypes.STRING_TYPE)); String className = random(); HierarchicalTypeDefinition<ClassType> classType = createClassTypeDef(className, ImmutableSet.<String>of(), createRequiredAttrDef(attrType, DataTypes.STRING_TYPE)); String traitName = random(); HierarchicalTypeDefinition<TraitType> traitType = createTraitTypeDef(traitName, ImmutableSet.<String>of(), createRequiredAttrDef(attrType, DataTypes.INT_TYPE)); ts.defineTypes(ImmutableList.of(orgLevelEnum), ImmutableList.of(structType), ImmutableList.of(traitType), ImmutableList.of(classType)); }
@Test public void testUTFNames() throws Exception { TypeSystem ts = getTypeSystem(); String enumType = random(); EnumTypeDefinition orgLevelEnum = new EnumTypeDefinition(enumType, new EnumValue(random(), 1), new EnumValue(random(), 2)); String structName = random(); String attrType = random(); StructTypeDefinition structType = createStructTypeDef(structName, createRequiredAttrDef(attrType, DataTypes.STRING_TYPE)); String className = random(); HierarchicalTypeDefinition<ClassType> classType = createClassTypeDef(className, ImmutableSet.<String>of(), createRequiredAttrDef(attrType, DataTypes.STRING_TYPE)); String traitName = random(); HierarchicalTypeDefinition<TraitType> traitType = createTraitTypeDef(traitName, ImmutableSet.<String>of(), createRequiredAttrDef(attrType, DataTypes.INT_TYPE)); ts.defineTypes(ImmutableList.of(orgLevelEnum), ImmutableList.of(structType), ImmutableList.of(traitType), ImmutableList.of(classType)); }
String description = "_description"; String typeDescription = typeName + description; getTypeSystem().defineEnumType(typeName, typeDescription, new EnumValue("0", 0), new EnumValue("1", 1), new EnumValue("2", 2), new EnumValue("3", 3)); assertTrue(getTypeSystem().getTypeNames().contains(typeName)); IDataType type = getTypeSystem().getDataType(EnumType.class, typeName);
@BeforeClass public void onetimeSetup() throws Exception { // init TypeSystem TypeSystem ts = TypeSystem.getInstance().reset(); // Customer ClassType customerType = ts.defineClassType(TypesUtil .createClassTypeDef(CLASSTYPE_CUSTOMER, ImmutableSet.<String>of(), TypesUtil.createRequiredAttrDef("name", DataTypes.STRING_TYPE), TypesUtil.createRequiredAttrDef("id", DataTypes.LONG_TYPE))); // Address StructType addressType = ts.defineStructType(STRUCTTYPE_ADDRESS, true, TypesUtil.createRequiredAttrDef("first line", DataTypes.STRING_TYPE), TypesUtil.createOptionalAttrDef("second line", DataTypes.STRING_TYPE), TypesUtil.createRequiredAttrDef("city", DataTypes.STRING_TYPE), TypesUtil.createRequiredAttrDef("pincode", DataTypes.INT_TYPE)); // Privileged TraitType privilegedTrait = ts.defineTraitType(TypesUtil .createTraitTypeDef(TRAITTYPE_PRIVILEGED, ImmutableSet.<String>of(), TypesUtil.createRequiredAttrDef("category", DataTypes.INT_TYPE))); // Shipping EnumType shippingEnum = ts.defineEnumType(TypesUtil.createEnumTypeDef(ENUMTYPE_SHIPPING, new EnumValue("Domestic", 1), new EnumValue("International", 2))); }
String description = "_description"; String typeDescription = typeName + description; getTypeSystem().defineEnumType(typeName, typeDescription, new EnumValue("0", 0), new EnumValue("1", 1), new EnumValue("2", 2), new EnumValue("3", 3)); assertTrue(getTypeSystem().getTypeNames().contains(typeName)); IDataType type = getTypeSystem().getDataType(EnumType.class, typeName);
@BeforeClass public void onetimeSetup() throws Exception { // init TypeSystem TypeSystem ts = TypeSystem.getInstance().reset(); // Customer ClassType customerType = ts.defineClassType(TypesUtil .createClassTypeDef(CLASSTYPE_CUSTOMER, ImmutableSet.<String>of(), TypesUtil.createRequiredAttrDef("name", DataTypes.STRING_TYPE), TypesUtil.createRequiredAttrDef("id", DataTypes.LONG_TYPE))); // Address StructType addressType = ts.defineStructType(STRUCTTYPE_ADDRESS, true, TypesUtil.createRequiredAttrDef("first line", DataTypes.STRING_TYPE), TypesUtil.createOptionalAttrDef("second line", DataTypes.STRING_TYPE), TypesUtil.createRequiredAttrDef("city", DataTypes.STRING_TYPE), TypesUtil.createRequiredAttrDef("pincode", DataTypes.INT_TYPE)); // Privileged TraitType privilegedTrait = ts.defineTraitType(TypesUtil .createTraitTypeDef(TRAITTYPE_PRIVILEGED, ImmutableSet.<String>of(), TypesUtil.createRequiredAttrDef("category", DataTypes.INT_TYPE))); // Shipping EnumType shippingEnum = ts.defineEnumType(TypesUtil.createEnumTypeDef(ENUMTYPE_SHIPPING, new EnumValue("Domestic", 1), new EnumValue("International", 2))); }