/* * 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 * * * Copyright 2006-2008 Sun Microsystems, Inc. */ package org.opends.server.protocols.ldap; import org.opends.server.types.*; import org.opends.server.core.DirectoryServer; import org.opends.server.protocols.asn1.*; import static org.opends.server.util.ServerConstants.EOL; import org.opends.messages.Message; import org.testng.annotations.BeforeClass; import org.testng.annotations.Test; import static org.testng.Assert.assertEquals; import static org.testng.Assert.assertTrue; import java.util.ArrayList; import java.util.Iterator; /** * This class defines a set of tests for the * org.opends.server.protocol.ldap.ModifyResponseProtocolOp class. */ public class TestModifyResponseProtocolOp extends LdapTestCase { /** * The protocol op type for modify requests. */ public static final byte OP_TYPE_MODIFY_REQUEST = 0x66; /** * The protocol op type for modify responses. */ public static final byte OP_TYPE_MODIFY_RESPONSE = 0x67; /** * The result code for add result operations. */ private static final int resultCode = 10; /** * The error message to use for add result operations. */ private static final Message resultMsg = Message.raw("Test Successful"); /** * The DN to use for add result operations */ private DN dn; @BeforeClass public void setupDN() { //Setup the DN to use in the response tests. AttributeType attribute = DirectoryServer.getDefaultAttributeType("testAttribute"); AttributeValue attributeValue = AttributeValues.create(attribute, "testValue"); RDN[] rdns = new RDN[1]; rdns[0] = RDN.create(attribute, attributeValue); dn = new DN(rdns); } /** * Test to make sure the class processes the right LDAP op type. * * @throws Exception If the test failed unexpectedly. */ @Test public void testOpType() throws Exception { ModifyResponseProtocolOp modifyResponse = new ModifyResponseProtocolOp( resultCode); assertEquals(modifyResponse.getType(), OP_TYPE_MODIFY_RESPONSE); } /** * Test to make sure the class returns the correct protocol name. * * @throws Exception If the test failed unexpectedly. */ @Test public void testProtocolOpName() throws Exception { ModifyResponseProtocolOp modifyResponse = new ModifyResponseProtocolOp( resultCode); assertEquals(modifyResponse.getProtocolOpName(), "Modify Response"); } /** * Test the constructors to make sure the right objects are constructed. * * @throws Exception If the test failed unexpectedly. */ @Test public void testConstructors() throws Exception { ModifyResponseProtocolOp modifyResponse; ArrayList<LDAPAttribute> attributes; //Test to make sure the constructor with result code param works. modifyResponse = new ModifyResponseProtocolOp(resultCode); assertEquals(modifyResponse.getResultCode(), resultCode); //Test to make sure the constructor with result code and error message //params works. modifyResponse = new ModifyResponseProtocolOp(resultCode, resultMsg); assertEquals(modifyResponse.getErrorMessage(), resultMsg); assertEquals(modifyResponse.getResultCode(), resultCode); //Test to make sure the constructor with result code, message, dn, and //referal params works. ArrayList<String> referralURLs = new ArrayList<String>(); referralURLs.add("ds1.example.com"); referralURLs.add("ds2.example.com"); referralURLs.add("ds3.example.com"); modifyResponse = new ModifyResponseProtocolOp(resultCode, resultMsg, dn, referralURLs); assertEquals(modifyResponse.getErrorMessage(), resultMsg); assertEquals(modifyResponse.getResultCode(), resultCode); assertEquals(modifyResponse.getMatchedDN(), dn); assertEquals(modifyResponse.getReferralURLs(), referralURLs); } /** * Test the decode method when an empty element is passed * * @throws Exception If the test failed unexpectedly. */ @Test(expectedExceptions = LDAPException.class) public void testDecodeEmptyElement() throws Exception { ByteStringBuilder builder = new ByteStringBuilder(); ASN1Writer writer = ASN1.getWriter(builder); writer.writeStartSequence(OP_TYPE_MODIFY_RESPONSE); writer.writeEndSequence(); ASN1Reader reader = ASN1.getReader(builder.toByteString()); LDAPReader.readProtocolOp(reader); } /** * Test the decode method when an element with a invalid result code is * passed * * @throws Exception If the test failed unexpectedly. */ @Test(expectedExceptions = LDAPException.class) public void testDecodeInvalidResultCode() throws Exception { ByteStringBuilder builder = new ByteStringBuilder(); ASN1Writer writer = ASN1.getWriter(builder); writer.writeStartSequence(OP_TYPE_MODIFY_RESPONSE); writer.writeOctetString("Invalid Data"); writer.writeNull(); writer.writeNull(); writer.writeEndSequence(); ASN1Reader reader = ASN1.getReader(builder.toByteString()); LDAPReader.readProtocolOp(reader); } /** * Test the decode method when an element with a invalid dn is * passed. Never throws an exception as long as the element is not null. * This is the expected behavior. * * @throws Exception If the test failed unexpectedly. */ @Test public void testDecodeInvalidDN() throws Exception { ByteStringBuilder builder = new ByteStringBuilder(); ASN1Writer writer = ASN1.getWriter(builder); writer.writeStartSequence(OP_TYPE_MODIFY_RESPONSE); writer.writeInteger(resultCode); writer.writeNull(); writer.writeNull(); writer.writeEndSequence(); ASN1Reader reader = ASN1.getReader(builder.toByteString()); LDAPReader.readProtocolOp(reader); } /** * Test the decode method when an element with a invalid result message is * passed. Never throws an exception as long as the element is not null. * This is the expected behavior. * * @throws Exception If the test failed unexpectedly. */ @Test public void testDecodeInvalidResultMsg() throws Exception { ByteStringBuilder builder = new ByteStringBuilder(); ASN1Writer writer = ASN1.getWriter(builder); writer.writeStartSequence(OP_TYPE_MODIFY_RESPONSE); writer.writeInteger(resultCode); writer.writeOctetString(dn.toString()); writer.writeNull(); writer.writeEndSequence(); ASN1Reader reader = ASN1.getReader(builder.toByteString()); LDAPReader.readProtocolOp(reader); } /** * Test the decode method when an element with a invalid referral URL is * passed. Never throws an exception as long as the element is not null. * This is the expected behavior. * * @throws Exception If the test failed unexpectedly. */ @Test public void testDecodeInvalidReferralURLs() throws Exception { ByteStringBuilder builder = new ByteStringBuilder(); ASN1Writer writer = ASN1.getWriter(builder); writer.writeStartSequence(OP_TYPE_MODIFY_RESPONSE); writer.writeInteger(resultCode); writer.writeOctetString(dn.toString()); writer.writeOctetString(resultMsg.toString()); writer.writeNull(); writer.writeEndSequence(); ASN1Reader reader = ASN1.getReader(builder.toByteString()); LDAPReader.readProtocolOp(reader); } /** * Test the encode and decode methods and corner cases. * * @throws Exception If the test failed unexpectedly. */ @Test public void testEncodeDecode() throws Exception { ByteStringBuilder builder = new ByteStringBuilder(); ASN1Writer writer = ASN1.getWriter(builder); ModifyResponseProtocolOp modifyEncoded; ModifyResponseProtocolOp modifyDecoded; ArrayList<String> referralURLs = new ArrayList<String>(); referralURLs.add("ds1.example.com"); referralURLs.add("ds2.example.com"); referralURLs.add("ds3.example.com"); //Test case for a full encode decode operation with normal params. modifyEncoded = new ModifyResponseProtocolOp(resultCode, resultMsg, dn, referralURLs); modifyEncoded.write(writer); ASN1Reader reader = ASN1.getReader(builder.toByteString()); modifyDecoded = (ModifyResponseProtocolOp)LDAPReader.readProtocolOp(reader); assertEquals(modifyEncoded.getType(), OP_TYPE_MODIFY_RESPONSE); assertEquals(modifyEncoded.getMatchedDN().compareTo( modifyDecoded.getMatchedDN()), 0); assertEquals(modifyEncoded.getErrorMessage(), modifyDecoded.getErrorMessage()); assertEquals(modifyEncoded.getResultCode(), modifyDecoded.getResultCode()); assertEquals(modifyEncoded.getReferralURLs(), modifyDecoded.getReferralURLs()); //Test case for a full encode decode operation with an empty DN params. modifyEncoded = new ModifyResponseProtocolOp(resultCode, resultMsg, DN.nullDN(), referralURLs); builder.clear(); modifyEncoded.write(writer); reader = ASN1.getReader(builder.toByteString()); modifyDecoded = (ModifyResponseProtocolOp)LDAPReader.readProtocolOp(reader); assertEquals(modifyDecoded.getMatchedDN(), null); //Test case for a full empty referral url param. ArrayList<String> emptyReferralURLs = new ArrayList<String>(); modifyEncoded = new ModifyResponseProtocolOp(resultCode, resultMsg, dn, emptyReferralURLs); builder.clear(); modifyEncoded.write(writer); reader = ASN1.getReader(builder.toByteString()); modifyDecoded = (ModifyResponseProtocolOp)LDAPReader.readProtocolOp(reader); assertTrue(modifyDecoded.getReferralURLs() == null); //Test case for a full encode decode operation with resultCode param only. modifyEncoded = new ModifyResponseProtocolOp(resultCode); builder.clear(); modifyEncoded.write(writer); reader = ASN1.getReader(builder.toByteString()); modifyDecoded = (ModifyResponseProtocolOp)LDAPReader.readProtocolOp(reader); assertEquals(modifyDecoded.getMatchedDN(), null); assertEquals(modifyDecoded.getErrorMessage(), null); assertEquals(modifyEncoded.getResultCode(), modifyDecoded.getResultCode()); assertTrue(modifyDecoded.getReferralURLs() == null); } /** * Test the toString (single line) method. * * @throws Exception If the test fails unexpectedly. */ @Test public void TestToStringSingleLine() throws Exception { ModifyResponseProtocolOp modifyResponse; StringBuilder buffer = new StringBuilder(); StringBuilder key = new StringBuilder(); ArrayList<String> referralURLs = new ArrayList<String>(); referralURLs.add("ds1.example.com"); referralURLs.add("ds2.example.com"); referralURLs.add("ds3.example.com"); modifyResponse = new ModifyResponseProtocolOp(resultCode, resultMsg, dn, referralURLs); modifyResponse.toString(buffer); key.append("ModifyResponse(resultCode="+resultCode+", " + "errorMessage="+resultMsg+", matchedDN="+dn.toString()+", " + "referralURLs={"); Iterator<String> iterator = referralURLs.iterator(); key.append(iterator.next()); while (iterator.hasNext()) { key.append(", "); key.append(iterator.next()); } key.append("})"); assertEquals(buffer.toString(), key.toString()); } /** * Test the toString (multi line) method. * * @throws Exception If the test fails unexpectedly. */ @Test public void TestToStringMultiLine() throws Exception { ModifyResponseProtocolOp modifyResponse; StringBuilder buffer = new StringBuilder(); StringBuilder key = new StringBuilder(); ArrayList<String> referralURLs = new ArrayList<String>(); referralURLs.add("ds1.example.com"); referralURLs.add("ds2.example.com"); referralURLs.add("ds3.example.com"); int indent = 5; int i; modifyResponse = new ModifyResponseProtocolOp(resultCode, resultMsg, dn, referralURLs); modifyResponse.toString(buffer, indent); StringBuilder indentBuf = new StringBuilder(indent); for (i=0 ; i < indent; i++) { indentBuf.append(' '); } key.append(indentBuf); key.append("Modify Response"); key.append(EOL); key.append(indentBuf); key.append(" Result Code: "); key.append(resultCode); key.append(EOL); key.append(indentBuf); key.append(" Error Message: "); key.append(resultMsg); key.append(EOL); key.append(indentBuf); key.append(" Matched DN: "); key.append(dn.toString()); key.append(EOL); key.append(indentBuf); key.append(" Referral URLs: "); key.append(EOL); for (String url : referralURLs) { key.append(indentBuf); key.append(" "); key.append(url); key.append(EOL); } assertEquals(buffer.toString(), key.toString()); } }