/*
 * Hibernate Validator, declare and validate application constraints
 *
 * License: Apache License, Version 2.0
 * See the license.txt file in the root directory or <http://www.apache.org/licenses/LICENSE-2.0>.
 */
package org.hibernate.validator.test.internal.engine.cascaded;

import static org.hibernate.validator.testutil.ConstraintViolationAssert.assertCorrectPropertyPaths;

import java.util.HashMap;
import java.util.Map;
import java.util.Set;

import javax.validation.ConstraintViolation;
import javax.validation.Valid;
import javax.validation.Validation;
import javax.validation.Validator;
import javax.validation.constraints.Size;

import org.hibernate.validator.HibernateValidator;
import org.hibernate.validator.constraints.Email;
import org.testng.annotations.Test;

/**
 * @author Gunnar Morling
 */
public class MapExtractorTest {

	@Test
	public void canUseCustomKeyExtractorForMaps() throws Exception {
		CustomerWithCascadingKeys bob = new CustomerWithCascadingKeys();

		Validator validator = Validation.byProvider( HibernateValidator.class )
				.configure()
				.buildValidatorFactory()
				.getValidator();

		Set<ConstraintViolation<CustomerWithCascadingKeys>> violations = validator.validate( bob );

		assertCorrectPropertyPaths( violations, "addressByType<K>[too short].type", "addressByType<K>[too small].type" );
	}

	@Test
	public void canUseCustomKeyExtractorForConstrainedMapKeys() throws Exception {
		CustomerWithConstrainedKeys bob = new CustomerWithConstrainedKeys();

		Validator validator = Validation.byProvider( HibernateValidator.class )
				.configure()
				.buildValidatorFactory()
				.getValidator();

		Set<ConstraintViolation<CustomerWithConstrainedKeys>> violations = validator.validate( bob );

		assertCorrectPropertyPaths( violations, "addressByType<K>[too short].<map key>", "addressByType<K>[too small].<map key>" );
	}

	@Test
	public void canUseKeyAndValueExtractorForCascadedValidationOfMap() throws Exception {
		CustomerWithCascadingKeyAndValueMap bob = new CustomerWithCascadingKeyAndValueMap();

		Validator validator = Validation.byProvider( HibernateValidator.class )
				.configure()
				.buildValidatorFactory()
				.getValidator();

		Set<ConstraintViolation<CustomerWithCascadingKeyAndValueMap>> violations = validator.validate( bob );

		assertCorrectPropertyPaths( violations, "addressByType[too small].email", "addressByType[long enough].email", "addressByType<K>[too small].type", "addressByType<K>[too short].type" );
	}

	@Test
	public void canUseKeyAndValueExtractorForTypeLevelConstraintsOfMap() throws Exception {
		CustomerWithConstrainedKeysAndValues bob = new CustomerWithConstrainedKeysAndValues();

		Validator validator = Validation.byProvider( HibernateValidator.class )
				.configure()
				.buildValidatorFactory()
				.getValidator();

		Set<ConstraintViolation<CustomerWithConstrainedKeysAndValues>> violations = validator.validate( bob );

		assertCorrectPropertyPaths( violations, "addressByType<K>[too short].<map key>", "addressByType<K>[too small].<map key>", "addressByType[long enough].<map value>" );
	}

	private static class CustomerWithCascadingKeys {

		Map<@Valid AddressType, String> addressByType;

		public CustomerWithCascadingKeys() {
			addressByType = new HashMap<>();
			addressByType.put( new AddressType( "too short" ), "work@bob.de" );
			addressByType.put( new AddressType( "too small" ), "work@bob.de" );
			addressByType.put( new AddressType( "long enough" ), "work@bob.de" );
		}
	}

	private static class CustomerWithConstrainedKeys {

		Map<@Size(min = 10) String, String> addressByType;

		public CustomerWithConstrainedKeys() {
			addressByType = new HashMap<>();
			addressByType.put( "too short", "work@bob.de" );
			addressByType.put( "too small", "work@bob.de" );
			addressByType.put( "long enough", "work@bob.de" );
		}
	}

	private static class CustomerWithConstrainedKeysAndValues {

		Map<@Size(min = 10) String, @Email String> addressByType;

		public CustomerWithConstrainedKeysAndValues() {
			addressByType = new HashMap<>();
			addressByType.put( "too short", "work@bob.de" );
			addressByType.put( "too small", "work@bob.de" );
			addressByType.put( "long enough", "invalid" );
		}
	}

	private static class CustomerWithCascadingKeyAndValueMap {

		Map<@Valid AddressType, @Valid EmailAddress> addressByType;

		public CustomerWithCascadingKeyAndValueMap() {
			addressByType = new HashMap<>();
			addressByType.put( new AddressType( "too short" ), new EmailAddress( "work@bob.de" ) );
			addressByType.put( new AddressType( "too small" ), new EmailAddress( "invalid" ) );
			addressByType.put( new AddressType( "long enough" ), new EmailAddress( "alsoinvalid" ) );
		}
	}
}