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

How to use withCodeSourceUrlOf method of org.mockitoutil.ClassLoaders class

Best Mockito code snippet using org.mockitoutil.ClassLoaders.withCodeSourceUrlOf

Source:ClassLoadersTest.java

...76    @Test(expected = ClassNotFoundException.class)77    public void excluding_class_loader_cannot_load_classes_when_no_correct_source_url_set() throws Exception {78        // given79        ClassLoader cl = excludingClassLoader()80                .withCodeSourceUrlOf(this.getClass())81                .build();82        // when83        cl.loadClass("org.mockito.Mockito");84        // then class CNFE85    }86    @Test87    public void excluding_class_loader_can_load_classes_when_correct_source_url_set() throws Exception {88        // given89        ClassLoader cl = excludingClassLoader()90                .withCodeSourceUrlOf(Mockito.class)91                .build();92        // when93        cl.loadClass("org.mockito.Mockito");94        // then class successfully loaded95    }96    @Test97    public void excluding_class_loader_cannot_load_class_when_excluded_prefix_match_class_to_load() throws Exception {98        // given99        ClassLoader cl = excludingClassLoader()100                .withCodeSourceUrlOf(Mockito.class)101                .without("org.mockito.BDDMockito")102                .build();103        cl.loadClass("org.mockito.Mockito");104        // when105        try {106            cl.loadClass("org.mockito.BDDMockito");107            fail("should have raise a ClassNotFoundException");108        } catch (ClassNotFoundException e) {109            assertThat(e.getMessage()).contains("org.mockito.BDDMockito");110        }111        // then class successfully loaded112    }113    @Test114    public void can_not_load_a_class_not_previously_registered_in_builder() throws Exception {115        // given116        ClassLoader cl = ClassLoaders117                .inMemoryClassLoader()118                .withClassDefinition("yop.Dude", SimpleClassGenerator.makeMarkerInterface("yop.Dude"))119                .build();120        // when121        try {122            cl.loadClass("not.Defined");123            fail();124        } catch (ClassNotFoundException e) {125            // then126            assertThat(e.getMessage()).contains("not.Defined");127        }128    }129    @Test130    public void can_load_a_class_in_memory_from_bytes() throws Exception {131        // given132        ClassLoader cl = ClassLoaders133                .inMemoryClassLoader()134                .withClassDefinition("yop.Dude", SimpleClassGenerator.makeMarkerInterface("yop.Dude"))135                .build();136        // when137        Class<?> aClass = cl.loadClass("yop.Dude");138        // then139        assertThat(aClass).isNotNull();140        assertThat(aClass.getClassLoader()).isEqualTo(cl);141        assertThat(aClass.getName()).isEqualTo("yop.Dude");142    }143    @Test144    public void cannot_load_a_class_file_not_in_parent() throws Exception {145        // given146        ClassLoader cl = ClassLoaders147                .inMemoryClassLoader()148                .withParent(jdkClassLoader())149                .build();150        cl.loadClass("java.lang.String");151        try {152            // when153            cl.loadClass("org.mockito.Mockito");154            fail("should have not found Mockito class");155        } catch (ClassNotFoundException e) {156            // then157            assertThat(e.getMessage()).contains("org.mockito.Mockito");158        }159    }160    @Test161    public void can_list_all_classes_reachable_in_a_classloader() throws Exception {162        ClassLoader classLoader = ClassLoaders.inMemoryClassLoader()163                .withParent(jdkClassLoader())164                .withClassDefinition("a.A", SimpleClassGenerator.makeMarkerInterface("a.A"))165                .withClassDefinition("a.b.B", SimpleClassGenerator.makeMarkerInterface("a.b.B"))166                .withClassDefinition("c.C", SimpleClassGenerator.makeMarkerInterface("c.C"))167//                .withCodeSourceUrlOf(ClassLoaders.class)168                .build();169        assertThat(ClassLoaders.in(classLoader).listOwnedClasses()).containsOnly("a.A", "a.b.B", "c.C");170        assertThat(ClassLoaders.in(classLoader).omit("b", "c").listOwnedClasses()).containsOnly("a.A");171    }172    @Test173    public void return_bootstrap_classloader() throws Exception {174        assertThat(jdkClassLoader()).isNotEqualTo(Mockito.class.getClassLoader());175        assertThat(jdkClassLoader()).isNotEqualTo(ClassLoaders.class.getClassLoader());176        assertThat(jdkClassLoader()).isEqualTo(Number.class.getClassLoader());177        assertThat(jdkClassLoader()).isEqualTo(null);178    }179    @Test180    public void return_current_classloader() throws Exception {181        assertThat(currentClassLoader()).isEqualTo(this.getClass().getClassLoader());...

withCodeSourceUrlOf

Using AI Code Generation

1import org.junit.Test;2import org.junit.runner.RunWith;3import org.mockito.Mock;4import org.mockito.runners.MockitoJUnitRunner;5import org.mockitoutil.ClassLoaders;6import java.io.File;7import java.io.IOException;8import java.net.URL;9import java.nio.file.Files;10import java.nio.file.Paths;11import java.util.List;12import static org.assertj.core.api.Assertions.assertThat;13import static org.mockito.Mockito.when;14@RunWith(MockitoJUnitRunner.class)15public class MockitoJunitRunnerTest {16    private List<String> list;17    public void shouldMockList() throws Exception {18        when(list.get(0)).thenReturn("first");19        assertThat(list.get(0)).isEqualTo("first");20    }21    public void shouldLoadClassWithCodeSourceUrlOf() throws Exception {22        ClassLoader classLoader = ClassLoaders.inMemoryClassLoader()23                .withClassDefinition("package org.mockito.test; public class ClassWithCodeSourceUrl {}")24                .withCodeSourceUrlOf(MockitoJunitRunnerTest.class)25                .build();26        Class<?> clazz = classLoader.loadClass("org.mockito.test.ClassWithCodeSourceUrl");27        URL codeSourceUrl = clazz.getProtectionDomain().getCodeSource().getLocation();28        assertThat(codeSourceUrl).isNotNull();29        assertThat(codeSourceUrl.toString()).isEqualTo("file:/Users/jkowalski/mockito/mockito-core/target/test-classes/");30    }31}

withCodeSourceUrlOf

Using AI Code Generation

1package org.mockitoutil;2import java.io.IOException;3import java.io.InputStream;4import java.net.URL;5import java.net.URLClassLoader;6import java.util.Enumeration;7import java.util.Map;8import java.util.concurrent.ConcurrentHashMap;9public class ClassLoaders {10    public static ClassLoader withCodeSourceUrlOf(final Class<?> clazz) {11        return new URLClassLoader(new URL[0], ClassLoader.getSystemClassLoader()) {12            public URL getResource(String name) {13                URL resource = super.getResource(name);14                if (resource != null) {15                    return resource;16                }17                return clazz.getResource(name);18            }19            public Enumeration<URL> getResources(String name) throws IOException {20                Enumeration<URL> resources = super.getResources(name);21                if (resources.hasMoreElements()) {22                    return resources;23                }24                return clazz.getClassLoader().getResources(name);25            }26            public InputStream getResourceAsStream(String name) {27                InputStream resourceStream = super.getResourceAsStream(name);28                if (resourceStream != null) {29                    return resourceStream;30                }31                return clazz.getResourceAsStream(name);32            }33            public String toString() {34                return "ClassLoader with code source URL of " + clazz;35            }36        };37    }38    public static ClassLoader withMockitoCoreInClasspath() {39        return new URLClassLoader(new URL[0], ClassLoader.getSystemClassLoader()) {40            private final Map<String, Class<?>> classes = new ConcurrentHashMap<String, Class<?>>();41            public Class<?> loadClass(String name) throws ClassNotFoundException {42                Class<?> clazz = classes.get(name);43                if (clazz != null) {44                    return clazz;45                }46                try {47                    clazz = super.loadClass(name);48                } catch (ClassNotFoundException e) {49                    clazz = Class.forName(name, true, getClass().getClassLoader());50                }51                classes.put(name, clazz);52                return clazz;53            }54            public String toString() {55                return "ClassLoader with mockito-core in classpath";56            }57        };58    }59}

withCodeSourceUrlOf

Using AI Code Generation

1package org.mockitoutil;2import java.io.*;3import java.net.*;4import java.util.*;5public class ClassLoaders {6    public static ClassLoader withCodeSourceUrlOf(final Class<?> clazz) {7        return new ClassLoader() {8            protected Class<?> findClass(String name) throws ClassNotFoundException {9                if (name.equals(clazz.getName())) {10                    return clazz;11                }12                throw new ClassNotFoundException(name);13            }14            protected URL findResource(String name) {15                if (name.equals(clazz.getName().replace('.', '/') + ".class")) {16                    return clazz.getResource(name);17                }18                return null;19            }20            protected Enumeration<URL> findResources(String name) throws IOException {21                return new Enumeration<URL>() {22                    private URL resource = findResource(name);23                    public boolean hasMoreElements() {24                        return resource != null;25                    }26                    public URL nextElement() {27                        if (resource == null) {28                            throw new NoSuchElementException();29                        }30                        URL result = resource;31                        resource = null;32                        return result;33                    }34                };35            }36        };37    }38}39package org.mockitoutil;40import java.io.IOException;41import java.net.URL;42import java.util.Enumeration;43import java.util.NoSuchElementException;44public class ClassLoaders {45    public static ClassLoader withCodeSourceUrlOf(final Class<?> clazz) {46        return new ClassLoader() {47            protected Class<?> findClass(String name) throws ClassNotFoundException {48                if (name.equals(clazz.getName())) {49                    return clazz;50                }51                throw new ClassNotFoundException(name);52            }53            protected URL findResource(String name) {54                if (name.equals(clazz.getName().replace('.', '/') + ".class")) {55                    return clazz.getResource(name);56                }57                return null;58            }59            protected Enumeration<URL> findResources(String name) throws IOException {60                return new Enumeration<URL>() {61                    private URL resource = findResource(name);62                    public boolean hasMoreElements() {63                        return resource != null;64                    }65                    public URL nextElement() {66                        if (resource == null) {67                            throw new NoSuchElementException();68                        }

withCodeSourceUrlOf

Using AI Code Generation

1package org.example;2public class Foo {3    public String bar() {4        return "bar";5    }6}7package org.example;8public class Foo {9    public String bar() {

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.