/* * 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 legal-notices/CDDLv1_0.txt * or http://forgerock.org/license/CDDLv1.0.html. * 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 legal-notices/CDDLv1_0.txt. * 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. * Portions Copyright 2014-2015 ForgeRock AS */ package org.opends.server.plugins; import java.io.ByteArrayInputStream; import java.util.ArrayList; import java.util.HashSet; import java.util.List; import java.util.UUID; import org.forgerock.opendj.config.server.ConfigException; import org.opends.server.TestCaseUtils; import org.opends.server.admin.server.AdminTestCaseUtils; import org.opends.server.admin.std.meta.EntryUUIDPluginCfgDefn; import org.opends.server.admin.std.server.EntryUUIDPluginCfg; import org.opends.server.api.plugin.PluginType; import org.opends.server.core.DirectoryServer; import org.opends.server.types.AttributeType; import org.opends.server.types.DN; import org.opends.server.types.Entry; import org.opends.server.types.LDIFImportConfig; import org.testng.annotations.BeforeClass; import org.testng.annotations.DataProvider; import org.testng.annotations.Test; import static org.testng.Assert.*; /** * This class defines a set of tests for the * org.opends.server.plugins.EntryUUIDPlugin class. */ public class EntryUUIDPluginTestCase extends PluginTestCase { /** * Ensures that the Directory Server is running. * * @throws Exception If an unexpected problem occurs. */ @BeforeClass public void startServer() throws Exception { TestCaseUtils.startServer(); } /** * Retrieves a set of valid configuration entries that may be used to * initialize the plugin. * * @throws Exception If an unexpected problem occurs. */ @DataProvider(name = "validConfigs") public Object[][] getValidConfigs() throws Exception { List<Entry> entries = TestCaseUtils.makeEntries( "dn: cn=Entry UUID,cn=Plugins,cn=config", "objectClass: top", "objectClass: ds-cfg-plugin", "objectClass: ds-cfg-entry-uuid-plugin", "cn: Entry UUID", "ds-cfg-java-class: org.opends.server.plugins.EntryUUIDPlugin", "ds-cfg-enabled: true", "ds-cfg-plugin-type: ldifImport", "ds-cfg-plugin-type: preOperationAdd", "", "dn: cn=Entry UUID,cn=Plugins,cn=config", "objectClass: top", "objectClass: ds-cfg-plugin", "objectClass: ds-cfg-entry-uuid-plugin", "cn: Entry UUID", "ds-cfg-java-class: org.opends.server.plugins.EntryUUIDPlugin", "ds-cfg-enabled: true", "ds-cfg-plugin-type: ldifImport", "", "dn: cn=Entry UUID,cn=Plugins,cn=config", "objectClass: top", "objectClass: ds-cfg-plugin", "objectClass: ds-cfg-entry-uuid-plugin", "cn: Entry UUID", "ds-cfg-java-class: org.opends.server.plugins.EntryUUIDPlugin", "ds-cfg-enabled: true", "ds-cfg-plugin-type: preOperationAdd"); Object[][] array = new Object[entries.size()][1]; for (int i=0; i < array.length; i++) { array[i] = new Object[] { entries.get(i) }; } return array; } /** * Tests the process of initializing the server with valid configurations. * * @param e The configuration entry to use for the initialization. * * @throws Exception If an unexpected problem occurs. */ @Test(dataProvider = "validConfigs") public void testInitializeWithValidConfigs(Entry e) throws Exception { HashSet<PluginType> pluginTypes = TestCaseUtils.getPluginTypes(e); EntryUUIDPluginCfg configuration = AdminTestCaseUtils.getConfiguration( EntryUUIDPluginCfgDefn.getInstance(), e); EntryUUIDPlugin plugin = new EntryUUIDPlugin(); plugin.initializePlugin(pluginTypes, configuration); plugin.finalizePlugin(); } /** * Tests the process of initializing the server with valid configurations but * without the entryUUID attribute type defined in the server. * * @param e The configuration entry to use for the initialization. * * @throws Exception If an unexpected problem occurs. */ @Test(dataProvider = "validConfigs") public void testInitializeWithValidConfigsWithoutSchema(Entry e) throws Exception { AttributeType entryUUIDType = DirectoryServer.getAttributeTypeOrNull("entryuuid"); DirectoryServer.deregisterAttributeType(entryUUIDType); HashSet<PluginType> pluginTypes = TestCaseUtils.getPluginTypes(e); EntryUUIDPluginCfg configuration = AdminTestCaseUtils.getConfiguration( EntryUUIDPluginCfgDefn.getInstance(), e); EntryUUIDPlugin plugin = new EntryUUIDPlugin(); plugin.initializePlugin(pluginTypes, configuration); plugin.finalizePlugin(); DirectoryServer.registerAttributeType(entryUUIDType, false); } /** * Retrieves a set of invalid configuration entries. * * @throws Exception If an unexpected problem occurs. */ @DataProvider(name = "invalidConfigs") public Object[][] getInvalidConfigs() throws Exception { ArrayList<Entry> entries = new ArrayList<>(); for (String s : PluginType.getPluginTypeNames()) { if (s.equalsIgnoreCase("ldifImport") || s.equalsIgnoreCase("preOperationAdd")) { continue; } Entry e = TestCaseUtils.makeEntry( "dn: cn=Entry UUID,cn=Plugins,cn=config", "objectClass: top", "objectClass: ds-cfg-plugin", "objectClass: ds-cfg-entry-uuid-plugin", "cn: Entry UUID", "ds-cfg-java-class: org.opends.server.plugins.EntryUUIDPlugin", "ds-cfg-enabled: true", "ds-cfg-plugin-type: " + s); entries.add(e); } Object[][] array = new Object[entries.size()][1]; for (int i=0; i < array.length; i++) { array[i] = new Object[] { entries.get(i) }; } return array; } /** * Tests the process of initializing the server with valid configurations. * * @param e The configuration entry to use for the initialization. * * @throws Exception If an unexpected problem occurs. */ @Test(dataProvider = "invalidConfigs", expectedExceptions = { ConfigException.class }) public void testInitializeWithInvalidConfigs(Entry e) throws Exception { HashSet<PluginType> pluginTypes = TestCaseUtils.getPluginTypes(e); EntryUUIDPluginCfg configuration = AdminTestCaseUtils.getConfiguration( EntryUUIDPluginCfgDefn.getInstance(), e); EntryUUIDPlugin plugin = new EntryUUIDPlugin(); plugin.initializePlugin(pluginTypes, configuration); } /** * Tests the <CODE>doLDIFImport</CODE> method. * * @throws Exception If an unexpected problem occurs. */ @Test public void testDoLDIFImport() throws Exception { TestCaseUtils.initializeTestBackend(true); String ldifString = TestCaseUtils.makeLdif("dn: o=test", "objectClass: top", "objectClass: organization", "o: test"); Entry e = TestCaseUtils.makeEntry("dn: o=test", "objectClass: top", "objectClass: organization", "o: test"); ByteArrayInputStream bais = new ByteArrayInputStream(ldifString.getBytes("UTF-8")); LDIFImportConfig importConfig = new LDIFImportConfig(bais); DN dn = DN.valueOf("cn=Entry UUID,cn=plugins,cn=config"); EntryUUIDPlugin plugin = (EntryUUIDPlugin) DirectoryServer.getPluginConfigManager().getRegisteredPlugin(dn); plugin.doLDIFImport(importConfig, e); assertNotNull(e.getAttribute("entryuuid")); } /** * Tests the <CODE>doLDIFImport</CODE> method with an entry that already has * the entryUUID operational attribute. * * @throws Exception If an unexpected problem occurs. */ @Test public void testDoLDIFImportWithExistingUUID() throws Exception { TestCaseUtils.initializeTestBackend(true); String uuid = UUID.randomUUID().toString(); String ldifString = TestCaseUtils.makeLdif("dn: o=test", "objectClass: top", "objectClass: organization", "o: test", "entryUUID: " + uuid); Entry e = TestCaseUtils.makeEntry("dn: o=test", "objectClass: top", "objectClass: organization", "o: test", "entryUUID: " + uuid); ByteArrayInputStream bais = new ByteArrayInputStream(ldifString.getBytes("UTF-8")); LDIFImportConfig importConfig = new LDIFImportConfig(bais); DN dn = DN.valueOf("cn=Entry UUID,cn=plugins,cn=config"); EntryUUIDPlugin plugin = (EntryUUIDPlugin) DirectoryServer.getPluginConfigManager().getRegisteredPlugin(dn); plugin.doLDIFImport(importConfig, e); assertNotNull(e.getAttribute("entryuuid")); } /** * Tests the <CODE>doPreOperationAdd</CODE> method. * * @throws Exception If an unexpected problem occurs. */ @Test public void testDoPreOperationAdd() throws Exception { TestCaseUtils.initializeTestBackend(true); Entry e = TestCaseUtils.addEntry("dn: cn=test,o=test", "objectClass: top", "objectClass: device", "cn: test"); assertNotNull(e.getAttribute("entryuuid")); } /** * Tests the <CODE>doPreOperationAdd</CODE> method with an entry that already * has the entryUUID operational attribute. * * @throws Exception If an unexpected problem occurs. */ @Test public void testDoPreOperationAddWithExistingUUID() throws Exception { TestCaseUtils.initializeTestBackend(true); Entry e = TestCaseUtils.addEntry("dn: cn=test,o=test", "objectClass: top", "objectClass: device", "cn: test", "entryUUID: " + UUID.randomUUID()); assertNotNull(e.getAttribute("entryuuid")); } }