protected AgentContextImpl mockAgentContext(String subDir) throws Exception {
if (m_contextDir != null) {
cleanDir(m_contextDir);
m_contextDir.delete();
}
m_contextDir = new File(getWorkDir(), subDir);
m_contextDir.mkdirs();
cleanDir(m_contextDir);
AgentContextImpl context = new AgentContextImpl(m_contextDir);
for (Class<?> handlerClass : AgentContextImpl.KNOWN_HANDLERS) {
if (ScheduledExecutorService.class.equals(handlerClass)) {
context.setHandler(ScheduledExecutorService.class, new SynchronousExecutorService());
}
else {
setMockedHandler(context, handlerClass);
}
}
return context;
}
|