我有一个 Singleton/Factory 对象，我想为其编写 JUnit 测试。 Factory 方法根据类路径上的属性文件中的类名来决定要实例化哪个实现类。如果没有找到属性文件，或者属性文件不包含类名键，则该类将实例化一个默认实现类。

由于工厂保留了单例的静态实例，以便在实例化后使用，为了能够在工厂方法中测试“故障转移”逻辑，我需要在不同的类加载器中运行每个测试方法。

JUnit(或其他单元测试包)有没有办法做到这一点？

编辑:这里是一些正在使用的工厂代码:

private static MyClass myClassImpl = instantiateMyClass();

private static MyClass instantiateMyClass() {
    MyClass newMyClass = null;
    String className = null;

    try {
        Properties props = getProperties();
        className = props.getProperty(PROPERTY_CLASSNAME_KEY);

        if (className == null) {
            log.warn("instantiateMyClass: Property [" + PROPERTY_CLASSNAME_KEY
                    + "] not found in properties, using default MyClass class [" + DEFAULT_CLASSNAME + "]");
            className = DEFAULT_CLASSNAME;
        }

        Class MyClassClass = Class.forName(className);
        Object MyClassObj = MyClassClass.newInstance();
        if (MyClassObj instanceof MyClass) {
            newMyClass = (MyClass) MyClassObj;
        }
    }
    catch (...) {
        ...
    }

    return newMyClass;
}

private static Properties getProperties() throws IOException {

    Properties props = new Properties();

    InputStream stream = Thread.currentThread().getContextClassLoader().getResourceAsStream(PROPERTIES_FILENAME);

    if (stream != null) {
        props.load(stream);
    }
    else {
        log.error("getProperties: could not load properties file [" + PROPERTIES_FILENAME + "] from classpath, file not found");
    }

    return props;
}

最佳答案

这个问题可能很老，但由于这是我遇到这个问题时找到的最接近的答案，所以我想描述一下我的解决方案。

使用 JUnit 4

拆分您的测试，以便每个类有一个测试方法(此解决方案仅更改类之间的类加载器，而不是方法之间的更改，因为父运行器每个类收集所有方法一次)

将 @RunWith(SeparateClassloaderTestRunner.class) 注释添加到您的测试类中。

创建 SeparateClassloaderTestRunner 如下所示:

public class SeparateClassloaderTestRunner extends BlockJUnit4ClassRunner {

    public SeparateClassloaderTestRunner(Class<?> clazz) throws InitializationError {
        super(getFromTestClassloader(clazz));
    }

    private static Class<?> getFromTestClassloader(Class<?> clazz) throws InitializationError {
        try {
            ClassLoader testClassLoader = new TestClassLoader();
            return Class.forName(clazz.getName(), true, testClassLoader);
        } catch (ClassNotFoundException e) {
            throw new InitializationError(e);
        }
    }

    public static class TestClassLoader extends URLClassLoader {
        public TestClassLoader() {
            super(((URLClassLoader)getSystemClassLoader()).getURLs());
        }

        @Override
        public Class<?> loadClass(String name) throws ClassNotFoundException {
            if (name.startsWith("org.mypackages.")) {
                return super.findClass(name);
            }
            return super.loadClass(name);
        }
    }
}

请注意，我必须这样做才能测试在无法更改的旧框架中运行的代码。如果可以选择，我会减少使用静态和/或放入测试 Hook 以允许重置系统。它可能不漂亮，但它让我可以测试大量否则会很困难的代码。

此外，此解决方案还打破了依赖于类加载技巧(例如 Mockito)的任何其他内容。

关于java - 为不同的 JUnit 测试使用不同的类加载器？，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/42102/