@Test
public void testReadDomainsAndGuids() throws Exception {
CustomerDAO dao = new CustomerDAO();
IConfiguration configuration = EasyMock.createStrictMock(IConfiguration.class);
dao.setConfiguration(configuration);
ResultSet resultSet = EasyMock.createStrictMock(ResultSet.class);
Set<String> domains = new HashSet<String>();
Set<GlobalIdentifier> guids = new HashSet<GlobalIdentifier>();
int custId = 34;
int cloudService = 2;
int replicationZone = 453;
int cloudService2 = 1;
int replicationZone2 = 13;
int cloudService3 = 3;
// first exec of loop
EasyMock.expect(resultSet.getString(10)).andReturn("domain123");
EasyMock.expect(resultSet.getString(11)).andReturn("guid123");
EasyMock.expect(resultSet.getInt(12)).andReturn(cloudService);
EasyMock.expect(resultSet.getInt(13)).andReturn(replicationZone);
EasyMock.expect(resultSet.next()).andReturn(true);
EasyMock.expect(resultSet.getInt(1)).andReturn(custId);
// second exec of loop
EasyMock.expect(resultSet.getString(10)).andReturn("domain456");
EasyMock.expect(resultSet.getString(11)).andReturn("guid456");
EasyMock.expect(resultSet.getInt(12)).andReturn(cloudService2);
EasyMock.expect(resultSet.getInt(13)).andReturn(replicationZone2);
EasyMock.expect(resultSet.next()).andReturn(true);
EasyMock.expect(resultSet.getInt(1)).andReturn(custId);
// third exec of loop (guid not valid with no cloud service)
EasyMock.expect(resultSet.getString(10)).andReturn("domain456");
EasyMock.expect(resultSet.getString(11)).andReturn("guid789");
EasyMock.expect(resultSet.getInt(12)).andReturn(cloudService3);
EasyMock.expect(resultSet.next()).andReturn(true);
EasyMock.expect(resultSet.getInt(1)).andReturn(custId + 1); // ends loop with mismatched custid
EasyMock.replay(resultSet);
assertTrue(
"Should have another item even.",
dao.readDomainsAndGuids(resultSet, custId, domains, guids));
EasyMock.verify(resultSet);
assertEquals("Should have 2 domains.", 2, domains.size());
assertTrue("Domain123 not found.", domains.contains("domain123"));
assertTrue("Domain456 not found.", domains.contains("domain456"));
assertEquals("Should have 2 guids.", 2, guids.size());
for (GlobalIdentifier guid : guids) {
if (guid.getGuid().equals("guid123")) {
assertEquals("Wrong cloud service in guid123", CloudService.GOOGLE, guid.getService());
assertEquals("Wrong replication zone.", replicationZone, guid.getReplicationZone());
} else {
assertEquals("Wrong cloud service in guid456", CloudService.OFFICE365, guid.getService());
assertEquals("Wrong replication zone.", replicationZone2, guid.getReplicationZone());
}
}
}
public List<ClientEventEntry> findLogEvents(
List<SearchConstraint> constraints, List<String> orderBy, int offset, int limit, Connection c)
throws SQLException {
StringBuffer sql =
new StringBuffer(
"select e.event_id, e.customer_id, e.user_id, e.event_time, e.description, e.has_log_file from dat_client_log_events e ");
appendWhere(sql, constraints, s_propToColumnMap);
appendOrderBy(sql, orderBy, "e.event_id desc", s_propToColumnMap);
appendLimits(sql, offset, limit);
Statement stmt = null;
ResultSet rs = null;
try {
stmt = c.createStatement();
rs = stmt.executeQuery(sql.toString());
List<ClientEventEntry> results = new ArrayList<ClientEventEntry>();
while (rs.next()) {
ClientEventEntry entry =
new ClientEventEntry(
rs.getInt(1),
rs.getInt(2),
rs.getInt(3),
resolveDate(rs.getTimestamp(4)),
rs.getString(5),
rs.getInt(6) != 0);
results.add(entry);
}
return results;
} finally {
DbUtils.safeClose(rs);
DbUtils.safeClose(stmt);
}
}
protected static void logPoliciesUsingRemovedTopologyObjs(
String operation,
String topologyType,
Collection<Integer> deletedNodes,
final String constraintName,
Connection c)
throws SQLException {
String query =
"select sets.customer_id, p.policy_name, pc.policy_id, sets.user_set_id, sets.constraint_value "
+ " from policies p, policy_criteria pc, dat_saved_user_sets sets "
+ " where sets.constraint_name = "
+ QueryUtils.literal(constraintName)
+ " and "
+ QueryUtils.dbCast("sets.constraint_value", QueryUtils.CastType.INTEGER)
+ " in "
+ QueryUtils.literal(deletedNodes)
+ " and sets.user_set_id = pc.userset_id and pc.policy_id = p.policy_id";
s_logger.debug(query);
Statement s = null;
ResultSet rs = null;
try {
s = c.createStatement();
rs = s.executeQuery(query);
while (rs.next()) {
int custID = rs.getInt(1);
String policyName = rs.getString(2);
int policyID = rs.getInt(3);
int userSetID = rs.getInt(4);
int nodeID = rs.getInt(5);
StringBuilder bld =
new StringBuilder(operation)
.append(" ")
.append(topologyType)
.append(" ")
.append(nodeID)
.append(" which is refered to by userset ")
.append(userSetID)
.append(" in policy ")
.append(policyName)
.append(" with id ")
.append(policyID)
.append(" for customer ")
.append(custID);
s_logger.warn(bld.toString());
}
} finally {
if (s != null) s.close();
if (rs != null) rs.close();
}
}
public void purgeLogEvents(final List<SearchConstraint> constraints, final int daysToKeep) {
blockHereIfBadNfs();
StringBuffer sql =
new StringBuffer(
"select u.mail_directory || '/logs/' || e.event_id || '.log' "
+ "from dat_client_log_events e, dat_user_account u "
+ "where event_time < current_timestamp - (? || ' days')::interval "
+ "and e.has_log_file != 0 "
+ "and e.user_id = u.object_id");
appendWhereConstraints(sql, constraints, s_propToColumnMap);
m_logCategory.info("Purge event logs query is\n" + sql);
StringBuffer sql2 =
new StringBuffer(
"delete from dat_client_log_events "
+ "where event_time < current_timestamp - (? || ' days')::interval");
if (!constraints.isEmpty()) {
sql2.append(" and event_id in (select e.event_id from dat_client_log_events e ");
appendWhere(sql2, constraints, s_propToColumnMap);
sql2.append(")");
}
m_logCategory.info("Purge event logs query is\n" + sql2);
try {
Connection c = m_txManager.getConnection();
PreparedStatement stmt = null;
ResultSet rs = null;
boolean needsRollback = true;
try {
stmt = c.prepareStatement(sql.toString());
stmt.setInt(1, daysToKeep);
rs = stmt.executeQuery();
while (rs.next()) {
File logFile = new File(getNfsRoot(), rs.getString(1));
// nfs usage, but DB transaction takes no locks.
if (logFile.exists()) {
boolean ok = logFile.delete();
if (!ok) {
m_logger
.log("Unable to delete")
.param(LoggingConsts.FILENAME, logFile.getAbsolutePath())
.warn();
}
} // file exists.
} // each row
// Below, no nfs usage occurs. We can use the same connection.
// the 'sql delete' may use DB locks.
stmt = c.prepareStatement(sql2.toString());
stmt.setInt(1, daysToKeep);
stmt.executeUpdate();
c.commit();
needsRollback = false;
} finally {
DbUtils.safeClose(rs);
DbUtils.safeClose(stmt);
if (needsRollback) {
c.rollback();
}
m_txManager.returnConnection(c);
}
} catch (Exception ex) {
m_logger.log("Error in purgeLogEvents").warn(ex);
}
}
@SuppressWarnings("unchecked")
@Test
public void testReadNextCustomer() throws Exception {
CustomerDAO dao =
EasyMock.createMockBuilder(CustomerDAO.class)
.addMockedMethod("readDomainsAndGuids")
.createStrictMock();
ResultSet resultSet = EasyMock.createStrictMock(ResultSet.class);
List<ICustomer> customerList = new LinkedList<ICustomer>();
int custID = 34;
int state = 3;
String name = "myName";
String fromAddress = "myFromAddr";
String clientID = "myClientId";
String clientKey = "myClientKey";
String backendId = "myBackendId";
String templateId = "myTemplateId";
String externalID = "myExternalId";
int estMailboxCount = 34453;
int estCloudMailboxCount = 345234;
int estOnpremisesJournalMailboxCount = 3523;
int estWelcomedCount = 938423;
int estNotificationCount = 3423;
int estAdminCount = 45345;
Date countsUpdatedTime = new Date();
int estTotalContacts = 345234;
int estTotalUsersWithContacts = 87432;
int estTotalCalendarEntries = 343460;
int estTotalUsersWithCalendar = 9892;
int estTotalDLists = 8932;
int estTotalDListMembers = 34982;
int estTotalPDLs = 12342142;
int estTotalPDLMembers = 8732324;
int estNoWelcomeResponseCount = 734232;
int estNoLoginCount = 90832;
int estNoActiveLoginCount = 89923;
int estIgnoredCount = 349823;
String channel = "myChannel";
EasyMock.expect(resultSet.getInt(1)).andReturn(custID);
EasyMock.expect(resultSet.getInt(2)).andReturn(state);
EasyMock.expect(resultSet.getString(3)).andReturn(name);
EasyMock.expect(resultSet.getString(4)).andReturn(fromAddress);
EasyMock.expect(resultSet.getString(5)).andReturn(clientID);
EasyMock.expect(resultSet.getString(6)).andReturn(clientKey);
EasyMock.expect(resultSet.getString(7)).andReturn(backendId);
EasyMock.expect(resultSet.getString(8)).andReturn(templateId);
EasyMock.expect(resultSet.getString(9)).andReturn(externalID);
EasyMock.expect(resultSet.getInt(14)).andReturn(estMailboxCount);
EasyMock.expect(resultSet.getInt(15)).andReturn(estCloudMailboxCount);
EasyMock.expect(resultSet.getInt(16)).andReturn(estOnpremisesJournalMailboxCount);
EasyMock.expect(resultSet.getInt(17)).andReturn(estWelcomedCount);
EasyMock.expect(resultSet.getInt(18)).andReturn(estNotificationCount);
EasyMock.expect(resultSet.getInt(19)).andReturn(estAdminCount);
EasyMock.expect(resultSet.getTimestamp(20))
.andReturn(new Timestamp(countsUpdatedTime.getTime()));
EasyMock.expect(resultSet.getInt(21)).andReturn(estTotalContacts);
EasyMock.expect(resultSet.getInt(22)).andReturn(estTotalUsersWithContacts);
EasyMock.expect(resultSet.getInt(23)).andReturn(estTotalCalendarEntries);
EasyMock.expect(resultSet.getInt(24)).andReturn(estTotalUsersWithCalendar);
EasyMock.expect(resultSet.getInt(25)).andReturn(estTotalDLists);
EasyMock.expect(resultSet.getInt(26)).andReturn(estTotalDListMembers);
EasyMock.expect(resultSet.getInt(27)).andReturn(estTotalPDLs);
EasyMock.expect(resultSet.getInt(28)).andReturn(estTotalPDLMembers);
EasyMock.expect(resultSet.getInt(29)).andReturn(estNoWelcomeResponseCount);
EasyMock.expect(resultSet.getInt(30)).andReturn(estNoLoginCount);
EasyMock.expect(resultSet.getInt(31)).andReturn(estNoActiveLoginCount);
EasyMock.expect(resultSet.getInt(32)).andReturn(estIgnoredCount);
EasyMock.expect(resultSet.getString(33)).andReturn(channel);
EasyMock.expect(
dao.readDomainsAndGuids(
EasyMock.eq(resultSet),
EasyMock.eq(custID),
EasyMock.isA(Set.class),
EasyMock.isA(Set.class)))
.andReturn(true);
EasyMock.replay(dao, resultSet);
assertTrue(
"Should have found another customer.", dao.readNextCustomer(resultSet, customerList));
EasyMock.verify(dao, resultSet);
assertEquals("Should have customer in list.", 1, customerList.size());
Customer cust = (Customer) customerList.get(0);
assertEquals("Wrong customer id.", custID, cust.getCustID());
assertEquals("Wrong customer state.", state, cust.getState().toInt());
assertEquals("Wrong customer name.", name, cust.getName());
assertEquals("Wrong from address.", fromAddress, cust.getFromAddress());
assertEquals("Wrong backend id.", backendId, cust.getBackendHostname());
assertEquals("Wrong template id.", templateId, cust.getTemplateId());
assertEquals("Wrong channel.", channel, cust.getChannel());
assertEquals("Wrong client id.", clientID, cust.getClientID());
assertEquals("Wrong client key.", clientKey, cust.getClientKey());
assertEquals("Wrong channel.", channel, cust.getChannel());
assertEquals("Wrong est mailbox count.", estMailboxCount, cust.getEstMailboxCount());
assertEquals(
"Wrong est cloud mailbox count.", estCloudMailboxCount, cust.getEstCloudMailboxCount());
assertEquals(
"Wrong est on premises journal mailbox count.",
estOnpremisesJournalMailboxCount,
cust.getEstOnpremisesJournalMailboxCount());
assertEquals("Wrong est not welcomed count.", estWelcomedCount, cust.getEstNotWelcomedCount());
assertEquals(
"Wrong est notification not set count.",
estNotificationCount,
cust.getEstNotificationNotSetCount());
assertEquals("Wrong est admin count.", estAdminCount, cust.getEstAdminCount());
assertEquals("Wrong counts updated time.", countsUpdatedTime, cust.getCountsUpdatedTime());
assertEquals("Wrong total contacts.", estTotalContacts, cust.getTotalContacts());
assertEquals(
"Wrong total users with contacts.",
estTotalUsersWithContacts,
cust.getTotalUsersWithContacts());
assertEquals(
"Wrong total calendar entries.", estTotalCalendarEntries, cust.getTotalCalendarEntries());
assertEquals(
"Wrong total users with calendar entries.",
estTotalUsersWithCalendar,
cust.getTotalUsersWithCalendarEntries());
assertEquals("Wrong total dlists.", estTotalDLists, cust.getTotalDlists());
assertEquals("Wrong total dlist members.", estTotalDListMembers, cust.getTotalDlistMembers());
assertEquals("Wrong total pdls.", estTotalPDLs, cust.getTotalPDLs());
assertEquals("Wrong total pdl members.", estTotalPDLMembers, cust.getTotalPDLMembers());
assertEquals(
"Wrong est no welcome reponse count.",
estNoWelcomeResponseCount,
cust.getEstNoWelcomeResponseCount());
assertEquals("Wrong est no login count.", estNoLoginCount, cust.getEstNoLoginCount());
assertEquals(
"Wrong est no active login count.", estNoActiveLoginCount, cust.getEstNoActiveLoginCount());
assertEquals("Wrong est ignored count.", estIgnoredCount, cust.getEstIgnoredCount());
assertEquals("Wrong external id.", externalID, cust.getExternalID());
}
