Add TestLDAPConnectionHandler.java test.

dugan

25.11.2006 50539d6456e29e659810196788e1741ce2acf307

 opendj-sdk/opends/tests/unit-tests-testng/src/server/org/opends/server/protocols/ldap/LdapTestCase.java

@@ -26,13 +26,19 @@
 */
package org.opends.server.protocols.ldap ;

import static org.opends.server.config.ConfigConstants.ATTR_LISTEN_PORT;
import org.opends.server.types.Entry;
import org.opends.server.DirectoryServerTestCase;
import org.opends.server.config.ConfigEntry;
import org.opends.server.protocols.asn1.ASN1Boolean;
import org.opends.server.protocols.asn1.ASN1Element;
import org.opends.server.protocols.asn1.ASN1Long;
import org.opends.server.protocols.asn1.ASN1Sequence;
import org.opends.server.types.Attribute;
import org.testng.annotations.Test;

import java.net.InetSocketAddress;
import java.net.ServerSocket;
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.ListIterator;
@@ -44,6 +50,9 @@
@Test(groups = { "precommit", "ldap" })
public abstract class LdapTestCase extends DirectoryServerTestCase
{
	
   private static String localHost = "127.0.0.1";
	
  /**
   * Determine whether one LDAPAttribute is equal to another.
   * The values of the attribute must be identical and in the same order.
@@ -136,4 +145,41 @@
     op.toString(sb);
     op.toString(sb, 1);
  }
  
  /**
   * Generate a LDAPConnectionHandler from a entry. The listen port is
   * determined automatiacally, so no ATTR_LISTEN_PORT should be in the
   * entry.
   * 
   * @param handlerEntry The entry to be used to configure the handle.
   * @return
   * @throws Exception if the handler cannot be initialized.
   */
  static LDAPConnectionHandler 
  getLDAPHandlerInstance(Entry handlerEntry) throws Exception {
     ServerSocket serverLdapSocket = new ServerSocket();
     serverLdapSocket.setReuseAddress(true);
     serverLdapSocket.bind(new InetSocketAddress(localHost, 0));
     long serverLdapPort = serverLdapSocket.getLocalPort();
     Attribute a=new Attribute(ATTR_LISTEN_PORT, String.valueOf(serverLdapPort));
     handlerEntry.addAttribute(a,null);
     String LDAPClassName=LDAPConnectionHandler.class.getName();
     Class LDAPConnHandlerClass = Class.forName(LDAPClassName);
     LDAPConnectionHandler LDAPConnHandler = 
        (LDAPConnectionHandler) LDAPConnHandlerClass.newInstance();
     LDAPConnHandler.initializeConnectionHandler(new ConfigEntry(handlerEntry, null ));
     return LDAPConnHandler;
  }
  
  /**
 * @return A free port number.
 * @throws Exception if socket cannot be created or bound to.
 */
static long
  getFreePort() throws Exception {
     ServerSocket serverLdapSocket = new ServerSocket();
     serverLdapSocket.setReuseAddress(true);
     serverLdapSocket.bind(new InetSocketAddress(localHost, 0));
     return serverLdapSocket.getLocalPort();
  }
}

 opendj-sdk/opends/tests/unit-tests-testng/src/server/org/opends/server/protocols/ldap/TestLDAPConnectionHandler.java

New file
@@ -0,0 +1,328 @@
/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License, Version 1.0 only
 * (the "License").  You may not use this file except in compliance
 * with the License.
 *
 * You can obtain a copy of the license at
 * trunk/opends/resource/legal-notices/OpenDS.LICENSE
 * or https://OpenDS.dev.java.net/OpenDS.LICENSE.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at
 * trunk/opends/resource/legal-notices/OpenDS.LICENSE.  If applicable,
 * add the following below this CDDL HEADER, with the fields enclosed
 * by brackets "[]" replaced with your own identifying * information:
 *      Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 *
 *
 *      Portions Copyright 2006 Sun Microsystems, Inc.
 */

package org.opends.server.protocols.ldap;

import java.util.LinkedHashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Collection;

import static org.opends.server.config.ConfigConstants.*;

import org.opends.server.api.ClientConnection;
import org.opends.server.TestCaseUtils;
import org.opends.server.types.*;
import org.opends.server.config.ConfigAttribute;
import org.opends.server.config.ConfigEntry;
import org.opends.server.core.DirectoryServer;
import org.opends.server.types.Attribute;
import org.opends.server.types.Entry;
import static org.testng.Assert.*;

