/*
 * Copyright © 2019 Christian Kaltepoth
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0, which is available at
 * http://www.eclipse.org/legal/epl-2.0.
 *
 * This Source Code may also be made available under the following Secondary
 * Licenses when the conditions for such availability set forth in the
 * Eclipse Public License v. 2.0 are satisfied: GNU General Public License,
 * version 2 with the GNU Classpath Exception, which is available at
 * https://www.gnu.org/software/classpath/license.html.
 *
 * SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
 */
package jakarta.mvc.tck.tests.security.csrf.base;

import com.gargoylesoftware.htmlunit.WebClient;
import com.gargoylesoftware.htmlunit.html.DomElement;
import com.gargoylesoftware.htmlunit.html.HtmlPage;
import org.hamcrest.CoreMatchers;
import org.jboss.arquillian.container.test.api.Deployment;
import org.jboss.arquillian.junit.Arquillian;
import org.jboss.arquillian.test.api.ArquillianResource;
import org.jboss.shrinkwrap.api.spec.WebArchive;
import org.jboss.test.audit.annotations.SpecAssertion;
import org.jboss.test.audit.annotations.SpecVersion;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import jakarta.mvc.tck.Sections;
import jakarta.mvc.tck.tests.security.CsrfConstants;
import jakarta.mvc.tck.util.Archives;
import jakarta.mvc.tck.util.MvcMatchers;

import java.io.IOException;
import java.net.URL;

import static org.hamcrest.CoreMatchers.equalTo;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertThat;

@RunWith(Arquillian.class)
@SpecVersion(spec = "mvc", version = "1.0")
public class CsrfBaseTest {

    @ArquillianResource
    private URL baseUrl;

    private WebClient webClient;

    @Deployment(testable = false)
    public static WebArchive createDeployment() {
        return Archives.getMvcArchive()
                .addClass(CsrfBaseController.class)
                .addView("csrf/base/base.jsp")
                .build();
    }

    @Before
    public void before() {
        webClient = new WebClient();
        webClient.getOptions().setThrowExceptionOnFailingStatusCode(false);
        webClient.getOptions().setRedirectEnabled(false);
    }

    /**
     * Encodes is available via injection
     */
    @Test
    @SpecAssertion(section = Sections.SECURITY_CSRF, id = "csrf-obj")
    public void csrfInstanceViaContext() throws IOException {

        HtmlPage page = webClient.getPage(baseUrl.toString() + "mvc/csrf/base");

        assertThat(page.getWebResponse().getStatusCode(), equalTo(200));
        assertThat(page.getElementById("csrf-injected").getTextContent(), equalTo("true"));

    }

    /**
     * Encodes is available via EL
     */
    @Test
    @SpecAssertion(section = Sections.SECURITY_CSRF, id = "csrf-obj")
    public void csrfInstanceViaEL() throws IOException {

        HtmlPage page = webClient.getPage(baseUrl.toString() + "mvc/csrf/base");

        assertThat(page.getWebResponse().getStatusCode(), equalTo(200));
        assertThat(page.getElementById("csrf-el").getTextContent(), equalTo("true"));

    }


    /**
     * CSRF token can be injected into hidden field
     */
    @Test
    @SpecAssertion(section = Sections.SECURITY_CSRF, id = "csrf-hidden-field")
    public void canInjectTokenIntoHiddenField() throws IOException {

        HtmlPage page = webClient.getPage(baseUrl.toString() + "mvc/csrf/base");

        assertThat(page.getWebResponse().getStatusCode(), equalTo(200));

        DomElement inputHidden = page.getElementById("token");
        assertNotNull("Hidden input not found", inputHidden);
        assertThat(inputHidden.getAttribute("name"), MvcMatchers.isNotBlank());
        assertThat(inputHidden.getAttribute("value"), MvcMatchers.isNotBlank());

    }

    /**
     * Token is available from response header
     */
    @Test
    @SpecAssertion(section = Sections.SECURITY_CSRF, id = "csrf-inject-header")
    public void tokenIsProvidedViaElAndResponseHeader() throws IOException {

        HtmlPage page = webClient.getPage(baseUrl.toString() + "mvc/csrf/base");

        assertThat(page.getWebResponse().getStatusCode(), equalTo(200));

        DomElement hiddenField = page.getElementById("token");

        String hiddenFieldValue = hiddenField.getAttribute("value");
        assertThat(hiddenFieldValue, MvcMatchers.isNotBlank());

        String headerValue = page.getWebResponse().getResponseHeaderValue(CsrfConstants.CSRF_TOKEN_HEADER_NAME);
        assertThat(headerValue, MvcMatchers.isNotBlank());
        assertThat(headerValue, CoreMatchers.equalTo(hiddenFieldValue));

    }


}
