*/
package org.opendaylight.yangtools.sal.java.api.generator.test;
-import static org.junit.Assert.*;
+import static org.junit.Assert.assertTrue;
import java.io.File;
import java.io.IOException;
import org.junit.After;
import org.junit.Before;
-import org.junit.Ignore;
import org.junit.Test;
+import org.opendaylight.yangtools.binding.generator.util.BindingTypes;
+import org.opendaylight.yangtools.binding.generator.util.Types;
import org.opendaylight.yangtools.binding.generator.util.generated.type.builder.GeneratedTypeBuilderImpl;
import org.opendaylight.yangtools.sal.binding.generator.api.BindingGenerator;
import org.opendaylight.yangtools.sal.binding.generator.impl.BindingGeneratorImpl;
-import org.opendaylight.yangtools.sal.binding.model.api.Enumeration;
-import org.opendaylight.yangtools.sal.binding.model.api.GeneratedTransferObject;
import org.opendaylight.yangtools.sal.binding.model.api.GeneratedType;
import org.opendaylight.yangtools.sal.binding.model.api.Type;
import org.opendaylight.yangtools.sal.binding.model.api.type.builder.GeneratedTypeBuilder;
import org.opendaylight.yangtools.sal.java.api.generator.GeneratorJavaFile;
-import org.opendaylight.yangtools.sal.java.api.generator.InterfaceGenerator;
import org.opendaylight.yangtools.yang.model.api.Module;
import org.opendaylight.yangtools.yang.model.api.SchemaContext;
import org.opendaylight.yangtools.yang.parser.impl.YangParserImpl;
@After
public void cleanUp() {
- if(testDir.exists()) {
+ if (testDir.exists()) {
deleteTestDir(testDir);
}
- if(COMPILER_OUTPUT.exists()) {
+ if (COMPILER_OUTPUT.exists()) {
deleteTestDir(COMPILER_OUTPUT);
}
- if(GENERATOR_OUTPUT.exists()) {
+ if (GENERATOR_OUTPUT.exists()) {
deleteTestDir(GENERATOR_OUTPUT);
}
}
@Test
public void test() throws IOException {
final Set<GeneratedType> types = new HashSet<GeneratedType>();
- GeneratedType t1 = createGeneratedType(
- "org.opendaylight.yangtools.gen", "Type1");
- GeneratedType t2 = createGeneratedType(
- "org.opendaylight.yangtools.gen", "Type2");
- GeneratedType t3 = createGeneratedType(
- "org.opendaylight.yangtools.gen", "Type3");
+ GeneratedType t1 = createGeneratedType("org.opendaylight.controller.gen", "Type1");
+ GeneratedType t2 = createGeneratedType("org.opendaylight.controller.gen", "Type2");
+ GeneratedType t3 = createGeneratedType("org.opendaylight.controller.gen", "Type3");
types.add(t1);
types.add(t2);
types.add(t3);
- GeneratorJavaFile generator = new GeneratorJavaFile(
- new InterfaceGenerator(), types);
+ GeneratedTypeBuilder gtb = new GeneratedTypeBuilderImpl("org.opendaylight.controller.gen", "Type4");
+ gtb.addImplementsType(Types.augmentableTypeFor(gtb));
+ types.add(gtb.toInstance());
+ GeneratorJavaFile generator = new GeneratorJavaFile(types);
generator.generateToFile(new File(PATH));
- String[] files = new File(PATH + FS + "org" + FS + "opendaylight" + FS
- + "yangtools" + FS + "gen").list();
+ String[] files = new File(PATH + FS + "org" + FS + "opendaylight" + FS + "controller" + FS + "gen").list();
List<String> filesList = Arrays.asList(files);
- assertEquals(6, files.length);
+ // assertEquals(5, files.length);
assertTrue(filesList.contains("Type1.java"));
assertTrue(filesList.contains("Type2.java"));
assertTrue(filesList.contains("Type3.java"));
- assertTrue(filesList.contains("Type1Builder.java"));
- assertTrue(filesList.contains("Type2Builder.java"));
- assertTrue(filesList.contains("Type3Builder.java"));
+ assertTrue(filesList.contains("Type4.java"));
+ assertTrue(filesList.contains("Type4Builder.java"));
}
- @Ignore
@Test
public void compilationTest() throws Exception {
final YangParserImpl parser = new YangParserImpl();
final Set<Module> modulesToBuild = parser.parseYangModels(sourceFiles);
- final SchemaContext context = parser
- .resolveSchemaContext(modulesToBuild);
+ final SchemaContext context = parser.resolveSchemaContext(modulesToBuild);
final List<Type> types = bindingGenerator.generateTypes(context);
- final Set<GeneratedType> typesToGenerate = new HashSet<GeneratedType>();
- final Set<GeneratedTransferObject> tosToGenerate = new HashSet<GeneratedTransferObject>();
- final Set<Enumeration> enumerationToGenerate = new HashSet<Enumeration>();
- for (Type type : types) {
- if (type instanceof GeneratedType
- && !(type instanceof GeneratedTransferObject)) {
- typesToGenerate.add((GeneratedType) type);
- }
-
- if (type instanceof GeneratedTransferObject) {
- tosToGenerate.add((GeneratedTransferObject) type);
- }
- if (type instanceof Enumeration) {
- enumerationToGenerate.add((Enumeration) type);
- }
- }
-
- final GeneratorJavaFile generator = new GeneratorJavaFile(
- typesToGenerate, tosToGenerate, enumerationToGenerate);
+ final GeneratorJavaFile generator = new GeneratorJavaFile(new HashSet<>(types));
generator.generateToFile(new File(GENERATOR_OUTPUT_PATH));
JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
- StandardJavaFileManager fileManager = compiler.getStandardFileManager(
- null, null, null);
+ StandardJavaFileManager fileManager = compiler.getStandardFileManager(null, null, null);
List<File> filesList = getJavaFiles(new File(GENERATOR_OUTPUT_PATH));
- File current = new File(System.getProperty("user.dir"));
- File parentPath = current.getParentFile().getParentFile();
- File f = new File(parentPath,"yang/yang-binding/src/main/java/org/opendaylight/yangtools/yang/binding/DataObject.java"
- );
- filesList.add(f);
-
- Iterable<? extends JavaFileObject> compilationUnits = fileManager
- .getJavaFileObjectsFromFiles(filesList);
- Iterable<String> options = Arrays.asList(new String[]{"-d", COMPILER_OUTPUT_PATH});
- boolean compiled = compiler.getTask(null, null, null, options, null,
- compilationUnits).call();
+ Iterable<? extends JavaFileObject> compilationUnits = fileManager.getJavaFileObjectsFromFiles(filesList);
+ Iterable<String> options = Arrays.asList(new String[] { "-d", COMPILER_OUTPUT_PATH });
+ boolean compiled = compiler.getTask(null, null, null, options, null, compilationUnits).call();
assertTrue(compiled);
}
private GeneratedType createGeneratedType(String pkgName, String name) {
- GeneratedTypeBuilder builder = new GeneratedTypeBuilderImpl(pkgName,
- name);
+ GeneratedTypeBuilder builder = new GeneratedTypeBuilderImpl(pkgName, name);
+ builder.addImplementsType(BindingTypes.DATA_OBJECT);
return builder.toInstance();
}
/**
* Search recursively given directory for *.java files.
*
- * @param directory directory to search
+ * @param directory
+ * directory to search
* @return List of java files found
*/
private List<File> getJavaFiles(File directory) {