hotspot/test/testlibrary/jittester/src/jdk/test/lib/jittester/jtreg/JitTesterDriver.java

/*
 * Copyright (c) 2015, 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.
 */

package jdk.test.lib.jittester.jtreg;

import jdk.test.lib.Asserts;
import jdk.test.lib.process.OutputAnalyzer;
import jdk.test.lib.process.ProcessTools;
import jdk.test.lib.Utils;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.Arrays;
import java.util.function.Predicate;
import java.util.regex.Pattern;
import java.util.stream.Collectors;
import java.util.stream.Stream;

public class JitTesterDriver {

    public static void main(String[] args) {
        if (args.length < 1) {
            throw new IllegalArgumentException(
                    "[TESTBUG]: wrong number of argument : " + args.length
                    + ". Expected at least 1 argument -- jit-tester test name.");
        }
        OutputAnalyzer oa;
        try {
            ProcessBuilder pb = ProcessTools.createJavaProcessBuilder(true, args);
            oa = new OutputAnalyzer(pb.start());
        } catch (Exception e) {
            throw new Error("Unexpected exception on test jvm start :" + e, e);
        }

        String name = args[args.length - 1];
        Pattern splitOut = Pattern.compile("\\n"); // tests use \n only in stdout
        Pattern splitErr = Pattern.compile("\\r?\\n"); // can handle both \r\n and \n
        Path testDir = Paths.get(Utils.TEST_SRC);
        String goldOut = formatOutput(streamGoldFile(testDir, name, "out"));
        String anlzOut = formatOutput(Arrays.stream(splitOut.split(oa.getStdout())));
        Asserts.assertEQ(anlzOut, goldOut, "Actual stdout isn't equal to golden one");
        String goldErr = formatOutput(streamGoldFile(testDir, name, "err"));
        String anlzErr = formatOutput(Arrays.stream(splitErr.split(oa.getStderr())));
        Asserts.assertEQ(anlzErr, goldErr, "Actual stderr isn't equal to golden one");

        int exitValue = Integer.parseInt(streamGoldFile(testDir, name, "exit").findFirst().get());
        oa.shouldHaveExitValue(exitValue);
    }

    private static String formatOutput(Stream<String> stream) {
        String result = stream.collect(Collectors.joining(Utils.NEW_LINE));
        if (result.length() > 0) {
            result += Utils.NEW_LINE;
        }
        return result;
    }

    private static Stream<String> streamGoldFile(Path dir, String name, String suffix) {
        try {
            return Files.lines(dir.resolve(name + ".gold." + suffix));
        } catch (IOException e) {
            throw new Error(String.format("Can't read golden %s for %s : %s", suffix, name, e), e);
        }
    }
}