package de.superx.util;

import java.io.File;
import java.io.IOException;
import java.nio.charset.Charset;

import org.junit.Assert;
import org.junit.Test;
import org.springframework.security.web.csrf.CsrfToken;
import org.springframework.security.web.csrf.DefaultCsrfToken;

import de.memtext.util.StringUtils;

public class CsrfTokenInsertionTest {

    private static final String TEST_FILE_PATH = String.join(File.separator, "test", "resources", "csrf_token_insertion", "test_file.html");
    private static final String REFERENCE_FILE_PATH = String.join(File.separator, "test", "resources", "csrf_token_insertion", "reference_file.html");

    @Test
    public void testInsertCsrfToken() throws IOException {
        CsrfToken token = new DefaultCsrfToken("bla", "csrfToken", "csrfTokenValue");
        File htmlFile = new File(TEST_FILE_PATH);
        File referenceFile = new File(REFERENCE_FILE_PATH);
        Charset charset = Charset.forName("UTF8");
        String htmlContent = StringUtils.readFile(htmlFile, charset);
        String referenceContent = StringUtils.readFile(referenceFile, charset);
        referenceContent = referenceContent.trim();
        String result = de.superx.util.HtmlUtils.insertCsrfToken(token, htmlContent);
        result = result.trim();
        Assert.assertEquals(referenceContent, result);
    }

}