ST_BasicTest.java

package nebula.simpletemplate;

import java.math.BigDecimal;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import com.google.common.collect.Lists;
import com.google.common.collect.Maps;


public class ST_BasicTest extends BasicTest {
    public static final String tmpdir = "tmp";//System.getProperty("java.io.tmpdir");

	@Override
	protected void setUp() throws Exception {
	}

	public void testMaps() {
		String text = "${name}\t${age}\t${male}\t${value}";
		String expected = "wangshilian\t10\ttrue\t3.1415";

		ST st = new ST(text,'$','}');

		// setUp
		Map<String, Object> root = Maps.newHashMap();
		root.put("name", "wangshilian");
		root.put("age", 10);
		root.put("male", true);
		root.put("value", new BigDecimal("3.1415"));

		assertEquals(expected, st.renderNamed(root));
	}

	public void testTypes() {
		String text = "${at.name}\t${at.age}\t${at.male}\t${at.value}";
		String expected = "wangshilian\t10\ttrue\t3.1415";

		ST st = new ST(text,'$','}');

		// setUp
		Person person = new Person();
		person.setName("wangshilian");
		person.setAge(10);
		person.setMale(true);
		person.setValue(new BigDecimal("3.1415"));

		assertEquals(expected, st.render(person));
	}

	public void testTypesList() {
		String text = "${at.name}\t${at.age}\t${at.male}\t${at.value}";
		String expected = "wangshilian\t10\ttrue\t3.1415wangshilian\t10\ttrue\t3.1415wangshilian\t10\ttrue\t3.1415";

		ST st = new ST(text,'$','}');

		// setUp
		List<Person> dataList = Lists.newArrayList();

		{
			Person person = new Person();
			person.setName("wangshilian");
			person.setAge(10);
			person.setMale(true);
			person.setValue(new BigDecimal("3.1415"));
			dataList.add(person);
		}
		{
			Person person = new Person();
			person.setName("wangshilian");
			person.setAge(10);
			person.setMale(true);
			person.setValue(new BigDecimal("3.1415"));
			dataList.add(person);
		}
		{
			Person person = new Person();
			person.setName("wangshilian");
			person.setAge(10);
			person.setMale(true);
			person.setValue(new BigDecimal("3.1415"));
			dataList.add(person);
		}

		assertEquals(expected, st.renderList(dataList));
	}

	String newline = "\n";

	public void testInclude() throws Exception {
		String template = "load ${box(xx)};";
		String expected = "load kewl" + newline + "daddy;";
		
		ST st = new ST(template,'$','}');
		
		 st.impl.nativeGroup.defineTemplate("box", "kewl" + newline + "daddy");

		Map<String, Object> root = Maps.newHashMap();
		root.put("xx", "wangshilian");
		
		String result = st.renderNamed(root);
		assertEquals(expected, result);
	}
	

	public void testIncludeSubTempalte() throws Exception {
		String template = "load ${xx: { 11(${xx})11 } : {x | 22(${x})22}};";
		String expected = "load 22( 11(wangshilian)11 )22;";
		
		ST st = new ST(template,'$','}');
		
		Map<String, Object> root = Maps.newHashMap();
		root.put("xx", "wangshilian");
		
		String result = st.renderNamed(root);
		assertEquals(expected, result);
	}
	
	public void testIncludeSubTempalte2() throws Exception {
		String template = "lo yy:${yy} ad ${xx: {kewl\nxx:${xx}daddy}} xx:${xx} yy:${yy};";
		String expected = "lo yy:yy ad kewl\nxx:xxdaddy xx:xx yy:yy;";
		
		ST st = new ST(template,'$','}');
		
		Map<String, Object> root = Maps.newHashMap();
		root.put("xx", "xx");
		root.put("yy", "yy");
		
		String result = st.renderNamed(root);
		assertEquals(expected, result);
	}
	

