X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=holmes-actions%2Fsrc%2Ftest%2Fjava%2Forg%2Fonap%2Fholmes%2Fcommon%2Fconfig%2FMicroServiceConfigTest.java;h=fbc22e51850fc07fb7137d543d73d0d11deb2b10;hb=59d22c8f5ed81e1221ee6e9c5364a8baf5bc3a20;hp=33431ff18297786a477e25c2da17668170b64933;hpb=8b3c9e2b928e3a0308974482e2645ace4838c468;p=holmes%2Fcommon.git

diff --git a/holmes-actions/src/test/java/org/onap/holmes/common/config/MicroServiceConfigTest.java b/holmes-actions/src/test/java/org/onap/holmes/common/config/MicroServiceConfigTest.java
index 33431ff..fbc22e5 100644
--- a/holmes-actions/src/test/java/org/onap/holmes/common/config/MicroServiceConfigTest.java
+++ b/holmes-actions/src/test/java/org/onap/holmes/common/config/MicroServiceConfigTest.java
@@ -1,12 +1,12 @@
 /**
- * Copyright 2017 ZTE Corporation.
- *
+ * Copyright 2017-2020 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
- *
- *     http://www.apache.org/licenses/LICENSE-2.0
- *
+ * <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.
@@ -16,24 +16,353 @@
 
 package org.onap.holmes.common.config;
 
+import org.apache.commons.lang3.StringUtils;
+import org.easymock.EasyMock;
+import org.junit.Ignore;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.powermock.api.easymock.PowerMock;
+import org.powermock.core.classloader.annotations.PowerMockIgnore;
+import org.powermock.core.classloader.annotations.PrepareForTest;
+import org.powermock.modules.junit4.PowerMockRunner;
+import org.powermock.reflect.internal.WhiteboxImpl;
+
+import static org.hamcrest.core.Is.is;
 import static org.hamcrest.core.IsEqual.equalTo;
+import static org.hamcrest.core.IsNull.nullValue;
 import static org.junit.Assert.assertThat;
+import static org.onap.holmes.common.config.MicroServiceConfig.*;
 
-import org.junit.Test;
-
+@PrepareForTest(MicroServiceConfig.class)
+@PowerMockIgnore({"javax.ws.*"})
+@RunWith(PowerMockRunner.class)
 public class MicroServiceConfigTest {
 
+    private static String ACTUAL_HOSTNAME = System.getenv(HOSTNAME);
+
+
     @Test
     public void getMsbServerAddrTest() {
-        System.setProperty("MSB_ADDR", "test");
-        assertThat("http://test", equalTo(MicroServiceConfig.getMsbServerAddr()));
-        System.clearProperty("MSB_ADDR");
+        System.setProperty(MSB_IAG_SERVICE_HOST, "test");
+        System.setProperty(MSB_IAG_SERVICE_PORT, "443");
+        assertThat("http://test:443", equalTo(getMsbServerAddrWithHttpPrefix()));
+        System.clearProperty(MicroServiceConfig.MSB_IAG_SERVICE_PORT);
+        System.clearProperty(MicroServiceConfig.MSB_IAG_SERVICE_HOST);
+    }
+
+    @Test
+    public void getMsbServerIpTest() {
+        System.setProperty(MSB_IAG_SERVICE_HOST, "10.54.23.79");
+        System.setProperty(MSB_IAG_SERVICE_PORT, "443");
+        System.setProperty(HOSTNAME, "rule-mgmt");
+        PowerMock.mockStaticPartial(MicroServiceConfig.class, "getServiceConfigInfoFromCBS", String.class);
+        EasyMock.expect(MicroServiceConfig.getServiceConfigInfoFromCBS(System.getProperty(HOSTNAME)))
+                .andReturn("{\"msb.hostname\": \"10.54.23.79:443\"}").times(2);
+        PowerMock.replayAll();
+        assertThat("10.54.23.79", equalTo(getMsbIpAndPort()[0]));
+        assertThat("443", equalTo(getMsbIpAndPort()[1]));
+        System.clearProperty(MicroServiceConfig.HOSTNAME);
+        System.clearProperty(MicroServiceConfig.MSB_IAG_SERVICE_PORT);
+        System.clearProperty(MicroServiceConfig.MSB_IAG_SERVICE_HOST);
     }
 
     @Test
     public void getServiceIpTest() {
-        System.setProperty("SERVICE_IP", "test");
-        assertThat("test", equalTo(MicroServiceConfig.getServiceIp()));
-        System.clearProperty("SERVICE_IP");
+        String ip = StringUtils.isEmpty(ACTUAL_HOSTNAME) ? "10.74.5.8" : ACTUAL_HOSTNAME;
+        String hostname = String.format("http://%s", ip);
+        System.setProperty(HOSTNAME, hostname);
+        assertThat(ip, equalTo(getMicroServiceIpAndPort()[0]));
+        assertThat("80", equalTo(getMicroServiceIpAndPort()[1]));
+        System.clearProperty(HOSTNAME);
+    }
+
+    @Test
+    public void getConsulAddrInfoTest() {
+        System.setProperty(CONSUL_HOST, "127.0.0.1");
+        assertThat("http://127.0.0.1:8500/v1/catalog/service/", equalTo(getConsulAddrInfo()));
+        System.clearProperty(CONSUL_HOST);
+    }
+
+    @Test
+    public void getConfigBindingServiceAddrInfoTest_consul_not_exist() throws Exception {
+        System.setProperty(CONFIG_BINDING_SERVICE, "config_binding_service");
+        PowerMock.mockStaticPartial(MicroServiceConfig.class, "execQuery", String.class);
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())
+                .andThrow(new RuntimeException("Invalid URL."));
+
+        PowerMock.replayAll();
+
+        assertThat(getServiceAddrInfoFromDcaeConsulByHostName(System.getProperty(CONFIG_BINDING_SERVICE))
+                , is(nullValue()));
+
+        PowerMock.verifyAll();
+        System.clearProperty(CONFIG_BINDING_SERVICE);
+    }
+
+    @Test
+    public void getServiceAddrInfoFromDcaeConsulByHostName_consul_exists() throws Exception {
+        System.setProperty(CONFIG_BINDING_SERVICE, "config_binding_service");
+        PowerMock.mockStaticPartial(MicroServiceConfig.class, "execQuery", String.class);
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())
+                .andReturn("[{\"ServiceAddress\": \"127.0.0.2\", \"ServicePort\": \"8080\"}]");
+        System.setProperty(CONSUL_HOST, "127.0.0.1");
+
+        PowerMock.replayAll();
+
+        assertThat(getServiceAddrInfoFromDcaeConsulByHostName(System.getProperty(CONFIG_BINDING_SERVICE)),
+                equalTo("http://127.0.0.2:8080"));
+
+        PowerMock.verifyAll();
+
+        System.clearProperty(CONSUL_HOST);
+        System.clearProperty(CONFIG_BINDING_SERVICE);
+    }
+
+    @Test
+    public void getConfigBindingServiceAddrInfoTest_consul_exists_propertie_not_exist() throws Exception {
+        System.setProperty(CONFIG_BINDING_SERVICE, "config_binding_service");
+        PowerMock.mockStaticPartial(MicroServiceConfig.class, "execQuery", String.class);
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())
+                .andReturn("[{\"ServiceAddress\": \"127.0.0.2\"}]");
+        System.setProperty(CONSUL_HOST, "127.0.0.1");
+
+        PowerMock.replayAll();
+
+        assertThat(getServiceAddrInfoFromDcaeConsulByHostName(System.getProperty(CONFIG_BINDING_SERVICE)),
+                is(nullValue()));
+
+        PowerMock.verifyAll();
+
+        System.clearProperty(CONSUL_HOST);
+        System.clearProperty(CONFIG_BINDING_SERVICE);
+    }
+
+    @Test
+    public void getServiceAddrInfoFromCBS_consul_not_exist() throws Exception {
+        PowerMock.mockStaticPartial(MicroServiceConfig.class, "execQuery", String.class);
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())
+                .andThrow(new RuntimeException("Invalid URL.")).times(2);
+
+        PowerMock.replayAll();
+
+        assertThat(getServiceConfigInfoFromCBS(HOSTNAME), is(nullValue()));
+
+        PowerMock.verifyAll();
+    }
+
+    @Test
+    public void getServiceAddrInfoFromDcaeConsulByHostName_consul_exists_service_not_exist() throws Exception {
+        PowerMock.mockStaticPartial(MicroServiceConfig.class, "execQuery", String.class);
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())
+                .andReturn("[]");
+
+        PowerMock.replayAll();
+        assertThat(getServiceAddrInfoFromDcaeConsulByHostName(HOSTNAME), is(nullValue()));
+        PowerMock.verifyAll();
+    }
+
+    @Ignore
+    public void getMsbAddrInfo_msb_registered() throws Exception {
+        System.setProperty(MSB_IAG_SERVICE_HOST, "10.74.5.8");
+        System.setProperty(MSB_IAG_SERVICE_PORT, "1545");
+        System.setProperty(HOSTNAME, "rule-mgmt");
+        PowerMock.mockStaticPartial(MicroServiceConfig.class, "getServiceConfigInfoFromCBS", String.class);
+        EasyMock.expect(MicroServiceConfig.getServiceConfigInfoFromCBS(System.getProperty(HOSTNAME)))
+                .andReturn("{\"msb.hostname\": \"127.0.0.3:5432\"}");
+
+        PowerMock.replayAll();
+        String[] msbInfo = getMsbIpAndPort();
+        PowerMock.verifyAll();
+
+        assertThat(msbInfo[0], equalTo("127.0.0.3"));
+        assertThat(msbInfo[1], equalTo("5432"));
+
+        System.clearProperty(HOSTNAME);
+        System.clearProperty(MSB_IAG_SERVICE_PORT);
+        System.clearProperty(MSB_IAG_SERVICE_HOST);
+    }
+
+    @Ignore
+    public void getMsbAddrInfo_msb_not_registered() throws Exception {
+        System.setProperty(MSB_IAG_SERVICE_HOST, "10.74.5.8");
+        System.setProperty(MSB_IAG_SERVICE_PORT, "1545");
+        System.setProperty(HOSTNAME, "rule-mgmt");
+        PowerMock.mockStaticPartial(MicroServiceConfig.class, "getServiceConfigInfoFromCBS", String.class);
+        EasyMock.expect(MicroServiceConfig.getServiceConfigInfoFromCBS(System.getProperty(HOSTNAME)))
+                .andReturn("{}");
+
+        PowerMock.replayAll();
+        String[] msbInfo = getMsbIpAndPort();
+        PowerMock.verifyAll();
+
+        assertThat(msbInfo[0], equalTo("10.74.5.8"));
+        assertThat(msbInfo[1], equalTo("1545"));
+
+        System.clearProperty(HOSTNAME);
+        System.clearProperty(MSB_IAG_SERVICE_PORT);
+        System.clearProperty(MSB_IAG_SERVICE_HOST);
+    }
+
+    @Test
+    public void getMicroServiceIpAndPort_service_registered_to_consul() throws Exception {
+        System.setProperty(HOSTNAME, "rule-mgmt");
+        PowerMock.mockStaticPartial(MicroServiceConfig.class, "execQuery", String.class);
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())
+                .andReturn("[{\"ServiceAddress\": \"127.0.0.3\", \"ServicePort\": \"5432\"}]");
+
+        PowerMock.replayAll();
+        String[] msbInfo = getMicroServiceIpAndPort();
+        PowerMock.verifyAll();
+
+        assertThat(msbInfo[0], equalTo("127.0.0.3"));
+        assertThat(msbInfo[1], equalTo("5432"));
+
+        System.clearProperty(HOSTNAME);
+    }
+
+    @Test
+    public void getMicroServiceIpAndPort_service_not_registered_to_consul() throws Exception {
+        String ip = StringUtils.isEmpty(ACTUAL_HOSTNAME) ? "10.74.5.8" : ACTUAL_HOSTNAME;
+        String port = StringUtils.isEmpty(ACTUAL_HOSTNAME) ? "1545" : "80";
+        String hostname = String.format("http://%s:%s", ip, port);
+        System.setProperty(HOSTNAME, hostname);
+        PowerMock.mockStaticPartial(MicroServiceConfig.class, "execQuery", String.class);
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())
+                .andReturn("[]");
+
+        PowerMock.replayAll();
+        String[] msbInfo = getMicroServiceIpAndPort();
+        PowerMock.verifyAll();
+
+        assertThat(msbInfo[0], equalTo(ip));
+        assertThat(msbInfo[1], equalTo(port));
+
+        System.clearProperty(HOSTNAME);
+    }
+
+    @Test
+    public void getMicroServiceIpAndPort_service_not_registered_full_addr() throws Exception {
+        String ip = StringUtils.isEmpty(ACTUAL_HOSTNAME) ? "10.74.5.8" : ACTUAL_HOSTNAME;
+        String port = StringUtils.isEmpty(ACTUAL_HOSTNAME) ? "1545" : "80";
+        String hostname = String.format("http://%s:%s", ip, port);
+        System.setProperty(HOSTNAME, hostname);
+        PowerMock.mockStaticPartial(MicroServiceConfig.class, "execQuery", String.class);
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())
+                .andReturn("{}");
+
+        PowerMock.replayAll();
+        String[] msbInfo = getMicroServiceIpAndPort();
+        PowerMock.verifyAll();
+
+        assertThat(msbInfo[0], equalTo(ip));
+        assertThat(msbInfo[1], equalTo(port));
+
+        System.clearProperty(HOSTNAME);
+    }
+
+    @Test
+    public void getMicroServiceIpAndPort_service_not_registered_no_port() throws Exception {
+        String ip = StringUtils.isEmpty(ACTUAL_HOSTNAME) ? "10.74.5.8" : ACTUAL_HOSTNAME;
+        String hostname = String.format("http://%s", ip);
+        System.setProperty(HOSTNAME, hostname);
+        PowerMock.mockStaticPartial(MicroServiceConfig.class, "execQuery", String.class);
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())
+                .andReturn("{}");
+
+        PowerMock.replayAll();
+        String[] msbInfo = getMicroServiceIpAndPort();
+        PowerMock.verifyAll();
+
+        assertThat(msbInfo[0], equalTo(ip));
+        assertThat(msbInfo[1], equalTo("80"));
+
+        System.clearProperty(HOSTNAME);
+    }
+
+    @Test
+    public void getMicroServiceIpAndPort_service_not_registered_only_ip() throws Exception {
+        String ip = StringUtils.isEmpty(ACTUAL_HOSTNAME) ? "10.74.5.8" : ACTUAL_HOSTNAME;
+        String hostname = String.format("http://%s", ip);
+        System.setProperty(HOSTNAME, hostname);
+        PowerMock.mockStaticPartial(MicroServiceConfig.class, "execQuery", String.class);
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())
+                .andReturn("{}");
+
+        PowerMock.replayAll();
+        String[] msbInfo = getMicroServiceIpAndPort();
+        PowerMock.verifyAll();
+
+        assertThat(msbInfo[0], equalTo(ip));
+        assertThat(msbInfo[1], equalTo("80"));
+
+        System.clearProperty(HOSTNAME);
+    }
+
+    @Test
+    public void getMicroServiceIpAndPort_service_not_registered_full_addr_https() throws Exception {
+        String ip = StringUtils.isEmpty(ACTUAL_HOSTNAME) ? "10.74.5.8" : ACTUAL_HOSTNAME;
+        String port = StringUtils.isEmpty(ACTUAL_HOSTNAME) ? "1545" : "80";
+        String hostname = String.format("http://%s:%s", ip, port);
+        System.setProperty(HOSTNAME, hostname);
+        PowerMock.mockStaticPartial(MicroServiceConfig.class, "execQuery", String.class);
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())
+                .andReturn("[]");
+
+        PowerMock.replayAll();
+        String[] msbInfo = getMicroServiceIpAndPort();
+        PowerMock.verifyAll();
+
+        assertThat(msbInfo[0], equalTo(ip));
+        assertThat(msbInfo[1], equalTo(port));
+
+        System.clearProperty(HOSTNAME);
+    }
+
+    @Test
+    public void isValidIpAddress_with_port() throws Exception {
+        boolean res = WhiteboxImpl.invokeMethod(MicroServiceConfig.class, "isIpAddress", "10.75.13.21:90");
+        assertThat(res, is(true));
+    }
+
+    @Test
+    public void isValidIpAddress_without_port() throws Exception {
+        boolean res = WhiteboxImpl.invokeMethod(MicroServiceConfig.class, "isIpAddress", "10.75.13.21");
+        assertThat(res, is(true));
+    }
+
+    @Test
+    public void isValidIpAddress_with_port_with_http_prefix() throws Exception {
+        boolean res = WhiteboxImpl.invokeMethod(MicroServiceConfig.class, "isIpAddress", "http://10.75.13.21:90");
+        assertThat(res, is(true));
+    }
+
+    @Test
+    public void isValidIpAddress_without_port_with_https_prefix() throws Exception {
+        boolean res = WhiteboxImpl.invokeMethod(MicroServiceConfig.class, "isIpAddress", "https://10.75.13.21");
+        assertThat(res, is(true));
+    }
+
+    @Test
+    public void isValidIpAddress_invalid_ip_without_port() throws Exception {
+        boolean res = WhiteboxImpl.invokeMethod(MicroServiceConfig.class, "isIpAddress", "holmes-rule-mgmt");
+        assertThat(res, is(false));
+    }
+
+    @Test
+    public void isValidIpAddress_invalid_ip_with_port() throws Exception {
+        boolean res = WhiteboxImpl.invokeMethod(MicroServiceConfig.class, "isIpAddress", "holmes-rule-mgmt:443");
+        assertThat(res, is(false));
+    }
+
+    @Test
+    public void isValidIpAddress_invalid_ip_without_port_with_http_prefix() throws Exception {
+        boolean res = WhiteboxImpl.invokeMethod(MicroServiceConfig.class, "isIpAddress", "http://holmes-rule-mgmt");
+        assertThat(res, is(false));
+    }
+
+    @Test
+    public void isValidIpAddress_invalid_ip_with_port_with_https_prefix() throws Exception {
+        boolean res = WhiteboxImpl.invokeMethod(MicroServiceConfig.class, "isIpAddress", "https://holmes-rule-mgmt:443");
+        assertThat(res, is(false));
     }
 }
\ No newline at end of file