Storyboard EnumerationInSimpleClassModel
ClassModel model = new ClassModel("org.sdmlib.test.examples.simpleEnumModel.model"); Clazz enumeration = model.createClazz("TEnum").enableEnumeration(); GraphUtil.setLiteral(enumeration, new Literal("T1"), new Literal("T2"), new Literal("12"), new Literal("T1000")); enumeration.withMethod("toString", DataType.STRING); Clazz alexClazz = model.createClazz("Alex"); alexClazz.withAttribute("Name", DataType.STRING); Clazz macClazz = model.createClazz("Mac"); macClazz.withAttribute("Name", DataType.STRING) .withAttribute("type", DataType.create(enumeration)) .withAttribute("owner", DataType.create(alexClazz)); macClazz.withMethod("concat", DataType.STRING, new Parameter(DataType.INT)); macClazz.withMethod("select", DataType.create(enumeration), new Parameter(DataType.INT)); model.generate("src/test/java");
Check: Number of Enumeration types in the model: 1 actual 1