review: test Template array access (#1353)

* test Template array access

* fix comments

Author: pvojtechovsky

src/test/java/spoon/test/template/TemplateTest.java

@@ -20,6 +20,8 @@
 import spoon.support.compiler.FileSystemFile;
 import spoon.support.template.Parameters;
 import spoon.template.TemplateMatcher;
+import spoon.template.TemplateParameter;
+import spoon.test.template.testclasses.ArrayAccessTemplate;
 import spoon.test.template.testclasses.InvocationTemplate;
 import spoon.test.template.testclasses.SecurityCheckerTemplate;
 import spoon.test.template.testclasses.bounds.CheckBound;
@@ -476,4 +478,29 @@ public void testTemplateInvocationSubstitution() throws Exception {
 		//iface.$method$() becomes iface.hashCode()
 		assertEquals("iface.hashCode()", templateRoot.toString());
 	}
+	@Test
+	public void testTemplateArrayAccess() throws Exception {
+		//contract: the template engine supports substitution of arrays of parameters.
+		Launcher spoon = new Launcher();
+		spoon.addTemplateResource(new FileSystemFile("./src/test/java/spoon/test/template/testclasses/ArrayAccessTemplate.java"));
+
+		spoon.buildModel();
+		Factory factory = spoon.getFactory();
+
+		CtClass<?> resultKlass = factory.Class().create("Result");
+		CtClass<?> templateClass = factory.Class().get(ArrayAccessTemplate.class);
+		//create array of template parameters, which contains CtBlocks
+		TemplateParameter[] params = templateClass.getMethod("sampleBlocks").getBody().getStatements().toArray(new TemplateParameter[0]);
+		new ArrayAccessTemplate(params).apply(resultKlass);
+		CtMethod<?> m = resultKlass.getMethod("method");
+		//check that both TemplateParameter usages were replaced by appropriate parameter value
+		assertEquals(2, m.getBody().getStatements().size());
+		assertTrue(m.getBody().getStatements().get(0) instanceof CtBlock);
+		assertEquals("int i = 0", ((CtBlock)m.getBody().getStatements().get(0)).getStatement(0).toString());
+		assertTrue(m.getBody().getStatements().get(1) instanceof CtBlock);
+		assertEquals("java.lang.String s = \"Spoon is cool!\"", ((CtBlock)m.getBody().getStatements().get(1)).getStatement(0).toString());
+		//check that both @Parameter usage was replaced by appropriate parameter value
+		CtMethod<?> m2 = resultKlass.getMethod("method2");
+		assertEquals("java.lang.System.out.println(\"second\")", m2.getBody().getStatement(0).toString());
+	}
 }

src/test/java/spoon/test/template/testclasses/ArrayAccessTemplate.java

@@ -0,0 +1,41 @@
+package spoon.test.template.testclasses;
+
+import spoon.reflect.code.CtBlock;
+import spoon.template.ExtensionTemplate;
+import spoon.template.Local;
+import spoon.template.Parameter;
+import spoon.template.TemplateParameter;
+
+public class ArrayAccessTemplate extends ExtensionTemplate {
+
+	public void method() throws Throwable {
+		blocks[0].S();
+		blocks[1].S();
+	}
+	
+	public void method2() throws Throwable {
+		System.out.println(strings[1]);
+	}
+
+	@Parameter
+	TemplateParameter<CtBlock<?>>[] blocks;
+
+	@Parameter
+	String[] strings;
+
+	@Local
+	public ArrayAccessTemplate(TemplateParameter<CtBlock<?>>[] blocks) {
+		this.blocks = blocks;
+		strings = new String[]{"first","second"};
+	}
+	
+	@Local
+	void sampleBlocks() {
+		{
+			int i=0;
+		}
+		{
+			String s="Spoon is cool!";
+		}
+	}
+}