sft.environment.TargetFolder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of SimpleFunctionalTest Show documentation
A JUnit extension to easily adopt functional testing and acceptance testing
There is a newer version: 1.9
/*******************************************************************************
 * Copyright (c) 2013, 2014 Sylvain Lézier.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *    Sylvain Lézier - initial implementation
 *******************************************************************************/
package sft.environment;

import sft.environment.ResourceFolder;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.URISyntaxException;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

public class TargetFolder extends ResourceFolder {
    public TargetFolder(String toProjectPath, String path) {
        super(toProjectPath, path);
    }

    public File createFileFromClass(Class aClass, String extension) {
        String filePath = getFilePath(aClass, extension);
        makeDir(filePath);
        return getFile(filePath);
    }

    public List copyFromResources(String fileName) throws IOException {
        try {
            File resourceFile = new File(this.getClass().getClassLoader().getResource(fileName).toURI());
            File targetDirectory = ensureExists();
            List paths = copyFromResources(targetDirectory, resourceFile, "");
            Collections.sort(paths);
            return paths;
        } catch (URISyntaxException e) {
            throw new RuntimeException(e);
        }
    }

    private List copyFromResources(File targetDirectory, File sourceFile,String relativePath) throws IOException {
        ArrayList filesCopied = new ArrayList();
        if (sourceFile.isDirectory()) {
            File newTargetDirectory = new File(targetDirectory, sourceFile.getName());
            if (!newTargetDirectory.exists()) {
                newTargetDirectory.mkdir();
            }
            for (File innerFile : sourceFile.listFiles()) {
                filesCopied.addAll(copyFromResources(newTargetDirectory, innerFile,relativePath+newTargetDirectory.getName()+"/"));
            }
        } else {
            String fileToInclude = copyFileFromResources(targetDirectory, sourceFile,relativePath);
            filesCopied.add(fileToInclude);
        }
        return filesCopied;
    }

    private String copyFileFromResources(File targetDirectory, File sourceFile,String relativePath) throws IOException {
        InputStream sourceAsStream = null;
        OutputStream targetAsStream = null;
        try {
            sourceAsStream = new FileInputStream(sourceFile);
            File targetFile = new File(targetDirectory, sourceFile.getName());
            targetAsStream = new FileOutputStream(targetFile);
            int readBytes;
            byte[] buffer = new byte[4096];
            while ((readBytes = sourceAsStream.read(buffer)) > 0) {
                targetAsStream.write(buffer, 0, readBytes);
            }
            return relativePath + targetFile.getName();
        } finally {
            if (sourceAsStream != null) {
                sourceAsStream.close();
            }
            if (targetAsStream != null) {
                targetAsStream.close();
            }
        }
    }

    private void makeDir(String path) {
        File parentDirectory = ensureExists();
        for (String file : path.split("/")) {
            if (!file.contains(".")) {
                parentDirectory = new File(parentDirectory, file);
                if (!parentDirectory.exists()) {
                    parentDirectory.mkdir();
                }
            }
        }
    }

    private File ensureExists() {
        File targetDirectory = new File(getResourceFolder() + path);
        if (!targetDirectory.exists()) {
            targetDirectory.mkdir();
        }
        return targetDirectory;
    }
}