import org.testng.annotations.BeforeClass;
import org.testng.annotations.Test;

/**
 * @author Administrator
 *
 */
public class TestLDAPConnectionHandler extends LdapTestCase {

   private static String reasonMsg="Don't need a reason.";

   /**
    * Once-only initialization.
    * 
    * @throws Exception
    *           If an unexpected error occurred.
    */
   @BeforeClass
   public void setUp() throws Exception {
      // This test suite depends on having the schema available, so we'll
      // start the server.
      TestCaseUtils.startServer();
   }

   @Test()
   /**
    *  Creates two handlers, one which is SSL type. Then change some values via the setter
    *  methods.
    *  
    * @throws Exception if the handler cannot be instantiated.
    */
   public void testLDAPConnectionHandler() throws Exception {
      Entry LDAPHandlerEntry=null;

      LDAPHandlerEntry=TestCaseUtils.makeEntry(
            "dn: cn=LDAP Connection Handler,cn=Connection Handlers,cn=config",
            "objectClass: top",
            "objectClass: ds-cfg-connection-handler",
            "objectClass: ds-cfg-ldap-connection-handler",
            "cn: LDAP Connection Handler",
            "ds-cfg-connection-handler-class: org.opends.server.protocols.ldap.LDAPConnectionHandler",
            "ds-cfg-connection-handler-enabled: true",
            "ds-cfg-listen-address: 0.0.0.0",
            "ds-cfg-accept-backlog: 128",
            "ds-cfg-allow-ldapv2: false",
            "ds-cfg-keep-stats: false",
            "ds-cfg-use-tcp-keepalive: true",
            "ds-cfg-use-tcp-nodelay: true",
            "ds-cfg-allow-tcp-reuse-address: true",
            "ds-cfg-send-rejection-notice: true",
            "ds-cfg-max-request-size: 5 megabytes",
            "ds-cfg-num-request-handlers: 2",
            "ds-cfg-allow-start-tls: false",
            "ds-cfg-use-ssl: false",
            "ds-cfg-ssl-client-auth-policy: optional",
      "ds-cfg-ssl-cert-nickname: server-cert");
      LDAPConnectionHandler LDAPConnHandler=getLDAPHandlerInstance(LDAPHandlerEntry);
      LDAPConnHandler.allowLDAPv2();
      LDAPConnHandler.allowStartTLS();
      LDAPConnHandler.setKeepStats(false);
      LDAPConnHandler.keepStats();
      LDAPConnHandler.toString(new StringBuilder());
      LDAPConnHandler.toString();
      LDAPStatistics tracker=LDAPConnHandler.getStatTracker();
      LinkedHashMap<String,String> alerts = LDAPConnHandler.getAlerts();
      String c=LDAPConnHandler.getClassName();
      DN dn = LDAPConnHandler.getComponentEntryDN();
      DN dn1 = LDAPConnHandler.getConfigurableComponentEntryDN();
      List<ConfigAttribute> atts = LDAPConnHandler.getConfigurationAttributes();
      String[] cips = LDAPConnHandler.getEnabledSSLCipherSuites();
      String[] protos = LDAPConnHandler.getEnabledSSLProtocols();
      int maxReqSize = LDAPConnHandler.getMaxRequestSize();
      String shutListName=LDAPConnHandler.getShutdownListenerName();
      SSLClientAuthPolicy policy = LDAPConnHandler.getSSLClientAuthPolicy();
      Collection<ClientConnection> cons=LDAPConnHandler.getClientConnections();
      LDAPConnHandler.processServerShutdown(reasonMsg);
      //Reset some things for the SSL handler	
      Attribute useSSL=new Attribute(ATTR_USE_SSL, String.valueOf(false));
      //MPD fix this in ConfigConstants
      Attribute startTls=new Attribute(ATTR_ALLOW_STARTTLS, String.valueOf(false));
      AttributeType attrType=DirectoryServer.getAttributeType(ATTR_LISTEN_PORT, true);
      Attribute a=new Attribute(attrType);
      LDAPHandlerEntry.removeAttribute(a, null);
      LDAPHandlerEntry.removeAttribute(useSSL, null);
      LDAPHandlerEntry.removeAttribute(startTls, null);
      Attribute useSSL1=new Attribute(ATTR_USE_SSL, String.valueOf(true));
      Attribute startTls1=new Attribute(ATTR_ALLOW_STARTTLS, String.valueOf(true));
      LDAPHandlerEntry.addAttribute(useSSL1,null);
      LDAPHandlerEntry.addAttribute(startTls1,null);
      LDAPConnectionHandler LDAPSConnHandler = getLDAPHandlerInstance(LDAPHandlerEntry);
      LDAPSConnHandler.finalizeConnectionHandler(reasonMsg, true);
      LDAPConnHandler.processServerShutdown(reasonMsg);
   }

