/* | |
* 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. | |
*/ | |
package org.apache.directory.server.core.integ; | |
import org.apache.directory.server.annotations.CreateLdapServer; | |
import org.apache.directory.server.core.api.DirectoryService; | |
import org.apache.directory.server.factory.ServerAnnotationProcessor; | |
import org.apache.directory.server.ldap.LdapServer; | |
import org.junit.runner.Description; | |
import org.junit.runners.model.Statement; | |
import org.slf4j.Logger; | |
import org.slf4j.LoggerFactory; | |
/** | |
* A {@link TestRule} for processing | |
* {@link CreateLdapServer @CreateLdapServer} annotations. | |
* | |
* @author <a href="mailto:dev@directory.apache.org">Apache Directory Project</a> | |
*/ | |
public class CreateLdapServerRule extends CreateDsRule | |
{ | |
private static final Logger LOG = LoggerFactory.getLogger( CreateLdapServerRule.class ); | |
private CreateLdapServerRule classCreateLdapServerRule; | |
private LdapServer ldapServer; | |
public CreateLdapServerRule() | |
{ | |
this( null ); | |
} | |
public CreateLdapServerRule( CreateLdapServerRule classCreateLdapServerRule ) | |
{ | |
super( classCreateLdapServerRule ); | |
this.classCreateLdapServerRule = classCreateLdapServerRule; | |
} | |
@Override | |
public Statement apply( final Statement base, final Description description ) | |
{ | |
return super.apply( buildStatement( base, description ), description ); | |
} | |
private Statement buildStatement( final Statement base, final Description description ) | |
{ | |
final CreateLdapServer createLdapServer = description.getAnnotation( CreateLdapServer.class ); | |
if ( createLdapServer == null ) | |
{ | |
return new Statement() | |
{ | |
@Override | |
public void evaluate() throws Throwable | |
{ | |
LdapServer ldapServer = getLdapServer(); | |
DirectoryService directoryService = getDirectoryService(); | |
if ( ldapServer != null && directoryService != ldapServer.getDirectoryService() ) | |
{ | |
LOG.trace( "Changing to new directory service" ); | |
DirectoryService oldDirectoryService = ldapServer.getDirectoryService(); | |
ldapServer.setDirectoryService( directoryService ); | |
try | |
{ | |
base.evaluate(); | |
} | |
finally | |
{ | |
LOG.trace( "Reverting to old directory service" ); | |
ldapServer.setDirectoryService( oldDirectoryService ); | |
} | |
} | |
else | |
{ | |
LOG.trace( "no @CreateLdapServer on: {}", description ); | |
base.evaluate(); | |
} | |
} | |
}; | |
} | |
else | |
{ | |
return new Statement() | |
{ | |
@Override | |
public void evaluate() throws Throwable | |
{ | |
LOG.trace( "Creating ldap server" ); | |
ldapServer = ServerAnnotationProcessor.createLdapServer( description, | |
getDirectoryService() ); | |
try | |
{ | |
base.evaluate(); | |
} | |
finally | |
{ | |
LOG.trace( "Stopping ldap server" ); | |
ldapServer.stop(); | |
} | |
} | |
}; | |
} | |
} | |
public LdapServer getLdapServer() | |
{ | |
return ldapServer == null | |
? ( classCreateLdapServerRule == null ? null : classCreateLdapServerRule.getLdapServer() ) | |
: ldapServer; | |
} | |
} |