org.cobol85.TestGenerator.java Source code

Introduction

Here is the source code for org.cobol85.TestGenerator.java

Source

/*
 * Copyright (C) 2016, Ulrich Wolffgang <u.wol@wwu.de>
 * All rights reserved.
 *
 * This software may be modified and distributed under the terms
 * of the BSD 3-clause license. See the LICENSE file for details.
 */

package org.cobol85;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.Arrays;

import org.apache.commons.io.FilenameUtils;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;

public class TestGenerator {

    private final static String[] cobolFileExtensions = new String[] { "cbl", "cob", "jcl", "txt", "" };

    private final static String[] directoriesExcluded = new String[] { "cics" };

    private final static String[] directoriesNonRepo = new String[] {};

    private final static File inputDirectory = new File("src/test/resources/org/cobol85");

    private final static Logger LOG = LogManager.getLogger(TestGenerator.class);

    private final static File outputDirectory = new File("src/test/java/org/cobol85");

    public static String firstToUpper(final String str) {
        return Character.toUpperCase(str.charAt(0)) + str.substring(1);
    }

    public static void generateTestClass(final File cobol85InputFile, final File outputDirectory,
            final String packageName, final boolean isNonRepoTest) throws IOException {
        if (cobol85InputFile.isFile() && !cobol85InputFile.isHidden()) {
            final String inputFilename = firstToUpper(FilenameUtils.removeExtension(cobol85InputFile.getName()));

            final File outputFile = new File(outputDirectory + "/" + inputFilename + "Test.java");

            LOG.info("Creating {}.", outputFile);
            final boolean createdNewFile = outputFile.createNewFile();

            if (createdNewFile) {
                final PrintWriter pWriter = new PrintWriter(new FileWriter(outputFile));

                final String defaultDirectoryName = "src/test/resources/org/cobol85/";
                final String cobol85InputFileName = cobol85InputFile.getPath().replace("\\", "/");

                pWriter.write("package " + packageName + ";\n");
                pWriter.write("\n");
                pWriter.write("import java.io.File;\n");
                pWriter.write("\n");
                pWriter.write("import org.junit.Test;\n");
                pWriter.write("import org.cobol85.applicationcontext.Cobol85GrammarContextFactory;\n");
                pWriter.write("import org.cobol85.runner.Cobol85ParseTestRunner;\n");
                pWriter.write("import org.cobol85.runner.impl.Cobol85ParseTestRunnerImpl;\n");
                pWriter.write("\n");
                pWriter.write("public class " + inputFilename + "Test {\n");
                pWriter.write("\n");
                pWriter.write("   @Test\n");
                pWriter.write("   public void test() throws Exception {\n");
                pWriter.write("      Cobol85GrammarContextFactory.configureDefaultApplicationContext();\n");
                pWriter.write("\n");

                if (isNonRepoTest) {
                    final String shortenedCobol85InputFileName = cobol85InputFileName.replace(defaultDirectoryName,
                            "");

                    pWriter.write(
                            "      final String testDirectoryString = System.getProperty(\"testDirectory\", \""
                                    + defaultDirectoryName + "\");\n");
                    pWriter.write("      final File inputFile = new File(testDirectoryString, \""
                            + shortenedCobol85InputFileName + "\");\n");
                } else {
                    pWriter.write("      final File inputFile = new File(\"" + cobol85InputFileName + "\");\n");
                }

                pWriter.write("      final Cobol85ParseTestRunner runner = new Cobol85ParseTestRunnerImpl();\n");
                pWriter.write("      runner.parseFile(inputFile, null);\n");
                pWriter.write("   }\n");
                pWriter.write("}");

                pWriter.flush();
                pWriter.close();
            }
        }
    }

    public static void generateTestClasses(final File inputDirectory, final File outputDirectory,
            final String packageName) throws IOException {
        final String outputDirectoryPath = outputDirectory.getPath();

        if (inputDirectory.isDirectory()) {
            // for each of the files in the directory
            for (final File inputDirectoryFile : inputDirectory.listFiles()) {
                // if the file is a Cobol85 relevant file
                if (isCobolFile(inputDirectoryFile)) {
                    final File parentDirectory = inputDirectoryFile.getParentFile();
                    final boolean isNonRepoTest = isDirectoryNonRepo(parentDirectory);
                    generateTestClass(inputDirectoryFile, outputDirectory, packageName, isNonRepoTest);
                }
                // else, if the file is a relevant directory
                else if (inputDirectoryFile.isDirectory() && !isDirectoryExcluded(inputDirectoryFile)) {
                    final File subInputDirectory = inputDirectoryFile;
                    final String subInputDirectoryName = subInputDirectory.getName();

                    if (!".".equals(subInputDirectoryName) && !"..".equals(subInputDirectoryName)) {
                        /*
                         * determine the output directory, where test classes
                         * should be placed
                         */
                        final File subOutputDirectory = new File(outputDirectoryPath + "/" + subInputDirectoryName);
                        subOutputDirectory.mkdirs();

                        // determine the package name of test classes
                        final String subPackageName = packageName + "." + subInputDirectoryName;

                        generateTestClasses(subInputDirectory, subOutputDirectory, subPackageName);
                    }
                }
            }
        }
    }

    protected static boolean isCobolFile(final File inputFile) {
        final String extension = FilenameUtils.getExtension(inputFile.getName()).toLowerCase();
        return inputFile.isFile() && Arrays.asList(cobolFileExtensions).contains(extension);
    }

    protected static boolean isDirectoryExcluded(final File directory) {
        final String directoryName = directory.getName();
        return Arrays.asList(directoriesExcluded).contains(directoryName);
    }

    protected static boolean isDirectoryNonRepo(final File directory) {
        final String directoryName = directory.getName();
        return Arrays.asList(directoriesNonRepo).contains(directoryName);
    }

    public static void main(final String[] args) throws IOException {
        generateTestClasses(inputDirectory, outputDirectory, "org.cobol85");
    }
}