hs.hsadmin/hsarback/test/de/hsadmin/remote/PacMigrationTest.java
2014-03-21 18:44:19 +01:00

295 lines
11 KiB
Java

package de.hsadmin.remote;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;
import java.util.Calendar;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;
import org.apache.xmlrpc.XmlRpcException;
import org.apache.xmlrpc.client.XmlRpcClient;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import de.hsadmin.core.util.Config;
import de.hsadmin.core.util.TextUtil;
import de.hsadmin.hostsharing.BasePacType;
public class PacMigrationTest {
private static final String MODULE = "pac";
private XmlRpcClient client;
private RemoteCASHelper cas;
private Config config;
@Before
public void setUp() throws Exception {
client = RemoteTestHelper.getClient();
cas = new RemoteCASHelper();
config = Config.getInstance();
}
@After
public void tearDown() throws Exception {
client = null;
cas = null;
}
@Test
public void testSW() {
testCreateSW();
testUpdateSW();
testDeleteSW();
}
@Test
public void testDW() {
testCreateDW();
testUpdateDW();
}
public void testCreateDW() {
int count = getPacsCount();
String user = "ad";
try {
String grantingTicketURL = cas.getGrantingTicketURL(user);
Object execute = addPac(user, grantingTicketURL,
"aaa04", BasePacType.PAC_DW, "176.9.242.77",
640, 4, 1);
assertEquals(count + 1, getPacsCount());
validateAddedPac(execute, BasePacType.PAC_DW, 640, 4, 1);
setPacAdminPassword(user, grantingTicketURL, "aaa04", "test123");
} catch (XmlRpcException e) {
fail(e.getMessage());
}
}
public void testCreateSW() {
int count = getPacsCount();
String user = "ad";
try {
String grantingTicketURL = cas.getGrantingTicketURL(user);
Object execute = addPac(user, grantingTicketURL,
"aaa05", BasePacType.PAC_SW, "176.9.242.77",
1024, 8, -1);
assertEquals(count + 1, getPacsCount());
validateAddedPac(execute, BasePacType.PAC_SW, 1024, 8, -1);
setPacAdminPassword(user, grantingTicketURL, "aaa05", "test123");
} catch (XmlRpcException e) {
fail(e.getMessage());
}
}
public void testUpdateDW() {
String user = "ad";
try {
String grantingTicketURL = cas.getGrantingTicketURL(user);
Map<String, Object> setParams = new HashMap<String, Object>();
Map<String, String> components = new HashMap<String, String>();
setParams.put("basepac", BasePacType.PAC_WEB);
setParams.put("components", components);
// components.put("QUOTA", "2048"); // does not yet work
Map<String, String> whereParams = new HashMap<String, String>();
whereParams.put("name", "aaa04");
Object[] params = new Object[] { user,
cas.getServiceTicket(grantingTicketURL, RemoteTestHelper.getBackendURL()),
setParams, whereParams };
Object execute = client.execute(MODULE + ".update", params);
assertNotNull(execute);
params = new Object[] { user,
cas.getServiceTicket(grantingTicketURL, RemoteTestHelper.getBackendURL()),
whereParams };
execute = client.execute(MODULE + ".search", params);
assertNotNull(execute);
assertTrue(execute instanceof Object[]);
Object[] untypedResultArray = (Object[]) execute;
assertEquals(1, untypedResultArray.length);
assertTrue(untypedResultArray[0] instanceof Map<?,?>);
@SuppressWarnings("unchecked")
Map<String, ?> pacResultMap = (Map<String, ?>) untypedResultArray[0];
assertEquals(BasePacType.PAC_WEB, pacResultMap.get("basepac"));
Object compMapObj = pacResultMap.get("components");
assertTrue(compMapObj instanceof Map<?,?>);
@SuppressWarnings("unchecked")
Map<String,String> compMap = (Map<String, String>) compMapObj;
assertEquals("1024", compMap.get("QUOTA"));
// assertEquals("2048", compMap.get("QUOTA")); // does not yet work
assertEquals("5", compMap.get("TRAFFIC"));
} catch (XmlRpcException e) {
fail(e.getMessage());
}
}
public void testUpdateSW() {
String user = "ad";
try {
String grantingTicketURL = cas.getGrantingTicketURL(user);
Map<String, Object> setParams = new HashMap<String, Object>();
Map<String, String> components = new HashMap<String, String>();
setParams.put("basepac", BasePacType.PAC_WEB);
setParams.put("components", components);
// components.put("QUOTA", "2048"); // does not yet work
Map<String, String> whereParams = new HashMap<String, String>();
whereParams.put("name", "aaa05");
Object[] params = new Object[] { user,
cas.getServiceTicket(grantingTicketURL, RemoteTestHelper.getBackendURL()),
setParams, whereParams };
Object execute = client.execute(MODULE + ".update", params);
assertNotNull(execute);
params = new Object[] { user,
cas.getServiceTicket(grantingTicketURL, RemoteTestHelper.getBackendURL()),
whereParams };
execute = client.execute(MODULE + ".search", params);
assertNotNull(execute);
assertTrue(execute instanceof Object[]);
Object[] untypedResultArray = (Object[]) execute;
assertEquals(1, untypedResultArray.length);
assertTrue(untypedResultArray[0] instanceof Map<?,?>);
@SuppressWarnings("unchecked")
Map<String, ?> pacResultMap = (Map<String, ?>) untypedResultArray[0];
assertEquals(BasePacType.PAC_WEB, pacResultMap.get("basepac"));
Object compMapObj = pacResultMap.get("components");
assertTrue(compMapObj instanceof Map<?,?>);
@SuppressWarnings("unchecked")
Map<String,String> compMap = (Map<String, String>) compMapObj;
assertEquals("1024", compMap.get("QUOTA"));
// assertEquals("2048", compMap.get("QUOTA")); // does not yet work
assertEquals("10", compMap.get("TRAFFIC"));
} catch (XmlRpcException e) {
fail(e.getMessage());
}
}
public void testDeleteSW() {
String user = "ad";
try {
int count = getPacsCount();
String grantingTicketURL = cas.getGrantingTicketURL(user);
Map<String, Object> setParams = new HashMap<String, Object>();
setParams.put("created", TextUtil.format(daysBack(10)));
Map<String, String> whereParams = new HashMap<String, String>();
whereParams.put("name", "aaa05");
Object[] params = new Object[] { user,
cas.getServiceTicket(grantingTicketURL, RemoteTestHelper.getBackendURL()),
setParams, whereParams };
Object execute = client.execute(MODULE + ".update", params);
assertNotNull(execute);
setParams = new HashMap<String, Object>();
setParams.put("cancelled", TextUtil.format(daysBack(1)));
whereParams = new HashMap<String, String>();
whereParams.put("name", "aaa05");
params = new Object[] { user,
cas.getServiceTicket(grantingTicketURL, RemoteTestHelper.getBackendURL()),
setParams, whereParams };
execute = client.execute(MODULE + ".update", params);
assertNotNull(execute);
whereParams.put("name", "aaa05");
params = new Object[] { user,
cas.getServiceTicket(grantingTicketURL, RemoteTestHelper.getBackendURL()),
whereParams };
execute = client.execute(MODULE + ".delete", params);
assertEquals(count - 1, getPacsCount());
} catch (XmlRpcException e) {
fail(e.getMessage());
}
}
private Date daysBack(int days) {
Calendar calendar = Calendar.getInstance();
calendar.add(Calendar.DAY_OF_YEAR, -days);
return calendar.getTime();
}
private int getPacsCount() {
int count = 0;
String user = "hsh00-aaa";
String grantingTicketURL = cas.getGrantingTicketURL(user);
Map<String, String> whereParams = new HashMap<String, String>();
whereParams.put("customer", config.getProperty("accountprefix.customer") + "-aaa");
Object[] params = new Object[] { user,
cas.getServiceTicket(grantingTicketURL, RemoteTestHelper.getBackendURL()),
whereParams };
try {
Object execute = client.execute(MODULE + ".search", params);
Object[] result = (Object[]) execute;
count = result.length;
} catch (XmlRpcException e) {
fail(e.getMessage());
}
return count;
}
private Object addPac(String user, String grantingTicketURL,
String pacName, String basepacName, String inetAddr,
int quota, int traffic, int daemons)
throws XmlRpcException {
Map<String, Object> setParams = new HashMap<String, Object>();
setParams.put("name", pacName);
setParams.put("hive", "h99");
setParams.put("customer", config.getProperty("accountprefix.customer") + "-aaa");
setParams.put("basepac", basepacName);
setParams.put("curinetaddr", inetAddr);
HashMap<String, String> components = new HashMap<String, String>();
components.put("QUOTA", Integer.toString(quota));
components.put("TRAFFIC", Integer.toString(traffic));
if (daemons >= 0) {
components.put("DAEMON", Integer.toString(daemons));
}
setParams.put("components", components);
Object[] params = new Object[] { user,
cas.getServiceTicket(grantingTicketURL, RemoteTestHelper.getBackendURL()),
setParams };
Object execute = client.execute(MODULE + ".add", params);
assertTrue(execute instanceof Map<?, ?>);
Map<String, Object> whereParams = new HashMap<String, Object>();
whereParams.put("name", pacName);
params = new Object[] { user,
cas.getServiceTicket(grantingTicketURL, RemoteTestHelper.getBackendURL()),
whereParams };
execute = client.execute(MODULE + ".search", params);
return execute;
}
private void validateAddedPac(Object execute, String basepacName, int quota, int traffic, int daemons) {
assertNotNull(execute);
assertTrue(execute instanceof Object[]);
Object[] untypedResultArray = (Object[]) execute;
assertEquals(1, untypedResultArray.length);
assertTrue(untypedResultArray[0] instanceof Map<?,?>);
@SuppressWarnings("unchecked")
Map<String, ?> pacResultMap = (Map<String, ?>) untypedResultArray[0];
assertEquals(basepacName, pacResultMap.get("basepac"));
Object compMapObj = pacResultMap.get("components");
assertTrue(compMapObj instanceof Map<?,?>);
@SuppressWarnings("unchecked")
Map<String,String> compMap = (Map<String, String>) compMapObj;
assertEquals(Integer.toString(quota), compMap.get("QUOTA"));
assertEquals(Integer.toString(traffic), compMap.get("TRAFFIC"));
if (daemons >= 0) {
assertEquals(Integer.toString(daemons), compMap.get("DAEMON"));
} else {
assertTrue(compMap.get("DAEMON") == null);
}
}
private void setPacAdminPassword(String user, String grantingTicketURL,
String pacName, String password)
throws XmlRpcException {
HashMap<String, Object> whereParams = new HashMap<String, Object>();
HashMap<String, Object> setParams = new HashMap<String, Object>();
whereParams.put("name", pacName);
setParams.put("password", password);
Object[] params = new Object[] { user,
cas.getServiceTicket(grantingTicketURL, RemoteTestHelper.getBackendURL()),
setParams, whereParams };
client.execute("user.update", params);
}
}