/*
* 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.backends.jeb;
import static org.testng.Assert.assertEquals;
import java.io.File;
import java.util.ArrayList;
import java.util.Random;
import java.util.TreeMap;
import org.opends.server.core.AddOperation;
import org.opends.server.core.DeleteOperation;
import org.opends.server.protocols.internal.InternalClientConnection;
import org.opends.server.tools.makeldif.MakeLDIFInputStream;
import org.opends.server.tools.makeldif.TemplateFile;
import org.opends.server.types.DN;
import org.opends.server.types.LDIFImportConfig;
import org.opends.server.types.ResultCode;
import org.opends.server.DirectoryServerTestCase;
import org.opends.server.TestCaseUtils;
import org.opends.messages.Message;
import org.opends.server.types.Entry;
import org.opends.server.util.LDIFReader;
import org.testng.annotations.Test;
/**
* An abstract base class for all Jeb backend test cases.
*/
@Test(groups = { "precommit", "jeb" }, sequential = true)
public abstract class JebTestCase extends DirectoryServerTestCase {
private TreeMap<DN,Entry> entryTreeMap = new TreeMap<DN,Entry>();
int numEntries;
/**
* This method takes an MakeLDIF template and a number of entries to create
* and adds the created entries into server.
*
* @param template MakeLDIF template to use.
* @param numEntries Number of entries to create and add.
* @throws Exception if the entries cannot be created or if the add
* fails.
*/
protected void
createLoadEntries(String[] template, int numEntries) throws Exception {
InternalClientConnection connection =
InternalClientConnection.getRootConnection();
String makeLDIFPath =
System.getProperty(TestCaseUtils.PROPERTY_BUILD_ROOT) +
File.separator + "resource"+File.separator+"MakeLDIF";
TemplateFile templateFile =
new TemplateFile(makeLDIFPath, new Random());
ArrayList<Message> warnings = new ArrayList<Message>();
templateFile.parse(template, warnings);
MakeLDIFInputStream ldifEntryStream =
new MakeLDIFInputStream(templateFile);
LDIFReader reader =
new LDIFReader(new LDIFImportConfig(ldifEntryStream));
for(int i =0; i<numEntries;i++) {
Entry entry = reader.readEntry(false);
entryTreeMap.put(entry.getDN(), entry);
AddOperation addOperation =
connection.processAdd(entry.getDN(),
entry.getObjectClasses(),
entry.getUserAttributes(),
entry.getOperationalAttributes());
assertEquals(addOperation.getResultCode(), ResultCode.SUCCESS,
"Add of this entry was not successful");
}
reader.close();
this.numEntries=numEntries;
}
/**This method should be used to remove the entries created in the
* above loadEntries method.
* Note that it starts at the last key and works backwards so that the leaf
* entries are removed first before the top level nodes.
*
* @throws Exception if the entries cannot be removed.
*/
protected void
removeLoadedEntries() throws Exception {
InternalClientConnection connection =
InternalClientConnection.getRootConnection();
for(int j =0; j < numEntries; j++) {
DN entryDN = entryTreeMap.lastKey();
DeleteOperation deleteOperation =
connection.processDelete(entryDN);
entryTreeMap.remove(entryDN);
assertEquals(deleteOperation.getResultCode(), ResultCode.SUCCESS);
}
}
}