package org.jboss.resteasy.test.providers.custom;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;

import javax.ws.rs.client.ClientBuilder;
import javax.ws.rs.core.Response;

import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
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.client.jaxrs.ResteasyClient;
import org.jboss.resteasy.spi.HttpResponseCodes;
import org.jboss.resteasy.test.providers.custom.resource.CustomClientConstrainedFeature;
import org.jboss.resteasy.test.providers.custom.resource.CustomConstrainedFeatureResource;
import org.jboss.resteasy.test.providers.custom.resource.CustomServerConstrainedFeature;
import org.jboss.resteasy.utils.PortProviderUtil;
import org.jboss.resteasy.utils.TestUtil;
import org.jboss.shrinkwrap.api.Archive;
import org.jboss.shrinkwrap.api.spec.WebArchive;
import org.junit.Test;
import org.junit.runner.RunWith;

/**
 * @tpSubChapter Core
 * @tpChapter Integration tests
 * @tpSince RESTEasy 3.6.1
 * @tpTestCaseDetails Regression test for RESTEASY-1861
 */
@RunWith(Arquillian.class)
@RunAsClient
public class CustomConstrainedFeatureTest {

    private static final String TEST_URI = generateURL("/test-custom-feature");
    private static final Logger LOGGER = LogManager.getLogger(CustomConstrainedFeatureTest.class.getName());
    private static final String CUSTOM_PROVIDERS_FILENAME = "CustomConstrainedFeature.Providers";

    @Deployment
    public static Archive<?> createTestArchive() {
        WebArchive war = TestUtil.prepareArchive(CustomConstrainedFeatureTest.class.getSimpleName());
        war.addAsResource(CustomConstrainedFeatureTest.class.getPackage(), CUSTOM_PROVIDERS_FILENAME,
                "META-INF/services/javax.ws.rs.ext.Providers");
        return TestUtil.finishContainerPrepare(war, null, CustomConstrainedFeatureResource.class,
                CustomServerConstrainedFeature.class, CustomClientConstrainedFeature.class);
    }

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

    /**
     * @tpTestDetails Call client with restricted feature for server runtime.
     * @tpSince RESTEasy 3.6.1
     */
    @Test
    public void testClientCall() {
        CustomServerConstrainedFeature.reset();
        CustomClientConstrainedFeature.reset();
        // This will register always in SERVER runtime
        // ResteasyProviderFactory providerFactory = ResteasyProviderFactory.newInstance();
        // providerFactory.register(CustomClientConstrainedFeature.class);
        // providerFactory.register(CustomServerConstrainedFeature.class);
        // ResteasyClientImpl client = new ResteasyClientBuilderImpl().build();
        // the line below does the same as if there is providers file in META-INF/services/javax.ws.rs.ext.Providers
        ResteasyClient client = (ResteasyClient) ClientBuilder.newBuilder().register(CustomClientConstrainedFeature.class)
                .register(CustomServerConstrainedFeature.class).build();
        assertTrue(CustomConstrainedFeatureResource.ERROR_CLIENT_FEATURE, CustomClientConstrainedFeature.wasInvoked());
        assertFalse(CustomConstrainedFeatureResource.ERROR_SERVER_FEATURE, CustomServerConstrainedFeature.wasInvoked());
        Response response = client.target(TEST_URI).request().get();
        LOGGER.info("Response from server: {}", response.readEntity(String.class));
        // server must return 200 if only registered feature was for server runtime
        assertEquals(HttpResponseCodes.SC_OK, response.getStatus());
        client.close();
    }
}
