import org.onap.holmes.rulemgt.bolt.enginebolt.EngineWrapper;\r
import org.onap.holmes.rulemgt.db.CorrelationRuleDao;\r
import org.onap.holmes.rulemgt.db.CorrelationRuleQueryDao;\r
-import org.onap.holmes.rulemgt.send.Ip4AddingRule;\r
+import org.onap.holmes.rulemgt.tools.EngineTools;\r
import org.powermock.api.easymock.PowerMock;\r
import org.powermock.modules.junit4.PowerMockRunner;\r
import org.powermock.reflect.Whitebox;\r
\r
private CorrelationRuleDao correlationRuleDaoMock;\r
\r
- private Ip4AddingRule ip4AddingRuleMock;\r
+ private EngineTools engineToolsMock;\r
\r
private static final String USER_NAME = "admin";\r
\r
correlationRuleQueryDaoMock = PowerMock.createMock(CorrelationRuleQueryDao.class);\r
dbDaoUtilMock = PowerMock.createMock(DbDaoUtil.class);\r
correlationRuleDaoMock = PowerMock.createMock(CorrelationRuleDao.class);\r
- ip4AddingRuleMock = PowerMock.createMock(Ip4AddingRule.class);\r
+ engineToolsMock = PowerMock.createMock(EngineTools.class);\r
\r
Whitebox.setInternalState(ruleMgtWrapper, "daoUtil", dbDaoUtilMock);\r
Whitebox.setInternalState(ruleMgtWrapper, "correlationRuleQueryDao", correlationRuleQueryDaoMock);\r
Whitebox.setInternalState(ruleMgtWrapper, "engineWarpper", engineWrapperMock);\r
Whitebox.setInternalState(ruleMgtWrapper, "correlationRuleDao", correlationRuleDaoMock);\r
- Whitebox.setInternalState(ruleMgtWrapper,"ip4AddingRule", ip4AddingRuleMock);\r
+ Whitebox.setInternalState(ruleMgtWrapper,"engineTools", engineToolsMock);\r
\r
PowerMock.resetAll();\r
}\r
correlationRuleRet.setRid("rule_" + System.currentTimeMillis());\r
\r
EasyMock.expect(correlationRuleDaoMock.queryRuleByRuleName(ruleName)).andReturn(null);\r
- EasyMock.expect(ip4AddingRuleMock.getEngineIp4AddRule()).andReturn("10.96.33.34");\r
+ EasyMock.expect(engineToolsMock.getEngineWithLeastRules()).andReturn("10.96.33.34");\r
EasyMock.expect(engineWrapperMock.checkRuleFromEngine(EasyMock.anyObject(CorrelationCheckRule4Engine.class)\r
, EasyMock.anyObject(String.class)))\r
.andReturn(true);\r