Cwikius-Spring
/
spring-security
mirror of https://github.com/spring-projects/spring-security.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Modifications to allow compilation with updated Apache DS version.

This commit is contained in:
Luke Taylor 2005-12-15 01:44:20 +00:00
parent ce3d6f2129
commit 12ebd4e814
1 changed files with 153 additions and 127 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

280
sandbox/src/test/java/org/acegisecurity/providers/dao/ldap/support/EmbeddedLdapServerController.java
View File

@ -19,10 +19,12 @@ import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.Hashtable;
import java.util.Properties;
import javax.naming.Context;
import javax.naming.Name;
import javax.naming.NamingException;
import javax.naming.InitialContext;
import javax.naming.directory.Attributes;
import javax.naming.directory.DirContext;
import javax.naming.directory.InitialDirContext;
@ -32,176 +34,200 @@ 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;
import org.apache.ldap.server.jndi.CoreContextFactory;
import org.apache.ldap.server.jndi.ServerContextFactory;
import org.apache.ldap.server.configuration.MutableStartupConfiguration;
import org.apache.ldap.server.configuration.MutableServerStartupConfiguration;
import org.apache.ldap.server.configuration.ShutdownConfiguration;
import org.apache.ldap.server.DirectoryService;
import org.springframework.core.io.Resource;
import org.springframework.core.io.ClassPathResource;
import org.acegisecurity.providers.ldap.LdapUtils;
/**
/**
* Used as static field in BaseLdapTestCase;
* responsible for global state during JUnit tests - since
* JUnit reinstantiates the test class for every method.
* responsible for global state during JUnit tests - since
* JUnit reinstantiates the test class for every method.
*
* @version $Id$
*
*/
public class EmbeddedLdapServerController {
private File tempDirectory;
private DirContext serverContext;
static final int LDAP_PORT = 10389;
private File tempDirectory;
private DirContext serverContext;
/**
*
*/
public EmbeddedLdapServerController() {
*
*/
public EmbeddedLdapServerController() {
// create temporary directory for directory-server to store files in
tempDirectory = initTempFiles();
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,
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()) {
*/
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 {
} 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
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
// 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();
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 );
serverContext.createSubcontext( dn, attributes );
}
}
/** 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 startServer() {
System.out.println("Creating embedded LDAP server on port " + LDAP_PORT);
MutableServerStartupConfiguration startup = new MutableServerStartupConfiguration();
startup.setWorkingDirectory(tempDirectory);
startup.setLdapPort(LDAP_PORT);
Hashtable env = startup.toJndiEnvironment();
env.putAll(getEnvironment());
env.put(Context.INITIAL_CONTEXT_FACTORY, ServerContextFactory.class.getName());
try {
serverContext = new InitialDirContext( env );
} catch (NamingException nx) {
nx.printStackTrace();
}
}
/**
* Stops the apache directory server, and attempts to remove
* the data files that the server creates.
*/
protected void shutdownServer() {
protected void shutdownServer() {
// close our internal instance of the server-context
try {
serverContext.close();
} catch (NamingException e) {
e.printStackTrace( System.err );
}
serverContext = null;
LdapUtils.closeContext(serverContext);
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.
*
Hashtable env = new ShutdownConfiguration().toJndiEnvironment();
env.putAll(getEnvironment());
try {
new InitialContext( env );
} catch (NamingException e) {
e.printStackTrace();
}
}
/**
* 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()) {
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.
}
}
}
/**
* 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);
public void finalize() throws Throwable {
System.out.println("Entering EmbeddedLdapServerController.finalize()");
shutdownServer();
cleanupTempFiles(tempDirectory);
tempDirectory.delete();
super.finalize();
System.out.println("Leaving LdapTestHelper.finalize()");
}
super.finalize();
System.out.println("Leaving EmbeddedLdapServerController.finalize()");
}
/**
* @return The directory that the directory server will use to store its data files.
*/
public File getTempDirectory() {
return tempDirectory;
}
// 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;
}
// public String getTempDirectoryPath() {
// return tempDirectory.getAbsolutePath();
// }
/** Create and return a Hashtable with standard JNDI settings for our tests. */
protected Properties getEnvironment() {
Properties env = new Properties();
env.setProperty(Context.SECURITY_AUTHENTICATION, "simple");
env.setProperty(Context.SECURITY_PRINCIPAL, "uid=admin,ou=system");
env.setProperty(Context.SECURITY_CREDENTIALS,"secret");
// env.setProperty(Context.INITIAL_CONTEXT_FACTORY, CoreContextFactory.class.getName());
env.setProperty( Context.PROVIDER_URL, "ou=system" );
// env.put( EnvKeys.WKDIR, tempDirectory.getAbsolutePath() );
return env;
}
/** Get our reference to the server-mode context. */
public DirContext getServerContext() {
return serverContext;
}
public DirContext getServerContext() {
return serverContext;
}
public static void main(String[] args) throws IOException {
EmbeddedLdapServerController server = new EmbeddedLdapServerController();
System.out.println(DirectoryService.getInstance());
}
}