2 * Copyright (c) 2013 Cisco Systems, Inc. and others. All rights reserved.
4 * This program and the accompanying materials are made available under the
5 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6 * and is available at http://www.eclipse.org/legal/epl-v10.html
8 package org.opendaylight.controller.sal.java.api.generator.test;
10 import static org.junit.Assert.*;
13 import java.util.ArrayList;
14 import java.util.Arrays;
15 import java.util.HashSet;
16 import java.util.List;
19 import javax.tools.JavaCompiler;
20 import javax.tools.JavaFileObject;
21 import javax.tools.StandardJavaFileManager;
22 import javax.tools.ToolProvider;
24 import org.junit.After;
25 import org.junit.Before;
26 import org.junit.Test;
27 import org.opendaylight.controller.binding.generator.util.generated.type.builder.GeneratedTypeBuilderImpl;
28 import org.opendaylight.controller.sal.binding.generator.api.BindingGenerator;
29 import org.opendaylight.controller.sal.binding.generator.impl.BindingGeneratorImpl;
30 import org.opendaylight.controller.sal.binding.model.api.GeneratedTransferObject;
31 import org.opendaylight.controller.sal.binding.model.api.GeneratedType;
32 import org.opendaylight.controller.sal.binding.model.api.Type;
33 import org.opendaylight.controller.sal.binding.model.api.type.builder.GeneratedTypeBuilder;
34 import org.opendaylight.controller.sal.java.api.generator.GeneratorJavaFile;
35 import org.opendaylight.controller.sal.java.api.generator.InterfaceGenerator;
36 import org.opendaylight.controller.yang.model.api.Module;
37 import org.opendaylight.controller.yang.model.api.SchemaContext;
38 import org.opendaylight.controller.yang.model.parser.impl.YangModelParserImpl;
40 public class GeneratorJavaFileTest {
41 private static final String FS = File.separator;
42 private static final String PATH = "test-dir";
43 private final File testDir = new File(PATH);
45 private static final String GENERATOR_OUTPUT_PATH = "src/test/resources/src";
46 private static final File GENERATOR_OUTPUT = new File(GENERATOR_OUTPUT_PATH);
47 private static final String COMPILER_OUTPUT_PATH = "src/test/resources/bin";
48 private static final File COMPILER_OUTPUT = new File(COMPILER_OUTPUT_PATH);
52 assertTrue(testDir.mkdir());
53 assertTrue(COMPILER_OUTPUT.mkdirs());
54 assertTrue(GENERATOR_OUTPUT.mkdirs());
58 public void cleanUp() {
59 deleteTestDir(testDir);
60 deleteTestDir(COMPILER_OUTPUT);
61 deleteTestDir(GENERATOR_OUTPUT);
66 final Set<GeneratedType> types = new HashSet<GeneratedType>();
67 GeneratedType t1 = createGeneratedType(
68 "org.opendaylight.controller.gen", "Type1");
69 GeneratedType t2 = createGeneratedType(
70 "org.opendaylight.controller.gen", "Type2");
71 GeneratedType t3 = createGeneratedType(
72 "org.opendaylight.controller.gen", "Type3");
76 GeneratorJavaFile generator = new GeneratorJavaFile(
77 new InterfaceGenerator(), types);
78 generator.generateToFile(PATH);
80 // path: test-dir/com/cisco/yang
81 String[] files = new File(PATH + FS + "org" + FS + "opendaylight" + FS + "controller" + FS + "gen").list();
82 List<String> filesList = Arrays.asList(files);
84 assertEquals(3, files.length);
85 assertTrue(filesList.contains("Type1.java"));
86 assertTrue(filesList.contains("Type2.java"));
87 assertTrue(filesList.contains("Type3.java"));
91 public void compilationTest() throws Exception {
92 final YangModelParserImpl parser = new YangModelParserImpl();
93 final BindingGenerator bindingGenerator = new BindingGeneratorImpl();
95 File sourcesDir = new File("src/test/resources/yang");
96 File[] sourceFiles = sourcesDir.listFiles();
97 String[] sourcesDirPaths = new String[sourceFiles.length];
98 for (int i = 0; i < sourceFiles.length; i++) {
99 sourcesDirPaths[i] = sourceFiles[i].getAbsolutePath();
101 final Set<Module> modulesToBuild = parser
102 .parseYangModels(sourcesDirPaths);
104 final SchemaContext context = parser
105 .resolveSchemaContext(modulesToBuild);
106 final List<Type> types = bindingGenerator.generateTypes(context);
107 final Set<GeneratedType> typesToGenerate = new HashSet<GeneratedType>();
108 final Set<GeneratedTransferObject> tosToGenerate = new HashSet<GeneratedTransferObject>();
109 for (Type type : types) {
110 if (type instanceof GeneratedType) {
111 typesToGenerate.add((GeneratedType) type);
114 if (type instanceof GeneratedTransferObject) {
115 tosToGenerate.add((GeneratedTransferObject) type);
119 final GeneratorJavaFile generator = new GeneratorJavaFile(
120 typesToGenerate, tosToGenerate);
121 generator.generateToFile(GENERATOR_OUTPUT_PATH);
123 JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
124 StandardJavaFileManager fileManager = compiler.getStandardFileManager(
127 List<File> filesList = getJavaFiles(new File(GENERATOR_OUTPUT_PATH));
129 Iterable<? extends JavaFileObject> compilationUnits = fileManager
130 .getJavaFileObjectsFromFiles(filesList);
131 Iterable<String> options = Arrays.asList(new String[] { "-d",
132 COMPILER_OUTPUT_PATH });
133 boolean compiled = compiler.getTask(null, null, null, options, null,
134 compilationUnits).call();
135 assertTrue(compiled);
138 private GeneratedType createGeneratedType(String pkgName, String name) {
139 GeneratedTypeBuilder builder = new GeneratedTypeBuilderImpl(pkgName,
141 return builder.toInstance();
144 private void deleteTestDir(File file) {
145 if (file.isDirectory()) {
146 for (File f : file.listFiles()) {
150 if (!file.delete()) {
151 throw new RuntimeException("Failed to clean up after test");
156 * Search recursively given directory for *.java files.
159 * directory to search
160 * @return List of java files found
162 private List<File> getJavaFiles(File directory) {
163 List<File> result = new ArrayList<File>();
164 for (File file : directory.listFiles()) {
165 if (file.isDirectory()) {
166 result.addAll(getJavaFiles(file));
168 String absPath = file.getAbsolutePath();
169 if (absPath.endsWith(".java")) {