summaryrefslogtreecommitdiffstats
path: root/rulemgt/src/test
diff options
context:
space:
mode:
authorGuangrongFu <fu.guangrong@zte.com.cn>2018-03-29 17:20:07 +0800
committerGuangrongFu <fu.guangrong@zte.com.cn>2018-03-29 18:30:01 +0800
commit82d2318187ea739a0b14442cbd82a3a7f6a419a7 (patch)
tree65f8503abef34470483bb89a585053b7f18d4d25 /rulemgt/src/test
parentadf983c3fa5c0be8fe05d349190b149da1effb2e (diff)
Fixed the Start-up Logic
Change-Id: I5646e771e05f5f774162373a2f69a0815acb53e5 Issue-ID: HOLMES-106 Signed-off-by: GuangrongFu <fu.guangrong@zte.com.cn>
Diffstat (limited to 'rulemgt/src/test')
-rw-r--r--rulemgt/src/test/java/org/onap/holmes/rulemgt/send/RuleAllocationTest.java57
1 files changed, 56 insertions, 1 deletions
diff --git a/rulemgt/src/test/java/org/onap/holmes/rulemgt/send/RuleAllocationTest.java b/rulemgt/src/test/java/org/onap/holmes/rulemgt/send/RuleAllocationTest.java
index 24c5f3a..696e32a 100644
--- a/rulemgt/src/test/java/org/onap/holmes/rulemgt/send/RuleAllocationTest.java
+++ b/rulemgt/src/test/java/org/onap/holmes/rulemgt/send/RuleAllocationTest.java
@@ -17,8 +17,22 @@
package org.onap.holmes.rulemgt.send;
+import org.easymock.EasyMock;
+import org.glassfish.hk2.api.ServiceLocator;
+import org.junit.After;
+import org.junit.Before;
import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.onap.holmes.common.dropwizard.ioc.utils.ServiceLocatorHolder;
+import org.onap.holmes.common.utils.DbDaoUtil;
+import org.onap.holmes.rulemgt.bolt.enginebolt.EngineWrapper;
+import org.onap.holmes.rulemgt.db.CorrelationRuleDao;
+import org.onap.holmes.rulemgt.msb.EngineIpList;
+import org.onap.holmes.rulemgt.wrapper.RuleMgtWrapper;
+import org.onap.holmes.rulemgt.wrapper.RuleQueryWrapper;
import org.powermock.api.easymock.PowerMock;
+import org.powermock.core.classloader.annotations.PrepareForTest;
+import org.powermock.modules.junit4.PowerMockRunner;
import org.powermock.reflect.Whitebox;
import java.util.ArrayList;
@@ -27,11 +41,50 @@ import java.util.List;
import static org.hamcrest.core.IsEqual.equalTo;
import static org.junit.Assert.assertThat;
+@RunWith(PowerMockRunner.class)
+@PrepareForTest({ServiceLocator.class, RuleMgtWrapper.class, RuleQueryWrapper.class, EngineWrapper.class,
+ EngineIpList.class, DbDaoUtil.class, RuleAllocation.class, ServiceLocatorHolder.class})
public class RuleAllocationTest {
- private RuleAllocation ruleAllocation = new RuleAllocation();
+ @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();
+ }
+
@Test
public void extendCompareIpTest() throws Exception{
+ RuleAllocation ruleAllocation = new RuleAllocation();
+
List<String> newList = new ArrayList<>();
newList.add("10.96.33.34");
newList.add("10.74.65.24");
@@ -47,6 +100,8 @@ public class RuleAllocationTest {
@Test
public void destroyCompareIpTest() throws Exception{
+ RuleAllocation ruleAllocation = new RuleAllocation();
+
List<String> newList = new ArrayList<>();
newList.add("10.96.33.34");