+ @Before
+ public void prepare() {
+
+ ServiceLocator locator = PowerMock.createMock(ServiceLocator.class);
+ RuleMgtWrapper ruleMgtWrapper = PowerMock.createMock(RuleMgtWrapper.class);
+ RuleQueryWrapper ruleQueryWrapper = PowerMock.createMock(RuleQueryWrapper.class);
+ EngineWrapper engineWrapper = PowerMock.createMock(EngineWrapper.class);
+ EngineIpList engineIpList = PowerMock.createMock(EngineIpList.class);
+ CorrelationRuleDao correlationRuleDao = PowerMock.createMock(CorrelationRuleDao.class);
+ DbDaoUtil daoUtil = PowerMock.createMock(DbDaoUtil.class);
+ PowerMock.mockStatic(ServiceLocatorHolder.class);
+
+ EasyMock.expect(ServiceLocatorHolder.getLocator()).andReturn(locator);
+ EasyMock.expect(locator.getService(RuleMgtWrapper.class)).andReturn(ruleMgtWrapper);
+ EasyMock.expect(locator.getService(RuleQueryWrapper.class)).andReturn(ruleQueryWrapper);
+ EasyMock.expect(locator.getService(EngineWrapper.class)).andReturn(engineWrapper);
+ EasyMock.expect(locator.getService(EngineIpList.class)).andReturn(engineIpList);
+ EasyMock.expect(locator.getService(DbDaoUtil.class)).andReturn(daoUtil);
+ EasyMock.expect(daoUtil.getJdbiDaoByOnDemand(CorrelationRuleDao.class)).andReturn(correlationRuleDao);
+ try {
+ EasyMock.expect(engineIpList.getServiceCount()).andReturn(new ArrayList());
+ } catch (Exception e) {
+ // Do nothing
+ }
+
+
+ PowerMock.replayAll();
+
+ }
+
+ @After
+ public void destroy() {
+ PowerMock.resetAll();
+ }
+