Allocate Rule

[holmes/rule-management.git] / rulemgt / src / test / java / org / onap / holmes / rulemgt / wrapper / RuleQueryWrapperTest.java

diff --git a/rulemgt/src/test/java/org/onap/holmes/rulemgt/wrapper/RuleQueryWrapperTest.java b/rulemgt/src/test/java/org/onap/holmes/rulemgt/wrapper/RuleQueryWrapperTest.java
new file mode 100644 (file)
index 0000000..b07efe9
--- /dev/null
+++ b/rulemgt/src/test/java/org/onap/holmes/rulemgt/wrapper/RuleQueryWrapperTest.java
@@ -0,0 +1,61 @@
+/**
+ * Copyright 2017 ZTE Corporation.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.onap.holmes.rulemgt.wrapper;
+
+import org.easymock.EasyMock;
+import org.hamcrest.core.IsNull;
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Test;
+import org.onap.holmes.common.api.entity.CorrelationRule;
+import org.onap.holmes.rulemgt.db.CorrelationRuleDao;
+import org.powermock.api.easymock.PowerMock;
+
+import java.util.ArrayList;
+import java.util.List;
+
+
+public class RuleQueryWrapperTest {
+    private CorrelationRuleDao correlationRuleDao;
+    private RuleQueryWrapper ruleQueryWrapper;
+
+    @Before
+    public void setUp() {
+        correlationRuleDao = PowerMock.createMock(CorrelationRuleDao.class);
+        ruleQueryWrapper = PowerMock.createMock(RuleQueryWrapper.class);
+    }
+
+    @Test
+    public void queryRuleByEnable() throws Exception{
+        int enable = 0;
+        EasyMock.expect(ruleQueryWrapper.queryRuleByEnable(EasyMock.anyInt())).andReturn(new ArrayList<CorrelationRule>());
+        PowerMock.replayAll();
+        List<CorrelationRule> correlationRules = ruleQueryWrapper.queryRuleByEnable(enable);
+        PowerMock.verifyAll();
+        Assert.assertThat(correlationRules, IsNull.<List<CorrelationRule>>notNullValue());
+    }
+
+    @Test
+    public void queryRuleByEngineInstance() throws Exception{
+        String engineInstance = "10.96.33.34";
+        EasyMock.expect(ruleQueryWrapper.queryRuleByEngineInstance(EasyMock.anyObject(String.class))).andReturn(new ArrayList<CorrelationRule>());
+        PowerMock.replayAll();
+        List<CorrelationRule> correlationRules = ruleQueryWrapper.queryRuleByEngineInstance(engineInstance);
+        PowerMock.verifyAll();
+        Assert.assertThat(correlationRules, IsNull.<List<CorrelationRule>>notNullValue());
+    }
+}