/*
* Copyright (c) 2010-2014 Evolveum
*
* Licensed 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 com.evolveum.midpoint.testing.longtest;
import com.evolveum.midpoint.model.test.AbstractModelIntegrationTest;
import com.evolveum.midpoint.prism.PrismObject;
import com.evolveum.midpoint.schema.constants.MidPointConstants;
import com.evolveum.midpoint.schema.result.OperationResult;
import com.evolveum.midpoint.task.api.Task;
import com.evolveum.midpoint.test.IntegrationTestTools;
import com.evolveum.midpoint.test.util.TestUtil;
import com.evolveum.midpoint.util.logging.Trace;
import com.evolveum.midpoint.util.logging.TraceManager;
import com.evolveum.midpoint.xml.ns._public.common.common_3.*;
import org.apache.commons.io.IOUtils;
import org.apache.commons.lang.ArrayUtils;
import org.opends.server.types.Entry;
import org.opends.server.types.LDIFImportConfig;
import org.opends.server.util.LDIFException;
import org.opends.server.util.LDIFReader;
import org.springframework.test.annotation.DirtiesContext;
import org.springframework.test.context.ContextConfiguration;
import org.testng.AssertJUnit;
import org.testng.annotations.AfterClass;
import org.testng.annotations.Test;
import javax.xml.namespace.QName;
import java.io.File;
import java.io.IOException;
import static com.evolveum.midpoint.test.IntegrationTestTools.display;
/**
* @author lazyman
*/
@ContextConfiguration(locations = {"classpath:ctx-longtest-test-main.xml"})
@DirtiesContext(classMode = DirtiesContext.ClassMode.AFTER_CLASS)
public class TestGenericSynchronization extends AbstractModelIntegrationTest {
private static final Trace LOGGER = TraceManager.getTrace(TestGenericSynchronization.class);
private static final File SYSTEM_CONFIGURATION_FILE = new File(COMMON_DIR, "system-configuration.xml");
private static final String SYSTEM_CONFIGURATION_OID = SystemObjectsType.SYSTEM_CONFIGURATION.value();
private static final File USER_ADMINISTRATOR_FILE = new File(COMMON_DIR, "user-administrator.xml");
private static final String USER_ADMINISTRATOR_OID = SystemObjectsType.USER_ADMINISTRATOR.value();
private static final String USER_ADMINISTRATOR_USERNAME = "administrator";
private static final File ROLE_SUPERUSER_FILE = new File(COMMON_DIR, "role-superuser.xml");
private static final String ROLE_SUPERUSER_OID = "00000000-0000-0000-0000-000000000004";
private static final File RESOURCE_OPENDJ_FILE = new File(COMMON_DIR, "resource-opendj-generic-sync.xml");
private static final String RESOURCE_OPENDJ_NAME = "Localhost OpenDJ";
private static final String RESOURCE_OPENDJ_OID = "10000000-0000-0000-0000-000000000030";
private static final String RESOURCE_OPENDJ_NAMESPACE = MidPointConstants.NS_RI;
public static final File OBJECT_TEMPLATE_ORG_FILE = new File(COMMON_DIR, "object-template-org.xml");
public static final String OBJECT_TEMPLATE_ORG_OID = "10000000-0000-0000-0000-000000000231";
//222 org. units, 2160 users
// private static final int[] TREE_LEVELS = {2, 5, 7, 2};
// private static final int[] TREE_LEVELS_USER = {5, 5, 20, 5};
//1378 org. units, 13286 users
// private static final int[] TREE_LEVELS = {2, 8, 5, 16};
// private static final int[] TREE_LEVELS_USER = {3, 5, 5, 10};
//86 org. units, 636 users
private static final int[] TREE_LEVELS = {2, 7, 5};
private static final int[] TREE_LEVELS_USER = {3, 5, 8};
//18 org. units, 86 users
// private static final int[] TREE_LEVELS = {2, 8};
// private static final int[] TREE_LEVELS_USER = {3, 5};
// We already have some users in LDAP instance
private int ldapdUserCount = 4;
private int ldapOrgCount = 0;
private PrismObject<ResourceType> resourceOpenDj;
private boolean logCreateEntry = true;
@Override
protected void startResources() throws Exception {
openDJController.startCleanServer();
}
@AfterClass
public static void stopResources() throws Exception {
openDJController.stop();
}
@Override
public void initSystem(Task initTask, OperationResult initResult) throws Exception {
super.initSystem(initTask, initResult);
loadOpenDJWithData();
modelService.postInit(initResult);
// System Configuration and administrator
repoAddObjectFromFile(SYSTEM_CONFIGURATION_FILE, initResult);
PrismObject<UserType> userAdministrator = repoAddObjectFromFile(USER_ADMINISTRATOR_FILE, initResult);
repoAddObjectFromFile(ROLE_SUPERUSER_FILE, initResult);
login(userAdministrator);
importObjectFromFile(OBJECT_TEMPLATE_ORG_FILE, initResult);
setDefaultObjectTemplate(OrgType.COMPLEX_TYPE, OBJECT_TEMPLATE_ORG_OID);
// Resources
resourceOpenDj = importAndGetObjectFromFile(ResourceType.class, RESOURCE_OPENDJ_FILE, RESOURCE_OPENDJ_OID,
initTask, initResult);
openDJController.setResource(resourceOpenDj);
assumeAssignmentPolicy(AssignmentPolicyEnforcementType.RELATIVE);
}
private void loadOpenDJWithData() throws IOException, LDIFException {
long ldapPopStart = System.currentTimeMillis();
int count = loadOpenDJ(TREE_LEVELS, TREE_LEVELS_USER, openDJController.getSuffixPeople(), 0);
long ldapPopEnd = System.currentTimeMillis();
IntegrationTestTools.display("Loaded " + count + " LDAP entries in " + ((ldapPopEnd - ldapPopStart) / 1000) + " seconds");
}
private int loadOpenDJ(int[] TREE_SIZE, int[] USER_COUNT, String dnSuffix, int count)
throws IOException, LDIFException {
if (TREE_SIZE.length == 0) {
return count;
}
for (int i = 0; i < TREE_SIZE[0]; i++) {
String ou = "L" + TREE_SIZE.length + "o" + i;
String newSuffix = "ou=" + ou + ',' + dnSuffix;
Entry org = createOrgEntry(ou, dnSuffix);
logCreateEntry(org);
ldapOrgCount++;
openDJController.addEntry(org);
count++;
for (int u = 0; u < USER_COUNT[0]; u++) {
// We have to make uid globally unique. Otherwise correlation takes place and it will
// "collapse" several accounts into one user
String uid = "L" + TREE_SIZE.length + "o" + i + "u" + u + "c" + ldapdUserCount;
String sn = "Doe" + uid;
Entry user = createUserEntry(uid, newSuffix, sn);
logCreateEntry(user);
openDJController.addEntry(user);
ldapdUserCount++;
count++;
}
count += loadOpenDJ(ArrayUtils.remove(TREE_SIZE, 0), ArrayUtils.remove(USER_COUNT, 0), newSuffix, 0);
}
return count;
}
private void logCreateEntry(Entry entry) {
if (logCreateEntry ) {
System.out.println("Creating LDAP entry: " + entry.getDN());
LOGGER.trace("Creating LDAP entry: {}", entry.getDN());
}
}
private Entry createUserEntry(String uid, String suffix, String sn) throws IOException, LDIFException {
StringBuilder sb = new StringBuilder();
String dn = "uid=" + uid + "," + suffix;
sb.append("dn: ").append(dn).append('\n');
sb.append("objectClass: inetOrgPerson\n");
sb.append("uid: ").append(uid).append('\n');
sb.append("givenName: ").append("John").append('\n');
sb.append("cn: ").append("John " + sn).append('\n');
sb.append("sn: ").append(sn).append('\n');
LDIFImportConfig importConfig = new LDIFImportConfig(IOUtils.toInputStream(sb.toString(), "utf-8"));
LDIFReader ldifReader = new LDIFReader(importConfig);
Entry ldifEntry = ldifReader.readEntry();
return ldifEntry;
}
private Entry createOrgEntry(String ou, String suffix) throws IOException, LDIFException {
StringBuilder sb = new StringBuilder();
String dn = "ou=" + ou + "," + suffix;
sb.append("dn: ").append(dn).append("\n");
sb.append("objectClass: organizationalUnit\n");
sb.append("ou: ").append(ou).append("\n");
sb.append("description: ").append("This is sparta! ...or " + ou).append("\n");
LDIFImportConfig importConfig = new LDIFImportConfig(IOUtils.toInputStream(sb.toString(), "utf-8"));
LDIFReader ldifReader = new LDIFReader(importConfig);
Entry ldifEntry = ldifReader.readEntry();
return ldifEntry;
}
@Test
public void test100TreeImport() throws Exception {
final String TEST_NAME = "test100TreeImport";
TestUtil.displayTestTile(this, TEST_NAME);
Task task = taskManager.createTaskInstance(TestLdap.class.getName() + "." + TEST_NAME);
task.setOwner(getUser(USER_ADMINISTRATOR_OID));
OperationResult result = task.getResult();
// WHEN
modelService.importFromResource(RESOURCE_OPENDJ_OID,
new QName(RESOURCE_OPENDJ_NAMESPACE, "inetOrgPerson"), task, result);
// THEN
TestUtil.displayThen(TEST_NAME);
OperationResult subresult = result.getLastSubresult();
TestUtil.assertInProgress("importAccountsFromResource result", subresult);
waitForTaskFinish(task, true, 20000 + (ldapdUserCount + ldapOrgCount) * 2000);
// THEN
TestUtil.displayThen(TEST_NAME);
int userCount = modelService.countObjects(UserType.class, null, null, task, result);
display("Users", userCount);
AssertJUnit.assertEquals("Unexpected number of users", ldapdUserCount, userCount);
}
@Test
public void test200MoveRootChild() throws Exception {
final String TEST_NAME = "test200MoveRootChild";
TestUtil.displayTestTile(this, TEST_NAME);
//todo move one child of one root to root position
}
}