	public void testIncludeSubTempalteList() throws Exception {
		String template = "lo yy:${yy} ad ${xx: {kewl( xx:${xx}daddy)}} xx:${xx} yy:${yy};";
		String expected = "lo yy:yy ad kewl( xx:xx001daddy)kewl( xx:xx002daddy)kewl( xx:xx003daddy) xx:xx001xx002xx003 yy:yy;";
		
		ST st = new ST(template,'$','}');
		
		Map<String, Object> root = Maps.newHashMap();
		
		List<String> list = Lists.newArrayList();
		list.add("xx001");
		list.add("xx002");
		list.add("xx003");		
		root.put("xx", list);
		root.put("yy", "yy");
		
		String result = st.renderNamed(root);
		assertEquals(expected, result);
	}


	public void testIncludeSubTempalteQueue() throws Exception {
		String template = "load ${xx: {x1 | 111 ${x1}kewl\ndaddy}:{xx | 222 ${xx}}};";
		String expected = "load 222 111 wangshiliankewl\ndaddy;";
		
		ST st = new ST(template,'$','}');
		
		Map<String, Object> root = Maps.newHashMap();
		root.put("xx", "wangshilian");
		
		String result = st.renderNamed(root);
		assertEquals(expected, result);
	}
	
	
	public void testSubTempalte() throws Exception {
		String template = "${{kewl\ndaddy}};";
		String expected = "kewl\ndaddy;";
		
		ST st = new ST(template,'$','}');
		
		Map<String, Object> root = Maps.newHashMap();
		root.put("xx", "wangshilian");
		
		String result = st.renderNamed(root);
		assertEquals(expected, result);
	}
	
	public void testGroupFile() throws Exception {		
		String template = "t(xx) ::= <<load <data(xx)> >>\n" +
				"data(name) ::= <<kewl\ndaddy;>>";
		String expected = "load kewl\ndaddy; ";
		writeFile(tmpdir, "t.stg", template);

		STGroup group =new STGroupFile(tmpdir + "/" + "t.stg");
		CompiledST tmp = group.lookupTemplate("t");
		
		
		Map<String, Object> root = Maps.newHashMap();
		root.put("xx", "wangshilian");
		String result = tmp.exec(root);
		assertEquals(expected, result);
	}
	
	
	public void testGroupPath() throws Exception {		
		String templateT = "t(xx) ::= << load <data(xx)> >>";
		String templateData	=	"data() ::= << kewl\ndaddy; >>";
		
		String expected = "load kewl\ndaddy;  ";
		writeFile(tmpdir, "t.st", templateT);
		writeFile(tmpdir, "data.st", templateData);

		STGroup group = new STGroupDir("tmp");
		CompiledST tmp = group.lookupTemplate("t");
		
		
		Map<String, Object> root = Maps.newHashMap();
		root.put("xx", "wangshilian");
		String result = tmp.exec(root);
		assertEquals(expected, result);
	}
	
	public void testTrueCond() throws Exception {
		String template = "<if(name)>works<endif>";
		ST st = new ST(template);
		st.add("name", "Ter");
		String expected = "works";
		String result = st.render();
		assertEquals(expected, result);
	}	

	public void testEmptyIFTemplate() throws Exception {
		String template = "<if(x)>fail<elseif(name)><endif>";
		ST st = new ST(template);
		st.add("name", "Ter");
		String expected = "";
		String result = st.render();
		assertEquals(expected, result);
	}

	public void testCondParens() throws Exception {
		String template = "<if(!(x||y)&&!z)>works<endif>";
		ST st = new ST(template);
		String expected = "works";
		String result = st.render();
		assertEquals(expected, result);
	}

	public void testFalseCond() throws Exception {
		String template = "<if(name)>works<endif>";
		ST st = new ST(template);
		String expected = "";
		String result = st.render();
		assertEquals(expected, result);
	}

	public void testFalseCond2() throws Exception {
		String template = "<if(name)>works<endif>";
		ST st = new ST(template);
		st.add("name", null);
		String expected = "";
		String result = st.render();
		assertEquals(expected, result);
	}

