URLClassLoader/closetest/GetResourceAsStream.java

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

/**
 * @test
 * @bug 6899919
 * @run shell build2.sh
 * @run main/othervm GetResourceAsStream
 */

import java.io.*;
import java.net.*;

public class GetResourceAsStream extends Common {

/*
 * We simply test various scenarios with class/resource files
 * and make sure the files can be deleted after closing
 * the loader. Therefore, the test will only really be verified
 * on Windows. It will still run correctly on other platforms
 */
    public static void main (String args[]) throws Exception {

        String workdir = System.getProperty("test.classes");
        if (workdir == null) {
            workdir = args[0];
        }

        /* the jar we copy for each test */
        File srcfile = new File (workdir, "foo.jar");

        /* the jar we use for the test */
        File testfile = new File (workdir, "test.jar");

        copyFile (srcfile, testfile);
        test (testfile, false, false);

        copyFile (srcfile, testfile);
        test (testfile, true, false);

        copyFile (srcfile, testfile);
        test (testfile, true, true);

        // repeat test using a directory of files

        File testdir= new File (workdir, "testdir");
        File srcdir= new File (workdir, "test3");

        copyDir (srcdir, testdir);
        test (testdir, true, false);

    }

    // create a loader on jarfile (or directory)
    // load a class , then look for a resource
    // then close the loader
    // check further new classes/resources cannot be loaded
    // check jar (or dir) can be deleted

    static void test (File file, boolean loadclass, boolean readall)
        throws Exception
    {
        URL[] urls = new URL[] {file.toURL()};
        System.out.println ("Doing tests with URL: " + urls[0]);
        URLClassLoader loader = new URLClassLoader (urls);
        if (loadclass) {
            Class testclass = loadClass ("com.foo.TestClass", loader, true);
        }
        InputStream s = loader.getResourceAsStream ("hello.txt");
        s.read();
        if (readall) {
            while (s.read() != -1) ;
            s.close();
        }

        loader.close ();

        // shouuld not find bye.txt now
        InputStream s1 = loader.getResourceAsStream("bye.txt");
        if (s1 != null) {
            throw new RuntimeException ("closed loader returned resource");
        }

        // now check we can delete the path
        rm_minus_rf (file);
        System.out.println (" ... OK");
    }
}