FailOnUnknownTokensTest.java

package com.hubspot.jinjava.tree;

import static org.assertj.core.api.Assertions.assertThat;
import static org.junit.Assert.*;

import java.util.HashMap;
import java.util.Map;

import org.junit.Before;
import org.junit.Test;

import com.hubspot.jinjava.Jinjava;
import com.hubspot.jinjava.JinjavaConfig;
import com.hubspot.jinjava.interpret.FatalTemplateErrorsException;

public class FailOnUnknownTokensTest {
  private static Jinjava jinjava;

  @Before
  public void setUp() throws Exception {
    JinjavaConfig.Builder builder = JinjavaConfig.newBuilder();
    builder.withFailOnUnknownTokens(true);
    JinjavaConfig config = builder.build();
    jinjava = new Jinjava(config);

  }

  @Test(expected = FatalTemplateErrorsException.class)
  public void itThrowsExceptionOnUnknownToken() {
    Map<String, String> context = new HashMap<String, String>();
    context.put("token1", "test");
    String template = "hello {{ token1 }} and {{ token2 }}";
    String str = jinjava.render(template, context);
  }

  @Test
  public void itReplaceTokensWithoutException() {
    Map<String, String> context = new HashMap<String, String>();
    context.put("token1", "test");
    context.put("token2", "test1");
    String template = "hello {{ token1 }} and {{ token2 }}";
    String renderedTemplate = jinjava.render(template, context);
    assertThat(renderedTemplate).isEqualTo("hello test and test1");
  }
}