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.assertEquals;
11 import static org.junit.Assert.assertTrue;
14 import java.io.IOException;
15 import java.util.ArrayList;
16 import java.util.Arrays;
17 import java.util.Collections;
18 import java.util.HashSet;
19 import java.util.List;
22 import javax.tools.JavaCompiler;
23 import javax.tools.JavaFileObject;
24 import javax.tools.StandardJavaFileManager;
25 import javax.tools.ToolProvider;
27 import org.junit.After;
28 import org.junit.Before;
29 import org.junit.Test;
30 import org.opendaylight.controller.binding.generator.util.generated.type.builder.GeneratedTypeBuilderImpl;
31 import org.opendaylight.controller.sal.binding.generator.api.BindingGenerator;
32 import org.opendaylight.controller.sal.binding.generator.impl.BindingGeneratorImpl;
33 import org.opendaylight.controller.sal.binding.model.api.GeneratedTransferObject;
34 import org.opendaylight.controller.sal.binding.model.api.GeneratedType;
35 import org.opendaylight.controller.sal.binding.model.api.Type;
36 import org.opendaylight.controller.sal.binding.model.api.type.builder.GeneratedTypeBuilder;
37 import org.opendaylight.controller.sal.java.api.generator.GeneratorJavaFile;
38 import org.opendaylight.controller.sal.java.api.generator.InterfaceGenerator;
39 import org.opendaylight.controller.yang.model.api.Module;
40 import org.opendaylight.controller.yang.model.api.SchemaContext;
41 import org.opendaylight.controller.yang.model.parser.impl.YangModelParserImpl;
43 public class GeneratorJavaFileTest {
44 private static final String FS = File.separator;
45 private static final String PATH = "test-dir";
46 private final File testDir = new File(PATH);
48 private static final String GENERATOR_OUTPUT_PATH = "src/test/resources/src";
49 private static final File GENERATOR_OUTPUT = new File(GENERATOR_OUTPUT_PATH);
50 private static final String COMPILER_OUTPUT_PATH = "src/test/resources/bin";
51 private static final File COMPILER_OUTPUT = new File(COMPILER_OUTPUT_PATH);
55 assertTrue(testDir.mkdir());
56 assertTrue(COMPILER_OUTPUT.mkdirs());
57 assertTrue(GENERATOR_OUTPUT.mkdirs());
61 public void cleanUp() {
62 deleteTestDir(testDir);
63 deleteTestDir(COMPILER_OUTPUT);
64 deleteTestDir(GENERATOR_OUTPUT);
68 public void test() throws IOException {
69 final Set<GeneratedType> types = new HashSet<GeneratedType>();
70 GeneratedType t1 = createGeneratedType(
71 "org.opendaylight.controller.gen", "Type1");
72 GeneratedType t2 = createGeneratedType(
73 "org.opendaylight.controller.gen", "Type2");
74 GeneratedType t3 = createGeneratedType(
75 "org.opendaylight.controller.gen", "Type3");
79 GeneratorJavaFile generator = new GeneratorJavaFile(
80 new InterfaceGenerator(), types);
81 generator.generateToFile(PATH);
83 String[] files = new File(PATH + FS + "org" + FS + "opendaylight" + FS + "controller" + FS + "gen").list();
84 List<String> filesList = Arrays.asList(files);
86 assertEquals(3, files.length);
87 assertTrue(filesList.contains("Type1.java"));
88 assertTrue(filesList.contains("Type2.java"));
89 assertTrue(filesList.contains("Type3.java"));
93 public void compilationTest() throws Exception {
94 final YangModelParserImpl parser = new YangModelParserImpl();
95 final BindingGenerator bindingGenerator = new BindingGeneratorImpl();
97 final File sourcesDir = new File("src/test/resources/yang");
98 final List<File> sourceFiles = new ArrayList<File>();
99 final File[] fileArray = sourcesDir.listFiles();
101 for (int i = 0; i < fileArray.length; ++i) {
102 sourceFiles.add(fileArray[i]);
105 // String[] sourcesDirPaths = new String[sourceFiles.length];
106 // for (int i = 0; i < sourceFiles.length; i++) {
107 // sourcesDirPaths[i] = sourceFiles[i].getAbsolutePath();
109 final Set<Module> modulesToBuild = parser
110 .parseYangModels(sourceFiles);
112 final SchemaContext context = parser
113 .resolveSchemaContext(modulesToBuild);
114 final List<Type> types = bindingGenerator.generateTypes(context);
115 final Set<GeneratedType> typesToGenerate = new HashSet<GeneratedType>();
116 final Set<GeneratedTransferObject> tosToGenerate = new HashSet<GeneratedTransferObject>();
117 for (Type type : types) {
118 if (type instanceof GeneratedType && !(type instanceof GeneratedTransferObject)) {
119 typesToGenerate.add((GeneratedType) type);
122 if (type instanceof GeneratedTransferObject) {
123 tosToGenerate.add((GeneratedTransferObject) type);
127 final GeneratorJavaFile generator = new GeneratorJavaFile(
128 typesToGenerate, tosToGenerate);
129 generator.generateToFile(GENERATOR_OUTPUT_PATH);
131 JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
132 StandardJavaFileManager fileManager = compiler.getStandardFileManager(
135 List<File> filesList = getJavaFiles(new File(GENERATOR_OUTPUT_PATH));
137 Iterable<? extends JavaFileObject> compilationUnits = fileManager
138 .getJavaFileObjectsFromFiles(filesList);
139 Iterable<String> options = Arrays.asList(new String[] { "-d",
140 COMPILER_OUTPUT_PATH });
141 boolean compiled = compiler.getTask(null, null, null, options, null,
142 compilationUnits).call();
143 assertTrue(compiled);
146 private GeneratedType createGeneratedType(String pkgName, String name) {
147 GeneratedTypeBuilder builder = new GeneratedTypeBuilderImpl(pkgName,
149 return builder.toInstance();
152 private void deleteTestDir(File file) {
153 if (file.isDirectory()) {
154 for (File f : file.listFiles()) {
158 if (!file.delete()) {
159 throw new RuntimeException("Failed to clean up after test");
164 * Search recursively given directory for *.java files.
167 * directory to search
168 * @return List of java files found
170 private List<File> getJavaFiles(File directory) {
171 List<File> result = new ArrayList<File>();
172 for (File file : directory.listFiles()) {
173 if (file.isDirectory()) {
174 result.addAll(getJavaFiles(file));
176 String absPath = file.getAbsolutePath();
177 if (absPath.endsWith(".java")) {