Change rule retrieval from CBS to ConfigMap

[holmes/rule-management.git] / rulemgt / src / test / java / org / onap / holmes / rulemgt / dcae / ConfigFileScanningTaskTest.java

diff --git a/rulemgt/src/test/java/org/onap/holmes/rulemgt/dcae/ConfigFileScanningTaskTest.java b/rulemgt/src/test/java/org/onap/holmes/rulemgt/dcae/ConfigFileScanningTaskTest.java
new file mode 100644 (file)
index 0000000..053612b
--- /dev/null
+++ b/rulemgt/src/test/java/org/onap/holmes/rulemgt/dcae/ConfigFileScanningTaskTest.java
@@ -0,0 +1,241 @@
+/**
+ * Copyright 2021 ZTE Corporation.
+ * <p>
+ * 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
+ * <p>
+ * http://www.apache.org/licenses/LICENSE-2.0
+ * <p>
+ * 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.dcae;
+
+import org.easymock.EasyMock;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.onap.holmes.common.ConfigFileScanner;
+import org.onap.holmes.common.utils.FileUtils;
+import org.onap.holmes.common.utils.JerseyClient;
+import org.onap.holmes.rulemgt.bean.response.RuleQueryListResponse;
+import org.onap.holmes.rulemgt.bean.response.RuleResult4API;
+import org.powermock.api.easymock.PowerMock;
+import org.powermock.core.classloader.annotations.PrepareForTest;
+import org.powermock.core.classloader.annotations.SuppressStaticInitializationFor;
+import org.powermock.modules.junit4.PowerMockRunner;
+import org.powermock.reflect.Whitebox;
+
+import java.util.HashMap;
+import java.util.Map;
+
+import static org.hamcrest.core.Is.is;
+import static org.hamcrest.core.IsEqual.equalTo;
+import static org.junit.Assert.assertThat;
+
+@RunWith(PowerMockRunner.class)
+@PrepareForTest({JerseyClient.class})
+@SuppressStaticInitializationFor({"org.onap.holmes.common.utils.JerseyClient"})
+public class ConfigFileScanningTaskTest {
+
+    @Test
+    public void run_add_rules() throws Exception {
+        String clName = "ControlLoop-VOLTE-2179b738-fd36-4843-a71a-a8c24c70c55b";
+        String indexPath = getFilePath("index-add.json");
+        String contents = FileUtils.readTextFile(indexPath);
+
+        ConfigFileScanningTask cfst = new ConfigFileScanningTask(null);
+        Whitebox.setInternalState(cfst, "configFile", indexPath);
+
+        // mock for getExistingRules
+        JerseyClient jcMock = PowerMock.createMock(JerseyClient.class);
+        PowerMock.expectNew(JerseyClient.class).andReturn(jcMock).anyTimes();
+        RuleQueryListResponse rqlr = new RuleQueryListResponse();
+        rqlr.getCorrelationRules().add(getRuleResult4API(clName, contents));
+        EasyMock.expect(jcMock.get(EasyMock.anyString(), EasyMock.anyObject())).andReturn(rqlr);
+
+        // mock for deployRule
+        EasyMock.expect(jcMock.header(EasyMock.anyString(), EasyMock.anyObject())).andReturn(jcMock);
+        EasyMock.expect(jcMock.put(EasyMock.anyString(), EasyMock.anyObject())).andReturn("");
+
+        PowerMock.replayAll();
+        cfst.run();
+        PowerMock.verifyAll();
+
+        Map<String, String> config = Whitebox.getInternalState(cfst, "configInEffect");
+        assertThat(config.size(), is(1));
+    }
+
+    @Test
+    public void run_remove_rules_normal() throws Exception {
+        String clName = "ControlLoop-VOLTE-2179b738-fd36-4843-a71a-a8c24c70c55b";
+        String indexPath = getFilePath("index-add.json");
+        String contents = FileUtils.readTextFile(indexPath);
+        Map<String, String> configInEffect = new HashMap<>();
+        configInEffect.put(clName, contents);
+
+        ConfigFileScanningTask cfst = new ConfigFileScanningTask(new ConfigFileScanner());
+        Whitebox.setInternalState(cfst, "configFile", getFilePath("index-empty.json"));
+        Whitebox.setInternalState(cfst, "configInEffect", configInEffect);
+
+        // mock for getExistingRules
+        JerseyClient jcMock = PowerMock.createMock(JerseyClient.class);
+        PowerMock.expectNew(JerseyClient.class).andReturn(jcMock).anyTimes();
+        RuleQueryListResponse rqlr = new RuleQueryListResponse();
+        rqlr.getCorrelationRules().add(getRuleResult4API(clName, contents));
+        EasyMock.expect(jcMock.get(EasyMock.anyString(), EasyMock.anyObject())).andReturn(rqlr);
+
+        // mock for deleteRule
+        EasyMock.expect(jcMock.delete(EasyMock.anyString())).andReturn("");
+
+        PowerMock.replayAll();
+        cfst.run();
+        PowerMock.verifyAll();
+
+        Map<String, String> config = Whitebox.getInternalState(cfst, "configInEffect");
+        assertThat(config.size(), is(0));
+    }
+
+    @Test
+    public void run_remove_rules_delete_null_pointer() throws Exception {
+        String clName = "ControlLoop-VOLTE-2179b738-fd36-4843-a71a-a8c24c70c55b";
+        String indexPath = getFilePath("index-add.json");
+        String contents = FileUtils.readTextFile(indexPath);
+        Map<String, String> configInEffect = new HashMap<>();
+        configInEffect.put(clName, contents);
+
+        ConfigFileScanningTask cfst = new ConfigFileScanningTask(new ConfigFileScanner());
+        Whitebox.setInternalState(cfst, "configFile", indexPath);
+        Whitebox.setInternalState(cfst, "configInEffect", configInEffect);
+
+        // mock for getExistingRules
+        JerseyClient jcMock = PowerMock.createMock(JerseyClient.class);
+        PowerMock.expectNew(JerseyClient.class).andReturn(jcMock).anyTimes();
+        RuleQueryListResponse rqlr = new RuleQueryListResponse();
+        rqlr.getCorrelationRules().add(getRuleResult4API("a-non-existing-rule", contents));
+        EasyMock.expect(jcMock.get(EasyMock.anyString(), EasyMock.anyObject())).andReturn(rqlr);
+
+        PowerMock.replayAll();
+        cfst.run();
+        PowerMock.verifyAll();
+
+        Map<String, String> config = Whitebox.getInternalState(cfst, "configInEffect");
+        assertThat(config.size(), is(1));
+    }
+
+    @Test
+    public void run_remove_rules_api_calling_returning_null() throws Exception {
+        String clName = "ControlLoop-VOLTE-2179b738-fd36-4843-a71a-a8c24c70c55b";
+        String indexPath = getFilePath("index-add.json");
+        String contents = FileUtils.readTextFile(indexPath);
+        Map<String, String> configInEffect = new HashMap<>();
+        configInEffect.put(clName, contents);
+
+        ConfigFileScanningTask cfst = new ConfigFileScanningTask(new ConfigFileScanner());
+        Whitebox.setInternalState(cfst, "configFile", indexPath);
+        Whitebox.setInternalState(cfst, "configInEffect", configInEffect);
+
+        // mock for getExistingRules
+        JerseyClient jcMock = PowerMock.createMock(JerseyClient.class);
+        PowerMock.expectNew(JerseyClient.class).andReturn(jcMock).anyTimes();
+        RuleQueryListResponse rqlr = new RuleQueryListResponse();
+        rqlr.getCorrelationRules().add(getRuleResult4API(clName, contents));
+        EasyMock.expect(jcMock.get(EasyMock.anyString(), EasyMock.anyObject())).andReturn(rqlr);
+
+        // mock for deleteRule
+        EasyMock.expect(jcMock.delete(EasyMock.anyString())).andReturn(null);
+
+        PowerMock.replayAll();
+        cfst.run();
+        PowerMock.verifyAll();
+
+        Map<String, String> config = Whitebox.getInternalState(cfst, "configInEffect");
+        assertThat(config.size(), is(1));
+    }
+
+    @Test
+    public void run_change_rules_normal() throws Exception {
+        String clName = "ControlLoop-VOLTE-2179b738-fd36-4843-a71a-a8c24c70c55b";
+        String oldDrlPath = getFilePath("ControlLoop-VOLTE-2179b738-fd36-4843-a71a-a8c24c70c55b.drl");
+        String oldDrlContents = FileUtils.readTextFile(oldDrlPath);
+        Map<String, String> configInEffect = new HashMap<>();
+        configInEffect.put(clName, oldDrlContents);
+
+        ConfigFileScanningTask cfst = new ConfigFileScanningTask(new ConfigFileScanner());
+        Whitebox.setInternalState(cfst, "configFile", getFilePath("index-rule-changed.json"));
+        Whitebox.setInternalState(cfst, "configInEffect", configInEffect);
+
+        // mock for getExistingRules
+        JerseyClient jcMock = PowerMock.createMock(JerseyClient.class);
+        PowerMock.expectNew(JerseyClient.class).andReturn(jcMock).anyTimes();
+        RuleQueryListResponse rqlr = new RuleQueryListResponse();
+        rqlr.getCorrelationRules().add(getRuleResult4API(clName, oldDrlContents));
+        EasyMock.expect(jcMock.get(EasyMock.anyString(), EasyMock.anyObject())).andReturn(rqlr);
+
+        // mock for deleteRule
+        EasyMock.expect(jcMock.delete(EasyMock.anyString())).andReturn("");
+
+        // mock for deployRule
+        EasyMock.expect(jcMock.header(EasyMock.anyString(), EasyMock.anyObject())).andReturn(jcMock);
+        EasyMock.expect(jcMock.put(EasyMock.anyString(), EasyMock.anyObject())).andReturn("");
+
+        PowerMock.replayAll();
+        cfst.run();
+        PowerMock.verifyAll();
+
+        Map<String, String> config = Whitebox.getInternalState(cfst, "configInEffect");
+        assertThat(config.size(), is(1));
+        assertThat(config.get(clName),
+                equalTo(FileUtils.readTextFile(
+                        getFilePath("ControlLoop-VOLTE-2179b738-fd36-4843-a71a-a8c24c70c55b-changed.drl"))));
+    }
+
+    @Test
+    public void run_change_rules_no_change_except_for_spaces() throws Exception {
+        String clName = "ControlLoop-VOLTE-2179b738-fd36-4843-a71a-a8c24c70c55b";
+        String oldDrlPath = getFilePath("ControlLoop-VOLTE-2179b738-fd36-4843-a71a-a8c24c70c55b.drl");
+        String oldDrlContents = FileUtils.readTextFile(oldDrlPath);
+        Map<String, String> configInEffect = new HashMap<>();
+        configInEffect.put(clName, oldDrlContents);
+
+        ConfigFileScanningTask cfst = new ConfigFileScanningTask(new ConfigFileScanner());
+        Whitebox.setInternalState(cfst, "configFile", getFilePath("index-rule-spaces-test.json"));
+        Whitebox.setInternalState(cfst, "configInEffect", configInEffect);
+
+        // mock for getExistingRules
+        JerseyClient jcMock = PowerMock.createMock(JerseyClient.class);
+        PowerMock.expectNew(JerseyClient.class).andReturn(jcMock).anyTimes();
+        RuleQueryListResponse rqlr = new RuleQueryListResponse();
+        rqlr.getCorrelationRules().add(getRuleResult4API(clName, oldDrlContents));
+        EasyMock.expect(jcMock.get(EasyMock.anyString(), EasyMock.anyObject())).andReturn(rqlr);
+
+        PowerMock.replayAll();
+        cfst.run();
+        PowerMock.verifyAll();
+
+        Map<String, String> config = Whitebox.getInternalState(cfst, "configInEffect");
+        assertThat(config.size(), is(1));
+        assertThat(config.get(clName),
+                equalTo(FileUtils.readTextFile(
+                        getFilePath("ControlLoop-VOLTE-2179b738-fd36-4843-a71a-a8c24c70c55b.drl"))));
+    }
+
+    private String getFilePath(String fileName) {
+        return ConfigFileScanningTaskTest.class.getResource("/" + fileName).getFile();
+    }
+
+    private RuleResult4API getRuleResult4API(String clName, String contents) {
+        RuleResult4API ruleResult4API = new RuleResult4API();
+        ruleResult4API.setRuleId(clName);
+        ruleResult4API.setRuleName(clName);
+        ruleResult4API.setLoopControlName(clName);
+        ruleResult4API.setContent(contents);
+        ruleResult4API.setDescription("");
+        ruleResult4API.setEnabled(1);
+        return ruleResult4API;
+    }
+}
\ No newline at end of file