package org.netbeans.lib.callgraph;

import java.io.File;

import java.io.IOException;

import java.util.ArrayList;

import java.util.List;

import org.junit.Test;

import static org.junit.Assert.*;

/**

 * Tests argument parsing.

 *

 * @author Tim Boudreau

 */

public class ArgumentsTest {

    @Test

    public void testBooleanSwitches() throws IOException {

        File tmpdir = new File(System.getProperty("java.io.tmpdir")).getCanonicalFile();

        String tmp = tmpdir.getAbsolutePath();

        Arguments args = new Arguments("-s", tmp);

        assertFalse(args.isMaven());

        assertTrue(args.isShortNames());

        assertTrue(args.isSelfReferences());

        assertTrue(args.folders().contains(new File(tmp)));

        args = new Arguments("-n", tmp);

        assertFalse(args.isShortNames());

        assertFalse(args.isSelfReferences());

        assertTrue(args.folders().contains(new File(tmp)));

        args = new Arguments("-n", "-o", tmp + File.separator + "out.txt", 

                "--exclude", "foo.bar,foo.baz", System.getProperty("java.io.tmpdir"));

        assertFalse(args.isShortNames());

        assertFalse(args.isSelfReferences());

        assertTrue("Looking for " + tmp + " but got " + args.folders().toString(), args.folders().contains(new File(tmp)));

        assertNotNull(args.methodGraphFile());

        assertEquals(new File(tmpdir, "out.txt"), args.methodGraphFile());

        assertFalse(args.excludePrefixes().isEmpty());

        assertTrue(args.excludePrefixes().contains("foo.bar"));

        assertTrue(args.excludePrefixes().contains("foo.baz"));

    }

    @Test

    public void testBuilder() throws IOException {

        File tmpdir = new File(System.getProperty("java.io.tmpdir"));

        String tmp = tmpdir.getAbsolutePath();

        CallgraphControl ctrl = Callgraph.configure().classGraphOutput(new File(tmpdir, "classes.txt"))

                .excludePrefix("foo.bar").excludePrefix("baz.quux")

                .ignoreSelfReferences().quiet().methodGraphOutput(new File(tmpdir, "methods.txt"))

                .addSourceParent(tmpdir).packageGraphOutput(new File(tmpdir, "pkgs.txt")).build();

    }

    @Test(expected = Arguments.InvalidArgumentsException.class)

    public void testNoArguments() throws IOException {

        new Arguments();

    }

    @Test(expected = Arguments.InvalidArgumentsException.class)

    public void testQuietAndNoOuput() throws IOException {

        new Arguments("-q", System.getProperty("java.io.tmpdir"));

    }

    @Test(expected = Arguments.InvalidArgumentsException.class)

    public void testNonExistentFolder() throws IOException {

        new Arguments("/" + System.currentTimeMillis());

    }

    @Test

    public void testGradleScan() throws IOException {

        File tmp = new File(System.getProperty("java.io.tmpdir"));

        File dir = new File(tmp, ArgumentsTest.class.getSimpleName() + "_" + System.currentTimeMillis());

        File project1 = new File(dir, "prj");

        File project2 = new File(dir, "prj2");

        File build1 = new File(project1, "a.gradle");

        File build2 = new File(project2, "b.gradle");

        File src1 = new File(project1, "src");

        File main1 = new File(src1, "main");

        File java1 = new File(main1, "java");

        File src2 = new File(project2, "src");

        File main2 = new File(src2, "main");

        File java2 = new File(main2, "java");

        assertTrue(java1.mkdirs());

        assertTrue(java2.mkdirs());

        build1.createNewFile();

        build2.createNewFile();

        java1 = java1.getCanonicalFile();

        java2 = java2.getCanonicalFile();

        try {

            Arguments args = new Arguments(dir.getAbsolutePath());

            assertTrue(args.hasGradleFile(project1));

            assertTrue(args.hasGradleFile(project2));

            List<String> errors = new ArrayList<>();

            args.findGradleSubfolders(errors);

            assertFalse(args.folders().isEmpty());

            assertEquals(args.folders().toString(), 2, args.folders().size());

            assertTrue(errors + "", errors.isEmpty());

            args = new Arguments("--gradle", "-s", dir.getAbsolutePath());

            assertTrue(args.isGradle());

            assertTrue(args.isShortNames());

            assertTrue(args.isSelfReferences());

            assertTrue(args.folders().contains(java1));

            assertTrue(args.folders().contains(java2));

        } finally {

            // Clean up

            assertTrue(build1.delete());

            assertTrue(build2.delete());

            assertTrue(java1.delete());

            assertTrue(java2.delete());

            assertTrue(main1.delete());

            assertTrue(main2.delete());

            assertTrue(src1.delete());

            assertTrue(src2.delete());

            assertTrue(project1.delete());

            assertTrue(project2.delete());

            assertTrue(dir.delete());

        }

    }

