package org.jellyfish.implementation;
   
   import java.util.ArrayList;
   import java.util.List;
   import java.util.SortedSet;
   
   import javax.naming.NamingException;
   import javax.naming.directory.Attribute;
   import javax.naming.directory.Attributes;
  import javax.naming.directory.BasicAttribute;
  import javax.naming.directory.BasicAttributes;
  
  import junit.framework.TestCase;
  import mock.org.apache.avalon.framework.logger.LoggerMock;
  import mock
  	.org
  	.apache
  	.avalon
  	.framework
  	.service
  	.ServiceManagerMock
  	.ServiceManagerMock;
  
  import org.jellyfish.ConfigurationManager;
  import org.jellyfish.ConfigurationManagerMock;
  import org.jellyfish.ConnectionManager;
  import org.jellyfish.ConnectionManagerMock;
  import org.jellyfish.MappingManager;
  import org.jellyfish.ObjectMapNotFoundException;
  import org.jellyfish.config.AttributeMapping;
  import org.jellyfish.config.Config;
  import org.jellyfish.config.ObjectMapping;
  import org.jellyfish.model.Person;
  
  public class MappingManagerDefaultTest extends TestCase {
  	private MappingManagerDefault mappingManager;
  	private Attributes attributes;
  	private ServiceManagerMock serviceManagerMock;
  	private ConfigurationManagerMock configurationManagerMock;
  	private ConnectionManagerMock connectionManagerMock;
  	private LoggerMock loggerMock;
  
  	public MappingManagerDefaultTest(String arg0) {
  		super(arg0);
  	}
  
  	protected void setUp() throws Exception {
  		configurationManagerMock = new ConfigurationManagerMock();
  		configurationManagerMock.getConfigurationReturn = createConfig();
  		connectionManagerMock = new ConnectionManagerMock();
  		attributes = createAttributes();
  		serviceManagerMock = new ServiceManagerMock();
  		serviceManagerMock.lookupMap.put(
  			ConfigurationManager.ROLE,
  			configurationManagerMock);
  		serviceManagerMock.lookupMap.put(
  			ConnectionManager.ROLE,
  			connectionManagerMock);
  		mappingManager = new MappingManagerDefault();
  		mappingManager.service(serviceManagerMock);
  		loggerMock = new LoggerMock();
  		mappingManager.enableLogging(loggerMock);
  	}
  
  	private Attributes createAttributes() {
  		Attributes attributes = new BasicAttributes();
  		attributes.put(MappingManager.OBJECTCLASS, "person");
  		attributes.put("cn", "Joe");
  		attributes.put("sn", "Smith");
  		Attribute telephoneNumbersAttribute =
  			new BasicAttribute("telephoneNumber");
  		telephoneNumbersAttribute.add("1111111111");
  		telephoneNumbersAttribute.add("9999999999");
  		attributes.put(telephoneNumbersAttribute);
  		return attributes;
  	}
  
  	private Config createConfig() {
  		Config config = new Config();
  		config.setBaseDn("dc=jellyfish,dc=org");
  		config.setUserDn("cn=${username},ou=jellyfish");
  		List objectMappings = new ArrayList();
  		ObjectMapping objectMapping = new ObjectMapping();
  		objectMapping.setJavaName(Person.class.getName());
  		objectMapping.setLdapName("person");
  		objectMapping.setDn("cn=${firstName},ou=people");
  		objectMapping.getAttributeMappings().add(
  			new AttributeMapping("firstName", "cn"));
  		objectMapping.getAttributeMappings().add(
  			new AttributeMapping("lastName", "sn"));
  		objectMapping.getAttributeMappings().add(
  			new AttributeMapping("manager", "mgr"));
  		objectMapping.getAttributeMappings().add(
  			new AttributeMapping(
  				"telephoneNumbers",
  				"telephoneNumber",
  				"java.util.TreeSet"));
  		objectMappings.add(objectMapping);
  		config.setObjectMappings(objectMappings);
 		return config;
 	}
 
 	private Person createPerson() {
 		Person person = new Person("Joe", "Smith");
 		person.setManager(new Person("Mike", "Jones"));
 		return person;
 	}
 
 	public void testMapToAttributes() throws NamingException {
 		Person person = createPerson();
 
 		Attributes attributes = mappingManager.mapToAttributes(person);
 
 		assertTrue(configurationManagerMock.getConfigurationCalled);
 
 		assertNotNull(attributes);
 		Attribute cnAttribute = attributes.get("cn");
 		assertNotNull(cnAttribute);
 		assertEquals(person.getFirstName(), cnAttribute.get());
 
 		Attribute snAttribute = attributes.get("sn");
 		assertNotNull(snAttribute);
 		assertEquals(person.getLastName(), snAttribute.get());
 	}
 
 	public void testMapToAttributesNullProperty() throws NamingException {
 		Person person = createPerson();
 		person.setLastName(null);
 
 		Attributes attributes = mappingManager.mapToAttributes(person);
 
 		assertTrue(configurationManagerMock.getConfigurationCalled);
 
 		assertNotNull(attributes);
 		Attribute cnAttribute = attributes.get("cn");
 		assertNotNull(cnAttribute);
 		assertEquals(person.getFirstName(), cnAttribute.get());
 
 		assertNull(attributes.get("sn"));
 	}
 
 	public void testMapNestedObjectToAttributes() throws NamingException {
 		Person person = createPerson();
 		Attributes attributes = mappingManager.mapToAttributes(person);
 		assertEquals(
 			"cn="
 				+ person.getManager().getFirstName()
 				+ ",ou=people,dc=jellyfish,dc=org",
 			attributes.get("mgr").get());
 	}
 
 	public void testGetObjectDn() {
 		Person person = createPerson();
 		String objectDn = mappingManager.getObjectDn(person);
 		assertEquals(
 			"cn=" + person.getFirstName() + ",ou=people,dc=jellyfish,dc=org",
 			objectDn);
 	}
 
 	public void testGetUserDn() {
 		String userDn = mappingManager.getUserDn("jellyfish");
 		assertEquals("cn=jellyfish,ou=jellyfish,dc=jellyfish,dc=org", userDn);
 	}
 
 	public void testGetObjectMappingFromAttributes()
 		throws ObjectMapNotFoundException {
 		ObjectMapping objectMapping =
 			mappingManager.getObjectMapping(attributes);
 		assertEquals(
 			configurationManagerMock
 				.getConfigurationReturn
 				.getObjectMappings()
 				.get(
 				0),
 			objectMapping);
 	}
 
 	public void testMapFromAttributes() throws NamingException {
 		Person mappedPerson =
 			(Person) mappingManager.mapFromAttributes(attributes);
 		assertEquals(attributes.get("cn").get(), mappedPerson.getFirstName());
 		assertEquals(attributes.get("sn").get(), mappedPerson.getLastName());
 		assertTrue(mappedPerson.getTelephoneNumbers() instanceof SortedSet);
 		assertTrue(mappedPerson.getTelephoneNumbers().contains("1111111111"));
 		assertTrue(mappedPerson.getTelephoneNumbers().contains("9999999999"));
 	}
 
 	public void testMapFromAttributesNullAttribute() throws NamingException {
 		attributes.remove("sn");
 		try {
 			Person mappedPerson =
 				(Person) mappingManager.mapFromAttributes(attributes);
 			assertEquals(
 				attributes.get("cn").get(),
 				mappedPerson.getFirstName());
 		} catch (NullPointerException nullPointerException) {
 			fail("NullPointerException should be avoided");
 		}
 	}
 
 	public void testMapFromAttributesNestedObject() throws NamingException {
 		Person mappedPerson =
 			(Person) mappingManager.mapFromAttributes(attributes);
 		assertEquals(attributes.get("cn").get(), mappedPerson.getFirstName());
 		assertEquals(attributes.get("sn").get(), mappedPerson.getLastName());
 	}
 
 	public void testMapFromAttributesNoObjectclass() {
 		attributes.remove(MappingManager.OBJECTCLASS);
 		Person mappedPerson =
 			(Person) mappingManager.mapFromAttributes(attributes);
 	}
 }