   @Test()
   /**
    *  Start a handler an then give its hasAcceptableConfiguration a ConfigEntry with
    *  numerous invalid cases.
    *  
    * @throws Exception if handler cannot be instantiated or the configuration is 
    *                   accepted.
    */
   public void testBadLDAPConnectionHandlerConfiguration() throws Exception
   {
      Entry BadHandlerEntry=TestCaseUtils.makeEntry(
            "dn: cn=LDAP Connection Handler,cn=Connection Handlers,cn=config",
            "objectClass: top",
            "objectClass: ds-cfg-connection-handler",
            "objectClass: ds-cfg-ldap-connection-handler",
            "cn: LDAP Connection Handler",
            "ds-cfg-connection-handler-class: org.opends.server.protocols.ldap.LDAPConnectionHandler",
            "ds-cfg-connection-handler-enabled: true",
            "ds-cfg-listen-address: 0.0.0.0",
            "ds-cfg-accept-backlog: 128",
            "ds-cfg-allow-ldapv2: false",
            "ds-cfg-keep-stats: false",
            "ds-cfg-use-tcp-keepalive: true",
            "ds-cfg-use-tcp-nodelay: true",
            "ds-cfg-allow-tcp-reuse-address: true",
            "ds-cfg-send-rejection-notice: true",
            "ds-cfg-max-request-size: 5 megabytes",
            "ds-cfg-num-request-handlers: 2",
            "ds-cfg-allow-start-tls: false",
            "ds-cfg-use-ssl: false",
            "ds-cfg-ssl-client-auth-policy: optional",
      "ds-cfg-ssl-cert-nickname: server-cert");
      LDAPConnectionHandler LDAPConnHandler=getLDAPHandlerInstance(BadHandlerEntry);
      //Add some invalid attrs and some duplicate attrs probably a better way but I feel
      //like typing
      Attribute a2=new Attribute(ATTR_LISTEN_PORT, String.valueOf(389));
      Attribute a2a=new Attribute(ATTR_LISTEN_PORT, String.valueOf(70000));
      Attribute a3=new Attribute(ATTR_LISTEN_ADDRESS, "localhost");
      Attribute a3a=new Attribute(ATTR_LISTEN_ADDRESS, "FAFASFSDFSADFASDFSDFSDAFAS");
      Attribute a4=new Attribute(ATTR_ACCEPT_BACKLOG, String.valueOf(Long.MAX_VALUE));
      Attribute a5=new Attribute(ATTR_ALLOWED_CLIENT, "129.800.990.45");
      Attribute a6=new Attribute(ATTR_DENIED_CLIENT, "129.");
      Attribute a7=new Attribute(ATTR_ALLOW_LDAPV2, "45");
      Attribute a8=new Attribute(ATTR_KEEP_LDAP_STATS, "45");
      Attribute a9=new Attribute(ATTR_SEND_REJECTION_NOTICE, "45");
      Attribute a10=new Attribute(ATTR_USE_TCP_KEEPALIVE, "45");
      Attribute a11=new Attribute(ATTR_USE_TCP_NODELAY, "45");
      Attribute a12=new Attribute(ATTR_ALLOW_REUSE_ADDRESS, "45");
      Attribute a13=new Attribute(ATTR_MAX_REQUEST_SIZE, "45 FLUBBERBYTES");
      Attribute a14=new Attribute(ATTR_USE_SSL, "45");
      Attribute a15=new Attribute(ATTR_ALLOW_STARTTLS, "45");
      BadHandlerEntry.addAttribute(a2, null);
      BadHandlerEntry.addAttribute(a3, null);
      BadHandlerEntry.addAttribute(a2a, null);
      BadHandlerEntry.addAttribute(a3a, null);
      BadHandlerEntry.addAttribute(a4, null);
      BadHandlerEntry.addAttribute(a5, null);
      BadHandlerEntry.addAttribute(a6, null);
      BadHandlerEntry.addAttribute(a7, null);
      BadHandlerEntry.addAttribute(a8, null);
      BadHandlerEntry.addAttribute(a9, null);
      BadHandlerEntry.addAttribute(a10, null);
      BadHandlerEntry.addAttribute(a11, null);
      BadHandlerEntry.addAttribute(a12, null);
      BadHandlerEntry.addAttribute(a13, null);
      BadHandlerEntry.addAttribute(a14, null);
      BadHandlerEntry.addAttribute(a15, null);
      LinkedList<String> reasons = new LinkedList<String>();
      ConfigEntry BadConfigEntry=new ConfigEntry(BadHandlerEntry, null );
      boolean ret=LDAPConnHandler.hasAcceptableConfiguration(BadConfigEntry, reasons);
      LDAPConnHandler.finalizeConnectionHandler(reasonMsg, true);
      LDAPConnHandler.processServerShutdown(reasonMsg);
      assertFalse(ret);
   }

