package org.apache.archiva.redback.components.apacheds;
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
*/
import junit.framework.TestCase;
import org.apache.directory.shared.ldap.util.AttributeUtils;
import org.apache.archiva.redback.components.apacheds.ApacheDs;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
import javax.inject.Inject;
import javax.naming.NamingEnumeration;
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 javax.naming.directory.DirContext;
import javax.naming.directory.InitialDirContext;
import javax.naming.directory.SearchControls;
import javax.naming.directory.SearchResult;
import java.io.File;
/**
* @author <a href="mailto:trygvis@inamo.no">Trygve Laugstøl</a>
* @author Olivier Lamy
*
*/
@RunWith( SpringJUnit4ClassRunner.class )
@ContextConfiguration( locations = { "classpath*:/META-INF/spring-context.xml", "classpath*:/spring-context.xml" } )
public class ApacheDsTest
extends TestCase
{
private String suffix = "dc=plexus,dc=codehaus,dc=org";
@Inject
private ApacheDs apacheDs;
private Logger logger = LoggerFactory.getLogger( getClass() );
protected void setUp()
throws Exception
{
super.setUp();
}
@Test
public void testBasic()
throws Exception
{
apacheDs.setBasedir( new File( "./target/plexus-home" ) );
apacheDs.addSimplePartition( "test", new String[]{"plexus", "codehaus", "org"} ).getSuffix();
apacheDs.startServer();
InitialDirContext context = apacheDs.getAdminContext();
String cn = "trygvis";
createUser( context, cn, createDn( cn ) );
assertExist( context, createDn( cn ), "cn", cn );
cn = "bolla";
createUser( context, cn, createDn( cn ) );
assertExist( context, createDn( cn ), "cn", cn );
SearchControls ctls = new SearchControls();
ctls.setDerefLinkFlag( true );
ctls.setSearchScope( SearchControls.ONELEVEL_SCOPE );
ctls.setReturningAttributes( new String[] { "*" } );
String filter = "(&(objectClass=inetOrgPerson)(cn=trygvis))";
NamingEnumeration<SearchResult> results = context.search( suffix, filter, ctls );
assertTrue( "a result should have been returned", results.hasMoreElements() );
SearchResult result = results.nextElement();
Attributes attrs = result.getAttributes();
logger.info("Attributes {}", AttributeUtils.toString( attrs ) );
assertFalse( "should only have one result returned", results.hasMoreElements() );
apacheDs.stopServer();
// ----------------------------------------------------------------------
// Start it again
// ----------------------------------------------------------------------
apacheDs.startServer();
context = apacheDs.getAdminContext();
assertExist( context, createDn( "trygvis" ), "cn", "trygvis" );
context.unbind( createDn( "trygvis" ) );
assertExist( context, createDn( "bolla" ), "cn", "bolla" );
context.unbind( createDn( "bolla" ) );
apacheDs.stopServer();
}
private void createUser( DirContext context, String cn, String dn )
throws NamingException
{
Attributes attributes = new BasicAttributes();
BasicAttribute objectClass = new BasicAttribute( "objectClass" );
objectClass.add( "top" );
objectClass.add( "inetOrgPerson" );
attributes.put( objectClass );
attributes.put( "cn", cn );
attributes.put( "sn", cn );
context.createSubcontext( dn, attributes );
}
private String createDn( String cn )
{
return "cn=" + cn + "," + suffix;
}
private void assertExist( DirContext context, String dn, String attribute, String value )
throws NamingException
{
SearchControls ctls = new SearchControls();
ctls.setDerefLinkFlag( true );
ctls.setSearchScope( SearchControls.ONELEVEL_SCOPE );
ctls.setReturningAttributes( new String[] { "*" } );
BasicAttributes matchingAttributes = new BasicAttributes();
matchingAttributes.put( attribute, value );
NamingEnumeration<SearchResult> results = context.search( suffix, matchingAttributes );
//NamingEnumeration<SearchResult> results = context.search( suffix, "(" + attribute + "=" + value + ")", ctls );
assertTrue( results.hasMoreElements() );
SearchResult result = results.nextElement();
Attributes attrs = result.getAttributes();
Attribute testAttr = attrs.get( attribute );
assertEquals( value, testAttr.get() );
}
}