package ensemble.search;

import ensemble.model.SampleInfo;
import java.io.*;
import junit.framework.Assert;
import org.junit.Test;

/**
 *
 * @author sg155630
 */
public class SampleInfoTest {

    private static File root = new File("src/ensemble/samples");

    @Test
    public void testSampleInfo() throws Exception {
        Assert.assertTrue(traverseSamples(root));
    }

    private boolean traverseSamples(File file) throws FileNotFoundException, IOException {
        boolean result = true;
        if (file.isDirectory()) {
            String[] files = file.list();
            if (files != null) {
                for (String fileName : files) {
                    result &= traverseSamples(new File(file, fileName));
                }
            }
        } else if (file.getName().toLowerCase().endsWith(".java")) {
            // read file contents into a string
            StringBuilder contentBuilder = new StringBuilder();
            BufferedReader br = new BufferedReader(new FileReader(file));
            try {
                String line;
                while ((line = br.readLine()) != null) {
                    contentBuilder.append(line);
                    contentBuilder.append('\n');
                }
            } finally {
                br.close();
            }
            final String fileContent = contentBuilder.toString();
            String sourceFileUrl = file.toURI().toString();
            String unqualifiedClassName = sourceFileUrl.substring(sourceFileUrl.lastIndexOf('/') + 1, sourceFileUrl.length() - 5);
            SampleInfo sampleInfo = new SampleInfo(sourceFileUrl, unqualifiedClassName, fileContent);

            for (String relPath : sampleInfo.getRelatesSamplePaths()) {
                File f = new File(root, relPath + "Sample.java");
                File f2 = new File(root, relPath + ".java");
                boolean exists = f.exists() || f2.exists();
                result &= exists;
                if (!exists) {
                    System.err.println(relPath + " not found for '" + sampleInfo.getName() + "'");
                }
            }

        }
        return result;
    }
}