1   package com.atlassian.plugins.codegen;
2   
3   import com.atlassian.plugins.codegen.modules.BasicClassModuleProperties;
4   import com.atlassian.plugins.codegen.modules.PluginModuleProperties;
5   import com.atlassian.plugins.codegen.util.CodeTemplateHelper;
6   
7   import org.apache.velocity.exception.ResourceNotFoundException;
8   import org.junit.Test;
9   
10  import static org.junit.Assert.assertEquals;
11  
12  /**
13   *
14   */
15  public class TestTemplateUtil
16  {
17  
18      @Test
19      public void simpleTextReplacement() throws Exception
20      {
21          PluginModuleProperties props = new BasicClassModuleProperties();
22          String expected = "i.work";
23          props.setProperty("PACKAGE_NAME", expected);
24          CodeTemplateHelper templateHelper = new CodeTemplateHelper();
25          String parsedTemplate = templateHelper.parseTemplate("templates/simple/Simple.java.vtl", props);
26  
27          assertEquals(expected, parsedTemplate);
28      }
29  
30      @Test(expected = ResourceNotFoundException.class)
31      public void throwsResourceNotFound() throws Exception
32      {
33          PluginModuleProperties props = new BasicClassModuleProperties();
34          String expected = "i.work";
35          props.setProperty("PACKAGE_NAME", expected);
36  
37          CodeTemplateHelper templateHelper = new CodeTemplateHelper();
38          String parsedTemplate = templateHelper.parseTemplate("/junk/simple/Simple.java.vtl", props);
39  
40          assertEquals(expected, parsedTemplate);
41      }
42  }