org.isda.cdm.validation.exists.CustomerAccountOnlyExistsValidator Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ingest-test-framework Show documentation

There is a newer version: 11.31.0

Show newest version

package org.isda.cdm.validation.exists;

import com.google.common.collect.ImmutableMap;
import com.rosetta.model.lib.path.RosettaPath;
import com.rosetta.model.lib.validation.ExistenceChecker;
import com.rosetta.model.lib.validation.ValidationResult;
import com.rosetta.model.lib.validation.ValidationResult.ValidationType;
import com.rosetta.model.lib.validation.ValidatorWithArg;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;
import org.isda.cdm.AccountTypeEnum2;
import org.isda.cdm.Client;
import org.isda.cdm.CustomerAccount;

import static com.rosetta.model.lib.validation.ValidationResult.failure;
import static com.rosetta.model.lib.validation.ValidationResult.success;

public class CustomerAccountOnlyExistsValidator implements ValidatorWithArg> {

	/* Casting is required to ensure types are output to ensure recompilation in Rosetta */
	@Override
	public  ValidationResult validate(RosettaPath path, T2 o, Set fields) {
		Map fieldExistenceMap = ImmutableMap.builder()
				.put("id", ExistenceChecker.isSet((String) o.getId()))
				.put("client", ExistenceChecker.isSet((List) o.getClient()))
				.put("accountType", ExistenceChecker.isSet((AccountTypeEnum2) o.getAccountType()))
				.build();
		
		// Find the fields that are set
		Set setFields = fieldExistenceMap.entrySet().stream()
				.filter(Map.Entry::getValue)
				.map(Map.Entry::getKey)
				.collect(Collectors.toSet());
		
		if (setFields.equals(fields)) {
			return success("CustomerAccount", ValidationType.ONLY_EXISTS, "CustomerAccount", path, "");
		}
		return failure("CustomerAccount", ValidationType.ONLY_EXISTS, "CustomerAccount", path, "",
				String.format("[%s] should only be set.  Set fields: %s", fields, setFields));
	}
}