code-generator/binding-java-api-generator/src/test/java/org/opendaylight/yangtools/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.yangtools.sal.java.api.generator.test;
   9
  10 import static org.junit.Assert.assertTrue;
  11
  12 import java.io.File;
  13 import java.io.IOException;
  14 import java.util.ArrayList;
  15 import java.util.Arrays;
  16 import java.util.HashSet;
  17 import java.util.List;
  18 import java.util.Set;
  19
  20 import javax.tools.JavaCompiler;
  21 import javax.tools.JavaFileObject;
  22 import javax.tools.StandardJavaFileManager;
  23 import javax.tools.ToolProvider;
  24
  25 import org.junit.After;
  26 import org.junit.Before;
  27 import org.junit.Test;
  28 import org.opendaylight.yangtools.binding.generator.util.BindingTypes;
  29 import org.opendaylight.yangtools.binding.generator.util.Types;
  30 import org.opendaylight.yangtools.binding.generator.util.generated.type.builder.GeneratedTypeBuilderImpl;
  31 import org.opendaylight.yangtools.sal.binding.generator.api.BindingGenerator;
  32 import org.opendaylight.yangtools.sal.binding.generator.impl.BindingGeneratorImpl;
  33 import org.opendaylight.yangtools.sal.binding.model.api.GeneratedType;
  34 import org.opendaylight.yangtools.sal.binding.model.api.Type;
  35 import org.opendaylight.yangtools.sal.binding.model.api.type.builder.GeneratedTypeBuilder;
  36 import org.opendaylight.yangtools.sal.java.api.generator.GeneratorJavaFile;
  37 import org.opendaylight.yangtools.yang.model.api.Module;
  38 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
  39 import org.opendaylight.yangtools.yang.parser.impl.YangParserImpl;
  40
  41 public class GeneratorJavaFileTest {
  42     private static final String FS = File.separator;
  43     private static final String PATH = "target/test/test-dir";
  44     private final File testDir = new File(PATH);
  45
  46     private static final String GENERATOR_OUTPUT_PATH = "target/test/src";
  47     private static final File GENERATOR_OUTPUT = new File(GENERATOR_OUTPUT_PATH);
  48     private static final String COMPILER_OUTPUT_PATH = "target/test/bin";
  49     private static final File COMPILER_OUTPUT = new File(COMPILER_OUTPUT_PATH);
  50
  51     @Before
  52     public void init() {
  53         assertTrue(testDir.mkdirs());
  54         assertTrue(COMPILER_OUTPUT.mkdirs());
  55         assertTrue(GENERATOR_OUTPUT.mkdirs());
  56     }
  57
  58     @After
  59     public void cleanUp() {
  60         if (testDir.exists()) {
  61             deleteTestDir(testDir);
  62         }
  63         if (COMPILER_OUTPUT.exists()) {
  64             deleteTestDir(COMPILER_OUTPUT);
  65         }
  66         if (GENERATOR_OUTPUT.exists()) {
  67             deleteTestDir(GENERATOR_OUTPUT);
  68         }
  69     }
  70
  71     @Test
  72     public void test() throws IOException {
  73         final Set<GeneratedType> types = new HashSet<GeneratedType>();
  74         GeneratedType t1 = createGeneratedType("org.opendaylight.controller.gen", "Type1");
  75         GeneratedType t2 = createGeneratedType("org.opendaylight.controller.gen", "Type2");
  76         GeneratedType t3 = createGeneratedType("org.opendaylight.controller.gen", "Type3");
  77         types.add(t1);
  78         types.add(t2);
  79         types.add(t3);
  80         GeneratedTypeBuilder gtb = new GeneratedTypeBuilderImpl("org.opendaylight.controller.gen", "Type4");
  81         gtb.addImplementsType(Types.augmentableTypeFor(gtb));
  82         types.add(gtb.toInstance());
  83         GeneratorJavaFile generator = new GeneratorJavaFile(types);
  84         generator.generateToFile(new File(PATH));
  85
  86         String[] files = new File(PATH + FS + "org" + FS + "opendaylight" + FS + "controller" + FS + "gen").list();
  87         List<String> filesList = Arrays.asList(files);
  88
  89         // assertEquals(5, files.length);
  90         assertTrue(filesList.contains("Type1.java"));
  91         assertTrue(filesList.contains("Type2.java"));
  92         assertTrue(filesList.contains("Type3.java"));
  93         assertTrue(filesList.contains("Type4.java"));
  94         assertTrue(filesList.contains("Type4Builder.java"));
  95     }
  96
  97     @Test
  98     public void compilationTest() throws Exception {
  99         final YangParserImpl parser = new YangParserImpl();
 100         final BindingGenerator bindingGenerator = new BindingGeneratorImpl();
 101
 102         final String resPath = getClass().getResource("/yang").getPath();
 103         final File sourcesDir = new File(resPath);
 104         final List<File> sourceFiles = new ArrayList<File>();
 105         final File[] fileArray = sourcesDir.listFiles();
 106
 107         for (int i = 0; i < fileArray.length; ++i) {
 108             sourceFiles.add(fileArray[i]);
 109         }
 110
 111         final Set<Module> modulesToBuild = parser.parseYangModels(sourceFiles);
 112
 113         final SchemaContext context = parser.resolveSchemaContext(modulesToBuild);
 114         final List<Type> types = bindingGenerator.generateTypes(context);
 115         final GeneratorJavaFile generator = new GeneratorJavaFile(new HashSet<>(types));
 116         generator.generateToFile(new File(GENERATOR_OUTPUT_PATH));
 117
 118         JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
 119         StandardJavaFileManager fileManager = compiler.getStandardFileManager(null, null, null);
 120
 121         List<File> filesList = getJavaFiles(new File(GENERATOR_OUTPUT_PATH));
 122         File current = new File(System.getProperty("user.dir"));
 123         File parentPath = current.getParentFile().getParentFile();
 124         File f = new File(parentPath,
 125                 "yang/yang-binding/src/main/java/org/opendaylight/yangtools/yang/binding/DataObject.java");
 126         filesList.add(f);
 127
 128         Iterable<? extends JavaFileObject> compilationUnits = fileManager.getJavaFileObjectsFromFiles(filesList);
 129         Iterable<String> options = Arrays.asList(new String[] { "-d", COMPILER_OUTPUT_PATH });
 130         boolean compiled = compiler.getTask(null, null, null, options, null, compilationUnits).call();
 131         assertTrue(compiled);
 132     }
 133
 134     private GeneratedType createGeneratedType(String pkgName, String name) {
 135         GeneratedTypeBuilder builder = new GeneratedTypeBuilderImpl(pkgName, name);
 136         builder.addImplementsType(BindingTypes.DATA_OBJECT);
 137         return builder.toInstance();
 138     }
 139
 140     private void deleteTestDir(File file) {
 141         if (file.isDirectory()) {
 142             for (File f : file.listFiles()) {
 143                 deleteTestDir(f);
 144             }
 145         }
 146         if (!file.delete()) {
 147             throw new RuntimeException("Failed to clean up after test");
 148         }
 149     }
 150
 151     /**
 152      * Search recursively given directory for *.java files.
 153      *
 154      * @param directory
 155      *            directory to search
 156      * @return List of java files found
 157      */
 158     private List<File> getJavaFiles(File directory) {
 159         List<File> result = new ArrayList<File>();
 160         for (File file : directory.listFiles()) {
 161             if (file.isDirectory()) {
 162                 result.addAll(getJavaFiles(file));
 163             } else {
 164                 String absPath = file.getAbsolutePath();
 165                 if (absPath.endsWith(".java")) {
 166                     result.add(file);
 167                 }
 168             }
 169         }
 170         return result;
 171     }
 172 }