KaneAI - World’s First E2E Software Testing Agent.Book a Demo Join TestMu Conference

How to use loadClass method of org.mockitoutil.SimplePerRealmReloadingClassLoader class

Best Mockito code snippet using org.mockitoutil.SimplePerRealmReloadingClassLoader.loadClass

Source:SimplePerRealmReloadingClassLoader.java

...11 * Each class can be reloaded in the realm if the LoadClassPredicate says so.12 */13public class SimplePerRealmReloadingClassLoader extends URLClassLoader {14    private final Map<String,Class> classHashMap = new HashMap<String, Class>();15    private ReloadClassPredicate reloadClassPredicate;16    public SimplePerRealmReloadingClassLoader(ReloadClassPredicate reloadClassPredicate) {17        super(getPossibleClassPathsUrls());18        this.reloadClassPredicate = reloadClassPredicate;19    }20    public SimplePerRealmReloadingClassLoader(ClassLoader parentClassLoader, ReloadClassPredicate reloadClassPredicate) {21        super(getPossibleClassPathsUrls(), parentClassLoader);22        this.reloadClassPredicate = reloadClassPredicate;23    }24    private static URL[] getPossibleClassPathsUrls() {25        return new URL[]{26                obtainClassPath(),27                obtainClassPath("org.mockito.Mockito"),28                obtainClassPath("org.mockito.cglib.proxy.Enhancer")29        };30    }31    private static URL obtainClassPath() {32        String className = SimplePerRealmReloadingClassLoader.class.getName();33        return obtainClassPath(className);34    }35    private static URL obtainClassPath(String className) {36        String path = className.replace('.', '/') + ".class";37        String url = SimplePerRealmReloadingClassLoader.class.getClassLoader().getResource(path).toExternalForm();38        try {39            return new URL(url.substring(0, url.length() - path.length()));40        } catch (MalformedURLException e) {41            throw new RuntimeException("Classloader couldn't obtain a proper classpath URL", e);42        }43    }44    @Override45    public Class<?> loadClass(String qualifiedClassName) throws ClassNotFoundException {46        if(reloadClassPredicate.acceptReloadOf(qualifiedClassName)) {47            // return customLoadClass(qualifiedClassName);48//            Class<?> loadedClass = findLoadedClass(qualifiedClassName);49            if(!classHashMap.containsKey(qualifiedClassName)) {50                Class<?> foundClass = findClass(qualifiedClassName);51                saveFoundClass(qualifiedClassName, foundClass);52                return foundClass;53            }54            return classHashMap.get(qualifiedClassName);55        }56        return useParentClassLoaderFor(qualifiedClassName);57    }58    private void saveFoundClass(String qualifiedClassName, Class<?> foundClass) {59        classHashMap.put(qualifiedClassName, foundClass);60    }61    private Class<?> useParentClassLoaderFor(String qualifiedName) throws ClassNotFoundException {62        return super.loadClass(qualifiedName);63    }64    public Object doInRealm(String callableCalledInClassLoaderRealm) throws Exception {65        ClassLoader current = Thread.currentThread().getContextClassLoader();66        try {67            Thread.currentThread().setContextClassLoader(this);68            Object instance = this.loadClass(callableCalledInClassLoaderRealm).getConstructor().newInstance();69            if (instance instanceof Callable) {70                Callable<?> callableInRealm = (Callable<?>) instance;71                return callableInRealm.call();72            }73        } finally {74            Thread.currentThread().setContextClassLoader(current);75        }76        throw new IllegalArgumentException("qualified name '" + callableCalledInClassLoaderRealm + "' should represent a class implementing Callable");77    }78    public Object doInRealm(String callableCalledInClassLoaderRealm, Class[] argTypes, Object[] args) throws Exception {79        ClassLoader current = Thread.currentThread().getContextClassLoader();80        try {81            Thread.currentThread().setContextClassLoader(this);82            Object instance = this.loadClass(callableCalledInClassLoaderRealm).getConstructor(argTypes).newInstance(args);83            if (instance instanceof Callable) {84                Callable<?> callableInRealm = (Callable<?>) instance;85                return callableInRealm.call();86            }87        } finally {88            Thread.currentThread().setContextClassLoader(current);89        }90        throw new IllegalArgumentException("qualified name '" + callableCalledInClassLoaderRealm + "' should represent a class implementing Callable");91    }92    public interface ReloadClassPredicate {93        boolean acceptReloadOf(String qualifiedName);94    }95}...

loadClass

Using AI Code Generation

1public class LoadClass {2    public static void main(String[] args) throws Exception {3        String className = "com.example.TestClass";4        String fileName = "C:\\TestClass.class";5        Class<?> clazz = new SimplePerRealmReloadingClassLoader().loadClass(className, fileName);6        System.out.println(clazz);7    }8}9package org.mockitoutil;10import java.io.*;11import java.net.*;12import java.util.*;13import java.util.concurrent.atomic.*;14import org.mockito.plugins.*;15import org.mockito.internal.util.*;16import org.mockito.exceptions.base.*;17import org.mockito.internal.exceptions.*;18public class SimplePerRealmReloadingClassLoader extends ClassLoader {19    private final Map<String, Class<?>> loadedClasses = new HashMap<String, Class<?>>();20    private final AtomicBoolean isInitialized = new AtomicBoolean(false);21    private final ClassLoader classLoaderMock;22    public SimplePerRealmReloadingClassLoader() {23        super(SimplePerRealmReloadingClassLoader.class.getClassLoader());24        classLoaderMock = new ClassLoaderMock();25    }26    public Class<?> loadClass(String name) throws ClassNotFoundException {27        return loadClass(name, false);28    }29    protected Class<?> loadClass(String name, boolean resolve) throws ClassNotFoundException {30        Class<?> clazz = loadedClasses.get(name);31        if (clazz == null) {32            try {33                clazz = super.loadClass(name, resolve);34            } catch (ClassNotFoundException e) {35                try {36                    clazz = loadClassFromMockClassLoader(name);37                } catch (ClassNotFoundException e1) {38                    throw new ClassNotFoundException("Cannot find class " + name, e);39                }40            }41        }42        if (resolve) {43            resolveClass(clazz);44        }45        return clazz;46    }47    private Class<?> loadClassFromMockClassLoader(String name) throws ClassNotFoundException {48        if (!isInitialized.getAndSet(true)) {49            initializeMockClassLoader();50        }51        return classLoaderMock.loadClass(name);52    }53    private void initializeMockClassLoader() {54        try {55            Class<?> classLoaderMockMaker = loadClass(ClassLoaderMockMaker.class.getName());56            ClassLoaderMockMaker maker = (ClassLoaderMockMaker) classLoaderMockMaker.newInstance();57            maker.setClassLoader(classLoaderMock);58        } catch (Exception e) {59            throw new MockitoException("Cannot initialize class loader mock", e);60        }61    }62    public Class<?> loadClass(String

loadClass

Using AI Code Generation

1import org.mockitoutil.SimplePerRealmReloadingClassLoader;2import org.mockitoutil.SimplePerRealmReloadingClassLoader.ClassLoaderProvider;3import java.lang.reflect.InvocationTargetException;4import java.lang.reflect.Method;5public class JUnitTestClass {6    public static void main(String[] args) throws ClassNotFoundException, NoSuchMethodException, InvocationTargetException, IllegalAccessException {7        SimplePerRealmReloadingClassLoader classLoader = SimplePerRealmReloadingClassLoader.createLoader(new ClassLoaderProvider() {8            public ClassLoader getClassLoader() {9                return JUnitTestClass.class.getClassLoader();10            }11        });12        Class<?> clazz = classLoader.loadClass("org.mockitoutil.SimplePerRealmReloadingClassLoader");13        Method method = clazz.getMethod("loadClass", String.class);14        String result = (String) method.invoke(clazz, "java.lang.String");15        System.out.println("Result of the method is "+result);16    }17}

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.