opendaylight/sal/yang-prototype/code-generator/binding-java-api-generator/src/test/java/org/opendaylight/controller/sal/java/api/generator/test/GeneratorJavaFileTest.java

   1 /*
   2  * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   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
   7  */
   8 package org.opendaylight.controller.sal.java.api.generator.test;
   9
  10 import static org.junit.Assert.*;
  11
  12 import java.io.File;
  13 import java.util.ArrayList;
  14 import java.util.Arrays;
  15 import java.util.HashSet;
  16 import java.util.List;
  17 import java.util.Set;
  18
  19 import javax.tools.JavaCompiler;
  20 import javax.tools.JavaFileObject;
  21 import javax.tools.StandardJavaFileManager;
  22 import javax.tools.ToolProvider;
  23
  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;
  39
  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);
  44
  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);
  49
  50     @Before
  51     public void init() {
  52         assertTrue(testDir.mkdir());
  53         assertTrue(COMPILER_OUTPUT.mkdirs());
  54         assertTrue(GENERATOR_OUTPUT.mkdirs());
  55     }
  56
  57     @After
  58     public void cleanUp() {
  59         deleteTestDir(testDir);
  60         deleteTestDir(COMPILER_OUTPUT);
  61         deleteTestDir(GENERATOR_OUTPUT);
  62     }
  63
  64     @Test
  65     public void test() {
  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");
  73         types.add(t1);
  74         types.add(t2);
  75         types.add(t3);
  76         GeneratorJavaFile generator = new GeneratorJavaFile(
  77                 new InterfaceGenerator(), types);
  78         generator.generateToFile(PATH);
  79
  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);
  83
  84         assertEquals(3, files.length);
  85         assertTrue(filesList.contains("Type1.java"));
  86         assertTrue(filesList.contains("Type2.java"));
  87         assertTrue(filesList.contains("Type3.java"));
  88     }
  89
  90     @Test
  91     public void compilationTest() throws Exception {
  92         final YangModelParserImpl parser = new YangModelParserImpl();
  93         final BindingGenerator bindingGenerator = new BindingGeneratorImpl();
  94
  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();
 100         }
 101         final Set<Module> modulesToBuild = parser
 102                 .parseYangModels(sourcesDirPaths);
 103
 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);
 112             }
 113
 114             if (type instanceof GeneratedTransferObject) {
 115                 tosToGenerate.add((GeneratedTransferObject) type);
 116             }
 117         }
 118
 119         final GeneratorJavaFile generator = new GeneratorJavaFile(
 120                 typesToGenerate, tosToGenerate);
 121         generator.generateToFile(GENERATOR_OUTPUT_PATH);
 122
 123         JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
 124         StandardJavaFileManager fileManager = compiler.getStandardFileManager(
 125                 null, null, null);
 126
 127         List<File> filesList = getJavaFiles(new File(GENERATOR_OUTPUT_PATH));
 128
 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);
 136     }
 137
 138     private GeneratedType createGeneratedType(String pkgName, String name) {
 139         GeneratedTypeBuilder builder = new GeneratedTypeBuilderImpl(pkgName,
 140                 name);
 141         return builder.toInstance();
 142     }
 143
 144     private void deleteTestDir(File file) {
 145         if (file.isDirectory()) {
 146             for (File f : file.listFiles()) {
 147                 deleteTestDir(f);
 148             }
 149         }
 150         if (!file.delete()) {
 151             throw new RuntimeException("Failed to clean up after test");
 152         }
 153     }
 154
 155     /**
 156      * Search recursively given directory for *.java files.
 157      *
 158      * @param directory
 159      *            directory to search
 160      * @return List of java files found
 161      */
 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));
 167             } else {
 168                 String absPath = file.getAbsolutePath();
 169                 if (absPath.endsWith(".java")) {
 170                     result.add(file);
 171                 }
 172             }
 173         }
 174         return result;
 175     }
 176
 177 }