| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, |
| * software distributed under the License is distributed on an |
| * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| * KIND, either express or implied. See the License for the |
| * specific language governing permissions and limitations |
| * under the License. |
| */ |
| package org.apache.openjpa.lib.util; |
| |
| import java.net.URL; |
| |
| import org.junit.Test; |
| import org.junit.Before; |
| |
| import static org.junit.Assert.*; |
| |
| /** |
| * Tests the {@link MultiClassLoader}. |
| * |
| * @author Abe White |
| */ |
| public class TestMultiClassLoader { |
| |
| private ClassLoader SYSTEM_LOADER = MultiClassLoader.class.getClassLoader(); |
| |
| private MultiClassLoader _loader = new MultiClassLoader(); |
| |
| @Before |
| public void setUp() { |
| _loader.addClassLoader(MultiClassLoader.THREAD_LOADER); |
| _loader.addClassLoader(SYSTEM_LOADER); |
| } |
| |
| /** |
| * Tests basic add/remove functionality. |
| */ |
| @Test |
| public void testBasic() { |
| assertEquals(2, _loader.size()); |
| assertTrue(!_loader.isEmpty()); |
| assertTrue(_loader.getClassLoaders()[0] == SYSTEM_LOADER); |
| assertEquals(Thread.currentThread().getContextClassLoader(), |
| _loader.getClassLoaders()[1]); |
| assertTrue(_loader.getClassLoader(0) == SYSTEM_LOADER); |
| assertTrue(!_loader.addClassLoader(MultiClassLoader.THREAD_LOADER)); |
| |
| FooLoader foo = new FooLoader(); |
| assertTrue(_loader.addClassLoader(foo)); |
| assertTrue(!_loader.addClassLoader(foo)); |
| assertEquals(3, _loader.size()); |
| assertEquals(foo, _loader.getClassLoaders()[2]); |
| |
| assertTrue(_loader.removeClassLoader(MultiClassLoader.THREAD_LOADER)); |
| assertTrue(!_loader.removeClassLoader(MultiClassLoader.THREAD_LOADER)); |
| assertEquals(2, _loader.size()); |
| assertTrue(_loader.getClassLoaders()[0] == SYSTEM_LOADER); |
| assertEquals(foo, _loader.getClassLoaders()[1]); |
| |
| assertTrue(_loader.removeClassLoader(foo)); |
| assertTrue(_loader.removeClassLoader(SYSTEM_LOADER)); |
| assertTrue(_loader.isEmpty()); |
| |
| MultiClassLoader loader2 = new MultiClassLoader(); |
| loader2.addClassLoader(MultiClassLoader.THREAD_LOADER); |
| loader2.addClassLoader(SYSTEM_LOADER); |
| assertTrue(_loader.addClassLoaders(loader2)); |
| assertTrue(!_loader.addClassLoaders(loader2)); |
| FooLoader foo2 = new FooLoader(); |
| loader2.addClassLoader(foo); |
| loader2.addClassLoader(foo2); |
| assertTrue(_loader.addClassLoaders(1, loader2)); |
| ClassLoader[] loaders = _loader.getClassLoaders(); |
| assertTrue(loaders[0] == SYSTEM_LOADER); |
| assertEquals(Thread.currentThread().getContextClassLoader(), |
| loaders[3]); |
| assertEquals(foo, loaders[1]); |
| assertEquals(foo2, loaders[2]); |
| } |
| |
| |
| /** |
| * Test finding resources. |
| */ |
| @Test |
| public void testGetResource() { |
| assertNull(_loader.getResource("foo")); |
| _loader.addClassLoader(new FooLoader()); |
| assertNotNull(_loader.getResource("foo")); |
| } |
| |
| |
| private static final class FooLoader extends ClassLoader { |
| |
| @Override |
| protected Class findClass(String name) throws ClassNotFoundException { |
| if ("foo".equals(name)) |
| return Integer.class; |
| throw new ClassNotFoundException(name); |
| } |
| |
| @Override |
| protected URL findResource(String name) { |
| try { |
| if ("foo".equals(name)) |
| return new URL("file:///dev/null"); |
| } catch (Exception e) { |
| } |
| return null; |
| } |
| } |
| } |