TemplateManagerTest.java

package io.robe.admin.util;

import freemarker.template.Configuration;
import freemarker.template.Template;
import org.junit.Assert;
import org.junit.Test;

import java.io.IOException;
import java.io.StringWriter;
import java.io.Writer;
import java.util.HashMap;
import java.util.Map;
import java.util.UUID;

/**
 * Created by hasanmumin on 11/10/2016.
 */
public class TemplateManagerTest {

    @Test
    public void templateShouldThrowRobeRuntimeException2() {

        try {
            new TemplateManager().process(new StringWriter());
            Assert.fail("Should throw Exception ");
        } catch (Exception e) {
            Assert.assertTrue("Throw Exception", true);
        }
    }

    @Test
    public void templateShouldThrowRobeRuntimeException3() {

        try {
            new TemplateManager("TEST", "<p>${\\undefined}</p>");
            Assert.fail("Should throw Exception ");
        } catch (Exception e) {
            Assert.assertTrue("Throw Exception", true);
        }
    }

    @Test
    public void templateManagerWithNameAndBody() {

        String content = "<p>${item}</p>";
        TemplateManager manager = new TemplateManager("test", content);

        Map<String, Object> parameters = new HashMap<>();

        String uuid = UUID.randomUUID().toString();
        parameters.put("item", uuid);

        manager.setParameter(parameters);
        Assert.assertTrue(manager.getParameter().get("item").equals(uuid));


        Writer out = new StringWriter();
        manager.process(out);

        String result = out.toString();

        Assert.assertTrue(result.equals("<p>" + uuid + "</p>"));
    }

    @Test
    public void templateManagerWithTemplate() throws IOException {

        String content = "<p>${item}</p>";

        Template template = new Template("name", content, new Configuration(Configuration.DEFAULT_INCOMPATIBLE_IMPROVEMENTS));
        TemplateManager manager = new TemplateManager(template);


        Map<String, Object> parameters = new HashMap<>();

        String uuid = UUID.randomUUID().toString();
        parameters.put("item", uuid);

        manager.setParameter(parameters);
        Assert.assertTrue(manager.getParameter().get("item").equals(uuid));

        Writer out = new StringWriter();
        manager.process(out);

        String result = out.toString();

        Assert.assertTrue(result.equals("<p>" + uuid + "</p>"));

    }
}