Multitenancy junit test : Using a global tenant

CurrentTenantIdentifierResolverImpl.java

/**
 * It specify what Tenant should be used:
 * <p>
 * <p>
 * There are 2 modes of resolving the identifier using {@link #resolveCurrentTenantIdentifier()}:
 * 
 * 	<li> Using a global tenant identifier for junit testing
 * 	<li> Using the http session to get the tenant for deployed applications
 * 
 * @author jm
 */
public class CurrentTenantIdentifierResolverImpl implements CurrentTenantIdentifierResolver {

// public TenantResolverStrategy tenantStrategy;
	private boolean contextInitialized;
	private static String globalTenant; //static because is for all the classes
	
	
	public CurrentTenantIdentifierResolverImpl() {
		contextInitialized = false;
		globalTenant = DEFAULT_TENANTID;
	}
	
	
	@Override
	public String resolveCurrentTenantIdentifier() {
		String tenant;
		if(contextInitialized){
				tenant = resolveTenantByHttpSession();
		}
		else{
			tenant = globalTenant;
		}
		return tenant;
	}
	
	/**
	 * Get tenantId in the session attribute KEY_TENANTID
	 * @return TenantId on KEY_TENANTID
	 */
	private String resolveTenantByHttpSession() {
	ServletRequestAttributes attr = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
	    	//If session attribute exists returns tenantId saved on the session
		if(attr != null){
			HttpSession session = attr.getRequest().getSession(false); // true == allow create
			if(session != null){
			    String tenant = (String) session.getAttribute(KEY_TENANTID_SESSION);
				if(tenant != null){
					return tenant;
				}
			}
		}
		//otherwise return default tenant
		return DEFAULT_TENANTID;
	}


  public static void setContextInitialized(boolean initialized) {
    contextInitialized = false;
    globalTenant = DEFAULT_TENANTID;
  }
	/**
	 * Used when create the initial entities
	 * @param tenant
	 */
	public static void setGlobalTenant(String tenant)
	{
		globalTenant = tenant;
	}
}