package org.jboss.resteasy.test.spring.inmodule;

import org.jboss.logging.Logger;
import org.jboss.resteasy.client.jaxrs.ResteasyClientBuilder;
import org.jboss.resteasy.util.HttpResponseCodes;
import org.jboss.shrinkwrap.api.ShrinkWrap;
import org.jboss.shrinkwrap.api.asset.StringAsset;
import org.junit.Assert;
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.resteasy.test.spring.inmodule.resource.SpringWebappContextResource;
import org.jboss.resteasy.utils.PortProviderUtil;
import org.jboss.shrinkwrap.api.Archive;
import org.jboss.shrinkwrap.api.spec.WebArchive;
import org.junit.Test;
import org.junit.runner.RunWith;
import javax.ws.rs.client.Client;
import javax.ws.rs.client.WebTarget;
import javax.ws.rs.core.Response;
import java.util.HashMap;
import java.util.Map;

/**
 * @tpSubChapter Spring
 * @tpChapter Integration tests
 * @tpTestCaseDetails Test basic header and uri info context injection with spring dependencies on the classpath
 * @tpSince RESTEasy 3.0.16
 */
@RunWith(Arquillian.class)
@RunAsClient
public class SpringWebappContextTest {

    private static Logger logger = Logger.getLogger(SpringWebappContextTest.class);
    private static final String BASE_URL = PortProviderUtil.generateBaseUrl(SpringWebappContextTest.class.getSimpleName());
    private static final String PATH = "/echo";
    private static final String EXPECTED_URI = BASE_URL + PATH + "/uri";
    private static final String EXPECTED_HEADERS = BASE_URL + PATH + "/headers" + ":text/plain";

    @Deployment
    private static Archive<?> deploy() {
        WebArchive archive = ShrinkWrap.create(WebArchive.class, SpringWebappContextTest.class.getSimpleName() + ".war")
                .addClass(SpringWebappContextResource.class)
                .addAsWebInfResource(SpringWebappContextTest.class.getPackage(), "web.xml", "web.xml")
                .addAsWebInfResource(SpringWebappContextTest.class.getPackage(), "springWebAppContext/applicationContext.xml", "applicationContext.xml");
        archive.addAsManifestResource(new StringAsset("Dependencies: org.springframework.spring meta-inf\n"), "MANIFEST.MF");
        return archive;
    }

    private String generateURL(String path) {
        return PortProviderUtil.generateURL(path, SpringWebappContextTest.class.getSimpleName());
    }

    /**
     * @tpTestDetails Get uri info from @Context injection
     * @tpSince RESTEasy 3.0.16
     */
    @Test
    public void testGetUri() throws Exception {
        doTestGet(PATH + "/uri", EXPECTED_URI, null);
    }

    /**
     * @tpTestDetails Get headers from @Context injection
     * @tpSince RESTEasy 3.0.16
     */
    @Test
    public void testGetHeaders() throws Exception {
        doTestGet(PATH + "/headers", EXPECTED_HEADERS, null);
    }

    /**
     * @tpTestDetails Test that the parameters given to the first request doesn't stick for the second request
     * @tpSince RESTEasy 3.0.16
     */
    @Test
    public void testParamsDontStick() throws Exception {
        Map<String, String> parameters = new HashMap<String, String>();
        parameters.put("param", "0");
        doTestGet(PATH + "/uri", EXPECTED_URI + "?param=0", parameters);
        parameters.put("param", "1");
        doTestGet(PATH + "/uri", EXPECTED_URI + "?param=1", parameters);
    }

    /**
     * @tpTestDetails Ensure concurrent invocations see different injected values
     * @tpSince RESTEasy 3.0.16
     */
    @Test
    public void testConcurrent() throws Exception {
        Thread uri = new Thread(new Runnable() {
            public void run() {
                for (int i = 0; i < 10; i++) {
                    try {
                        doTestGet(PATH + "/uri", EXPECTED_URI, null);
                    } catch (Exception e) {
                        Assert.fail(e.toString());
                    }
                }
            }
        });
        Thread headers = new Thread(new Runnable() {
            public void run() {
                for (int i = 0; i < 10; i++) {
                    try {
                        doTestGet(PATH + "/headers", EXPECTED_HEADERS, null);
                    } catch (Exception e) {
                        Assert.fail(e.toString());
                    }
                }
            }
        });
        uri.start();
        headers.start();
        uri.join();
        headers.join();
    }

    private void doTestGet(String context, String expectedReponsePattern, Map<String, String> parameters) throws Exception {

        Client client = ResteasyClientBuilder.newClient();
        WebTarget target = client.target(generateURL(context));

        WebTarget newTarget = null;
        if (parameters != null) {
            for (Map.Entry<String, String> entry : parameters.entrySet()) {
                logger.info("Entry parameters: " + entry.getKey() + ", " + entry.getValue());
                newTarget = target.queryParam(entry.getKey(), entry.getValue());
            }
        } else {
            newTarget = target;
        }

        verify(newTarget.request().accept("text/plain").get(), HttpResponseCodes.SC_OK, expectedReponsePattern);
        client.close();
    }

    private void verify(Response response, int expectedStatus, String expectedResponsePattern) throws Exception {
        Assert.assertEquals("Unexpected response code", expectedStatus, response.getStatus());

        if (expectedResponsePattern != null) {
            String entity = response.readEntity(String.class);
            Assert.assertTrue("Unexpected response: " + entity + ", no match for: "
                    + expectedResponsePattern, entity.indexOf(expectedResponsePattern) != -1);
        }
    }
}