Divide the MSB source codes into two repos

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

diff --git a/apiroute/apiroute-service/src/test/java/org/onap/msb/apiroute/wrapper/consulextend/expose/WatchCatalogServicesTaskTest.java b/apiroute/apiroute-service/src/test/java/org/onap/msb/apiroute/wrapper/consulextend/expose/WatchCatalogServicesTaskTest.java
new file mode 100644 (file)
index 0000000..14455ee
--- /dev/null
+++ b/apiroute/apiroute-service/src/test/java/org/onap/msb/apiroute/wrapper/consulextend/expose/WatchCatalogServicesTaskTest.java
@@ -0,0 +1,138 @@
+package org.onap.msb.apiroute.wrapper.consulextend.expose;
+
+import java.math.BigInteger;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+import org.apache.http.HttpEntity;
+import org.junit.Before;
+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.Consul;
+import org.onap.msb.apiroute.wrapper.consulextend.async.ConsulResponseCallback;
+import org.onap.msb.apiroute.wrapper.consulextend.expose.WatchCatalogServicesTask;
+import org.onap.msb.apiroute.wrapper.consulextend.expose.WatchTask;
+import org.onap.msb.apiroute.wrapper.consulextend.expose.WatchTask.Filter;
+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 WatchCatalogServicesTaskTest {
+       private static final Logger LOGGER = LoggerFactory
+                       .getLogger(WatchCatalogServicesTaskTest.class);
+
+       private Consul consul;
+
+       @SuppressWarnings({ "unchecked", "rawtypes" })
+       @Before
+       public void init() {
+
+               Map<String, List<String>> catalogservices = new HashMap<String, List<String>>();
+               String servicename = "huangleibo";
+               List<String> tags = new ArrayList<String>();
+               tags.add("1111");
+               tags.add("2222");
+               catalogservices.put(servicename, tags);
+
+               long lastContact = 1;
+               boolean knownLeader = true;
+               BigInteger index = BigInteger.valueOf(1);
+               final ConsulResponse<Map<String, List<String>>> response = new ConsulResponse<Map<String, List<String>>>(
+                               catalogservices, lastContact, knownLeader, index);
+
+               //
+               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(response);
+                                               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);
+
+               consul = Consul.newClient();
+
+               WatchCatalogServicesTask task0 = new WatchCatalogServicesTask(
+                               consul.catalogClient(), CatalogOptions.BLANK,
+                               QueryOptions.BLANK, 10);
+
+               WatchCatalogServicesTask task1 = new WatchCatalogServicesTask(
+                               consul.catalogClient(), 10);
+
+               WatchCatalogServicesTask task2 = new WatchCatalogServicesTask(
+                               consul.catalogClient());
+
+       }
+       
+       public class StopHandler implements WatchTask.Handler<HttpEntity>
+       {
+               
+               private WatchCatalogServicesTask task;
+               
+               StopHandler(WatchCatalogServicesTask task)
+               {
+                       this.task = task;
+               }
+
+               @Override
+               public void handle(ConsulResponse<HttpEntity> object) {
+                       // TODO Auto-generated method stub
+//                     Map<String, List<String>> map = (Map<String, List<String>>)object.getResponse();
+                       LOGGER.debug("handler is here");
+                       task.stopWatch();
+               }
+       }
+
+       @SuppressWarnings({ "unchecked", "rawtypes" })
+       @Test
+       public void teststartWatch() {
+               WatchCatalogServicesTask task0 = new WatchCatalogServicesTask(
+                               consul.catalogClient(), CatalogOptions.BLANK,
+                               QueryOptions.BLANK, 10);
+                               
+               task0.addFilter(new Filter() {
+                       @Override
+                       public boolean filter(ConsulResponse object) {
+                               // TODO Auto-generated method stub
+//                             Map<String, List<String>> map = (Map<String, List<String>>)object.getResponse();
+                               LOGGER.debug("filter is here");
+                               return true;
+                       }
+                       
+               });
+               
+               task0.addHandler(new StopHandler(task0));
+               
+               task0.startWatch();
+       }
+       
+}