1 /*
2  * Copyright (c) 2016, Oracle and/or its affiliates. All rights reserved.
3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4  *
5  * This code is free software; you can redistribute it and/or modify it
6  * under the terms of the GNU General Public License version 2 only, as
7  * published by the Free Software Foundation.
8  *
9  * This code is distributed in the hope that it will be useful, but WITHOUT
10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
12  * version 2 for more details (a copy is included in the LICENSE file that
13  * accompanied this code).
14  *
15  * You should have received a copy of the GNU General Public License version
16  * 2 along with this work; if not, write to the Free Software Foundation,
17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18  *
19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20  * or visit www.oracle.com if you need additional information or have any
21  * questions.
22  */
23 
24 import java.io.IOException;
25 import java.io.UncheckedIOException;
26 import java.net.URL;
27 import java.util.Collections;
28 import java.util.Enumeration;
29 import java.util.List;
30 import java.util.stream.Collectors;
31 import java.util.stream.Stream;
32 
33 /*
34  * @test
35  * @bug 8161230
36  * @summary Test java.lang.ClassLoader.resources() method
37  *
38  * @build ResourcesStreamTest
39  * @run main ResourcesStreamTest
40  */
41 public class ResourcesStreamTest {
42 
main(String[] args)43     public static void main(String[] args) throws Exception {
44         testSuccess();
45         testFailure();
46     }
47 
testSuccess()48     public static void testSuccess() throws Exception {
49         // failing part first
50         try {
51             ClassLoader cl = new FailingClassLoader();
52             // should create the stream pipe
53             Stream<URL> stream = cl.resources("the name");
54             // expect function to throw an exception when calling the method
55             stream.forEach(System.out::println);
56             throw new Exception("expected UncheckedIOException not thrown");
57         } catch (UncheckedIOException uio) {
58             String causeMessage = uio.getCause().getMessage();
59             if (!"the name".equals(causeMessage))
60                 throw new Exception("unexpected cause message: " + causeMessage);
61         }
62     }
63 
testFailure()64     public static void testFailure() throws Exception {
65         ClassLoader cl = new SuccessClassLoader();
66         long count = cl.resources("the name").count();
67         if (count != 1)
68             throw new Exception("expected resource is null or empty");
69 
70         cl.resources("the name")
71           .filter(url -> "file:/somefile".equals(url.toExternalForm()))
72           .findFirst()
73           .orElseThrow(() -> new Exception("correct URL not found"));
74     }
75 
76     public static class SuccessClassLoader extends ClassLoader {
77         @Override
getResources(String name)78         public Enumeration<URL> getResources(String name) throws IOException {
79             URL url = new URL("file:/somefile");
80             return Collections.enumeration(Collections.singleton(url));
81         }
82     }
83 
84     public static class FailingClassLoader extends ClassLoader {
85         @Override
getResources(String name)86         public Enumeration<URL> getResources(String name) throws IOException {
87             throw new IOException(name);
88         }
89     }
90 }
91