/*
 *  Licensed to the Apache Software Foundation (ASF) under one or more
 *  contributor license agreements.  See the NOTICE file distributed with
 *  this work for additional information regarding copyright ownership.
 *  The ASF licenses this file to You 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
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 *
 */

package org.apache.tools.ant.taskdefs;

import org.apache.tools.ant.MagicNames;
import org.apache.tools.ant.Project;
import org.apache.tools.ant.types.Path;
import org.apache.tools.ant.types.Commandline;

import org.junit.Before;
import org.junit.Test;

import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;

/**
 * Simple testcase for the ExecuteJava class - mostly stolen from
 * ExecuteWatchdogTest.
 *
 */
public class ExecuteJavaTest {

    private static final int TIME_OUT = 5000;
    private static final int CLOCK_ERROR = 200;
    private static final int TIME_OUT_TEST = TIME_OUT - CLOCK_ERROR;

    private ExecuteJava ej;
    private Project project;
    private Path cp;

    @Before
    public void setUp() {
        ej = new ExecuteJava();
        ej.setTimeout((long) TIME_OUT);
        project = new Project();
        project.setBasedir(".");
        project.setProperty(MagicNames.ANT_HOME, System.getProperty(MagicNames.ANT_HOME));
        cp = new Path(project, getTestClassPath());
        ej.setClasspath(cp);
    }

    private Commandline getCommandline(int timetorun) {
        Commandline cmd = new Commandline();
        cmd.setExecutable(TimeProcess.class.getName());
        cmd.createArgument().setValue(String.valueOf(timetorun));
        return cmd;
    }

    @Test
    public void testNoTimeOut() {
        Commandline cmd = getCommandline(TIME_OUT / 2);
        ej.setJavaCommand(cmd);
        ej.execute(project);
        assertFalse("process should not have been killed", ej.killedProcess());
    }

    // test that the watchdog ends the process
    @Test
    public void testTimeOut() {
        Commandline cmd = getCommandline(TIME_OUT * 2);
        ej.setJavaCommand(cmd);
        long now = System.currentTimeMillis();
        ej.execute(project);
        long elapsed = System.currentTimeMillis() - now;
        assertTrue("process should have been killed", ej.killedProcess());

        assertTrue("elapse time of " + elapsed
                   + " ms is less than timeout value of " + TIME_OUT_TEST + " ms",
                   elapsed >= TIME_OUT_TEST);
        assertTrue("elapse time of " + elapsed
                   + " ms is greater than run value of " + (TIME_OUT * 2) + " ms",
                   elapsed < TIME_OUT * 2);
    }

    @Test
    public void testNoTimeOutForked() {
        Commandline cmd = getCommandline(TIME_OUT / 2);
        ej.setJavaCommand(cmd);
        ej.fork(cp);
        assertFalse("process should not have been killed", ej.killedProcess());
    }

    // test that the watchdog ends the process
    @Test
    public void testTimeOutForked() {
        Commandline cmd = getCommandline(TIME_OUT * 2);
        ej.setJavaCommand(cmd);
        long now = System.currentTimeMillis();
        ej.fork(cp);
        long elapsed = System.currentTimeMillis() - now;
        assertTrue("process should have been killed", ej.killedProcess());

        assertTrue("elapse time of " + elapsed
                   + " ms is less than timeout value of " + TIME_OUT_TEST + " ms",
                   elapsed >= TIME_OUT_TEST);
        assertTrue("elapse time of " + elapsed
                   + " ms is greater than run value of " + (TIME_OUT * 2) + " ms",
                   elapsed < TIME_OUT * 2);
    }

    /**
     * Dangerous method to obtain the classpath for the test.
     * This is severely dependent on the build.xml properties.
     */
    private static String getTestClassPath() {
        String classpath = System.getProperty("build.tests");
        if (classpath == null) {
            System.err.println("WARNING: 'build.tests' property is not available!");
            classpath = System.getProperty("java.class.path");
        }

        return classpath;
    }

}
