- 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");
- }