test/jdk/java/net/URL/HandlersPkgPrefix/HandlersPkgPrefix.java

/*
 * Copyright (c) 2015, 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 java.net.MalformedURLException;
import java.net.URL;
import java.util.function.Consumer;

/*
 * @test
 * @bug 8075139
 * @summary Basic test for java.protocol.handler.pkgs
 * @compile handlers/foo/Handler.java handlers/bar/Handler.java HandlersPkgPrefix.java
 * @run main/othervm HandlersPkgPrefix
 */

public class HandlersPkgPrefix {
    static final Consumer<Result> KNOWN = r -> {
        if (r.exception != null)
            throw new RuntimeException("Unexpected exception " + r.exception);
        String p = r.url.getProtocol();
        if (!r.protocol.equals(p))
            throw new RuntimeException("Expected:" + r.protocol + ", got:" + p);
    };
    static final Consumer<Result> UNKNOWN = r -> {
        if (r.url != null)
            throw new RuntimeException("Unexpected url:" + r.url);
        if (!(r.exception instanceof MalformedURLException))
            throw new RuntimeException("Expected MalformedURLException, got:"
                                       + r.exception);
    };

    public static void main(String[] args) {
        withPrefix("unknown", "", UNKNOWN);
        withPrefix("unknown", "handlers", UNKNOWN);

        withPrefix("foo", "", UNKNOWN);
        withPrefix("foo", "xxx|yyy|zzz", UNKNOWN);
        withPrefix("foo", "||||", UNKNOWN);
        withPrefix("foo", "|a|b|c|handlers", KNOWN);

        withPrefix("bar", "", UNKNOWN);
        withPrefix("bar", "x.y.z|y.y.y|z.z.z", UNKNOWN);
        withPrefix("bar", " x.y.z | y.y.y | z.z.z| |  ", UNKNOWN);
        withPrefix("bar", "| a | b | c | handlers | d | e", KNOWN);
    }

    static void withPrefix(String protocol, String pkgPrefix,
                           Consumer<Result> resultChecker) {
        System.out.println("Testing, " + protocol + ", " + pkgPrefix);

        // The long standing implementation behavior is that the
        // property is read multiple times, not cached.
        System.setProperty("java.protocol.handler.pkgs", pkgPrefix);
        URL url = null;
        Exception exception = null;
        try {
            url = new URL(protocol + "://");
        } catch (MalformedURLException x) {
            exception = x;
        }
        resultChecker.accept(new Result(protocol, url, exception));
    }

    static class Result {
        final String protocol;
        final URL url;
        final Exception exception;
        Result(String protocol, URL url, Exception exception) {
            this.protocol = protocol;
            this.url = url;
            this.exception = exception;
        }
    }
}