1 files changed, 5 insertions, 5 deletions

diff --git a/rulemgt/src/test/java/org/onap/holmes/rulemgt/wrapper/RuleMgtWrapperTest.java b/rulemgt/src/test/java/org/onap/holmes/rulemgt/wrapper/RuleMgtWrapperTest.java
index 7033aff..ef26936 100644
--- a/rulemgt/src/test/java/org/onap/holmes/rulemgt/wrapper/RuleMgtWrapperTest.java
+++ b/rulemgt/src/test/java/org/onap/holmes/rulemgt/wrapper/RuleMgtWrapperTest.java
@@ -32,7 +32,7 @@ import org.onap.holmes.rulemgt.bean.response.RuleQueryListResponse;
 import org.onap.holmes.rulemgt.bolt.enginebolt.EngineWrapper;

 import org.onap.holmes.rulemgt.db.CorrelationRuleDao;

 import org.onap.holmes.rulemgt.db.CorrelationRuleQueryDao;

-import org.onap.holmes.rulemgt.send.Ip4AddingRule;

+import org.onap.holmes.rulemgt.tools.EngineTools;

 import org.powermock.api.easymock.PowerMock;

 import org.powermock.modules.junit4.PowerMockRunner;

 import org.powermock.reflect.Whitebox;

@@ -61,7 +61,7 @@ public class RuleMgtWrapperTest {
 

     private CorrelationRuleDao correlationRuleDaoMock;

 

-    private Ip4AddingRule ip4AddingRuleMock;

+    private EngineTools engineToolsMock;

 

     private static final String USER_NAME = "admin";

 

@@ -74,13 +74,13 @@ public class RuleMgtWrapperTest {
         correlationRuleQueryDaoMock = PowerMock.createMock(CorrelationRuleQueryDao.class);

         dbDaoUtilMock = PowerMock.createMock(DbDaoUtil.class);

         correlationRuleDaoMock = PowerMock.createMock(CorrelationRuleDao.class);

-        ip4AddingRuleMock = PowerMock.createMock(Ip4AddingRule.class);

+        engineToolsMock = PowerMock.createMock(EngineTools.class);

 

         Whitebox.setInternalState(ruleMgtWrapper, "daoUtil", dbDaoUtilMock);

         Whitebox.setInternalState(ruleMgtWrapper, "correlationRuleQueryDao", correlationRuleQueryDaoMock);

         Whitebox.setInternalState(ruleMgtWrapper, "engineWarpper", engineWrapperMock);

         Whitebox.setInternalState(ruleMgtWrapper, "correlationRuleDao", correlationRuleDaoMock);

-        Whitebox.setInternalState(ruleMgtWrapper,"ip4AddingRule", ip4AddingRuleMock);

+        Whitebox.setInternalState(ruleMgtWrapper,"engineTools", engineToolsMock);

 

         PowerMock.resetAll();

     }

@@ -166,7 +166,7 @@ public class RuleMgtWrapperTest {
         correlationRuleRet.setRid("rule_" + System.currentTimeMillis());

 

         EasyMock.expect(correlationRuleDaoMock.queryRuleByRuleName(ruleName)).andReturn(null);

-        EasyMock.expect(ip4AddingRuleMock.getEngineIp4AddRule()).andReturn("10.96.33.34");

+        EasyMock.expect(engineToolsMock.getEngineWithLeastRules()).andReturn("10.96.33.34");

         EasyMock.expect(engineWrapperMock.checkRuleFromEngine(EasyMock.anyObject(CorrelationCheckRule4Engine.class)

                 , EasyMock.anyObject(String.class)))

                 .andReturn(true);