au.com.agic.apptesting.utils.impl.ApplicationUrlLoaderImpl Maven / Gradle / Ivy

Go to download
package au.com.agic.apptesting.utils.impl;

import static com.google.common.base.Preconditions.checkNotNull;

import au.com.agic.apptesting.constants.Constants;
import au.com.agic.apptesting.profiles.FileProfileAccess;
import au.com.agic.apptesting.profiles.configuration.Configuration;
import au.com.agic.apptesting.profiles.configuration.FeatureGroup;
import au.com.agic.apptesting.profiles.configuration.UrlMapping;
import au.com.agic.apptesting.profiles.dataset.DataSet;
import au.com.agic.apptesting.profiles.dataset.DatasetsRootElement;
import au.com.agic.apptesting.profiles.dataset.Setting;
import au.com.agic.apptesting.utils.ApplicationUrlLoader;
import au.com.agic.apptesting.utils.SystemPropertyUtils;

import org.apache.commons.lang3.StringUtils;

import java.security.NoSuchAlgorithmException;
import java.security.SecureRandom;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Optional;

import javax.validation.constraints.NotNull;

/**
 * Loads the application urls from configuration
 */
public class ApplicationUrlLoaderImpl implements ApplicationUrlLoader {

	private static final SystemPropertyUtils SYSTEM_PROPERTY_UTILS = new SystemPropertyUtilsImpl();

	private static final FileProfileAccess PROFILE_ACCESS = new FileProfileAccess<>(
		SYSTEM_PROPERTY_UTILS.getProperty(Constants.CONFIGURATION),
		Configuration.class);

	private static final FileProfileAccess DATASET_ACCESS = new FileProfileAccess<>(
		SYSTEM_PROPERTY_UTILS.getProperty(Constants.DATA_SETS_PROFILE_SYSTEM_PROPERTY),
		DatasetsRootElement.class);

	private String getAppUrl() {
		final String appUrl = SYSTEM_PROPERTY_UTILS.getProperty(Constants.APP_URL_OVERRIDE_SYSTEM_PROPERTY);
		if (StringUtils.isNotBlank(appUrl)) {
			return appUrl;
		}

		return null;
	}

	@Override
	public List getAppUrls(final String featureGroup) {

        /*
			Deal with the override
         */
		final String appUrlOverride = getAppUrl();
		if (appUrlOverride != null) {
			return Arrays.asList(new UrlMapping(appUrlOverride));
		}

		final Optional configuration = PROFILE_ACCESS.getProfile();

		if (configuration.isPresent()) {
			final List retValue = getUrlMappings(configuration.get(), featureGroup);
			return getLimitedAppUrls(retValue);
		}

		/*
			There are no mappings to return
		 */
		return new ArrayList<>();
	}

	private List getLimitedAppUrls(@NotNull final List completeList) {
		checkNotNull(completeList);

		final String limitedUrls = SYSTEM_PROPERTY_UTILS.getProperty(Constants.NUMBER_URLS_SYSTEM_PROPERTY);
		if (StringUtils.isNoneBlank(limitedUrls)) {
			try {
				Collections.shuffle(completeList, SecureRandom.getInstance("SHA1PRNG"));

				final Integer limit = Integer.parseInt(limitedUrls);

				final List subList = new ArrayList<>();
				for (int i = 0; i < Math.min(limit, completeList.size()); ++i) {
					subList.add(completeList.get(i));
				}

				return subList;
			} catch (final NumberFormatException | NoSuchAlgorithmException ignored) {
				/*
					Invalid input that we ignore
				 */
			}
		}

		return completeList;
	}

	@Override
	public Map> getDatasets() {
		final Optional dataset = DATASET_ACCESS.getProfile();

		/*
			It is possible that a profile does not exist with data sets for this featureGroup
		 */
		if (!dataset.isPresent()) {
			return new HashMap<>();
		}

		return getDatasets(dataset.get());

	}

	private Map> getDatasets(@NotNull final DatasetsRootElement profile) {
		checkNotNull(profile);

		final Map commonDataSet = getCommonDataset(profile);

		final Map> dataSets = new HashMap<>();

		int index = 0;
		for (final DataSet dataSet : profile.getDataSets().getDataSets()) {

			if (!dataSets.containsKey(index)) {
				final Map newMap = new HashMap<>(commonDataSet);
				dataSets.put(index, newMap);
			}

			for (final Setting setting : dataSet.getSettings()) {
				dataSets.get(index).put(setting.getName(), setting.getValue());
			}

			++index;
		}

		return dataSets;
	}

	private List getUrlMappings(@NotNull final Configuration configuration, final String app) {
		checkNotNull(configuration);

		return configuration.getUrlMappings().getFeatureGroups().stream()
			.filter(e -> StringUtils.endsWithIgnoreCase(app, e.getName()))
			.findFirst()
			.map(FeatureGroup::getUrlMappings)
			.orElse(new ArrayList<>());
	}

	/**
	 * @param profile The combined profile
	 * @return The common data set values to be applied to all other datasets
	 */
	private Map getCommonDataset(@NotNull final DatasetsRootElement profile) {
		checkNotNull(profile);

		final Map commonDataSet = new HashMap<>();

		profile.getDataSets().getCommonDataSet().getSettings().stream()
			/*
				Ensure we add the data set to a sequential index
			 */
			.forEach(e -> commonDataSet.put(e.getName(), e.getValue()));

		return commonDataSet;
	}
}