/*******************************************************************************
 * Copyright 2016-2017 ZTE, Inc. and others.
 * 
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
 * in compliance with the License. You may obtain a copy of the License at
 * 
 * http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software distributed under the License
 * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
 * or implied. See the License for the specific language governing permissions and limitations under
 * the License.
 ******************************************************************************/
package org.onap.msb.apiroute.wrapper.consulextend.expose;

import java.math.BigInteger;
import java.util.ArrayList;
import java.util.List;

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.WatchServiceHealthTask;
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.model.health.ImmutableService;
import org.onap.msb.apiroute.wrapper.consulextend.model.health.ImmutableServiceHealth;
import org.onap.msb.apiroute.wrapper.consulextend.model.health.Service;
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.model.health.ImmutableNode;
import com.orbitz.consul.option.CatalogOptions;
import com.orbitz.consul.option.QueryOptions;

@RunWith(PowerMockRunner.class)
@PrepareForTest({Http.class})
@PowerMockIgnore({"javax.net.ssl.*", "jdk.internal.reflect.*"})
public class WatchServiceHealthTaskTest {
    private static final Logger LOGGER = LoggerFactory.getLogger(WatchServiceHealthTaskTest.class);

    private Consul consul;

    @SuppressWarnings({"unchecked", "rawtypes"})
    @Before
    public void init() {

        List<ServiceHealth> list = new ArrayList<ServiceHealth>();

        Service service = ImmutableService.builder().id("").port(0).address("").service("huangleibo").addTags("")
                        .createIndex(1).modifyIndex(1).build();
        ServiceHealth serviceHealth = ImmutableServiceHealth.builder().service(service)
                        .node(ImmutableNode.builder().node("").address("").build()).build();
        list.add(serviceHealth);

        long lastContact = 1;
        boolean knownLeader = true;
        BigInteger index = BigInteger.valueOf(1);
        final ConsulResponse<List<ServiceHealth>> response =
                        new ConsulResponse<List<ServiceHealth>>(list, 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);

    }

    @Test
    public void testgetServiceName() {
        consul = Consul.newClient();

        WatchServiceHealthTask task0 = new WatchServiceHealthTask(consul.healthClient(), "huangleibo_task0", true,
                        CatalogOptions.BLANK, 10, QueryOptions.BLANK);

        LOGGER.info("service name:" + task0.getServiceName());

        WatchServiceHealthTask task1 = new WatchServiceHealthTask(consul.healthClient(), "huangleibo_task1", true, 10);

        LOGGER.debug("service name:" + task1.getServiceName());

        WatchServiceHealthTask task2 = new WatchServiceHealthTask(consul.healthClient(), "huangleibo_task2", 10);

        LOGGER.debug("service name:" + task2.getServiceName());

    }

    public class StopHandler implements WatchTask.Handler<List<ServiceHealth>> {

        private WatchServiceHealthTask task;

        StopHandler(WatchServiceHealthTask task) {
            this.task = task;
        }

        @Override
        public void handle(ConsulResponse<List<ServiceHealth>> object) {
            // TODO Auto-generated method stub
            List<ServiceHealth> list = (List<ServiceHealth>) object.getResponse();
            LOGGER.debug("handler:" + list.get(0).getService().getService());
            task.stopWatch();
        }
    }

    @SuppressWarnings({"unchecked", "rawtypes"})
    @Test
    public void teststartWatch() {
        Consul consul = Consul.newClient();
        String serviceName = "huangleibo";

        WatchServiceHealthTask task0 = new WatchServiceHealthTask(consul.healthClient(), serviceName, true,
                        CatalogOptions.BLANK, 10, QueryOptions.BLANK);

        task0.addFilter(new Filter() {

            @Override
            public boolean filter(ConsulResponse object) {
                // TODO Auto-generated method stub
                List<ServiceHealth> list = (List<ServiceHealth>) object.getResponse();
                LOGGER.debug("filter:" + list.get(0).getService().getService());
                return true;
            }

        });

        task0.addHandler(new StopHandler(task0));

        task0.startWatch();
    }
}