|
@@ -1,193 +0,0 @@
|
|
|
-package net.sf.acegisecurity.providers.dao.ldap;
|
|
|
-
|
|
|
-import java.io.File;
|
|
|
-import java.io.FileInputStream;
|
|
|
-import java.io.IOException;
|
|
|
-import java.io.InputStream;
|
|
|
-import java.util.Hashtable;
|
|
|
-
|
|
|
-import javax.naming.Context;
|
|
|
-import javax.naming.Name;
|
|
|
-import javax.naming.NamingException;
|
|
|
-import javax.naming.directory.Attributes;
|
|
|
-import javax.naming.directory.DirContext;
|
|
|
-import javax.naming.directory.InitialDirContext;
|
|
|
-
|
|
|
-import org.apache.ldap.common.ldif.LdifIterator;
|
|
|
-import org.apache.ldap.common.ldif.LdifParser;
|
|
|
-import org.apache.ldap.common.ldif.LdifParserImpl;
|
|
|
-import org.apache.ldap.common.message.LockableAttributesImpl;
|
|
|
-import org.apache.ldap.common.name.LdapName;
|
|
|
-import org.apache.ldap.server.jndi.EnvKeys;
|
|
|
-
|
|
|
-/**
|
|
|
- * LdapTestHelper - used as static field in BaseLdapTestCase;
|
|
|
- * responsible for global state during JUnit tests - since
|
|
|
- * JUnit reinstantiates the test class for every method.
|
|
|
- *
|
|
|
- */
|
|
|
-public class LdapTestHelper {
|
|
|
-
|
|
|
- private File tempDirectory;
|
|
|
-
|
|
|
- private DirContext serverContext;
|
|
|
-
|
|
|
- /**
|
|
|
- *
|
|
|
- */
|
|
|
- public LdapTestHelper() {
|
|
|
- // create temporary directory for directory-server to store files in
|
|
|
- tempDirectory = initTempFiles();
|
|
|
- // start the apache directory server
|
|
|
- startServer();
|
|
|
- }
|
|
|
-
|
|
|
- /**
|
|
|
- * Creates if needed a temporary directory to store the apache directory
|
|
|
- * server files. Since I can't get the class to shutdown cleanly,
|
|
|
- * it also ensures a clean start by removing any files in the temp. directory.
|
|
|
- *
|
|
|
- * @return The directory that should be used to store temporary files in.
|
|
|
- */
|
|
|
- protected File initTempFiles() {
|
|
|
- String tmpDir = System.getProperty("java.io.tmpdir");
|
|
|
- File dir = new File(tmpDir);
|
|
|
- File tmp = new File(dir, "apacheds_tmp");
|
|
|
- if (tmp.exists()) {
|
|
|
- cleanupTempFiles(tmp);
|
|
|
- } else {
|
|
|
- tmp.mkdir();
|
|
|
- }
|
|
|
- System.out.println("Directory temp files at: " + tmp.getAbsolutePath());
|
|
|
- return tmp;
|
|
|
- }
|
|
|
-
|
|
|
- /** Attempts to open the file and import the contents as LDIF entries
|
|
|
- * into the test directory.
|
|
|
- *
|
|
|
- * @param file The LDIF file to import
|
|
|
- * @throws IOException
|
|
|
- * @throws NamingException
|
|
|
- */
|
|
|
- public void importLDIF(File file) throws IOException, NamingException {
|
|
|
- FileInputStream fis = new FileInputStream(file);
|
|
|
- importLDIF(fis);
|
|
|
- }
|
|
|
-
|
|
|
- /** Attempts to read the provided InputStream for LDIF entries
|
|
|
- * and adds those entries to the test directory server.
|
|
|
- *
|
|
|
- * @param in InputStream of LDIF data.
|
|
|
- * @throws NamingException
|
|
|
- * @throws IOException
|
|
|
- */
|
|
|
- public void importLDIF(InputStream in) throws NamingException, IOException {
|
|
|
- DirContext ctx = new InitialDirContext( getServerEnvironment() );
|
|
|
- try {
|
|
|
- LdifParser parser = new LdifParserImpl();
|
|
|
- LdifIterator iterator = new LdifIterator( in );
|
|
|
- while ( iterator.hasNext() ) {
|
|
|
- Attributes attributes = new LockableAttributesImpl();
|
|
|
- String ldif = ( String ) iterator.next();
|
|
|
- parser.parse( attributes, ldif );
|
|
|
- Name dn = new LdapName( ( String ) attributes.remove( "dn" ).get() );
|
|
|
- dn.remove( 0 );
|
|
|
- ctx.createSubcontext( dn, attributes );
|
|
|
- }
|
|
|
- } finally {
|
|
|
- ctx.close();
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- /** starts the apache directory server. */
|
|
|
- protected void startServer() {
|
|
|
- try {
|
|
|
- serverContext = new InitialDirContext( getServerEnvironment() );
|
|
|
- } catch (NamingException nx) {
|
|
|
- nx.printStackTrace( System.err );
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- /** stops the apache directory server, and attempts to remove
|
|
|
- * the data files that the server creates.
|
|
|
- */
|
|
|
- protected void shutdownServer() {
|
|
|
- // close our internal instance of the server-context
|
|
|
- try {
|
|
|
- serverContext.close();
|
|
|
- } catch (NamingException e) {
|
|
|
- e.printStackTrace( System.err );
|
|
|
- }
|
|
|
- serverContext = null;
|
|
|
-
|
|
|
- // signal the server that its time to say goodbye
|
|
|
- Hashtable env = getServerEnvironment();
|
|
|
- env.put(EnvKeys.SHUTDOWN, "true");
|
|
|
- try {
|
|
|
- new InitialDirContext( env );
|
|
|
- } catch (NamingException e) {
|
|
|
- e.printStackTrace( System.err );
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- /** Utility method to remove any files in the temporary directory
|
|
|
- * that we use to store the directory server's data files.
|
|
|
- *
|
|
|
- * @param tempDir The temporary directory.
|
|
|
- */
|
|
|
- protected void cleanupTempFiles(File tempDir) {
|
|
|
- if ((null != tempDir) && (tempDir.exists())) {
|
|
|
- File[] files = tempDir.listFiles();
|
|
|
- for (int i = 0; i < files.length; i++) {
|
|
|
- if (!files[i].delete()) {
|
|
|
- System.err.println("Error: unable to cleanup Apache Directory Server file: " + files[i]);
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- /**
|
|
|
- * This isn't working, probably because I am referencing the class
|
|
|
- * as a static field, but maybe someone can figure out a way to
|
|
|
- * implement this correctly.
|
|
|
- */
|
|
|
- public void finalize() throws Throwable {
|
|
|
- System.out.println("Entering LdapTestHelper.finalize()");
|
|
|
- shutdownServer();
|
|
|
- cleanupTempFiles(tempDirectory);
|
|
|
- tempDirectory.delete();
|
|
|
- super.finalize();
|
|
|
- System.out.println("Leaving LdapTestHelper.finalize()");
|
|
|
- }
|
|
|
-
|
|
|
- /**
|
|
|
- * @return The directory that the directory server will use to store its data files.
|
|
|
- */
|
|
|
- public File getTempDirectory() {
|
|
|
- return tempDirectory;
|
|
|
- }
|
|
|
-
|
|
|
- /**
|
|
|
- * @return The directory that the directory server will use to store its data files.
|
|
|
- */
|
|
|
- public String getTempDirectoryPath() {
|
|
|
- return tempDirectory.getAbsolutePath();
|
|
|
- }
|
|
|
-
|
|
|
- /** Create and return a Hashtable with standard JNDI settings for our tests. */
|
|
|
- protected Hashtable getServerEnvironment() {
|
|
|
- Hashtable env = new Hashtable();
|
|
|
- env.put( Context.PROVIDER_URL, "ou=system" );
|
|
|
- env.put( Context.INITIAL_CONTEXT_FACTORY, "org.apache.ldap.server.jndi.ServerContextFactory" );
|
|
|
- env.put( Context.SECURITY_PRINCIPAL, "uid=admin,ou=system" );
|
|
|
- env.put( Context.SECURITY_CREDENTIALS, "secret" );
|
|
|
- env.put( EnvKeys.WKDIR, tempDirectory.getAbsolutePath() );
|
|
|
- return env;
|
|
|
- }
|
|
|
-
|
|
|
- /** Get our reference to the server-mode context. */
|
|
|
- public DirContext getServerContext() {
|
|
|
- return serverContext;
|
|
|
- }
|
|
|
-
|
|
|
-}
|