com.github.zhengframework.test.WithZhengApplicationHolder 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.configuration.Configuration;
import com.github.zhengframework.core.ModuleProvider;
import com.github.zhengframework.test.WithZhengApplication.NotDefineConfiguration;
import com.google.inject.Module;
import lombok.AccessLevel;
import lombok.Getter;
import lombok.NoArgsConstructor;
import lombok.Setter;
import lombok.ToString;
import org.apache.commons.lang3.ArrayUtils;
import org.apache.commons.lang3.StringUtils;

@ToString
@NoArgsConstructor
@Setter(AccessLevel.PACKAGE)
@Getter(AccessLevel.PACKAGE)
@SuppressWarnings("unchecked")
class WithZhengApplicationHolder {

  private String configFile;
  private Class configurationClass;
  private String[] arguments = new String[0];
  private Class[] moduleClass = new Class[0];
  private Class[] excludeModuleClass = new Class[0];
  private Class[] excludeModuleProviderClass = new Class[0];
  private boolean autoLoadModule = true;

  public WithZhengApplicationHolder(WithZhengApplication withZhengApplication) {
    configFile = StringUtils.trimToNull(withZhengApplication.configFile());
    if (!withZhengApplication.configurationClass().equals(NotDefineConfiguration.class)) {
      configurationClass = withZhengApplication.configurationClass();
    }
    arguments = withZhengApplication.arguments();
    moduleClass = withZhengApplication.moduleClass();
    excludeModuleClass = withZhengApplication.excludeModuleClass();
    excludeModuleProviderClass = withZhengApplication.excludeModuleProviderClass();
    autoLoadModule = withZhengApplication.autoLoadModule();
  }

  public static WithZhengApplicationHolder create(WithZhengApplication withZhengApplication) {
    return new WithZhengApplicationHolder(withZhengApplication);
  }

  public WithZhengApplicationHolder merge(WithZhengApplicationHolder merged) {
    WithZhengApplicationHolder holder = new WithZhengApplicationHolder();
    holder.setArguments(getArguments());
    holder.setConfigFile(getConfigFile());
    holder.setConfigurationClass(getConfigurationClass());
    holder.setAutoLoadModule(isAutoLoadModule());
    holder.setExcludeModuleClass(getExcludeModuleClass());
    holder.setExcludeModuleProviderClass(getExcludeModuleProviderClass());
    holder.setModuleClass(getModuleClass());

    if (merged.getConfigFile() != null) {
      holder.setConfigFile(merged.getConfigFile());
    }
    if (merged.getConfigurationClass() != null) {
      holder.setConfigurationClass(merged.getConfigurationClass());
    }
    holder.setExcludeModuleClass(
        ArrayUtils.addAll(holder.getExcludeModuleClass(), merged.getExcludeModuleClass()));
    holder.setExcludeModuleProviderClass(ArrayUtils
        .addAll(holder.getExcludeModuleProviderClass(), merged.getExcludeModuleProviderClass()));
    holder.setModuleClass(ArrayUtils.addAll(holder.getModuleClass(), merged.getModuleClass()));
    holder.setAutoLoadModule(merged.isAutoLoadModule());
    holder.setArguments(ArrayUtils.addAll(holder.getArguments(), merged.getArguments()));
    return holder;
  }

  public WithZhengApplicationHolder merge(WithZhengApplication mergedAnnotation) {
    WithZhengApplicationHolder merged = new WithZhengApplicationHolder(mergedAnnotation);
    return merge(merged);
  }
}