package org.apache.struts2.views.freemarker; import com.opensymphony.xwork2.ActionContext; import com.opensymphony.xwork2.mock.MockActionInvocation; import com.opensymphony.xwork2.util.ClassLoaderUtil; import com.opensymphony.xwork2.util.ValueStack; import com.opensymphony.xwork2.util.fs.DefaultFileManagerFactory; import freemarker.template.Configuration; import freemarker.template.TemplateException; import freemarker.template.TemplateExceptionHandler; import org.apache.struts2.ServletActionContext; import org.apache.struts2.StrutsInternalTestCase; import org.apache.struts2.StrutsStatics; import org.apache.struts2.dispatcher.mapper.ActionMapper; import org.apache.struts2.dispatcher.mapper.ActionMapping; import org.apache.struts2.views.jsp.StrutsMockHttpServletResponse; import org.easymock.EasyMock; import org.springframework.mock.web.MockHttpServletRequest; import javax.servlet.ServletContext; import java.io.File; import java.io.PrintWriter; import java.io.StringWriter; import java.net.MalformedURLException; import java.net.URISyntaxException; import static org.apache.struts2.views.jsp.AbstractUITagTest.normalize; public class FreemarkerResultMockedTest extends StrutsInternalTestCase { ValueStack stack; MockActionInvocation invocation; ActionContext context; StrutsMockHttpServletResponse response; PrintWriter writer; StringWriter stringWriter; ServletContext servletContext; MockHttpServletRequest request; Configuration freemarkerConfig; public void testActionThatThrowsExceptionTag() throws Exception { File file = new File(FreeMarkerResultTest.class.getResource("callActionFreeMarker2.ftl").toURI()); EasyMock.expect(servletContext.getRealPath("/tutorial/org/apache/struts2/views/freemarker/callActionFreeMarker.ftl")).andReturn(file.getAbsolutePath()); file = new File(FreeMarkerResultTest.class.getResource("nested.ftl").toURI()); EasyMock.expect(servletContext.getRealPath("/tutorial/org/apache/struts2/views/freemarker/nested.ftl")).andReturn(file.getAbsolutePath()); EasyMock.replay(servletContext); init(); request.setRequestURI("/tutorial/test2.action"); ActionMapping mapping = container.getInstance(ActionMapper.class).getMapping(request, configurationManager); dispatcher.serviceAction(request, response, mapping); assertEquals("beforenestedafter", stringWriter.toString()); } public void testActionThatSucceedsTag() throws Exception { File file = new File(FreeMarkerResultTest.class.getResource("callActionFreeMarker2.ftl").toURI()); EasyMock.expect(servletContext.getRealPath("/tutorial/org/apache/struts2/views/freemarker/callActionFreeMarker2.ftl")).andReturn(file.getAbsolutePath()); file = new File(FreeMarkerResultTest.class.getResource("nested.ftl").toURI()); EasyMock.expect(servletContext.getRealPath("/tutorial/org/apache/struts2/views/freemarker/nested.ftl")).andReturn(file.getAbsolutePath()); EasyMock.replay(servletContext); init(); request.setRequestURI("/tutorial/test5.action"); ActionMapping mapping = container.getInstance(ActionMapper.class).getMapping(request, configurationManager); dispatcher.serviceAction(request, response, mapping); assertEquals("beforenestedafter", stringWriter.toString()); } public void testDynamicAttributesSupport() throws Exception { File file = new File(FreeMarkerResultTest.class.getResource("dynaAttributes.ftl").toURI()); EasyMock.expect(servletContext.getRealPath("/tutorial/org/apache/struts2/views/freemarker/dynaAttributes.ftl")).andReturn(file.getAbsolutePath()); file = new File(ClassLoaderUtil.getResource("template/simple/text.ftl", getClass()).toURI()); EasyMock.expect(servletContext.getRealPath("/template/simple/text.ftl")).andReturn(file.getAbsolutePath()); file = new File(ClassLoaderUtil.getResource("template/simple/css.ftl", getClass()).toURI()); EasyMock.expect(servletContext.getRealPath("/template/simple/css.ftl")).andReturn(file.getAbsolutePath()); EasyMock.expect(servletContext.getRealPath("/template/~~~simple/css.ftl")).andReturn(file.getAbsolutePath()); file = new File(ClassLoaderUtil.getResource("template/simple/scripting-events.ftl", getClass()).toURI()); EasyMock.expect(servletContext.getRealPath("/template/simple/scripting-events.ftl")).andReturn(file.getAbsolutePath()); EasyMock.expect(servletContext.getRealPath("/template/~~~simple/scripting-events.ftl")).andReturn(file.getAbsolutePath()); file = new File(ClassLoaderUtil.getResource("template/simple/common-attributes.ftl", getClass()).toURI()); EasyMock.expect(servletContext.getRealPath("/template/simple/common-attributes.ftl")).andReturn(file.getAbsolutePath()); EasyMock.expect(servletContext.getRealPath("/template/~~~simple/common-attributes.ftl")).andReturn(file.getAbsolutePath()); file = new File(ClassLoaderUtil.getResource("template/simple/dynamic-attributes.ftl", getClass()).toURI()); EasyMock.expect(servletContext.getRealPath("/template/simple/dynamic-attributes.ftl")).andReturn(file.getAbsolutePath()); EasyMock.expect(servletContext.getRealPath("/template/~~~simple/dynamic-attributes.ftl")).andReturn(file.getAbsolutePath()); EasyMock.replay(servletContext); init(); request.setRequestURI("/tutorial/test6.action"); ActionMapping mapping = container.getInstance(ActionMapper.class).getMapping(request, configurationManager); dispatcher.serviceAction(request, response, mapping); // TODO lukaszlenart: remove expectedJDK15 and if() after switching to Java 1.6 String expectedJDK17 = "<input type=\"text\" name=\"test\" value=\"\" id=\"test\" placeholder=\"input\" foo=\"bar\"/>" + "<input type=\"text\" name=\"test\" value=\"\" id=\"test\" placeholder=\"input\" foo=\"bar\"/>" + "<input type=\"text\" name=\"test\" value=\"\" id=\"test\" break=\"true\"/>" + "<input type=\"text\" name=\"required\" value=\"\" id=\"required\" required=\"true\"/>"; String expectedJDK18 = "<input type=\"text\" name=\"test\" value=\"\" id=\"test\" foo=\"bar\" placeholder=\"input\"/>" + "<input type=\"text\" name=\"test\" value=\"\" id=\"test\" foo=\"bar\" placeholder=\"input\"/>" + "<input type=\"text\" name=\"test\" value=\"\" id=\"test\" break=\"true\"/>" + "<input type=\"text\" name=\"required\" value=\"\" id=\"required\" required=\"true\"/>"; String result = stringWriter.toString(); if (result.contains("id=\"test\" foo=\"bar\"")) { assertEquals(expectedJDK18, result); } else { assertEquals(expectedJDK17, result); } } public void testManualListInTemplate() throws Exception { File file = new File(FreeMarkerResultTest.class.getResource("manual-list.ftl").toURI()); EasyMock.expect(servletContext.getRealPath("/tutorial/org/apache/struts2/views/freemarker/manual-list.ftl")).andReturn(file.getAbsolutePath()); file = new File(ClassLoaderUtil.getResource("template/simple/radiomap.ftl", getClass()).toURI()); EasyMock.expect(servletContext.getRealPath("/template/simple/radiomap.ftl")).andReturn(file.getAbsolutePath()); file = new File(ClassLoaderUtil.getResource("template/simple/css.ftl", getClass()).toURI()); EasyMock.expect(servletContext.getRealPath("/template/simple/css.ftl")).andReturn(file.getAbsolutePath()); EasyMock.expect(servletContext.getRealPath("/template/~~~simple/css.ftl")).andReturn(file.getAbsolutePath()); file = new File(ClassLoaderUtil.getResource("template/simple/scripting-events.ftl", getClass()).toURI()); EasyMock.expect(servletContext.getRealPath("/template/simple/scripting-events.ftl")).andReturn(file.getAbsolutePath()); EasyMock.expect(servletContext.getRealPath("/template/~~~simple/scripting-events.ftl")).andReturn(file.getAbsolutePath()); file = new File(ClassLoaderUtil.getResource("template/simple/common-attributes.ftl", getClass()).toURI()); EasyMock.expect(servletContext.getRealPath("/template/simple/common-attributes.ftl")).andReturn(file.getAbsolutePath()); EasyMock.expect(servletContext.getRealPath("/template/~~~simple/common-attributes.ftl")).andReturn(file.getAbsolutePath()); file = new File(ClassLoaderUtil.getResource("template/simple/dynamic-attributes.ftl", getClass()).toURI()); EasyMock.expect(servletContext.getRealPath("/template/simple/dynamic-attributes.ftl")).andReturn(file.getAbsolutePath()); EasyMock.expect(servletContext.getRealPath("/template/~~~simple/dynamic-attributes.ftl")).andReturn(file.getAbsolutePath()); EasyMock.replay(servletContext); init(); request.setRequestURI("/tutorial/test7.action"); ActionMapping mapping = container.getInstance(ActionMapper.class).getMapping(request, configurationManager); dispatcher.serviceAction(request, response, mapping); String expected = "<input type=\"radio\" name=\"client\" id=\"client_foo\" value=\"foo\"/><label for=\"client_foo\">foo</label>\n" + "<input type=\"radio\" name=\"client\" id=\"client_bar\" value=\"bar\"/><label for=\"client_bar\">bar</label>\n" + "\n" + "<input type=\"radio\" name=\"car\" id=\"carford\" value=\"ford\"/><label for=\"carford\">Ford Motor Co</label>\n" + "<input type=\"radio\" name=\"car\" id=\"cartoyota\" value=\"toyota\"/><label for=\"cartoyota\">Toyota</label>\n"; assertEquals(normalize(expected), normalize(stringWriter.toString())); } public void testDynamicAttributesInTheme() throws Exception { File file = new File(FreeMarkerResultTest.class.getResource("customTextField.ftl").toURI()); EasyMock.expect(servletContext.getRealPath("/tutorial/org/apache/struts2/views/freemarker/customTextField.ftl")).andReturn(file.getAbsolutePath()); file = new File(ClassLoaderUtil.getResource("template/test/text.ftl", getClass()).toURI()); EasyMock.expect(servletContext.getRealPath("/template/test/text.ftl")).andReturn(file.getAbsolutePath()); EasyMock.replay(servletContext); init(); request.setRequestURI("/tutorial/test8.action"); ActionMapping mapping = container.getInstance(ActionMapper.class).getMapping(request, configurationManager); dispatcher.serviceAction(request, response, mapping); String expected = "<input type=\"text\"autofocus=\"autofocus\"/>"; assertEquals(expected, stringWriter.toString()); } private void init() throws MalformedURLException, URISyntaxException { stringWriter = new StringWriter(); writer = new PrintWriter(stringWriter); response = new StrutsMockHttpServletResponse(); response.setWriter(writer); request = new MockHttpServletRequest(); stack = ActionContext.getContext().getValueStack(); context = new ActionContext(stack.getContext()); context.put(StrutsStatics.HTTP_RESPONSE, response); context.put(StrutsStatics.HTTP_REQUEST, request); context.put(StrutsStatics.SERVLET_CONTEXT, servletContext); ServletActionContext.setServletContext(servletContext); ServletActionContext.setRequest(request); ServletActionContext.setResponse(response); servletContext.setAttribute(FreemarkerManager.CONFIG_SERVLET_CONTEXT_KEY, null); invocation = new MockActionInvocation(); invocation.setStack(stack); invocation.setInvocationContext(context); //get fm config to use it in mock servlet context FreemarkerManager freemarkerManager = container.getInstance(FreemarkerManager.class); freemarkerConfig = freemarkerManager.getConfiguration(ServletActionContext.getServletContext()); freemarkerConfig.setTemplateExceptionHandler(TemplateExceptionHandler.RETHROW_HANDLER); freemarkerConfig.setServletContextForTemplateLoading(servletContext, null); } @Override public void setUp() throws Exception { super.setUp(); servletContext = EasyMock.createNiceMock(ServletContext.class); EasyMock.expect(servletContext.getInitParameter("TemplatePath")).andReturn(null); EasyMock.expect(servletContext.getInitParameter("templatePath")).andReturn(null); EasyMock.expect(servletContext.getAttribute(FreemarkerManager.CONFIG_SERVLET_CONTEXT_KEY)).andReturn(freemarkerConfig).anyTimes(); } }