Get the MSB info from DCAE Consul
[holmes/common.git] / holmes-actions / src / test / java / org / onap / holmes / common / config / MicroServiceConfigTest.java
index b29e490..4b697b0 100644 (file)
 \r
 package org.onap.holmes.common.config;\r
 \r
+import static org.hamcrest.core.Is.is;\r
 import static org.hamcrest.core.IsEqual.equalTo;\r
+import static org.hamcrest.core.IsNull.nullValue;\r
 import static org.junit.Assert.assertThat;\r
 import static org.junit.Assert.assertTrue;\r
+import static org.onap.holmes.common.config.MicroServiceConfig.*;\r
 \r
+import org.easymock.EasyMock;\r
+import org.junit.Rule;\r
 import org.junit.Test;\r
+import org.powermock.api.easymock.PowerMock;\r
+import org.powermock.core.classloader.annotations.PowerMockIgnore;\r
+import org.powermock.core.classloader.annotations.PrepareForTest;\r
+import org.powermock.modules.junit4.rule.PowerMockRule;\r
 \r
+@PrepareForTest(MicroServiceConfig.class)\r
+@PowerMockIgnore({"javax.ws.*"})\r
 public class MicroServiceConfigTest {\r
 \r
+    @Rule\r
+    public PowerMockRule powerMockRule = new PowerMockRule();\r
+\r
     @Test\r
     public void getMsbServerAddrTest() {\r
-        System.setProperty("MSB_ADDR", "test");\r
-        assertThat("http://test", equalTo(MicroServiceConfig.getMsbServerAddr()));\r
-        System.clearProperty("MSB_ADDR");\r
+        System.setProperty(MSB_ADDR, "test:80");\r
+        assertThat("http://test:80", equalTo(getMsbServerAddr()));\r
+        System.clearProperty(MicroServiceConfig.MSB_ADDR);\r
     }\r
 \r
     @Test\r
     public void getMsbServerIpTest() {\r
-        System.setProperty("MSB_ADDR", "10.54.23.79");\r
-        assertThat("10.54.23.79", equalTo(MicroServiceConfig.getMsbServerIp()));\r
-        System.clearProperty("MSB_ADDR");\r
+        System.setProperty(MSB_ADDR, "10.54.23.79:80");\r
+        assertThat("10.54.23.79", equalTo(getMsbAddrInfo()[0]));\r
+        assertThat("80", equalTo(getMsbAddrInfo()[1]));\r
+        System.clearProperty(MSB_ADDR);\r
     }\r
 \r
     @Test\r
-    public void getMsbPortTest() {\r
-        System.setProperty("MSB_PORT", "110");\r
-        assertTrue(110 == MicroServiceConfig.getMsbServerPort());\r
-        System.clearProperty("MSB_PORT");\r
+    public void getServiceIpTest() {\r
+        System.setProperty(HOSTNAME, "127.0.0.1");\r
+        assertThat("127.0.0.1", equalTo(getServiceAddrInfo()[0]));\r
+        assertThat("80", equalTo(getServiceAddrInfo()[1]));\r
+        System.clearProperty(HOSTNAME);\r
     }\r
 \r
     @Test\r
-    public void getMsbPortTestNonnumeric() {\r
-        System.setProperty("MSB_PORT", "test");\r
-        assertTrue(80 == MicroServiceConfig.getMsbServerPort());\r
-        System.clearProperty("MSB_PORT");\r
+    public void getConsulAddrInfoTest() {\r
+        System.setProperty(CONSUL_HOST, "127.0.0.1");\r
+        assertThat("127.0.0.1:8500/v1/catalog/service/", equalTo(getConsulAddrInfo()));\r
+        System.clearProperty(CONSUL_HOST);\r
     }\r
 \r
     @Test\r
-    public void getMsbPortTestNullValue() {\r
-        assertTrue(80 == MicroServiceConfig.getMsbServerPort());\r
+    public void getConfigBindingServiceAddrInfoTest_consul_not_exist() throws Exception {\r
+        PowerMock.mockStaticPartial(MicroServiceConfig.class, "execQuery", String.class);\r
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())\r
+                .andThrow(new RuntimeException("Invalid URL."));\r
+\r
+        PowerMock.replayAll();\r
+\r
+        assertThat(getConfigBindingServiceAddrInfo(), is(nullValue()));\r
+\r
+        PowerMock.verifyAll();\r
     }\r
 \r
     @Test\r
-    public void getServiceIpTest() {\r
-        System.setProperty("SERVICE_IP", "test");\r
-        assertThat("test", equalTo(MicroServiceConfig.getServiceIp()));\r
-        System.clearProperty("SERVICE_IP");\r
+    public void getConfigBindingServiceAddrInfoTest_consul_exists() throws Exception {\r
+        PowerMock.mockStaticPartial(MicroServiceConfig.class, "execQuery", String.class);\r
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())\r
+                .andReturn("[{\"ServiceAddress\": \"127.0.0.2\", \"ServicePort\": \"8080\"}]");\r
+        System.setProperty(CONSUL_HOST, "127.0.0.1");\r
+\r
+        PowerMock.replayAll();\r
+\r
+        assertThat(getConfigBindingServiceAddrInfo(), equalTo("127.0.0.2:8080"));\r
+\r
+        PowerMock.verifyAll();\r
+\r
+        System.clearProperty(CONSUL_HOST);\r
+    }\r
+\r
+    @Test\r
+    public void getConfigBindingServiceAddrInfoTest_consul_exists_propertie_not_exist() throws Exception {\r
+        PowerMock.mockStaticPartial(MicroServiceConfig.class, "execQuery", String.class);\r
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())\r
+                .andReturn("[{\"ServiceAddress\": \"127.0.0.2\"}]");\r
+        System.setProperty(CONSUL_HOST, "127.0.0.1");\r
+\r
+        PowerMock.replayAll();\r
+\r
+        assertThat(getConfigBindingServiceAddrInfo(), is(nullValue()));\r
+\r
+        PowerMock.verifyAll();\r
+\r
+        System.clearProperty(CONSUL_HOST);\r
+    }\r
+\r
+    @Test\r
+    public void getServiceAddrInfoFromCBS_consul_not_exist() throws Exception {\r
+        PowerMock.mockStaticPartial(MicroServiceConfig.class, "execQuery", String.class);\r
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())\r
+                .andThrow(new RuntimeException("Invalid URL.")).times(2);\r
+\r
+        PowerMock.replayAll();\r
+\r
+        assertThat(getServiceAddrInfoFromCBS(HOSTNAME), is(nullValue()));\r
+\r
+        PowerMock.verifyAll();\r
+    }\r
+\r
+    @Test\r
+    public void getServiceAddrInfoFromCBS_consul_exists_service_not_exist() throws Exception {\r
+        PowerMock.mockStaticPartial(MicroServiceConfig.class, "execQuery", String.class);\r
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())\r
+                .andReturn("[{\"ServiceAddress\": \"127.0.0.2\", \"ServicePort\": \"8080\"}]");\r
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())\r
+                .andReturn("{}");\r
+\r
+        PowerMock.replayAll();\r
+\r
+        assertThat(getServiceAddrInfoFromCBS(HOSTNAME), is(nullValue()));\r
+\r
+        PowerMock.verifyAll();\r
+    }\r
+\r
+    @Test\r
+    public void getServiceAddrInfoFromCBS_normal() throws Exception {\r
+        System.setProperty(HOSTNAME, "rule-mgmt");\r
+        PowerMock.mockStaticPartial(MicroServiceConfig.class, "execQuery", String.class);\r
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())\r
+                .andReturn("[{\"ServiceAddress\": \"127.0.0.2\", \"ServicePort\": \"8080\"}]");\r
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())\r
+                .andReturn("{\"rule-mgmt\": \"[\\\"127.0.0.3:5432\\\"]\"}");\r
+\r
+        PowerMock.replayAll();\r
+\r
+        assertThat(getServiceAddrInfoFromCBS("rule-mgmt"), equalTo("127.0.0.3:5432"));\r
+\r
+        PowerMock.verifyAll();\r
+\r
+        System.clearProperty(HOSTNAME);\r
+    }\r
+\r
+    @Test\r
+    public void getMsbAddrInfo_msb_registered() throws Exception {\r
+        System.setProperty(MSB_ADDR, "10.74.5.8:1545");\r
+        PowerMock.mockStaticPartial(MicroServiceConfig.class, "execQuery", String.class);\r
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())\r
+                .andReturn("[{\"ServiceAddress\": \"127.0.0.2\", \"ServicePort\": \"8080\"}]");\r
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())\r
+                .andReturn("{\"" + MSB_ADDR + "\": \"[\\\"127.0.0.3:5432\\\"]\"}");\r
+\r
+        PowerMock.replayAll();\r
+        String[] msbInfo = getMsbAddrInfo();\r
+        assertThat(msbInfo[0], equalTo("127.0.0.3"));\r
+        assertThat(msbInfo[1], equalTo("5432"));\r
+\r
+        PowerMock.verifyAll();\r
+\r
+        System.clearProperty(MSB_ADDR);\r
+    }\r
+\r
+    @Test\r
+    public void getMsbAddrInfo_msb_not_registered() throws Exception {\r
+        System.setProperty(MSB_ADDR, "10.74.5.8:1545");\r
+        PowerMock.mockStaticPartial(MicroServiceConfig.class, "execQuery", String.class);\r
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())\r
+                .andReturn("[{\"ServiceAddress\": \"127.0.0.2\", \"ServicePort\": \"8080\"}]");\r
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())\r
+                .andReturn("{}");\r
+\r
+        PowerMock.replayAll();\r
+        String[] msbInfo = getMsbAddrInfo();\r
+        assertThat(msbInfo[0], equalTo("10.74.5.8"));\r
+        assertThat(msbInfo[1], equalTo("1545"));\r
+\r
+        PowerMock.verifyAll();\r
+\r
+        System.clearProperty(MSB_ADDR);\r
+    }\r
+\r
+    @Test\r
+    public void getServiceAddrInfo_msb_registered() throws Exception {\r
+        System.setProperty(HOSTNAME, "rule-mgmt");\r
+        PowerMock.mockStaticPartial(MicroServiceConfig.class, "execQuery", String.class);\r
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())\r
+                .andReturn("[{\"ServiceAddress\": \"127.0.0.2\", \"ServicePort\": \"8080\"}]");\r
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())\r
+                .andReturn("{\"rule-mgmt\": \"[\\\"127.0.0.3:5432\\\"]\"}");\r
+\r
+        PowerMock.replayAll();\r
+        String[] msbInfo = getServiceAddrInfo();\r
+        assertThat(msbInfo[0], equalTo("127.0.0.3"));\r
+        assertThat(msbInfo[1], equalTo("5432"));\r
+\r
+        PowerMock.verifyAll();\r
+\r
+        System.clearProperty(HOSTNAME);\r
+    }\r
+\r
+    @Test\r
+    public void getServiceAddrInfo_msb_not_registered() throws Exception {\r
+        System.setProperty(HOSTNAME, "10.74.5.8:1545");\r
+        PowerMock.mockStaticPartial(MicroServiceConfig.class, "execQuery", String.class);\r
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())\r
+                .andReturn("[{\"ServiceAddress\": \"127.0.0.2\", \"ServicePort\": \"8080\"}]");\r
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())\r
+                .andReturn("{}");\r
+\r
+        PowerMock.replayAll();\r
+        String[] msbInfo = getServiceAddrInfo();\r
+        assertThat(msbInfo[0], equalTo("10.74.5.8"));\r
+        assertThat(msbInfo[1], equalTo("1545"));\r
+\r
+        PowerMock.verifyAll();\r
+\r
+        System.clearProperty(HOSTNAME);\r
+    }\r
+\r
+    @Test\r
+    public void getServiceAddrInfo_msb_not_registered_full_addr() throws Exception {\r
+        System.setProperty(HOSTNAME, "http://10.74.5.8:1545");\r
+        PowerMock.mockStaticPartial(MicroServiceConfig.class, "execQuery", String.class);\r
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())\r
+                .andReturn("[{\"ServiceAddress\": \"127.0.0.2\", \"ServicePort\": \"8080\"}]");\r
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())\r
+                .andReturn("{}");\r
+\r
+        PowerMock.replayAll();\r
+        String[] msbInfo = getServiceAddrInfo();\r
+        assertThat(msbInfo[0], equalTo("10.74.5.8"));\r
+        assertThat(msbInfo[1], equalTo("1545"));\r
+\r
+        PowerMock.verifyAll();\r
+\r
+        System.clearProperty(MSB_ADDR);\r
+    }\r
+\r
+    @Test\r
+    public void getServiceAddrInfo_msb_not_registered_no_port() throws Exception {\r
+        System.setProperty(HOSTNAME, "http://10.74.5.8");\r
+        PowerMock.mockStaticPartial(MicroServiceConfig.class, "execQuery", String.class);\r
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())\r
+                .andReturn("[{\"ServiceAddress\": \"127.0.0.2\", \"ServicePort\": \"8080\"}]");\r
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())\r
+                .andReturn("{}");\r
+\r
+        PowerMock.replayAll();\r
+        String[] msbInfo = getServiceAddrInfo();\r
+        assertThat(msbInfo[0], equalTo("10.74.5.8"));\r
+        assertThat(msbInfo[1], equalTo("80"));\r
+\r
+        PowerMock.verifyAll();\r
+\r
+        System.clearProperty(MSB_ADDR);\r
+    }\r
+\r
+    @Test\r
+    public void getServiceAddrInfo_msb_not_registered_only_ip() throws Exception {\r
+        System.setProperty(HOSTNAME, "10.74.5.8");\r
+        PowerMock.mockStaticPartial(MicroServiceConfig.class, "execQuery", String.class);\r
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())\r
+                .andReturn("[{\"ServiceAddress\": \"127.0.0.2\", \"ServicePort\": \"8080\"}]");\r
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())\r
+                .andReturn("{}");\r
+\r
+        PowerMock.replayAll();\r
+        String[] msbInfo = getServiceAddrInfo();\r
+        assertThat(msbInfo[0], equalTo("10.74.5.8"));\r
+        assertThat(msbInfo[1], equalTo("80"));\r
+\r
+        PowerMock.verifyAll();\r
+\r
+        System.clearProperty(MSB_ADDR);\r
+    }\r
+\r
+    @Test\r
+    public void getServiceAddrInfo_msb_not_registered_full_addr_https() throws Exception {\r
+        System.setProperty(HOSTNAME, "https://10.74.5.8:5432");\r
+        PowerMock.mockStaticPartial(MicroServiceConfig.class, "execQuery", String.class);\r
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())\r
+                .andReturn("[{\"ServiceAddress\": \"127.0.0.2\", \"ServicePort\": \"8080\"}]");\r
+        PowerMock.expectPrivate(MicroServiceConfig.class, "execQuery", EasyMock.anyObject())\r
+                .andReturn("{}");\r
+\r
+        PowerMock.replayAll();\r
+        String[] msbInfo = getServiceAddrInfo();\r
+        assertThat(msbInfo[0], equalTo("10.74.5.8"));\r
+        assertThat(msbInfo[1], equalTo("5432"));\r
+\r
+        PowerMock.verifyAll();\r
+\r
+        System.clearProperty(MSB_ADDR);\r
     }\r
 }
\ No newline at end of file