/**
* Copyright (C) 2008 - Abiquo Holdings S.L. All rights reserved.
*
* Please see /opt/abiquo/tomcat/webapps/legal/ on Abiquo server
* or contact contact@abiquo.com for licensing information.
*/
package com.abiquo.testng;
import java.io.File;
import java.util.List;
import org.apache.directory.server.configuration.ApacheDS;
import org.apache.directory.server.constants.ServerDNConstants;
import org.apache.directory.server.core.DefaultDirectoryService;
import org.apache.directory.server.core.factory.JdbmPartitionFactory;
import org.apache.directory.server.core.partition.Partition;
import org.apache.directory.server.core.partition.impl.btree.jdbm.JdbmPartition;
import org.apache.directory.server.ldap.LdapServer;
import org.apache.directory.server.protocol.shared.transport.TcpTransport;
import org.apache.directory.shared.ldap.constants.SchemaConstants;
import org.apache.directory.shared.ldap.exception.LdapInvalidDnException;
import org.apache.directory.shared.ldap.schema.ldif.extractor.SchemaLdifExtractor;
import org.apache.directory.shared.ldap.schema.ldif.extractor.impl.DefaultSchemaLdifExtractor;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.testng.ISuite;
import org.testng.ISuiteListener;
/**
* Starts and stops an ApacheDS server for integration tests.
*
* @author ssedano
*/
public class TestLdapListener implements ISuiteListener
{
/** apache DS port. */
public static final String DEFAULT_LDAP_PORT = "7007";
/** ldifDirectory. */
public static final String DEFAULT_LDIF_DIRECTORY = "src/test/resources/ldap/users.ldif";
/** workingDirectory. */
public static final String DEFAULT_WORKING_DIRECTORY = "target/apache-ds";
// Utility methods
public static String getParameter(final String name, final String defaultValue)
{
return System.getProperty(name, defaultValue);
}
private static final Logger LOGGER = LoggerFactory.getLogger(TestLdapListener.class);
private static final String LDAP_LDIF_DIR = "ldap.ldif.dir";
private static final String LDAP_WORKING_DIR = "ldap.working.dir";
private static final String LDAP_PORT = "ldap.port";
private DefaultDirectoryService service;
private LdapServer ldapService;
private ApacheDS apacheDs;
@Override
public void onStart(final ISuite suite)
{
LOGGER.info("Starting apacheDS server...");
int port = Integer.valueOf(getParameter(LDAP_PORT, DEFAULT_LDAP_PORT));
long start = System.currentTimeMillis();
try
{
JdbmPartition partition = getDefaultPartition();
reset(new File(getParameter(LDAP_WORKING_DIR, DEFAULT_WORKING_DIRECTORY)));
service.addPartition(partition);
configureLdapServer(port);
configureLdifDirectory(new File(getParameter(LDAP_LDIF_DIR, DEFAULT_LDIF_DIRECTORY)));
apacheDs.startup();
}
catch (Exception ex)
{
throw new RuntimeException("Could not start test server", ex);
}
LOGGER.info("ApacheDS started in {} milliseconds", System.currentTimeMillis() - start);
}
private JdbmPartition getDefaultPartition() throws LdapInvalidDnException
{
JdbmPartition partition = new JdbmPartition();
partition.setId("springframework.org");
partition.setSuffix("dc=springframework,dc=org");
return partition;
}
public void configureLdapServer(final int port) throws Exception
{
ldapService = new LdapServer();
ldapService.setTransports(new TcpTransport(port));
ldapService.setDirectoryService(service);
apacheDs = new ApacheDS(ldapService);
}
public void reset(final File workingDirectory) throws Exception
{
SchemaLdifExtractor extractor = new DefaultSchemaLdifExtractor(workingDirectory);
extractor.extractOrCopy(true);
service = new DefaultDirectoryService();
service.setWorkingDirectory(workingDirectory);
service.getChangeLog().setEnabled(false);
service.setSystemPartition(createSystemPartition(service, workingDirectory));
}
public void addPartitions(final List<JdbmPartition> partitions) throws Exception
{
for (Partition partition : partitions)
{
service.addPartition(partition);
}
}
public void configureLdifDirectory(final File ldifDirectory)
{
apacheDs.setLdifDirectory(ldifDirectory);
}
private JdbmPartition createSystemPartition(final DefaultDirectoryService service,
final File workingDirectory) throws Exception
{
JdbmPartitionFactory partitionFactory = new JdbmPartitionFactory();
JdbmPartition systemPartition =
partitionFactory.createPartition("system", ServerDNConstants.SYSTEM_DN, 500,
new File(workingDirectory, "system"));
partitionFactory.addIndex(systemPartition, SchemaConstants.OBJECT_CLASS_AT, 100);
systemPartition.setSchemaManager(service.getSchemaManager());
return systemPartition;
}
@Override
public void onFinish(final ISuite suite)
{
LOGGER.info("Stopping apacheDS server...");
long start = System.currentTimeMillis();
try
{
apacheDs.shutdown();
}
catch (Exception ex)
{
throw new RuntimeException("Could not stop apacheDS server", ex);
}
LOGGER.info("ApacheDS stopped in {} milliseconds", System.currentTimeMillis() - start);
}
}