Convert DocumentationGenerator to FileGenerator

[mdsal.git] / binding / maven-sal-api-gen-plugin / src / test / java / org / opendaylight / mdsal / binding / yang / wadl / generator / WadlGenTest.java

diff --git a/binding/maven-sal-api-gen-plugin/src/test/java/org/opendaylight/mdsal/binding/yang/wadl/generator/WadlGenTest.java b/binding/maven-sal-api-gen-plugin/src/test/java/org/opendaylight/mdsal/binding/yang/wadl/generator/WadlGenTest.java
index f3aa38301a6efb803da499421034d9be1edcca02..97758b1fb2b331e574480c160268f1e86d4bc695 100644 (file)
--- a/binding/maven-sal-api-gen-plugin/src/test/java/org/opendaylight/mdsal/binding/yang/wadl/generator/WadlGenTest.java
+++ b/binding/maven-sal-api-gen-plugin/src/test/java/org/opendaylight/mdsal/binding/yang/wadl/generator/WadlGenTest.java
@@ -8,99 +8,25 @@
 package org.opendaylight.mdsal.binding.yang.wadl.generator;
 
 import static org.junit.Assert.assertEquals;
-import static org.junit.Assert.assertFalse;
-import static org.junit.Assert.assertTrue;
 
-import java.io.File;
-import java.io.FileNotFoundException;
-import java.net.URI;
-import java.util.ArrayList;
-import java.util.Arrays;
-import java.util.Collection;
-import java.util.List;
+import com.google.common.collect.Table;
 import java.util.Optional;
 import java.util.Set;
-import org.junit.After;
-import org.junit.Before;
 import org.junit.Test;
+import org.opendaylight.yangtools.plugin.generator.api.GeneratedFile;
+import org.opendaylight.yangtools.plugin.generator.api.GeneratedFilePath;
+import org.opendaylight.yangtools.plugin.generator.api.GeneratedFileType;
 import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
 import org.opendaylight.yangtools.yang.test.util.YangParserTestUtils;
-import org.sonatype.plexus.build.incremental.DefaultBuildContext;
 
 public class WadlGenTest {
-
-    private static final String FS = File.separator;
-    private static final String TEST_PATH = "target" + FS + "test" + FS + "site";
-    private static final File GENERATOR_OUTPUT_DIR = new File(TEST_PATH);
-
-    @Before
-    public void init() {
-        if (GENERATOR_OUTPUT_DIR.exists()) {
-            deleteTestDir(GENERATOR_OUTPUT_DIR);
-        }
-        assertTrue(GENERATOR_OUTPUT_DIR.mkdirs());
-    }
-
-    @After
-    public void cleanUp() {
-        if (GENERATOR_OUTPUT_DIR.exists()) {
-            deleteTestDir(GENERATOR_OUTPUT_DIR);
-        }
-    }
-
     @Test
-    public void testListGeneration() throws Exception {
-        final List<File> sourceFiles = getSourceFiles("/wadl-gen");
-        final EffectiveModelContext context = YangParserTestUtils.parseYangFiles(sourceFiles);
+    public void testListGeneration() {
         final WadlGenerator generator = new WadlGenerator();
-        generator.setBuildContext(new DefaultBuildContext());
-        Collection<File> generatedWadlFiles = generator.generateSources(context, GENERATOR_OUTPUT_DIR,
+        final EffectiveModelContext context = YangParserTestUtils.parseYangResourceDirectory("/wadl-gen");
+        Table<GeneratedFileType, GeneratedFilePath, GeneratedFile> generatedWadlFiles = generator.generateFiles(context,
             Set.copyOf(context.getModules()), (module, representation) -> Optional.empty());
         assertEquals(3, generatedWadlFiles.size());
-        generatedWadlFiles.forEach(file -> assertTrue(file.exists()));
-    }
-
-    @Test
-    public void testListGenerationWithoutPath() throws Exception {
-        final List<File> sourceFiles = getSourceFiles("/wadl-gen");
-        final EffectiveModelContext context = YangParserTestUtils.parseYangFiles(sourceFiles);
-        final WadlGenerator generator = new WadlGenerator();
-        generator.setBuildContext(new DefaultBuildContext());
-        Collection<File> generatedWadlFiles = generator.generateSources(context, null, Set.copyOf(context.getModules()),
-            (module, representation) -> Optional.empty());
-        assertEquals(3, generatedWadlFiles.size());
-        generatedWadlFiles.forEach(file -> {
-            deleteTestDir(file);
-            assertFalse(file.exists());
-        });
-    }
-
-    private static List<File> getSourceFiles(final String path) throws Exception {
-        final URI resPath = WadlGenTest.class.getResource(path).toURI();
-        final File sourcesDir = new File(resPath);
-        if (!sourcesDir.exists()) {
-            throw new FileNotFoundException("Testing files were not found(" + sourcesDir.getName() + ")");
-        }
-        final List<File> sourceFiles = new ArrayList<>();
-        final File[] fileArray = sourcesDir.listFiles();
-        if (fileArray == null) {
-            throw new IllegalArgumentException("Unable to locate files in " + sourcesDir);
-        }
-        sourceFiles.addAll(Arrays.asList(fileArray));
-        return sourceFiles;
-    }
-
-    private static void deleteTestDir(final File file) {
-        if (file.isDirectory()) {
-            File[] filesToDelete = file.listFiles();
-            if (filesToDelete != null) {
-                for (File f : filesToDelete) {
-                    deleteTestDir(f);
-                }
-            }
-        }
-        if (!file.delete()) {
-            throw new RuntimeException("Failed to clean up after test");
-        }
+        // TODO: more asserts
     }
 }