com.github.zhengframework.test.ZhengApplicationRunner Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of zheng-test Show documentation
zheng framework module: junit support
There is a newer version: 1.8.0
package com.github.zhengframework.test;

import com.github.zhengframework.bootstrap.ZhengApplication;
import com.github.zhengframework.bootstrap.ZhengApplicationBuilder;
import com.github.zhengframework.configuration.Configuration;
import com.github.zhengframework.configuration.ConfigurationBuilder;
import com.github.zhengframework.configuration.source.FileConfigurationSource;
import com.google.inject.Injector;
import com.google.inject.Module;
import java.lang.reflect.InvocationTargetException;
import java.util.ArrayList;
import java.util.List;
import java.util.Optional;
import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.junit.runner.notification.RunNotifier;
import org.junit.runners.BlockJUnit4ClassRunner;
import org.junit.runners.model.FrameworkMethod;
import org.junit.runners.model.InitializationError;

@Slf4j
public class ZhengApplicationRunner extends BlockJUnit4ClassRunner {

  private final WithZhengApplicationHolder classHolder;
  private volatile Injector injector;

  public ZhengApplicationRunner(final Class testClass) throws InitializationError {
    super(testClass);
    WithZhengApplication withZhengApplication = testClass.getAnnotation(WithZhengApplication.class);
    if (withZhengApplication != null) {
      classHolder = WithZhengApplicationHolder.create(withZhengApplication);
    } else {
      classHolder = new WithZhengApplicationHolder();
    }

  }

  private Optional getWithZhengApplicationFor(
      final FrameworkMethod method) {
    final WithZhengApplication annotation = method.getAnnotation(WithZhengApplication.class);
    return Optional.ofNullable(annotation).map(WithZhengApplicationHolder::create);
  }

  @Override
  protected Object createTest() throws Exception {
    Class javaClass = getTestClass().getJavaClass();

    return injector.getInstance(javaClass);
  }

  private ZhengApplication createZhengApplication(WithZhengApplicationHolder holder)
      throws InitializationError {
    ZhengApplicationBuilder builder = ZhengApplicationBuilder.create();
    if (holder.getConfigurationClass() == null) {
      if (StringUtils.isNotEmpty(holder.getConfigFile())) {
        log.info("use ConfigFile={}", holder.getConfigFile());
        Configuration configuration = new ConfigurationBuilder()
            .withConfigurationSource(new FileConfigurationSource(holder.getConfigFile()))
            .build();
        builder.withConfiguration(configuration);
      }
    } else {
      try {
        log.info("user Configuration={}", holder.getConfigurationClass().getName());
        Configuration configuration = holder.getConfigurationClass().getConstructor().newInstance();
        builder.withConfiguration(configuration);
      } catch (InstantiationException | IllegalAccessException | InvocationTargetException | NoSuchMethodException e) {
        throw new InitializationError(e);
      }
    }
    builder.withArguments(holder.getArguments());
    builder.withAutoLoadModule(holder.isAutoLoadModule());
    List moduleList = new ArrayList<>(holder.getModuleClass().length);
    for (Class moduleClass : holder.getModuleClass()) {
      try {
        moduleList.add(moduleClass.getConstructor().newInstance());
      } catch (InstantiationException | IllegalAccessException | InvocationTargetException | NoSuchMethodException e) {
        log.error("create ZhengApplication fail, holder={}", holder, e);
        throw new InitializationError(e);
      }
    }
    builder.addModule(moduleList);
    builder.excludeModule(holder.getExcludeModuleClass());
    builder.excludeModuleProvider(holder.getExcludeModuleProviderClass());
    return builder.build();
  }

  @SneakyThrows
  @Override
  protected void runChild(final FrameworkMethod method, final RunNotifier notifier) {
    WithZhengApplicationHolder current = classHolder
        .merge(getWithZhengApplicationFor(method).orElse(new WithZhengApplicationHolder()));
    ZhengApplication application = createZhengApplication(current);
    application.start();
    injector = application.getInjector();
    super.runChild(method, notifier);
    application.stop();
  }
}