/test/integration-tests-openldap/src/test/java/org/springframework/ldap/itest/core/LdapTemplateSearchResultITest.java

/*

 * Copyright 2005-2010 the original author or authors.

 *

 * Licensed under the Apache License, Version 2.0 (the "License");

 * you may not use this file except in compliance with the License.

 * You may obtain a copy of the License at

 *

 *      http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 */

package org.springframework.ldap.itest.core;



import org.junit.After;

import org.junit.Before;

import org.junit.Test;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.core.io.ClassPathResource;

import org.springframework.ldap.core.ContextSource;

import org.springframework.ldap.core.DirContextAdapter;

import org.springframework.ldap.core.DistinguishedName;

import org.springframework.ldap.core.LdapTemplate;

import org.springframework.ldap.support.LdapUtils;

import org.springframework.ldap.test.AttributeCheckAttributesMapper;

import org.springframework.ldap.test.AttributeCheckContextMapper;

import org.springframework.ldap.test.LdapTestUtils;

import org.springframework.test.context.ContextConfiguration;

import org.springframework.test.context.junit4.AbstractJUnit4SpringContextTests;



import javax.naming.Name;

import javax.naming.directory.SearchControls;

import java.util.List;



import static junit.framework.Assert.assertEquals;



/**

 * Verifies that LdapTemplate search methods work against OpenLDAP with TLS.

 * 

 * @author Mattias Hellborg Arthursson

 */

@ContextConfiguration(locations = { "/conf/ldapTemplateTestContext-tls.xml" })

public class LdapTemplateSearchResultITest extends AbstractJUnit4SpringContextTests {



	@Autowired

	private LdapTemplate tested;



    @Autowired

    private ContextSource contextSource;



	private AttributeCheckAttributesMapper attributesMapper;



	private AttributeCheckContextMapper contextMapper;



	private static final String[] ALL_ATTRIBUTES = { "cn", "sn", "description", "telephoneNumber" };



	private static final String[] CN_SN_ATTRS = { "cn", "sn" };



	private static final String[] ABSENT_ATTRIBUTES = { "description", "telephoneNumber" };



	private static final String[] CN_SN_VALUES = { "Some Person2", "Person2" };



	private static final String[] ALL_VALUES = { "Some Person2", "Person2", "Sweden, Company1, Some Person2",

			"+46 555-123458" };



	private static final String BASE_STRING = "";



	private static final String FILTER_STRING = "(&(objectclass=person)(sn=Person2))";



	private static final Name BASE_NAME = new DistinguishedName(BASE_STRING);



    @Before

	public void prepareTestedInstance() throws Exception {

        LdapTestUtils.cleanAndSetup(

                contextSource,

                LdapUtils.newLdapName("ou=People"),

                new ClassPathResource("/setup_data.ldif"));



		attributesMapper = new AttributeCheckAttributesMapper();

		contextMapper = new AttributeCheckContextMapper();

	}



	@After

	public void cleanup() throws Exception {

        LdapTestUtils.clearSubContexts(contextSource, LdapUtils.newLdapName("ou=People"));

		attributesMapper = null;

		contextMapper = null;

	}



	@Test

	public void testSearch_AttributesMapper() {

		attributesMapper.setExpectedAttributes(ALL_ATTRIBUTES);

		attributesMapper.setExpectedValues(ALL_VALUES);

		List<Object> list = tested.search(BASE_STRING, FILTER_STRING, attributesMapper);

		assertEquals(1, list.size());

	}



	@Test

	public void testSearch_SearchScope_AttributesMapper() {

		attributesMapper.setExpectedAttributes(ALL_ATTRIBUTES);

		attributesMapper.setExpectedValues(ALL_VALUES);

        List<Object> list = tested.search(BASE_STRING, FILTER_STRING, SearchControls.SUBTREE_SCOPE, attributesMapper);

		assertEquals(1, list.size());

	}



	@Test

	public void testSearch_SearchScope_LimitedAttrs_AttributesMapper() {

		attributesMapper.setExpectedAttributes(CN_SN_ATTRS);

		attributesMapper.setExpectedValues(CN_SN_VALUES);

		attributesMapper.setAbsentAttributes(ABSENT_ATTRIBUTES);

        List<Object> list = tested.search(BASE_STRING, FILTER_STRING, SearchControls.SUBTREE_SCOPE, CN_SN_ATTRS,

				attributesMapper);

		assertEquals(1, list.size());

	}



