import java.awt.Component;
import java.io.*;

import static org.junit.Assert.*;

import org.netbeans.jemmy.util.PNGEncoder;

public class BaseClassSwing {

    protected String outputFolder = "";
    protected String inputFolder = "";
    protected String baselineFolder = "";

    public BaseClassSwing() {
    }

    public void testFiles() {

        try {

            /**
             * open all files
             */
            File filesToTest = new File(inputFolder);

            String[] fileList = filesToTest.list();

            //check we have some files
            if (fileList == null) {
                fail("No files in " + inputFolder);
            }

            for (String fileName : fileList) {
                test(fileName);
            }

            //Compare all output and check for changes
            if (compareImages(outputFolder, baselineFolder)) {
                fail("Current output does not match baseline.");
            }

        } catch (Exception e) {
            e.printStackTrace();
            fail(e.toString());
        }
    }

    public void getScreenshot(Component com, String filename) {
        PNGEncoder.captureScreen(com, filename, PNGEncoder.COLOR_MODE);
    }

    private boolean compareImages(String output, String baseline) {
        /**
         * ADD CODE TO GET FILES FROM FOLDER NAMES AND COMPARE THEM
         */
        return false;
    }

    public void test(String testFile) {
        //Do nothing for generic test
    }

}