org.ebayopensource.turmeric.tools.AbstractCodegenTestCase.java Source code

Introduction

Here is the source code for org.ebayopensource.turmeric.tools.AbstractCodegenTestCase.java

Source

/*******************************************************************************
 * Copyright (c) 2006-2010 eBay Inc. All Rights Reserved.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *******************************************************************************/
package org.ebayopensource.turmeric.tools;

import static org.hamcrest.Matchers.*;

import java.io.File;
import java.io.FileReader;
import java.io.FileWriter;
import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.util.Collections;
import java.util.Enumeration;
import java.util.List;
import java.util.Properties;

import org.apache.commons.io.IOUtils;
import org.ebayopensource.turmeric.junit.AbstractTurmericTestCase;
import org.ebayopensource.turmeric.junit.logging.UKernelLoggingUtils;
import org.ebayopensource.turmeric.junit.rules.TestingDir;
import org.ebayopensource.turmeric.junit.utils.MavenTestingUtils;
import org.ebayopensource.turmeric.tools.codegen.NonInteractiveCodeGen;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Rule;

public class AbstractCodegenTestCase extends AbstractTurmericTestCase {
    @Rule
    public TestingDir testingdir = new TestingDir();

    @Before
    public void initLogs() {
        UKernelLoggingUtils.initTesting(MavenTestingUtils.getTargetDir());
    }

    /**
     * Load {@link Properties} from disk.
     * 
     * @param propFile
     *            the properties file to load.
     * @return the loaded properties (or empty properties if file does not exist)
     * @throws IOException
     *             if unable to read existing file
     */
    protected Properties loadProperties(File propFile) throws IOException {
        Properties props = new Properties();
        if (!propFile.exists()) {
            return props;
        }
        FileReader reader = null;
        try {
            reader = new FileReader(propFile);
            props.load(reader);
        } finally {
            IOUtils.closeQuietly(reader);
        }
        return props;
    }

    /**
     * Load {@link Properties} from classpath.
     * 
     * @param resource
     *            the properties resource to load.
     * @return the loaded properties (or empty properties if resource not found)
     * @throws IOException
     *             if unable to read resource
     */
    protected Properties loadProperties(String resource) throws IOException {
        Properties props = new Properties();
        Enumeration<URL> enurls = this.getClass().getClassLoader().getResources(resource);
        if (!enurls.hasMoreElements()) {
            return props;
        }
        List<URL> urls = Collections.list(enurls);
        Assert.assertThat("Encountered multiple hits for resource: " + resource, urls.size(), is(1));
        InputStream stream = null;
        try {
            stream = urls.get(0).openStream();
            props.load(stream);
        } finally {
            IOUtils.closeQuietly(stream);
        }
        return props;
    }

    /**
     * Write {@link Properties} to disk.
     * 
     * @param propFile
     *            the file to write to
     * @param props
     *            the properties to write.
     * @throws IOException
     *             if unable to create the file
     */
    protected void writeProperties(File propFile, Properties props) throws IOException {
        FileWriter writer = null;
        try {
            writer = new FileWriter(propFile);
            props.store(writer, "Created by testcase");
        } finally {
            IOUtils.closeQuietly(writer);
        }
    }

    @Before
    public final void preventBadTests() {
        mavenTestingRules.setStrictReadPaths(true);
        mavenTestingRules.setStrictWritePaths(true);
        mavenTestingRules.setFailOnViolation(false);
    }

    public void performDirectCodeGen(String args[], File binDir) throws Exception {
        NonInteractiveCodeGen gen = new NonInteractiveCodeGen();
        MavenTestingUtils.ensureDirExists(binDir);
        gen.addExtraClassPath(binDir);
        gen.addJavaToolsClassPath();
        gen.execute(args);
    }

    public void performDirectCodeGen(String args[]) throws Exception {
        performDirectCodeGen(args, testingdir.getFile("bin"));
    }
}