/*
* 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.xdbm.impl.avl;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;
import java.io.File;
import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.UUID;
import javax.naming.directory.Attributes;
import org.apache.directory.server.constants.ApacheSchemaConstants;
import org.apache.directory.server.core.entry.ClonedServerEntry;
import org.apache.directory.server.xdbm.GenericIndex;
import org.apache.directory.server.xdbm.Index;
import org.apache.directory.server.xdbm.IndexEntry;
import org.apache.directory.server.xdbm.IndexNotFoundException;
import org.apache.directory.server.xdbm.StoreUtils;
import org.apache.directory.shared.ldap.model.constants.SchemaConstants;
import org.apache.directory.shared.ldap.model.csn.CsnFactory;
import org.apache.directory.shared.ldap.model.cursor.Cursor;
import org.apache.directory.shared.ldap.model.entry.Attribute;
import org.apache.directory.shared.ldap.model.entry.DefaultAttribute;
import org.apache.directory.shared.ldap.model.entry.DefaultEntry;
import org.apache.directory.shared.ldap.model.entry.DefaultModification;
import org.apache.directory.shared.ldap.model.entry.Entry;
import org.apache.directory.shared.ldap.model.entry.Modification;
import org.apache.directory.shared.ldap.model.entry.ModificationOperation;
import org.apache.directory.shared.ldap.model.exception.LdapNoSuchObjectException;
import org.apache.directory.shared.ldap.model.exception.LdapSchemaViolationException;
import org.apache.directory.shared.ldap.model.name.Dn;
import org.apache.directory.shared.ldap.model.name.Rdn;
import org.apache.directory.shared.ldap.model.schema.AttributeType;
import org.apache.directory.shared.ldap.model.schema.SchemaManager;
import org.apache.directory.shared.ldap.schemaextractor.SchemaLdifExtractor;
import org.apache.directory.shared.ldap.schemaextractor.impl.DefaultSchemaLdifExtractor;
import org.apache.directory.shared.ldap.schemaloader.LdifSchemaLoader;
import org.apache.directory.shared.ldap.schemamanager.impl.DefaultSchemaManager;
import org.apache.directory.shared.util.exception.Exceptions;
import org.junit.After;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Test;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
/**
* Unit test cases for AvlStore
*
* @author <a href="mailto:dev@directory.apache.org">Apache Directory Project</a>
*/
@SuppressWarnings("unchecked")
public class AvlStoreTest
{
private static final Logger LOG = LoggerFactory.getLogger( AvlStoreTest.class.getSimpleName() );
private static File wkdir;
private static AvlStore<Entry> store;
private static SchemaManager schemaManager = null;
private static Dn EXAMPLE_COM;
/** The OU AttributeType instance */
private static AttributeType OU_AT;
/** The DC AttributeType instance */
private static AttributeType DC_AT;
/** The ApacheAlias AttributeType instance */
private static AttributeType APACHE_ALIAS_AT;
@BeforeClass
public static void setup() throws Exception
{
String workingDirectory = System.getProperty( "workingDirectory" );
if ( workingDirectory == null )
{
String path = AvlStoreTest.class.getResource( "" ).getPath();
int targetPos = path.indexOf( "target" );
workingDirectory = path.substring( 0, targetPos + 6 );
}
File schemaRepository = new File( workingDirectory, "schema" );
SchemaLdifExtractor extractor = new DefaultSchemaLdifExtractor( new File( workingDirectory ) );
extractor.extractOrCopy( true );
LdifSchemaLoader loader = new LdifSchemaLoader( schemaRepository );
schemaManager = new DefaultSchemaManager( loader );
boolean loaded = schemaManager.loadAllEnabled();
if ( !loaded )
{
fail( "Schema load failed : " + Exceptions.printErrors(schemaManager.getErrors()) );
}
EXAMPLE_COM = new Dn( schemaManager, "dc=example,dc=com" );
OU_AT = schemaManager.getAttributeType( SchemaConstants.OU_AT );
DC_AT = schemaManager.getAttributeType( SchemaConstants.DC_AT );
APACHE_ALIAS_AT = schemaManager.getAttributeType( ApacheSchemaConstants.APACHE_ALIAS_AT );
}
@Before
public void createStore() throws Exception
{
destroyStore();
// initialize the store
store = new AvlStore<Entry>();
store.setId( "example" );
store.setSyncOnWrite( false );
store.addIndex( new AvlIndex( SchemaConstants.OU_AT_OID ) );
store.addIndex( new AvlIndex( SchemaConstants.UID_AT_OID ) );
StoreUtils.loadExampleData( store, schemaManager );
LOG.debug( "Created new store" );
}
@After
public void destroyStore() throws Exception
{
}
@Test
public void testSimplePropertiesUnlocked() throws Exception
{
AvlStore<Attributes> store = new AvlStore<Attributes>();
store.setSyncOnWrite( true ); // for code coverage
assertNull( store.getAliasIndex() );
store.addIndex( new AvlIndex<String, Attributes>( ApacheSchemaConstants.APACHE_ALIAS_AT_OID ) );
assertNotNull( store.getAliasIndex() );
assertEquals( 0, store.getCacheSize() );
assertNull( store.getPresenceIndex() );
store.addIndex( new AvlIndex<String, Attributes>( ApacheSchemaConstants.APACHE_EXISTENCE_AT_OID ) );
assertNotNull( store.getPresenceIndex() );
assertNull( store.getOneLevelIndex() );
store.addIndex( new AvlIndex<Long, Attributes>( ApacheSchemaConstants.APACHE_ONE_LEVEL_AT_OID ) );
assertNotNull( store.getOneLevelIndex() );
assertNull( store.getSubLevelIndex() );
store.addIndex( new AvlIndex<Long, Attributes>( ApacheSchemaConstants.APACHE_SUB_LEVEL_AT_OID ) );
assertNotNull( store.getSubLevelIndex() );
assertNull( store.getId() );
store.setId( "foo" );
assertEquals( "foo", store.getId() );
assertNull( store.getRdnIndex() );
store.addIndex( new AvlRdnIndex( ApacheSchemaConstants.APACHE_RDN_AT_OID ) );
assertNotNull( store.getRdnIndex() );
assertNull( store.getOneAliasIndex() );
store.addIndex( new AvlIndex<Long, Attributes>( ApacheSchemaConstants.APACHE_ONE_ALIAS_AT_OID ) );
assertNotNull( store.getOneAliasIndex() );
assertNull( store.getSubAliasIndex() );
store.addIndex( new AvlIndex<Long, Attributes>( ApacheSchemaConstants.APACHE_SUB_ALIAS_AT_OID ) );
assertNotNull( store.getSubAliasIndex() );
assertNull( store.getSuffixDn() );
store.setSuffixDn( EXAMPLE_COM );
assertEquals( "dc=example,dc=com", store.getSuffixDn().getName() );
assertNotNull( store.getSuffixDn() );
assertEquals( 0, store.getUserIndices().size() );
store.addIndex( new AvlIndex<Object, Attributes>( "1.2.3.4" ) );
assertEquals( 1, store.getUserIndices().size() );
assertNull( store.getPartitionPath() );
store.setPartitionPath( new File( "." ).toURI() );
assertNull( store.getPartitionPath() );
assertFalse( store.isInitialized() );
assertFalse( store.isSyncOnWrite() );
store.setSyncOnWrite( false );
assertFalse( store.isSyncOnWrite() );
store.sync();
store.destroy();
}
@Test
public void testSimplePropertiesLocked() throws Exception
{
assertNotNull( store.getAliasIndex() );
try
{
store.addIndex( new AvlIndex<String, Entry>( ApacheSchemaConstants.APACHE_ALIAS_AT_OID ) );
fail();
}
catch ( IllegalStateException e )
{
}
assertEquals( 0, store.getCacheSize() );
assertNotNull( store.getPresenceIndex() );
try
{
store.addIndex( new AvlIndex<String, Entry>( ApacheSchemaConstants.APACHE_EXISTENCE_AT_OID ) );
fail();
}
catch ( IllegalStateException e )
{
}
assertNotNull( store.getOneLevelIndex() );
try
{
store.addIndex( new AvlIndex<Long, Entry>( ApacheSchemaConstants.APACHE_ONE_LEVEL_AT_OID ) );
fail();
}
catch ( IllegalStateException e )
{
}
assertNotNull( store.getSubLevelIndex() );
try
{
store.addIndex( new AvlIndex<Long, Entry>( ApacheSchemaConstants.APACHE_SUB_LEVEL_AT_OID ) );
fail();
}
catch ( IllegalStateException e )
{
}
assertNotNull( store.getId() );
try
{
store.setId( "foo" );
fail();
}
catch ( IllegalStateException e )
{
}
assertNotNull( store.getNdnIndex() );
assertNotNull( store.getRdnIndex() );
try
{
store.addIndex( new AvlRdnIndex( ApacheSchemaConstants.APACHE_RDN_AT_OID ) );
fail();
}
catch ( IllegalStateException e )
{
}
assertNotNull( store.getOneAliasIndex() );
try
{
store.addIndex( new AvlIndex<Long, Entry>( ApacheSchemaConstants.APACHE_ONE_ALIAS_AT_OID ) );
fail();
}
catch ( IllegalStateException e )
{
}
assertNotNull( store.getSubAliasIndex() );
try
{
store.addIndex( new AvlIndex<Long, Entry>( ApacheSchemaConstants.APACHE_SUB_ALIAS_AT_OID ) );
fail();
}
catch ( IllegalStateException e )
{
}
assertNotNull( store.getSuffixDn() );
try
{
store.setSuffixDn( EXAMPLE_COM );
fail();
}
catch ( IllegalStateException e )
{
}
Iterator<String> systemIndices = store.systemIndices();
for ( int i = 0; i < 10; i++ )
{
assertTrue( systemIndices.hasNext() );
assertNotNull( systemIndices.next() );
}
assertFalse( systemIndices.hasNext() );
assertNotNull( store.getSystemIndex( APACHE_ALIAS_AT ) );
try
{
store.getSystemIndex( "bogus" );
fail();
}
catch ( IndexNotFoundException e )
{
}
try
{
store.getSystemIndex( DC_AT );
fail();
}
catch ( IndexNotFoundException e )
{
}
assertNotNull( store.getSuffixDn() );
assertEquals( 2, store.getUserIndices().size() );
assertFalse( store.hasUserIndexOn( DC_AT ) );
assertTrue( store.hasUserIndexOn( OU_AT ) );
assertTrue( store.hasSystemIndexOn( APACHE_ALIAS_AT ) );
Iterator<String> userIndices = store.userIndices();
assertTrue( userIndices.hasNext() );
assertNotNull( userIndices.next() );
assertTrue( userIndices.hasNext() );
assertNotNull( userIndices.next() );
assertFalse( userIndices.hasNext() );
assertNotNull( store.getUserIndex( OU_AT ) );
try
{
store.getUserIndex( "bogus" );
fail();
}
catch ( IndexNotFoundException e )
{
}
try
{
store.getUserIndex( "dc" );
fail();
}
catch ( IndexNotFoundException e )
{
}
assertNull( store.getPartitionPath() );
assertTrue( store.isInitialized() );
assertFalse( store.isSyncOnWrite() );
store.sync();
}
@Test
public void testPersistentProperties() throws Exception
{
assertNull( store.getProperty( "foo" ) );
store.setProperty( "foo", "bar" );
assertEquals( "bar", store.getProperty( "foo" ) );
}
@Test
public void testFreshStore() throws Exception
{
Dn dn = new Dn( schemaManager, "o=Good Times Co." );
assertEquals( 1L, ( long ) store.getEntryId( dn ) );
assertEquals( 11, store.count() );
// note that the suffix entry returns 0 for it's parent which does not exist
assertEquals( 0L, ( long ) store.getParentId( store.getEntryId( dn ) ) );
assertNull( store.getParentId( 0L ) );
// should NOW be allowed
store.delete( 1L );
}
@Test
public void testEntryOperations() throws Exception
{
assertEquals( 3, store.getChildCount( 1L ) );
Cursor<IndexEntry<Long, Entry, Long>> cursor = store.list( 1L );
assertNotNull( cursor );
cursor.beforeFirst();
assertTrue( cursor.next() );
assertEquals( 2L, ( long ) cursor.get().getId() );
assertTrue( cursor.next() );
assertEquals( 3, store.getChildCount( 1L ) );
store.delete( 2L );
assertEquals( 2, store.getChildCount( 1L ) );
assertEquals( 10, store.count() );
// add an alias and delete to test dropAliasIndices method
Dn dn = new Dn( schemaManager, "commonName=Jack Daniels,ou=Apache,ou=Board of Directors,o=Good Times Co." );
DefaultEntry entry = new DefaultEntry( schemaManager, dn );
entry.add( "objectClass", "top", "alias", "extensibleObject" );
entry.add( "ou", "Apache" );
entry.add( "commonName", "Jack Daniels" );
entry.add( "aliasedObjectName", "cn=Jack Daniels,ou=Engineering,o=Good Times Co." );
entry.add( "entryCSN", new CsnFactory( 1 ).newInstance().toString() );
entry.add( "entryUUID", UUID.randomUUID().toString() );
store.add( entry );
store.delete( 12L ); // drops the alias indices
}
@Test
public void testSubLevelIndex() throws Exception
{
Index idx = store.getSubLevelIndex();
assertEquals( 19, idx.count() );
Cursor<IndexEntry<Long, Attributes, Long>> cursor = idx.forwardCursor( 2L );
assertTrue( cursor.next() );
assertEquals( 2, ( long ) cursor.get().getId() );
assertTrue( cursor.next() );
assertEquals( 5, ( long ) cursor.get().getId() );
assertTrue( cursor.next() );
assertEquals( 6, ( long ) cursor.get().getId() );
assertFalse( cursor.next() );
idx.drop( 5L );
cursor = idx.forwardCursor( 2L );
assertTrue( cursor.next() );
assertEquals( 2, ( long ) cursor.get().getId() );
assertTrue( cursor.next() );
assertEquals( 6, ( long ) cursor.get().getId() );
assertFalse( cursor.next() );
// dn id 12
Dn martinDn = new Dn( schemaManager, "cn=Marting King,ou=Sales,o=Good Times Co." );
DefaultEntry entry = new DefaultEntry( schemaManager, martinDn );
entry.add( "objectClass", "top", "person", "organizationalPerson" );
entry.add( "ou", "Sales" );
entry.add( "cn", "Martin King" );
entry.add( "entryCSN", new CsnFactory( 1 ).newInstance().toString() );
entry.add( "entryUUID", UUID.randomUUID().toString() );
store.add( entry );
cursor = idx.forwardCursor( 2L );
cursor.afterLast();
assertTrue( cursor.previous() );
assertEquals( 12, ( long ) cursor.get().getId() );
Dn newParentDn = new Dn( schemaManager, "ou=Board of Directors,o=Good Times Co." );
Dn newDn = newParentDn.add( martinDn.getRdn() );
store.move( martinDn, newParentDn, newDn, new ClonedServerEntry( entry ) );
cursor = idx.forwardCursor( 3L );
cursor.afterLast();
assertTrue( cursor.previous() );
assertEquals( 12, ( long ) cursor.get().getId() );
// dn id 13
Dn marketingDn = new Dn( schemaManager, "ou=Marketing,ou=Sales,o=Good Times Co." );
entry = new DefaultEntry( schemaManager, marketingDn );
entry.add( "objectClass", "top", "organizationalUnit" );
entry.add( "ou", "Marketing" );
entry.add( "entryCSN", new CsnFactory( 1 ).newInstance().toString() );
entry.add( "entryUUID", UUID.randomUUID().toString() );
store.add( entry );
// dn id 14
Dn jimmyDn = new Dn( schemaManager, "cn=Jimmy Wales,ou=Marketing, ou=Sales,o=Good Times Co." );
entry = new DefaultEntry( schemaManager, jimmyDn );
entry.add( "objectClass", "top", "person", "organizationalPerson" );
entry.add( "ou", "Marketing" );
entry.add( "cn", "Jimmy Wales" );
entry.add( "entryCSN", new CsnFactory( 1 ).newInstance().toString() );
entry.add( "entryUUID", UUID.randomUUID().toString() );
store.add( entry );
newDn = newParentDn.add( marketingDn.getRdn() );
store.move( marketingDn, newParentDn, newDn, new ClonedServerEntry( entry ) );
cursor = idx.forwardCursor( 3L );
cursor.afterLast();
assertTrue( cursor.previous() );
assertEquals( 14, ( long ) cursor.get().getId() );
assertTrue( cursor.previous() );
assertEquals( 13, ( long ) cursor.get().getId() );
assertTrue( cursor.previous() );
assertEquals( 12, ( long ) cursor.get().getId() );
assertTrue( cursor.previous() );
assertEquals( 10, ( long ) cursor.get().getId() );
assertTrue( cursor.previous() );
assertEquals( 9, ( long ) cursor.get().getId() );
assertTrue( cursor.previous() );
assertEquals( 7, ( long ) cursor.get().getId() );
assertTrue( cursor.previous() );
assertEquals( 3, ( long ) cursor.get().getId() );
assertFalse( cursor.previous() );
}
@Test
public void testConvertIndex() throws Exception
{
Index nonAvlIndex = new GenericIndex( "ou", 10, new File( "." ).toURI() );
Method convertIndex = store.getClass().getDeclaredMethod( "convertAndInit", Index.class );
convertIndex.setAccessible( true );
Object obj = convertIndex.invoke( store, nonAvlIndex );
assertNotNull( obj );
assertEquals( AvlIndex.class, obj.getClass() );
}
@Test(expected = LdapNoSuchObjectException.class)
public void testAddWithoutParentId() throws Exception
{
Dn dn = new Dn( schemaManager, "cn=Marting King,ou=Not Present,o=Good Times Co." );
DefaultEntry entry = new DefaultEntry( schemaManager, dn );
entry.add( "objectClass", "top", "person", "organizationalPerson" );
entry.add( "ou", "Not Present" );
entry.add( "cn", "Martin King" );
store.add( entry );
}
@Test(expected = LdapSchemaViolationException.class)
public void testAddWithoutObjectClass() throws Exception
{
Dn dn = new Dn( schemaManager, "cn=Martin King,ou=Sales,o=Good Times Co." );
DefaultEntry entry = new DefaultEntry( schemaManager, dn );
entry.add( "ou", "Sales" );
entry.add( "cn", "Martin King" );
store.add( entry );
}
@Test
public void testModifyAddOUAttrib() throws Exception
{
Dn dn = new Dn( schemaManager, "cn=JOhnny WAlkeR,ou=Sales,o=Good Times Co." );
List<Modification> mods = new ArrayList<Modification>();
Attribute attrib = new DefaultAttribute( SchemaConstants.OU_AT, OU_AT );
attrib.add( "Engineering" );
Modification add = new DefaultModification( ModificationOperation.ADD_ATTRIBUTE, attrib );
mods.add( add );
store.modify( dn, mods );
}
@Test
public void testRename() throws Exception
{
Dn dn = new Dn( schemaManager, "cn=Pivate Ryan,ou=Engineering,o=Good Times Co." );
DefaultEntry entry = new DefaultEntry( schemaManager, dn );
entry.add( "objectClass", "top", "person", "organizationalPerson" );
entry.add( "ou", "Engineering" );
entry.add( "cn", "Private Ryan" );
entry.add( "entryCSN", new CsnFactory( 1 ).newInstance().toString() );
entry.add( "entryUUID", UUID.randomUUID().toString() );
store.add( entry );
Rdn rdn = new Rdn( "sn=James" );
store.rename( dn, rdn, true );
}
@Test
public void testRenameEscaped() throws Exception
{
Dn dn = new Dn( schemaManager, "cn=Pivate Ryan,ou=Engineering,o=Good Times Co." );
DefaultEntry entry = new DefaultEntry( schemaManager, dn );
entry.add( "objectClass", "top", "person", "organizationalPerson" );
entry.add( "ou", "Engineering" );
entry.add( "cn", "Private Ryan" );
entry.add( "entryCSN", new CsnFactory( 1 ).newInstance().toString() );
entry.add( "entryUUID", UUID.randomUUID().toString() );
store.add( entry );
Rdn rdn = new Rdn( "sn=Ja\\+es" );
store.rename( dn, rdn, true );
Dn dn2 = new Dn( schemaManager, "sn=Ja\\+es,ou=Engineering,o=Good Times Co." );
Long id = store.getEntryId( dn2 );
assertNotNull( id );
Entry entry2 = store.lookup( id );
assertEquals( "ja+es", entry2.get( "sn" ).getString() );
}
@Test
public void testMove() throws Exception
{
Dn childDn = new Dn( schemaManager, "cn=Pivate Ryan,ou=Engineering,o=Good Times Co." );
DefaultEntry childEntry = new DefaultEntry( schemaManager, childDn );
childEntry.add( "objectClass", "top", "person", "organizationalPerson" );
childEntry.add( "ou", "Engineering" );
childEntry.add( "cn", "Private Ryan" );
childEntry.add( "entryCSN", new CsnFactory( 1 ).newInstance().toString() );
childEntry.add( "entryUUID", UUID.randomUUID().toString() );
store.add( childEntry );
Dn parentDn = new Dn( schemaManager, "ou=Sales,o=Good Times Co." );
Rdn rdn = new Rdn( "cn=Ryan" );
store.moveAndRename( childDn, parentDn, rdn, new ClonedServerEntry( childEntry ), true );
// to drop the alias indices
childDn = new Dn( schemaManager, "commonName=Jim Bean,ou=Apache,ou=Board of Directors,o=Good Times Co." );
parentDn = new Dn( schemaManager, "ou=Engineering,o=Good Times Co." );
assertEquals( 3, store.getSubAliasIndex().count() );
Dn newDn = parentDn.add( childDn.getRdn() );
store.move( childDn, parentDn, newDn, childEntry );
assertEquals( 4, store.getSubAliasIndex().count() );
}
@Test
public void testModifyAdd() throws Exception
{
Dn dn = new Dn( schemaManager, "cn=JOhnny WAlkeR,ou=Sales,o=Good Times Co." );
List<Modification> mods = new ArrayList<Modification>();
Attribute attrib = new DefaultAttribute( SchemaConstants.SURNAME_AT, schemaManager
.lookupAttributeTypeRegistry( SchemaConstants.SURNAME_AT ) );
String attribVal = "Walker";
attrib.add( attribVal );
Modification add = new DefaultModification( ModificationOperation.ADD_ATTRIBUTE, attrib );
mods.add( add );
Entry lookedup = store.lookup( store.getEntryId( dn ) );
store.modify( dn, mods );
assertTrue( lookedup.get( "sn" ).contains( attribVal ) );
// testing the store.modify( dn, mod, entry ) API
Entry entry = new DefaultEntry( schemaManager, dn );
attribVal = "+1974045779";
entry.add( "telephoneNumber", attribVal );
store.modify( dn, ModificationOperation.ADD_ATTRIBUTE, entry );
lookedup = store.lookup( store.getEntryId( dn ) );
assertTrue( lookedup.get( "telephoneNumber" ).contains( attribVal ) );
}
@Test
public void testModifyReplace() throws Exception
{
Dn dn = new Dn( schemaManager, "cn=JOhnny WAlkeR,ou=Sales,o=Good Times Co." );
List<Modification> mods = new ArrayList<Modification>();
Attribute attrib = new DefaultAttribute( SchemaConstants.SN_AT, schemaManager
.lookupAttributeTypeRegistry( SchemaConstants.SN_AT_OID ) );
String attribVal = "Johnny";
attrib.add( attribVal );
Modification add = new DefaultModification( ModificationOperation.REPLACE_ATTRIBUTE, attrib );
mods.add( add );
Entry lookedup = store.lookup( store.getEntryId( dn ) );
assertEquals( "WAlkeR", lookedup.get( "sn" ).get().getString() ); // before replacing
store.modify( dn, mods );
assertEquals( attribVal, lookedup.get( "sn" ).get().getString() );
// testing the store.modify( dn, mod, entry ) API
Entry entry = new DefaultEntry( schemaManager, dn );
attribVal = "JWalker";
entry.add( "sn", attribVal );
store.modify( dn, ModificationOperation.REPLACE_ATTRIBUTE, entry );
assertEquals( attribVal, lookedup.get( "sn" ).get().getString() );
}
@Test
public void testModifyRemove() throws Exception
{
Dn dn = new Dn( schemaManager, "cn=JOhnny WAlkeR,ou=Sales,o=Good Times Co." );
List<Modification> mods = new ArrayList<Modification>();
Attribute attrib = new DefaultAttribute( SchemaConstants.SN_AT, schemaManager
.lookupAttributeTypeRegistry( SchemaConstants.SN_AT_OID ) );
Modification add = new DefaultModification( ModificationOperation.REMOVE_ATTRIBUTE, attrib );
mods.add( add );
Entry lookedup = store.lookup( store.getEntryId( dn ) );
assertNotNull( lookedup.get( "sn" ).get() );
store.modify( dn, mods );
assertNull( lookedup.get( "sn" ) );
// testing the store.modify( dn, mod, entry ) API
Entry entry = new DefaultEntry( schemaManager, dn );
// add an entry for the sake of testing the remove operation
entry.add( "sn", "JWalker" );
store.modify( dn, ModificationOperation.ADD_ATTRIBUTE, entry );
assertNotNull( lookedup.get( "sn" ) );
store.modify( dn, ModificationOperation.REMOVE_ATTRIBUTE, entry );
assertNull( lookedup.get( "sn" ) );
}
@Test
public void testModifyReplaceNonExistingIndexAttribute() throws Exception
{
Dn dn = new Dn( schemaManager, "cn=Tim B,ou=Sales,o=Good Times Co." );
DefaultEntry entry = new DefaultEntry( schemaManager, dn );
entry.add( "objectClass", "top", "person", "organizationalPerson" );
entry.add( "cn", "Tim B" );
entry.add( "entryCSN", new CsnFactory( 1 ).newInstance().toString() );
entry.add( "entryUUID", UUID.randomUUID().toString() );
store.add( entry );
List<Modification> mods = new ArrayList<Modification>();
Attribute attrib = new DefaultAttribute( SchemaConstants.OU_AT, OU_AT );
String attribVal = "Marketing";
attrib.add( attribVal );
Modification add = new DefaultModification( ModificationOperation.REPLACE_ATTRIBUTE, attrib );
mods.add( add );
Entry lookedup = store.lookup( store.getEntryId( dn ) );
assertNull( lookedup.get( "ou" ) ); // before replacing
store.modify( dn, mods );
assertEquals( attribVal, lookedup.get( "ou" ).get().getString() );
}
}