package org.jboss.resteasy.test.client;

import java.util.concurrent.CountDownLatch;

import javax.ws.rs.client.Client;
import javax.ws.rs.client.ClientBuilder;
import javax.ws.rs.client.Entity;
import javax.ws.rs.client.InvocationCallback;
import javax.ws.rs.client.WebTarget;
import javax.ws.rs.core.Response;

import org.jboss.arquillian.container.test.api.Deployment;
import org.jboss.arquillian.container.test.api.RunAsClient;
import org.jboss.arquillian.junit.Arquillian;
import org.jboss.logging.Logger;
import org.jboss.resteasy.client.jaxrs.ResteasyClientBuilder;
import org.jboss.resteasy.test.client.resource.AsyncInvokeResource;
import org.jboss.resteasy.test.client.resource.InContainerClientResource;
import org.jboss.resteasy.utils.TestUtil;
import org.jboss.shrinkwrap.api.Archive;
import org.jboss.shrinkwrap.api.spec.WebArchive;
import org.junit.After;
import org.junit.Assert;
import org.junit.Ignore;
import org.junit.Test;
import org.junit.runner.RunWith;

/**
 * @tpChapter Client tests
 * @tpSubChapter Performances
 * @tpTestCaseDetails https://issues.jboss.org/browse/RESTEASY-XYZ
 * @tpSince RESTEasy 4.0.0
 */
@RunWith(Arquillian.class)
@RunAsClient
@Ignore("Not a functional test")
public class InContainerClientBenchTest extends ClientTestBase {

    static Client nioClient;
    private static Logger log = Logger.getLogger(InContainerClientBenchTest.class);

    static final int ITERATIONS = 4000;
    static final int MAX_CONNECTIONS = 40;

    @Deployment
    public static Archive<?> deploy() {
        WebArchive war = TestUtil.prepareArchive(InContainerClientBenchTest.class.getSimpleName());
        war.addClass(InContainerClientBenchTest.class);
        war.addClass(ClientTestBase.class);
        return TestUtil.finishContainerPrepare(war, null, AsyncInvokeResource.class, InContainerClientResource.class);
    }

    @After
    public void after() throws Exception {
        if (nioClient != null)
            nioClient.close();
    }

    @Test
    public void testAsyncComposedPost() throws Exception {
        long start = System.currentTimeMillis();
        final String oldProp = System.getProperty("http.maxConnections");
        System.setProperty("http.maxConnections", String.valueOf(MAX_CONNECTIONS));
        nioClient = ((ResteasyClientBuilder) ClientBuilder.newBuilder()).useAsyncHttpEngine().build();
        WebTarget wt = nioClient.target(generateURL("/test-client"));
        runCallback(wt, "NIO", "client-post post ");
        long end = System.currentTimeMillis() - start;
        log.info("TEST COMPOSED NON BLOCKING IO - " + ITERATIONS + " iterations took " + end + "ms");
        if (oldProp != null) {
            System.setProperty("http.maxConnections", oldProp);
        } else {
            System.clearProperty("http.maxConnections");
        }
    }

    private void runCallback(WebTarget wt, String msg, String expectedResultPrefix) throws Exception {
        CountDownLatch latch = new CountDownLatch(ITERATIONS);
        for (int i = 0; i < ITERATIONS; i++) {
            final String m = msg + i;
            wt.request().async().post(Entity.text(m), new InvocationCallback<Response>() {
                @Override
                public void completed(Response response) {
                    String entity = response.readEntity(String.class);
                    Assert.assertEquals("Got: " + entity, expectedResultPrefix + m, entity);
                    latch.countDown();
                }

                @Override
                public void failed(Throwable error) {
                    throw new RuntimeException(error);
                }
            });
        }
        latch.await();
    }
}
