Divide the MSB source codes into two repos

[msb/apigateway.git] / apiroute / apiroute-service / src / test / java / org / onap / msb / apiroute / wrapper / consulextend / ConsulTest.java

diff --git a/apiroute/apiroute-service/src/test/java/org/onap/msb/apiroute/wrapper/consulextend/ConsulTest.java b/apiroute/apiroute-service/src/test/java/org/onap/msb/apiroute/wrapper/consulextend/ConsulTest.java
new file mode 100644 (file)
index 0000000..abf5e09
--- /dev/null
+++ b/apiroute/apiroute-service/src/test/java/org/onap/msb/apiroute/wrapper/consulextend/ConsulTest.java
@@ -0,0 +1,149 @@
+package org.onap.msb.apiroute.wrapper.consulextend;
+
+import java.util.List;
+
+import org.apache.http.HttpEntity;
+import org.junit.BeforeClass;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.Mockito;
+import org.mockito.invocation.InvocationOnMock;
+import org.mockito.stubbing.Answer;
+import org.onap.msb.apiroute.wrapper.consulextend.CatalogClient;
+import org.onap.msb.apiroute.wrapper.consulextend.Consul;
+import org.onap.msb.apiroute.wrapper.consulextend.HealthClient;
+import org.onap.msb.apiroute.wrapper.consulextend.async.ConsulResponseCallback;
+import org.onap.msb.apiroute.wrapper.consulextend.async.OriginalConsulResponse;
+import org.onap.msb.apiroute.wrapper.consulextend.model.health.ServiceHealth;
+import org.onap.msb.apiroute.wrapper.consulextend.util.Http;
+import org.powermock.api.mockito.PowerMockito;
+import org.powermock.core.classloader.annotations.PowerMockIgnore;
+import org.powermock.core.classloader.annotations.PrepareForTest;
+import org.powermock.modules.junit4.PowerMockRunner;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import com.fasterxml.jackson.core.type.TypeReference;
+import com.orbitz.consul.model.ConsulResponse;
+import com.orbitz.consul.option.CatalogOptions;
+import com.orbitz.consul.option.QueryOptions;
+
+@RunWith(PowerMockRunner.class)
+@PrepareForTest({ Http.class })
+@PowerMockIgnore({ "javax.net.ssl.*" })
+public class ConsulTest {
+       private static Consul consul10081;
+       private static Consul consul8500;
+
+       private static final Logger LOGGER = LoggerFactory
+                       .getLogger(ConsulTest.class);
+
+       @SuppressWarnings({ "unchecked", "rawtypes" })
+       @BeforeClass
+       public static void setUpBeforeClass() throws Exception {
+               
+               Http http = PowerMockito.mock(Http.class);
+               
+               PowerMockito
+                               .doAnswer(new Answer() {
+                                       @Override
+                                       public Object answer(InvocationOnMock invocation)
+                                                       throws Throwable {
+                                               Object[] args = invocation.getArguments();
+                                               ((ConsulResponseCallback) args[2]).onComplete(null);                                            
+                                               return null;
+                                       }
+                               })
+                               .when(http)
+                               .asyncGet(Mockito.anyString(),
+                                               Mockito.any(TypeReference.class),
+                                               Mockito.any(ConsulResponseCallback.class));
+               
+               //
+               PowerMockito.spy(Http.class);
+               PowerMockito.when(Http.getInstance()).thenReturn(http);
+                               
+               //
+               consul10081 = Consul.builder().withHostAndPort("10.74.148.111", 10081)
+                               .build();
+               consul8500 = Consul.builder().withHostAndPort("10.74.148.111", 8500)
+                               .build();
+       }
+
+       @Test
+       public void testcatalogClient() {
+               
+               ConsulResponseCallback<HttpEntity> callback = new ConsulResponseCallback<HttpEntity>() {
+
+                       @Override
+                       public void onComplete(
+                                       ConsulResponse<HttpEntity> consulResponse) {
+                               LOGGER.info("service list complete!");
+                       }
+
+                       @Override
+                       public void onFailure(Throwable throwable) {
+                               LOGGER.info("service list failure!");
+                       }
+
+                       @Override
+                       public void onDelayComplete(
+                                       OriginalConsulResponse<HttpEntity> originalConsulResponse) {
+                               // TODO Auto-generated method stub
+                               LOGGER.info("service list complete!");
+                       }
+
+               };
+
+               // 10081
+               CatalogClient catalogclient10081 = consul10081.catalogClient();
+               catalogclient10081.getServices(CatalogOptions.BLANK, QueryOptions.BLANK, callback);
+
+               // 8500
+               CatalogClient catalogclient8500 = consul8500.catalogClient();
+               catalogclient8500.getServices(CatalogOptions.BLANK, QueryOptions.BLANK, callback);
+       }
+
+       @Test
+       public void testhealthClient() {
+
+               ConsulResponseCallback<List<ServiceHealth>> callback = new ConsulResponseCallback<List<ServiceHealth>>() {
+
+                       @Override
+                       public void onComplete(
+                                       ConsulResponse<List<ServiceHealth>> consulResponse) {
+                               LOGGER.info("health service complete!");
+
+                       }
+
+                       @Override
+                       public void onFailure(Throwable throwable) {
+                               LOGGER.info("health service failure!");
+                       }
+
+                       @Override
+                       public void onDelayComplete(
+                                       OriginalConsulResponse<List<ServiceHealth>> originalConsulResponse) {
+                               // TODO Auto-generated method stub
+                               LOGGER.info("health service complete!");
+                       }
+
+               };
+
+               // 10081
+               HealthClient healthClient10081 = consul10081.healthClient();
+               healthClient10081.getAllServiceInstances("apigateway-default", CatalogOptions.BLANK,
+                               QueryOptions.BLANK, callback);
+
+               healthClient10081.getHealthyServiceInstances("apigateway-default",
+                               CatalogOptions.BLANK, QueryOptions.BLANK, callback);
+
+               // 8500
+               HealthClient healthClient8500 = consul8500.healthClient();
+               healthClient8500.getAllServiceInstances("apigateway-default", CatalogOptions.BLANK,
+                               QueryOptions.BLANK, callback);
+               healthClient8500.getHealthyServiceInstances("apigateway-default", CatalogOptions.BLANK,
+                               QueryOptions.BLANK, callback);
+
+       }
+}