+ 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 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 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 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
+ assertThat(getServiceAddrInfoFromCBS(HOSTNAME), is(nullValue()));\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
+ PowerMock.verifyAll();\r
+\r
+ assertThat(msbInfo[0], equalTo("127.0.0.3"));\r
+ assertThat(msbInfo[1], equalTo("5432"));\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
+ PowerMock.verifyAll();\r
+\r
+ assertThat(msbInfo[0], equalTo("10.74.5.8"));\r
+ assertThat(msbInfo[1], equalTo("1545"));\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
+ PowerMock.verifyAll();\r
+\r
+ assertThat(msbInfo[0], equalTo("127.0.0.3"));\r
+ assertThat(msbInfo[1], equalTo("5432"));\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
+ PowerMock.verifyAll();\r
+\r
+ assertThat(msbInfo[0], equalTo("10.74.5.8"));\r
+ assertThat(msbInfo[1], equalTo("1545"));\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
+ PowerMock.verifyAll();\r
+\r
+ assertThat(msbInfo[0], equalTo("10.74.5.8"));\r
+ assertThat(msbInfo[1], equalTo("1545"));\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
+ PowerMock.verifyAll();\r
+\r
+ assertThat(msbInfo[0], equalTo("10.74.5.8"));\r
+ assertThat(msbInfo[1], equalTo("80"));\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
+ PowerMock.verifyAll();\r
+\r
+ assertThat(msbInfo[0], equalTo("10.74.5.8"));\r
+ assertThat(msbInfo[1], equalTo("80"));\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
+ PowerMock.verifyAll();\r
+\r
+ assertThat(msbInfo[0], equalTo("10.74.5.8"));\r
+ assertThat(msbInfo[1], equalTo("5432"));\r
+\r
+ System.clearProperty(MSB_ADDR);\r