modules/gemfire-modules-session-external/src/test/java/com/gemstone/gemfire/modules/session/junit/SeparateClassloaderTestRunner.java

package com.gemstone.gemfire.modules.session.junit;

import org.junit.runners.BlockJUnit4ClassRunner;
import org.junit.runners.model.InitializationError;

import java.net.URLClassLoader;

/**
 * @author StackOverflow
 */
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("com.gemstone.gemfire.modules.session.")) {
        return super.findClass(name);
      }
      return super.loadClass(name);
    }
  }
}