    @Test

    public void testMavenScan() throws IOException {

        File tmp = new File(System.getProperty("java.io.tmpdir"));

        File dir = new File(tmp, ArgumentsTest.class.getSimpleName() + "_" + System.currentTimeMillis());

        File project1 = new File(dir, "prj");

        File project2 = new File(dir, "prj2");

        File pom1 = new File(project1, "pom.xml");

        File pom2 = new File(project2, "pom.xml");

        File src1 = new File(project1, "src");

        File main1 = new File(src1, "main");

        File java1 = new File(main1, "java");

        File src2 = new File(project2, "src");

        File main2 = new File(src2, "main");

        File java2 = new File(main2, "java");

        assertTrue(java1.mkdirs());

        assertTrue(java2.mkdirs());

        pom1.createNewFile();

        pom2.createNewFile();

        java1 = java1.getCanonicalFile();

        java2 = java2.getCanonicalFile();

        try {

            Arguments args = new Arguments("--maven", "-s", dir.getAbsolutePath());

            assertTrue(args.isMaven());

            assertTrue(args.isShortNames());

            assertTrue(args.isSelfReferences());

            assertTrue(args.folders().contains(java1));

            assertTrue(args.folders().contains(java2));

        } finally {

            // Clean up

            assertTrue(pom1.delete());

            assertTrue(pom2.delete());

            assertTrue(java1.delete());

            assertTrue(java2.delete());

            assertTrue(main1.delete());

            assertTrue(main2.delete());

            assertTrue(src1.delete());

            assertTrue(src2.delete());

            assertTrue(project1.delete());

            assertTrue(project2.delete());

            assertTrue(dir.delete());

        }

    }

    @Test

    public void testAntScan() throws IOException {

        File tmp = new File(System.getProperty("java.io.tmpdir"));

        File dir = new File(tmp, ArgumentsTest.class.getSimpleName() + "_" + System.currentTimeMillis());

        File project1 = new File(dir, "prj");

        File project2 = new File(dir, "prj2");

        File pom1 = new File(project1, "build.xml");

        File pom2 = new File(project2, "build.xml");

        File java1 = new File(project1, "src");

        File java2 = new File(project2, "src");

        assertTrue(java1.mkdirs());

        assertTrue(java2.mkdirs());

        pom1.createNewFile();

        pom2.createNewFile();

        dir = dir.getCanonicalFile();

        java1 = java1.getCanonicalFile();

        java2 = java2.getCanonicalFile();

        Arguments a = new Arguments("-v", dir.getAbsolutePath());

        assertEquals(1, a.folders().size());

        assertEquals(dir, a.folders().iterator().next());

        List<String> errors = new ArrayList<>();

        a.findAntSubfolders(errors);

        assertTrue(errors.isEmpty());

        System.out.println("FLDS: ");

        for (File f : a.folders()) {

            System.out.println("  " + f.getAbsolutePath());

        }

        assertEquals(2, a.folders().size());

        assertFalse(a.folders().isEmpty());

        assertTrue(a.folders().contains(java1));

        assertTrue(a.folders().contains(java2));

        try {

            Arguments args = new Arguments("--ant", "-s", dir.getAbsolutePath());

            assertTrue(args.isAnt());

            assertTrue(args.isShortNames());

            assertTrue(args.isSelfReferences());

            assertTrue(args.folders().contains(java1));

            assertTrue(args.folders().contains(java2));

        } finally {

            // Clean up

            assertTrue(pom1.delete());

            assertTrue(pom2.delete());

            assertTrue(java1.delete());

            assertTrue(java2.delete());

            assertTrue(project1.delete());

            assertTrue(project2.delete());

            assertTrue(dir.delete());

        }

    }

}