	public void testElseIf2() throws Exception {
		String template = "<if(x)>fail1<elseif(y)>fail2<elseif(z)>works<else>fail3<endif>";
		ST st = new ST(template);
		st.add("z", "blort");
		String expected = "works";
		String result = st.render();
		assertEquals(expected, result);
	}

	public void testElseIf3() throws Exception {
		String template = "<if(x)><elseif(y)><elseif(z)>works<else><endif>";
		ST st = new ST(template);
		st.add("z", "blort");
		String expected = "works";
		String result = st.render();
		assertEquals(expected, result);
	}

	public void testNotTrueCond() throws Exception {
		String template = "<if(!name)>works<endif>";
		ST st = new ST(template);
		st.add("name", "Ter");
		String expected = "";
		String result = st.render();
		assertEquals(expected, result);
	}

	public void testNotFalseCond() throws Exception {
		String template = "<if(!name)>works<endif>";
		ST st = new ST(template);
		String expected = "works";
		String result = st.render();
		assertEquals(expected, result);
	}

	public void testParensInConditonal() throws Exception {
		String template = "<if((a||b)&&(c||d))>works<endif>";
		ST st = new ST(template);
		st.add("a", true);
		st.add("b", true);
		st.add("c", true);
		st.add("d", true);
		String expected = "works";
		String result = st.render();
		assertEquals(expected, result);
	}

	public void testParensInConditonal2() throws Exception {
		String template = "<if((!a||b)&&!(c||d))>broken<else>works<endif>";
		ST st = new ST(template);
		st.add("a", true);
		st.add("b", true);
		st.add("c", true);
		st.add("d", true);
		String expected = "works";
		String result = st.render();
		assertEquals(expected, result);
	}

	public void testTrueCondWithElse() throws Exception {
		String template = "<if(name)>works<else>fail<endif>";
		ST st = new ST(template);
		st.add("name", "Ter");
		String expected = "works";
		String result = st.render();
		assertEquals(expected, result);
	}

	public void testFalseCondWithElse() throws Exception {
		String template = "<if(name)>fail<else>works<endif>";
		ST st = new ST(template);
		String expected = "works";
		String result = st.render();
		assertEquals(expected, result);
	}

	public void testElseIf() throws Exception {
		String template = "<if(name)>fail<elseif(id)>works<else>fail<endif>";
		ST st = new ST(template);
		st.add("id", "2DF3DF");
		String expected = "works";
		String result = st.render();
		assertEquals(expected, result);
	}

	public void testElseIfNoElseAllFalse() throws Exception {
		String template = "<if(name)>fail<elseif(id)>fail<endif>";
		ST st = new ST(template);
		String expected = "";
		String result = st.render();
		assertEquals(expected, result);
	}

	public void testElseIfAllExprFalse() throws Exception {
		String template = "<if(name)>fail<elseif(id)>fail<else>works<endif>";
		ST st = new ST(template);
		String expected = "works";
		String result = st.render();
		assertEquals(expected, result);
	}

	public void testOr() throws Exception {
		String template = "<if(name||notThere)>works<else>fail<endif>";
		ST st = new ST(template);
		st.add("name", "Ter");
		String expected = "works";
		String result = st.render();
		assertEquals(expected, result);
	}

	public void testMapConditionAndEscapeInside() throws Exception {
		String template = "<if(m.name)>works \\\\<endif>";
		ST st = new ST(template);
		Map<String, String> m = new HashMap<String, String>();
		m.put("name", "Ter");
		st.add("m", m);
		String expected = "works \\";
		String result = st.render();
		assertEquals(expected, result);
	}

	public void testAnd() throws Exception {
		String template = "<if(name&¬There)>fail<else>works<endif>";
		ST st = new ST(template);
		st.add("name", "Ter");
		String expected = "works";
		String result = st.render();
		assertEquals(expected, result);
	}

	public void testAndNot() throws Exception {
		String template = "<if(name&&!notThere)>works<else>fail<endif>";
		ST st = new ST(template);
		st.add("name", "Ter");
		String expected = "works";
		String result = st.render();
		assertEquals(expected, result);
	}

}