	@Test

	public void testSearch_AttributesMapper_Name() {

		attributesMapper.setExpectedAttributes(ALL_ATTRIBUTES);

		attributesMapper.setExpectedValues(ALL_VALUES);

        List<Object> list = tested.search(BASE_NAME, FILTER_STRING, attributesMapper);

		assertEquals(1, list.size());

	}



	@Test

	public void testSearch_SearchScope_AttributesMapper_Name() {

		attributesMapper.setExpectedAttributes(ALL_ATTRIBUTES);

		attributesMapper.setExpectedValues(ALL_VALUES);

        List<Object> list = tested.search(BASE_NAME, FILTER_STRING, SearchControls.SUBTREE_SCOPE, attributesMapper);

		assertEquals(1, list.size());

	}



	@Test

	public void testSearch_SearchScope_LimitedAttrs_AttributesMapper_Name() {

		attributesMapper.setExpectedAttributes(CN_SN_ATTRS);

		attributesMapper.setExpectedValues(CN_SN_VALUES);

		attributesMapper.setAbsentAttributes(ABSENT_ATTRIBUTES);

        List<Object> list = tested

				.search(BASE_NAME, FILTER_STRING, SearchControls.SUBTREE_SCOPE, CN_SN_ATTRS, attributesMapper);

		assertEquals(1, list.size());

	}



	@Test

	public void testSearch_ContextMapper() {

		contextMapper.setExpectedAttributes(ALL_ATTRIBUTES);

		contextMapper.setExpectedValues(ALL_VALUES);

		List<DirContextAdapter> list = tested.search(BASE_STRING, FILTER_STRING, contextMapper);

		assertEquals(1, list.size());

	}



	@Test

	public void testSearch_SearchScope_ContextMapper() {

		contextMapper.setExpectedAttributes(ALL_ATTRIBUTES);

		contextMapper.setExpectedValues(ALL_VALUES);

        List<DirContextAdapter> list = tested.search(BASE_STRING, FILTER_STRING, SearchControls.SUBTREE_SCOPE, contextMapper);

		assertEquals(1, list.size());

	}



	@Test

	public void testSearch_SearchScope_LimitedAttrs_ContextMapper() {

		contextMapper.setExpectedAttributes(CN_SN_ATTRS);

		contextMapper.setExpectedValues(CN_SN_VALUES);

		contextMapper.setAbsentAttributes(ABSENT_ATTRIBUTES);

        List<DirContextAdapter> list = tested.search(BASE_STRING, FILTER_STRING, SearchControls.SUBTREE_SCOPE, CN_SN_ATTRS, contextMapper);

		assertEquals(1, list.size());

	}



	@Test

	public void testSearch_ContextMapper_Name() {

		contextMapper.setExpectedAttributes(ALL_ATTRIBUTES);

		contextMapper.setExpectedValues(ALL_VALUES);

        List<DirContextAdapter> list = tested.search(BASE_NAME, FILTER_STRING, contextMapper);

		assertEquals(1, list.size());

	}



	@Test

	public void testSearch_SearchScope_ContextMapper_Name() {

		contextMapper.setExpectedAttributes(ALL_ATTRIBUTES);

		contextMapper.setExpectedValues(ALL_VALUES);

        List<DirContextAdapter> list = tested.search(BASE_NAME, FILTER_STRING, SearchControls.SUBTREE_SCOPE, contextMapper);

		assertEquals(1, list.size());

	}



	@Test

	public void testSearch_SearchScope_LimitedAttrs_ContextMapper_Name() {

		contextMapper.setExpectedAttributes(CN_SN_ATTRS);

		contextMapper.setExpectedValues(CN_SN_VALUES);

		contextMapper.setAbsentAttributes(ABSENT_ATTRIBUTES);

        List<DirContextAdapter> list = tested.search(BASE_NAME, FILTER_STRING, SearchControls.SUBTREE_SCOPE, CN_SN_ATTRS, contextMapper);

		assertEquals(1, list.size());

	}

}