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