/*
 * Copyright (c) 2014, 2016, Oracle and/or its affiliates. All rights reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 only, as
 * published by the Free Software Foundation.
 *
 * This code is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
 * or visit www.oracle.com if you need additional information or have any
 * questions.
 */

import static org.testng.Assert.assertEquals;
import static org.testng.Assert.assertFalse;
import static org.testng.Assert.assertTrue;
import static org.testng.Assert.fail;

import java.io.IOException;
import java.util.List;
import java.util.Optional;
import java.util.stream.Collectors;

import org.testng.TestNG;
import org.testng.annotations.Test;

/*
 * @test
 * @library /test/lib/share/classes
 * @modules jdk.management
 * @run testng Basic
 * @summary Basic tests for ProcessHandler
 * @author Roger Riggs
 */
public class Basic {
    /**
     * Tests of ProcessHandle.current.
     */
    @Test
    public static void test1() {
        try {
            ProcessHandle self = ProcessHandle.current();
            ProcessHandle self1 = ProcessHandle.current();
            assertEquals(self, self1); //, "get pid twice should be same %d: %d");
        } finally {
            // Cleanup any left over processes
            ProcessHandle.current().children().forEach(ProcessHandle::destroy);
        }
    }

    /**
     * Tests of ProcessHandle.get.
     */
    @Test
    public static void test2() {
        try {
            ProcessHandle self = ProcessHandle.current();
            long pid = self.getPid();       // known native process id
            Optional<ProcessHandle> self1 = ProcessHandle.of(pid);
            assertEquals(self1.get(), self,
                    "ProcessHandle.of(x.getPid()) should be equal getPid() %d: %d");

            Optional<ProcessHandle> ph = ProcessHandle.of(pid);
            assertEquals(pid, ph.get().getPid());
        } finally {
            // Cleanup any left over processes
            ProcessHandle.current().children().forEach(ProcessHandle::destroy);
        }
    }

    @Test
    public static void test3() {
        // Test can get parent of current
        ProcessHandle ph = ProcessHandle.current();
        try {
            Optional<ProcessHandle> pph = ph.parent();
            assertTrue(pph.isPresent(), "Current has a Parent");
        } finally {
            // Cleanup any left over processes
            ProcessHandle.current().children().forEach(ProcessHandle::destroy);
        }
    }

    @Test
    public static void test4() {
        try {
            Process p = new ProcessBuilder("sleep", "0").start();
            p.waitFor();

            long deadPid = p.getPid();
            p = null;               // Forget the process

            Optional<ProcessHandle> t = ProcessHandle.of(deadPid);
            assertFalse(t.isPresent(), "Handle created for invalid pid:" + t);
        } catch (IOException | InterruptedException ex) {
            fail("Unexpected exception", ex);
        } finally {
            // Cleanup any left over processes
            ProcessHandle.current().children().forEach(ProcessHandle::destroy);
        }
    }

    @Test
    public static void test5() {
        // Always contains itself.
        ProcessHandle current = ProcessHandle.current();
        List<ProcessHandle> list = ProcessHandle.allProcesses().collect(Collectors.toList());
        if (!list.stream()
                .anyMatch(ph -> ph.equals(ProcessHandle.current()))) {
            System.out.printf("current: %s%n", current);
            System.out.printf("all processes.size: %d%n", list.size());
            list.forEach(p -> ProcessUtil.printProcess(p, "   allProcesses: "));
            fail("current process not found in all processes");
        }
    }

    @Test(expectedExceptions = IllegalStateException.class)
    public static void test6() {
        ProcessHandle.current().onExit();
    }

    @Test(expectedExceptions = IllegalStateException.class)
    public static void test7() {
        ProcessHandle.current().destroyForcibly();
    }

    // Main can be used to run the tests from the command line with only testng.jar.
    @SuppressWarnings("raw_types")
    public static void main(String[] args) {
        Class<?>[] testclass = {TreeTest.class};
        TestNG testng = new TestNG();
        testng.setTestClasses(testclass);
        testng.run();
    }

}