opendaylight/sal/yang-prototype/code-generator/maven-yang-plugin/src/test/java/org/opendaylight/controller/yang2sources/plugin/GenerateSourcesTest.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.yang2sources.plugin;
   9
  10 import static org.hamcrest.core.Is.is;
  11 import static org.junit.Assert.*;
  12 import static org.mockito.Mockito.doReturn;
  13
  14 import java.io.File;
  15 import java.io.IOException;
  16 import java.util.Collection;
  17 import java.util.Map;
  18 import java.util.Set;
  19
  20 import org.apache.maven.plugin.logging.Log;
  21 import org.apache.maven.project.MavenProject;
  22 import org.junit.Before;
  23 import org.junit.Ignore;
  24 import org.junit.Test;
  25 import org.mockito.Mock;
  26 import org.mockito.MockitoAnnotations;
  27 import org.opendaylight.controller.yang.model.api.Module;
  28 import org.opendaylight.controller.yang.model.api.SchemaContext;
  29 import org.opendaylight.controller.yang2sources.plugin.ConfigArg.CodeGeneratorArg;
  30 import org.opendaylight.controller.yang2sources.spi.CodeGenerator;
  31
  32 import com.google.common.collect.Lists;
  33
  34 public class GenerateSourcesTest {
  35
  36     private String yang;
  37     private YangToSourcesMojo mojo;
  38     private File outDir;
  39     @Mock
  40     private MavenProject project;
  41
  42     @Before
  43     public void setUp() {
  44         MockitoAnnotations.initMocks(this);
  45
  46         yang = new File(getClass().getResource("/mock.yang").getFile())
  47                 .getParent();
  48         outDir = new File("/outputDir");
  49         mojo = new YangToSourcesMojo(
  50                 new CodeGeneratorArg[] { new CodeGeneratorArg(
  51                         GeneratorMock.class.getName(), "outputDir") }, yang);
  52         doReturn(new File("")).when(project).getBasedir();
  53         mojo.project = project;
  54     }
  55
  56     @Ignore
  57     @Test
  58     public void test() throws Exception {
  59         mojo.execute();
  60         assertThat(GeneratorMock.called, is(1));
  61         assertThat(GeneratorMock.outputDir, is(outDir));
  62         assertNotNull(GeneratorMock.log);
  63         assertTrue(GeneratorMock.additionalCfg.isEmpty());
  64     }
  65
  66     public static class GeneratorMock implements CodeGenerator {
  67
  68         private static int called = 0;
  69         private static File outputDir;
  70         private static Log log;
  71         private static Map<String, String> additionalCfg;
  72
  73         @Override
  74         public Collection<File> generateSources(SchemaContext context,
  75                 File outputBaseDir, Set<Module> currentModules,
  76                 File projectBaseDir) throws IOException {
  77             called++;
  78             outputDir = outputBaseDir;
  79             return Lists.newArrayList();
  80         }
  81
  82         @Override
  83         public void setLog(Log log) {
  84             this.log = log;
  85         }
  86
  87         @Override
  88         public void setAdditionalConfig(
  89                 Map<String, String> additionalConfiguration) {
  90             this.additionalCfg = additionalConfiguration;
  91         }
  92     }
  93
  94 }