Add unit test for MSB JAVA SDK

[msb/java-sdk.git] / src / test / java / org / onap / msb / sdk / httpclient / msb / MSBServiceClientTest.java

diff --git a/src/test/java/org/onap/msb/sdk/httpclient/msb/MSBServiceClientTest.java b/src/test/java/org/onap/msb/sdk/httpclient/msb/MSBServiceClientTest.java
index 71acd1a..403e44b 100644 (file)
--- a/src/test/java/org/onap/msb/sdk/httpclient/msb/MSBServiceClientTest.java
+++ b/src/test/java/org/onap/msb/sdk/httpclient/msb/MSBServiceClientTest.java
@@ -16,6 +16,7 @@ package org.onap.msb.sdk.httpclient.msb;
 import org.junit.Assert;
 import org.junit.Test;
 import org.junit.runner.RunWith;
+import org.onap.msb.sdk.discovery.MSBService;
 import org.onap.msb.sdk.discovery.common.RouteException;
 import org.onap.msb.sdk.discovery.entity.MicroServiceFullInfo;
 import org.onap.msb.sdk.discovery.entity.MicroServiceInfo;
@@ -89,6 +90,20 @@ public class MSBServiceClientTest {
         Assert.assertTrue(microServiceFullInfo.getVisualRange().equals("1"));
     }
 
+    @Test
+    public void test_unregistration() throws Exception {
+        mockHttpDel();
+        MSBServiceClient msbClient = new MSBServiceClient("127.0.0.1", 10081);
+        msbClient.cancelMicroServiceInfo("aai", "v8");
+    }
+
+    @Test
+    public void test_unregistration_a_instance() throws Exception {
+        mockHttpDel();
+        MSBServiceClient msbClient = new MSBServiceClient("127.0.0.1", 10081);
+        msbClient.cancelMicroServiceInfo("aai", "v8", "10.74.44.1", "8443");
+    }
+
     private MicroServiceFullInfo mockMicroServiceFullInfo(MicroServiceInfo info) {
         MicroServiceFullInfo serviceInfo = new MicroServiceFullInfo();
         serviceInfo.setServiceName(info.getServiceName());
@@ -113,4 +128,8 @@ public class MSBServiceClientTest {
         PowerMockito.when(HttpClientUtil.httpGet(mockMSBUrl)).thenReturn(mockServiceInfoJson);
     }
 
+    private void mockHttpDel() throws Exception {
+        PowerMockito.mockStatic(HttpClientUtil.class);
+
+    }
 }