/*
* 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.shared.dsmlv2.searchResponse.searchResultReference;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;
import java.util.Collection;
import java.util.Map;
import org.apache.directory.shared.dsmlv2.AbstractResponseTest;
import org.apache.directory.shared.dsmlv2.DsmlControl;
import org.apache.directory.shared.dsmlv2.Dsmlv2ResponseParser;
import org.apache.directory.shared.dsmlv2.reponse.SearchResponse;
import org.apache.directory.shared.ldap.model.exception.LdapURLEncodingException;
import org.apache.directory.shared.ldap.model.message.Control;
import org.apache.directory.shared.ldap.model.message.SearchResultReference;
import org.apache.directory.shared.ldap.model.url.LdapUrl;
import org.apache.directory.shared.util.Strings;
import org.junit.Test;
import org.junit.runner.RunWith;
import com.mycila.junit.concurrent.Concurrency;
import com.mycila.junit.concurrent.ConcurrentJunitRunner;
/**
* Tests for the Search Result Reference Response parsing
*
* @author <a href="mailto:dev@directory.apache.org">Apache Directory Project</a>
*/
@RunWith(ConcurrentJunitRunner.class)
@Concurrency()
public class SearchResultReferenceTest extends AbstractResponseTest
{
/**
* Test parsing of a response with a (optional) Control element
*/
@Test
public void testResponseWith1Control()
{
Dsmlv2ResponseParser parser = null;
try
{
parser = new Dsmlv2ResponseParser( getCodec() );
parser.setInput( SearchResultReferenceTest.class.getResource( "response_with_1_control.xml" ).openStream(),
"UTF-8" );
parser.parse();
}
catch ( Exception e )
{
fail( e.getMessage() );
}
SearchResultReference searchResultReference = ( ( SearchResponse ) parser.getBatchResponse()
.getCurrentResponse().getDecorated() ).getCurrentSearchResultReference();
Map<String, Control> controls = searchResultReference.getControls();
assertEquals( 1, searchResultReference.getControls().size() );
Control control = controls.get( "1.2.840.113556.1.4.643" );
assertNotNull( control );
assertTrue( control.isCritical() );
assertEquals( "1.2.840.113556.1.4.643", control.getOid() );
assertEquals( "Some text", Strings.utf8ToString( ( byte[] ) ( ( DsmlControl<?> ) control ).getValue() ) );
}
/**
* Test parsing of a response with a (optional) Control element with empty value
*/
@Test
public void testResponseWith1ControlEmptyValue()
{
Dsmlv2ResponseParser parser = null;
try
{
parser = new Dsmlv2ResponseParser( getCodec() );
parser.setInput( SearchResultReferenceTest.class.getResource( "response_with_1_control_empty_value.xml" )
.openStream(), "UTF-8" );
parser.parse();
}
catch ( Exception e )
{
fail( e.getMessage() );
}
SearchResultReference searchResultReference = ( ( SearchResponse ) parser.getBatchResponse()
.getCurrentResponse().getDecorated() ).getCurrentSearchResultReference();
Map<String, Control> controls = searchResultReference.getControls();
assertEquals( 1, searchResultReference.getControls().size() );
Control control = controls.get( "1.2.840.113556.1.4.643" );
assertNotNull( control );
assertTrue( control.isCritical() );
assertEquals( "1.2.840.113556.1.4.643", control.getOid() );
assertFalse( ( ( DsmlControl<?> ) control ).hasValue() );
}
/**
* Test parsing of a response with 2 (optional) Control elements
*/
@Test
public void testResponseWith2Controls()
{
Dsmlv2ResponseParser parser = null;
try
{
parser = new Dsmlv2ResponseParser( getCodec() );
parser.setInput(
SearchResultReferenceTest.class.getResource( "response_with_2_controls.xml" ).openStream(), "UTF-8" );
parser.parse();
}
catch ( Exception e )
{
fail( e.getMessage() );
}
SearchResultReference searchResultReference = ( ( SearchResponse ) parser.getBatchResponse()
.getCurrentResponse().getDecorated() ).getCurrentSearchResultReference();
Map<String, Control> controls = searchResultReference.getControls();
assertEquals( 2, searchResultReference.getControls().size() );
Control control = controls.get( "1.2.840.113556.1.4.789" );
assertNotNull( control );
assertEquals( "1.2.840.113556.1.4.789", control.getOid() );
assertEquals( "Some other text", Strings.utf8ToString( ( byte[] ) ( ( DsmlControl<?> ) control ).getValue() ) );
}
/**
* Test parsing of a response with 3 (optional) Control elements without value
*/
@Test
public void testResponseWith3ControlsWithoutValue()
{
Dsmlv2ResponseParser parser = null;
try
{
parser = new Dsmlv2ResponseParser( getCodec() );
parser.setInput( SearchResultReferenceTest.class.getResource( "response_with_3_controls_without_value.xml" )
.openStream(), "UTF-8" );
parser.parse();
}
catch ( Exception e )
{
fail( e.getMessage() );
}
SearchResultReference searchResultReference = ( ( SearchResponse ) parser.getBatchResponse()
.getCurrentResponse().getDecorated() ).getCurrentSearchResultReference();
Map<String, Control> controls = searchResultReference.getControls();
assertEquals( 3, searchResultReference.getControls().size() );
Control control = controls.get( "1.2.840.113556.1.4.456" );
assertNotNull( control );
assertTrue( control.isCritical() );
assertEquals( "1.2.840.113556.1.4.456", control.getOid() );
assertFalse( ( ( DsmlControl<?> ) control ).hasValue() );
}
/**
* Test parsing of a Response with the (optional) requestID attribute
*/
@Test
public void testResponseWithRequestId()
{
Dsmlv2ResponseParser parser = null;
try
{
parser = new Dsmlv2ResponseParser( getCodec() );
parser.setInput( SearchResultReferenceTest.class.getResource( "response_with_requestID_attribute.xml" )
.openStream(), "UTF-8" );
parser.parse();
}
catch ( Exception e )
{
fail( e.getMessage() );
}
SearchResultReference searchResultReference = ( ( SearchResponse ) parser.getBatchResponse()
.getCurrentResponse().getDecorated() ).getCurrentSearchResultReference();
assertEquals( 456, searchResultReference.getMessageId() );
}
/**
* Test parsing of a Response with the (optional) requestID attribute equals 0
*/
@Test
public void testResponseWithRequestIdEquals0()
{
testParsingFail( SearchResultReferenceTest.class, "response_with_requestID_equals_0.xml" );
}
/**
* Test parsing of a response with 0 Ref
*/
@Test
public void testResponseWith0Ref()
{
testParsingFail( SearchResultReferenceTest.class, "response_with_0_ref.xml" );
}
/**
* Test parsing of a Response with 1 Ref
*/
@Test
public void testResponseWith1Ref()
{
Dsmlv2ResponseParser parser = null;
try
{
parser = new Dsmlv2ResponseParser( getCodec() );
parser.setInput( SearchResultReferenceTest.class.getResource( "response_with_1_ref.xml" ).openStream(),
"UTF-8" );
parser.parse();
}
catch ( Exception e )
{
fail( e.getMessage() );
}
SearchResultReference searchResultReference = ( ( SearchResponse ) parser.getBatchResponse()
.getCurrentResponse().getDecorated() ).getCurrentSearchResultReference();
Collection<String> references = searchResultReference.getReferral().getLdapUrls();
assertEquals( 1, references.size() );
try
{
assertTrue( references.contains( new LdapUrl( "ldap://localhost" ).toString() ) );
}
catch ( LdapURLEncodingException e )
{
fail();
}
}
/**
* Test parsing of a Response with 1 Ref
*/
@Test
public void testResponseWith1EmptyRef()
{
Dsmlv2ResponseParser parser = null;
try
{
parser = new Dsmlv2ResponseParser( getCodec() );
parser.setInput( SearchResultReferenceTest.class.getResource( "response_with_1_empty_ref.xml" )
.openStream(), "UTF-8" );
parser.parse();
}
catch ( Exception e )
{
fail( e.getMessage() );
}
SearchResultReference searchResultReference = ( ( SearchResponse ) parser.getBatchResponse()
.getCurrentResponse().getDecorated() ).getCurrentSearchResultReference();
Collection<String> references = searchResultReference.getReferral().getLdapUrls();
assertEquals( 0, references.size() );
}
/**
* Test parsing of a Response with 2 Ref
*/
@Test
public void testResponseWith2Ref()
{
Dsmlv2ResponseParser parser = null;
try
{
parser = new Dsmlv2ResponseParser( getCodec() );
parser.setInput( SearchResultReferenceTest.class.getResource( "response_with_2_ref.xml" ).openStream(),
"UTF-8" );
parser.parse();
}
catch ( Exception e )
{
fail( e.getMessage() );
}
SearchResultReference searchResultReference = ( ( SearchResponse ) parser.getBatchResponse()
.getCurrentResponse().getDecorated() ).getCurrentSearchResultReference();
Collection<String> references = searchResultReference.getReferral().getLdapUrls();
assertEquals( 2, references.size() );
try
{
assertTrue( references.contains( new LdapUrl( "ldap://localhost" ).toString() ) );
}
catch ( LdapURLEncodingException e )
{
fail();
}
try
{
assertTrue( references.contains( new LdapUrl( "ldap://www.apache.org" ).toString() ) );
}
catch ( LdapURLEncodingException e )
{
fail();
}
}
/**
* Test parsing of a response with 1 wrong Ref
*/
@Test
public void testResponseWith1WrongRef()
{
testParsingFail( SearchResultReferenceTest.class, "response_with_1_wrong_ref.xml" );
}
}