   /**
    * Create handler and then change most of its values and see if
    * it is acceptable and applied.
    * @throws Exception if handler cannot be instantiated.
    */
   @Test()
   public void testGoodLDAPConnectionHandlerConfiguration() throws Exception
   {
      Entry GoodHandlerEntry=TestCaseUtils.makeEntry(
            "dn: cn=LDAP Connection Handler,cn=Connection Handlers,cn=config",
            "objectClass: top",
            "objectClass: ds-cfg-connection-handler",
            "objectClass: ds-cfg-ldap-connection-handler",
            "cn: LDAP Connection Handler",
            "ds-cfg-connection-handler-class: org.opends.server.protocols.ldap.LDAPConnectionHandler",
            "ds-cfg-connection-handler-enabled: true",
            "ds-cfg-listen-address: 0.0.0.0",
            "ds-cfg-accept-backlog: 128",
            "ds-cfg-allow-ldapv2: false",
            "ds-cfg-keep-stats: false",
            "ds-cfg-use-tcp-keepalive: true",
            "ds-cfg-use-tcp-nodelay: true",
            "ds-cfg-allow-tcp-reuse-address: true",
            "ds-cfg-send-rejection-notice: true",
            "ds-cfg-max-request-size: 5 megabytes",
            "ds-cfg-num-request-handlers: 2",
            "ds-cfg-allow-start-tls: false",
            "ds-cfg-use-ssl: true",
            "ds-cfg-ssl-client-auth-policy: optional",
      "ds-cfg-ssl-cert-nickname: server-cert");
      LDAPConnectionHandler LDAPConnHandler=getLDAPHandlerInstance(GoodHandlerEntry);
      //Make attrTypes to remove
      AttributeType at0=DirectoryServer.getAttributeType(ATTR_LISTEN_PORT, true);
      AttributeType at1=DirectoryServer.getAttributeType(ATTR_LISTEN_ADDRESS, true);
      AttributeType at2=DirectoryServer.getAttributeType(ATTR_ALLOW_LDAPV2, true);
      AttributeType at3=DirectoryServer.getAttributeType(ATTR_ALLOW_LDAPV2, true);
      AttributeType at4=DirectoryServer.getAttributeType(ATTR_KEEP_LDAP_STATS, true);
      AttributeType at5=DirectoryServer.getAttributeType(ATTR_SEND_REJECTION_NOTICE,true);
      AttributeType at6=DirectoryServer.getAttributeType(ATTR_USE_TCP_KEEPALIVE,true);
      AttributeType at7=DirectoryServer.getAttributeType(ATTR_USE_TCP_NODELAY,true);
      AttributeType at8=DirectoryServer.getAttributeType(ATTR_ALLOW_REUSE_ADDRESS,true);
      AttributeType at9=DirectoryServer.getAttributeType(ATTR_USE_SSL,true);
      AttributeType at10=DirectoryServer.getAttributeType(ATTR_ALLOW_STARTTLS,true);
      AttributeType at11=DirectoryServer.getAttributeType(ATTR_MAX_REQUEST_SIZE,true);
      AttributeType at12=DirectoryServer.getAttributeType(ATTR_ACCEPT_BACKLOG,true);
      //Remove them
      Attribute rAttr0=new Attribute(at0);
      GoodHandlerEntry.removeAttribute(rAttr0, null);
      Attribute rAttr1=new Attribute(at1);
      GoodHandlerEntry.removeAttribute(rAttr1, null);
      Attribute rAttr2=new Attribute(at2);
      GoodHandlerEntry.removeAttribute(rAttr2, null);
      Attribute rAttr3=new Attribute(at3);
      GoodHandlerEntry.removeAttribute(rAttr3, null);
      Attribute rAttr4=new Attribute(at4);
      GoodHandlerEntry.removeAttribute(rAttr4, null);
      Attribute rAttr5=new Attribute(at5);
      GoodHandlerEntry.removeAttribute(rAttr5, null);
      Attribute rAttr6=new Attribute(at6);
      GoodHandlerEntry.removeAttribute(rAttr6, null);
      Attribute rAttr7=new Attribute(at7);		
      GoodHandlerEntry.removeAttribute(rAttr7, null);
      Attribute rAttr8=new Attribute(at8);
      Attribute rAttr9=new Attribute(at9);
      Attribute rAttr10=new Attribute(at10);
      Attribute rAttr11=new Attribute(at11);
      Attribute rAttr12=new Attribute(at12);
      GoodHandlerEntry.removeAttribute(rAttr8, null);
      GoodHandlerEntry.removeAttribute(rAttr9, null);
      GoodHandlerEntry.removeAttribute(rAttr10, null);
      GoodHandlerEntry.removeAttribute(rAttr11, null);
      GoodHandlerEntry.removeAttribute(rAttr12, null);
      //Make new AttrTypes with different values
      long newPort=getFreePort();
      Attribute a2=new Attribute(ATTR_LISTEN_PORT, String.valueOf(newPort));
      Attribute a3=new Attribute(ATTR_LISTEN_ADDRESS, "localhost");	
      Attribute a4=new Attribute(ATTR_ACCEPT_BACKLOG, String.valueOf(25));
      Attribute a5=new Attribute(ATTR_ALLOWED_CLIENT, "129.56.56.45");
      Attribute a6=new Attribute(ATTR_DENIED_CLIENT, "129.*.*.90");
      Attribute a7=new Attribute(ATTR_ALLOW_LDAPV2, "true");
      Attribute a8=new Attribute(ATTR_KEEP_LDAP_STATS, "true");
      Attribute a9=new Attribute(ATTR_SEND_REJECTION_NOTICE, "false");
      Attribute a10=new Attribute(ATTR_USE_TCP_KEEPALIVE, "false");
      Attribute a11=new Attribute(ATTR_USE_TCP_NODELAY, "false");
      Attribute a12=new Attribute(ATTR_ALLOW_REUSE_ADDRESS, "false");
      Attribute a13=new Attribute(ATTR_MAX_REQUEST_SIZE, "45 kb");
      Attribute a14=new Attribute(ATTR_USE_SSL, "false");
      Attribute a15=new Attribute(ATTR_ALLOW_STARTTLS, "true");
      //Add them
      GoodHandlerEntry.addAttribute(a2, null);
      GoodHandlerEntry.addAttribute(a3, null);
      GoodHandlerEntry.addAttribute(a4, null);
      GoodHandlerEntry.addAttribute(a5, null);
      GoodHandlerEntry.addAttribute(a6, null);
      GoodHandlerEntry.addAttribute(a7, null);
      GoodHandlerEntry.addAttribute(a8, null);
      GoodHandlerEntry.addAttribute(a9, null);
      GoodHandlerEntry.addAttribute(a10, null);
      GoodHandlerEntry.addAttribute(a11, null);
      GoodHandlerEntry.addAttribute(a12, null);
      GoodHandlerEntry.addAttribute(a13, null);
      GoodHandlerEntry.addAttribute(a14, null);
      GoodHandlerEntry.addAttribute(a15, null);
      LinkedList<String> reasons = new LinkedList<String>();
      ConfigEntry newConfigEntry=new ConfigEntry(GoodHandlerEntry, null );
      //see if we're ok
      boolean ret=LDAPConnHandler.hasAcceptableConfiguration(newConfigEntry, reasons);
      assertTrue(ret);	
      LDAPConnHandler.applyNewConfiguration(newConfigEntry, true);
      LDAPConnHandler.finalizeConnectionHandler(reasonMsg, true);

   }
}