diff options
Diffstat (limited to 'src/test/java/org/onap/msb/sdk/discovery/MSBServiceTest.java')
| -rw-r--r-- | src/test/java/org/onap/msb/sdk/discovery/MSBServiceTest.java | 64 |
1 files changed, 54 insertions, 10 deletions
diff --git a/src/test/java/org/onap/msb/sdk/discovery/MSBServiceTest.java b/src/test/java/org/onap/msb/sdk/discovery/MSBServiceTest.java index a09e64b..b202ff4 100644 --- a/src/test/java/org/onap/msb/sdk/discovery/MSBServiceTest.java +++ b/src/test/java/org/onap/msb/sdk/discovery/MSBServiceTest.java @@ -19,6 +19,7 @@ import java.util.Set; import org.junit.Assert; import org.junit.Test; import org.junit.runner.RunWith; +import org.onap.msb.sdk.discovery.common.RouteConst; import org.onap.msb.sdk.discovery.common.RouteException; import org.onap.msb.sdk.discovery.entity.MicroServiceFullInfo; import org.onap.msb.sdk.discovery.entity.MicroServiceInfo; @@ -32,20 +33,25 @@ import org.powermock.modules.junit4.PowerMockRunner; @RunWith(PowerMockRunner.class) @PrepareForTest({HttpClientUtil.class}) public class MSBServiceTest { - private static final String mockMSBUrl1 = + private static final String MOCK_MSB_URL_REG_UPDATE_TRUE = "http://127.0.0.1:10081/api/microservices/v1/services?createOrUpdate=true"; - private static final String mockMSBUrl2 = + private static final String MOCK_MSB_URL_REG_UPDATE_FALSE = "http://127.0.0.1:10081/api/microservices/v1/services?createOrUpdate=false"; - private static final String mockRegistrationJson = + private static final String MOCK_MSB_URL_DIS = + "http://127.0.0.1:10081/api/microservices/v1/services/aai/version/v8?ifPassStatus=true"; + private static final String MOCK_MSB_URL_UNREG = + "http://127.0.0.1:10081/api/microservices/v1/services/aai/version/v8"; + + private static final String MOCK_REG_SERVICE_JSON = "{\"serviceName\":\"aai\",\"version\":\"v8\",\"url\":\"/aai/v8\",\"protocol\":\"REST\",\"visualRange\":\"1\",\"lb_policy\":\"\",\"path\":\"/aai/v8\",\"nodes\":[{\"ip\":\"10.74.44.1\",\"port\":\"8443\",\"ttl\":\"\"}],\"metadata\":null}"; @Test - public void test_registration() throws RouteException { + public void test_registration_update_true() throws RouteException { String msbAddress = "127.0.0.1:10081"; MicroServiceInfo microServiceInfo = - (MicroServiceInfo) JacksonJsonUtil.jsonToBean(mockRegistrationJson, MicroServiceInfo.class); + (MicroServiceInfo) JacksonJsonUtil.jsonToBean(MOCK_REG_SERVICE_JSON, MicroServiceInfo.class); MicroServiceFullInfo microServiceFullInfo = mockMicroServiceFullInfo(microServiceInfo); - mockGetRest(mockMSBUrl1, JacksonJsonUtil.beanToJson(microServiceFullInfo)); + mockHttpPost(MOCK_MSB_URL_REG_UPDATE_TRUE, JacksonJsonUtil.beanToJson(microServiceFullInfo)); MSBService msbService = new MSBService(); microServiceFullInfo = msbService.registerMicroServiceInfo(msbAddress, microServiceInfo); Assert.assertTrue(microServiceFullInfo.getPath().equals("/aai/v8")); @@ -60,9 +66,9 @@ public class MSBServiceTest { public void test_registration_update_false() throws RouteException { String msbAddress = "127.0.0.1:10081"; MicroServiceInfo microServiceInfo = - (MicroServiceInfo) JacksonJsonUtil.jsonToBean(mockRegistrationJson, MicroServiceInfo.class); + (MicroServiceInfo) JacksonJsonUtil.jsonToBean(MOCK_REG_SERVICE_JSON, MicroServiceInfo.class); MicroServiceFullInfo microServiceFullInfo = mockMicroServiceFullInfo(microServiceInfo); - mockGetRest(mockMSBUrl2, JacksonJsonUtil.beanToJson(microServiceFullInfo)); + mockHttpPost(MOCK_MSB_URL_REG_UPDATE_FALSE, JacksonJsonUtil.beanToJson(microServiceFullInfo)); MSBService msbService = new MSBService(); microServiceFullInfo = msbService.registerMicroServiceInfo(msbAddress, microServiceInfo, false); Assert.assertTrue(microServiceFullInfo.getPath().equals("/aai/v8")); @@ -73,6 +79,32 @@ public class MSBServiceTest { Assert.assertTrue(microServiceFullInfo.getVisualRange().equals("1")); } + @Test + public void test_discovery() throws RouteException { + String msbAddress = "127.0.0.1:10081"; + MicroServiceInfo microServiceInfo = + (MicroServiceInfo) JacksonJsonUtil.jsonToBean(MOCK_REG_SERVICE_JSON, MicroServiceInfo.class); + MicroServiceFullInfo microServiceFullInfo = mockMicroServiceFullInfo(microServiceInfo); + mockHttpGet(MOCK_MSB_URL_DIS, JacksonJsonUtil.beanToJson(microServiceFullInfo)); + + MSBService msbService = new MSBService(); + microServiceFullInfo = msbService.queryMicroServiceInfo(msbAddress, "aai", "v8"); + Assert.assertTrue(microServiceFullInfo.getPath().equals("/aai/v8")); + Assert.assertTrue(microServiceFullInfo.getProtocol().equals("REST")); + Assert.assertTrue(microServiceFullInfo.getServiceName().equals("aai")); + Assert.assertTrue(microServiceFullInfo.getUrl().equals("/aai/v8")); + Assert.assertTrue(microServiceFullInfo.getVersion().equals("v8")); + Assert.assertTrue(microServiceFullInfo.getVisualRange().equals("1")); + } + + // @Test + // public void test_unregistration() throws Exception { + // String msbAddress = "127.0.0.1:10081"; + // mockHttpDel(); + // MSBService msbService = new MSBService(); + // msbService.cancelMicroServiceInfo(msbAddress, "aai", "v8"); + // } + private MicroServiceFullInfo mockMicroServiceFullInfo(MicroServiceInfo info) { MicroServiceFullInfo serviceInfo = new MicroServiceFullInfo(); @@ -87,9 +119,21 @@ public class MSBServiceTest { } - private void mockGetRest(String mockMSBUrl, String mockServiceInfoJson) throws RouteException { + private void mockHttpPost(String mockMSBUrl, String mockServiceInfoJson) throws RouteException { PowerMockito.mockStatic(HttpClientUtil.class); - PowerMockito.when(HttpClientUtil.httpPostWithJSON(mockMSBUrl, mockRegistrationJson)) + PowerMockito.when(HttpClientUtil.httpPostWithJSON(mockMSBUrl, MOCK_REG_SERVICE_JSON)) .thenReturn(mockServiceInfoJson); } + + private void mockHttpGet(String mockMSBUrl, String mockServiceInfoJson) throws RouteException { + PowerMockito.mockStatic(HttpClientUtil.class); + PowerMockito.when(HttpClientUtil.httpGet(mockMSBUrl)).thenReturn(mockServiceInfoJson); + } + + // private void mockHttpDel() throws Exception { + // PowerMockito.mockStatic(HttpClientUtil.class); + // HttpClientUtil myClass = PowerMockito.spy(new HttpClientUtil()); + // PowerMockito.doNothing().when(myClass, HttpClientUtil.class.getMethod("delete", String.class, + // String.class)); + // } } |