diff options
Diffstat (limited to 'rulemgt/src/test/java/org/onap/holmes/rulemgt/db/CorrelationRuleQueryServiceTest.java')
| -rw-r--r-- | rulemgt/src/test/java/org/onap/holmes/rulemgt/db/CorrelationRuleQueryServiceTest.java | 159 |
1 files changed, 159 insertions, 0 deletions
diff --git a/rulemgt/src/test/java/org/onap/holmes/rulemgt/db/CorrelationRuleQueryServiceTest.java b/rulemgt/src/test/java/org/onap/holmes/rulemgt/db/CorrelationRuleQueryServiceTest.java new file mode 100644 index 0000000..a2bd863 --- /dev/null +++ b/rulemgt/src/test/java/org/onap/holmes/rulemgt/db/CorrelationRuleQueryServiceTest.java @@ -0,0 +1,159 @@ +/**
+ * Copyright 2017-2020 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.db;
+
+import org.easymock.EasyMock;
+import org.jdbi.v3.core.Handle;
+import org.jdbi.v3.core.Jdbi;
+import org.jdbi.v3.core.result.ResultIterable;
+import org.jdbi.v3.core.statement.Query;
+import org.junit.Before;
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.rules.ExpectedException;
+import org.junit.runner.RunWith;
+import org.onap.holmes.common.api.entity.CorrelationRule;
+import org.onap.holmes.common.exception.CorrelationException;
+import org.onap.holmes.rulemgt.bean.request.RuleQueryCondition;
+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.*;
+
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.is;
+
+@RunWith(PowerMockRunner.class)
+@PrepareForTest({Handle.class, Query.class})
+public class CorrelationRuleQueryServiceTest {
+
+ @Rule
+ public ExpectedException thrown = ExpectedException.none();
+ private Handle handle;
+ private Query query;
+ private ResultIterable resultIterable;
+ private CorrelationRuleQueryService correlationRuleQueryDao;
+ private RuleQueryCondition ruleQueryCondition;
+ private Jdbi jdbi;
+
+ @Before
+ public void setUp() throws Exception {
+ correlationRuleQueryDao = new CorrelationRuleQueryService();
+
+ jdbi = PowerMock.createMock(Jdbi.class);
+ handle = PowerMock.createMock(Handle.class);
+ query = PowerMock.createMock(Query.class);
+ resultIterable = PowerMock.createMock(ResultIterable.class);
+
+ Whitebox.setInternalState(correlationRuleQueryDao, "jdbi", jdbi);
+
+ ruleQueryCondition = createRuleQueryCondition();
+ }
+
+
+ @Test
+ public void getCorrelationRulesByCondition_db_exception() throws Exception {
+
+ thrown.expect(CorrelationException.class);
+ thrown.expectMessage("Failed to query the rule.");
+
+ EasyMock.expect(jdbi.open()).andReturn(handle);
+ EasyMock.expect(handle.createQuery(EasyMock.anyObject(String.class))).andReturn(query);
+ EasyMock.expect(query.mapToMap()).andThrow(new RuntimeException()).anyTimes();
+ handle.close();
+ EasyMock.expectLastCall();
+
+ PowerMock.replayAll();
+
+ correlationRuleQueryDao.getCorrelationRulesByCondition(ruleQueryCondition);
+
+ PowerMock.verifyAll();
+ }
+
+ @Test
+ public void getCorrelationRulesByCondition_normal() throws Exception {
+ EasyMock.expect(jdbi.open()).andReturn(handle);
+ EasyMock.expect(handle.createQuery(EasyMock.anyObject(String.class))).andReturn(query);
+ EasyMock.expect(query.mapToMap()).andReturn(resultIterable);
+ EasyMock.expect(resultIterable.list()).andReturn(createQueryResult());
+ handle.close();
+ EasyMock.expectLastCall();
+
+ PowerMock.replayAll();
+
+ List<CorrelationRule> result = correlationRuleQueryDao.getCorrelationRulesByCondition(ruleQueryCondition);
+ assertThat(result.size(), is(1));
+
+ PowerMock.verifyAll();
+ }
+
+ @Test
+ public void getCorrelationRulesByCondition_get_where_sql_exception() throws Exception {
+ thrown.expect(CorrelationException.class);
+ thrown.expectMessage("An error occurred while building the query SQL.");
+
+ EasyMock.expect(jdbi.open()).andReturn(handle);
+ EasyMock.expect(handle.createQuery(EasyMock.anyObject(String.class))).andReturn(query);
+ EasyMock.expect(query.mapToMap()).andReturn(resultIterable);
+ EasyMock.expect(resultIterable.list()).andReturn(createQueryResult());
+ handle.close();
+ EasyMock.expectLastCall();
+
+ PowerMock.replayAll();
+
+ correlationRuleQueryDao.getCorrelationRulesByCondition(null);
+
+ PowerMock.verifyAll();
+ }
+
+ private List<Map<String, Object>> createQueryResult() {
+ List<Map<String, Object>> list = new ArrayList<>();
+ Map<String, Object> value = new HashMap<>();
+ value.put("name", "Rule-001");
+ value.put("rid", "rule_" + System.currentTimeMillis());
+ value.put("description", "desc");
+ value.put("enable", 0);
+ value.put("templateid", 1L);
+ value.put("engineid", "engine-001");
+ value.put("enginetype", "engineType-001");
+ value.put("creator", "admin");
+ value.put("createtime", new Date());
+ value.put("updator", "admin");
+ value.put("updatetime", new Date());
+ value.put("params", new Properties());
+ value.put("domain", "Domain");
+ value.put("ismanual", 0);
+ value.put("vendor", "Vendor");
+ value.put("content", "Contents");
+ value.put("package", "package");
+ list.add(value);
+ return list;
+ }
+
+ private RuleQueryCondition createRuleQueryCondition() {
+ RuleQueryCondition ruleQueryCondition = new RuleQueryCondition();
+ ruleQueryCondition.setRid("rule_" + System.currentTimeMillis());
+ ruleQueryCondition.setName("Rule-001");
+ ruleQueryCondition.setEnabled(0);
+ ruleQueryCondition.setCreator("admin");
+ ruleQueryCondition.setModifier("admin");
+ return ruleQueryCondition;
+ }
+
+}
|