check expectedErrorCode and allow to modify a list of attributes

smaguin

21.42.2007 322a22bf908b3d4c9c53a92b3698d1027ea85767

check expectedErrorCode and allow to modify a list of attributes

 opends/tests/functional-tests/shared/java/modifyAnAttribute.java

@@ -37,12 +37,18 @@
import   javax.naming.directory.BasicAttributes;
import  javax.naming.*;
import  javax.naming.directory.ModificationItem;
import  java.util.HashSet;
import  java.util.StringTokenizer;
import java.util.Iterator;



/**
  *  modify an entry with an attribute 
  *  the operation can be a replace, delete or a add new attribute
  *  if expectedErrorCode is set, we compare it with the ldap error code returned in the exception
  *  the function returns 0 if the code are equals
  *  otherwise, returns 1 
  */

public class modifyAnAttribute {
@@ -57,6 +63,20 @@
     String dnToModify=null;
     String newAttributeValue=null;
     String changetype=null;
     String errorCode=null;
     String errorMessage=null;
     String listAttributesToModify=null;
     String expectedErrorCode="0";
     
     int ind1;
     String attributeName;
     String attributeValue;
     Hashtable envLdap  = new Hashtable();
     LdapContext ctx;  
     
     Attributes attributes = new BasicAttributes(); 
     HashSet attributeSet = new HashSet(); 

     
     for (int k=0; k< args.length; k++) {
         String opt1 = args[k]; 
@@ -86,14 +106,56 @@
         if (opt1.equals("-t")) {
             changetype = val1;
         }
         if (opt1.equals("-E")) {
             expectedErrorCode = val1;
         }
         if (opt1.equals("-l")) {
                listAttributesToModify = val1;
     		 
                ind1= val1.indexOf (":");

                attributeName=val1.substring (0,ind1);
                attributeValue=val1.substring (ind1+1);

                BasicAttribute attrToComplete = null;
    		 
                Iterator it = attributeSet.iterator();
                while (attrToComplete == null && it.hasNext()) {
                    BasicAttribute attr = (BasicAttribute) it.next();
                    if ((attr.getID()).equalsIgnoreCase(attributeName)) {
                        attrToComplete = attr;
                    }
                }
    		 	
                if (attrToComplete == null) {
                    attrToComplete = new BasicAttribute(attributeName);
                    attributeSet.add(attrToComplete);
                }
                attributeValue=attributeValue.replaceAll("QUOT","\\\"");
                attrToComplete.add(attributeValue);
         }
         k++;
     }
 
    newAttributeValue=newAttributeValue.replaceAll("QUOT","\\\"");
   
     
     if ( attributeToModify != null && newAttributeValue != null ) {
    	 
            BasicAttribute attrToComplete = null;
    	    
            attrToComplete = new BasicAttribute(attributeToModify);
            attributeSet.add(attrToComplete);
            newAttributeValue=newAttributeValue.replaceAll("QUOT","\\\"");
            attrToComplete.add(newAttributeValue);
     }
    
     
     Iterator it2 = attributeSet.iterator();
     while (it2.hasNext()) {
         BasicAttribute attr = (BasicAttribute)it2.next();
         attributes.put(attr);
     } 
     
    String provider = "ldap://"  + hostname + ":" + ldapPort  + "/";

    Hashtable envLdap  = new Hashtable();
    
    envLdap.put("java.naming.factory.initial", "com.sun.jndi.ldap.LdapCtxFactory");
    envLdap.put(Context.SECURITY_AUTHENTICATION, "simple");
@@ -101,9 +163,9 @@
    envLdap.put(Context.SECURITY_CREDENTIALS, credential);
    envLdap.put(Context.PROVIDER_URL, provider);

    LdapContext ctx;
	

    try {

        
        CompositeName entryDN = new CompositeName (dnToModify);
        System.out.println(changetype + " attribute " + attributeToModify + " for entry " + dnToModify);
@@ -111,36 +173,50 @@
        // connect to server
        ctx = new InitialLdapContext(envLdap, null);
        
        Attributes attrs = new BasicAttributes(true);
        Attribute attr = new BasicAttribute(attributeToModify);
        attr.add(newAttributeValue);
        attrs.put (attr);

        // replace attribute
         if (changetype.equals("replace")) {
            ctx.modifyAttributes(entryDN, LdapContext.REPLACE_ATTRIBUTE , attrs);
            ctx.modifyAttributes(entryDN, LdapContext.REPLACE_ATTRIBUTE , attributes);
         }
         else if (changetype.equals("add")) { 
             // add attribute   
             ctx.modifyAttributes(entryDN, LdapContext.ADD_ATTRIBUTE , attrs);
             ctx.modifyAttributes(entryDN, LdapContext.ADD_ATTRIBUTE , attributes);
         }
         else if (changetype.equals("delete")) { 
             // add attribute
             ctx.modifyAttributes(entryDN, LdapContext.REMOVE_ATTRIBUTE , attrs);
             ctx.modifyAttributes(entryDN, LdapContext.REMOVE_ATTRIBUTE , attributes);
         }
        
        ctx.close();

    } catch (CommunicationException e1) {
        String error = e1.getMessage();
        System.out.println(" Catch exception : " + e1.getMessage());
        System.exit(1);
        errorMessage = e1.getMessage();

    } catch (NamingException e2) {
        System.out.println(" Catch exception : " + e2.getMessage());
        System.exit(1);
        errorMessage = e2.getMessage();

    } catch (Exception e3) {
            System.out.println(" Catch exception : " + e3.getMessage());
            System.exit(1);
        errorMessage= e3.getMessage();
    }
    // No error, the modify is success 
    if ( errorMessage == null ) {
        errorCode="0";
    } 
    else {
        System.out.println (errorMessage);
        int ind=errorMessage.indexOf("-");
        errorCode=errorMessage.substring(18, ind-1);
    }
    // Compare the errorCode and the expected error Code
    int diff=expectedErrorCode.compareTo(errorCode); 
    
    if (diff == 0 ){
        // Got the expected error Code. The test is success
        System.exit(0);
    }
    else {
        System.out.println ("Error: modify request didn't return the expected error Code");
        System.out.println ("Expected " + expectedErrorCode + "; Got" + errorCode);
        System.exit(1);
    }
 
}
}