/* * $Id$ * * Licensed to the Apache Software Foundation (ASF) under one * or more contributor license agreements. See the NOTICE file * distributed with this work for additional information * regarding copyright ownership. The ASF licenses this file * to you under the Apache License, Version 2.0 (the * "License"); you may not use this file except in compliance * with the License. You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, * software distributed under the License is distributed on an * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY * KIND, either express or implied. See the License for the * specific language governing permissions and limitations * under the License. */ package org.apache.struts2.views.xslt; import com.opensymphony.xwork2.Action; import com.opensymphony.xwork2.ActionContext; import com.opensymphony.xwork2.mock.MockActionInvocation; import com.opensymphony.xwork2.util.ClassLoaderUtil; import com.opensymphony.xwork2.util.ValueStack; import org.apache.struts2.ServletActionContext; import org.apache.struts2.StrutsInternalTestCase; import org.springframework.mock.web.MockHttpServletRequest; import org.springframework.mock.web.MockHttpServletResponse; import org.springframework.mock.web.MockServletContext; import javax.xml.transform.Source; import javax.xml.transform.TransformerException; import javax.xml.transform.URIResolver; import javax.xml.transform.stream.StreamSource; import java.util.ArrayList; import java.util.List; /** * Unit test for {@link XSLTResult}. * */ public class XSLTResultTest extends StrutsInternalTestCase { private XSLTResult result; private MockHttpServletResponse response; private MockHttpServletRequest request; private MockServletContext servletContext; private MockActionInvocation mai; private ValueStack stack; public void testNoLocation() throws Exception { try { result.setParse(false); result.setLocation(null); result.execute(mai); fail("Should have thrown an IllegalArgumentException"); } catch (IllegalArgumentException e) { // success } } public void testNoFileFound() throws Exception { try { result.setParse(false); result.setLocation("nofile.xsl"); result.execute(mai); fail("Should have thrown a TransformerException"); } catch (TransformerException e) { // success } } public void testSimpleTransform() throws Exception { result.setParse(false); result.setLocation("XSLTResultTest.xsl"); result.execute(mai); String out = response.getContentAsString(); assertTrue(out.startsWith("<?xml version=\"1.0\" encoding=\"UTF-8\"?>")); assertTrue(out.indexOf("<result xmlns=\"http://www.w3.org/TR/xhtml1/strict\"") > -1); } public void testSimpleTransformParse() throws Exception { result.setParse(true); result.setLocation("${top.myLocation}"); result.execute(mai); String out = response.getContentAsString(); assertTrue(out.startsWith("<?xml version=\"1.0\" encoding=\"UTF-8\"?>")); assertTrue(out.indexOf("<result xmlns=\"http://www.w3.org/TR/xhtml1/strict\"") > -1); } public void testTransform2() throws Exception { result.setParse(false); result.setLocation("XSLTResultTest2.xsl"); result.execute(mai); String out = response.getContentAsString(); assertTrue(out.startsWith("<?xml version=\"1.0\" encoding=\"UTF-8\"?>")); assertTrue(out.indexOf("<html xmlns=\"http://www.w3.org/TR/xhtml1/strict\"") > -1); assertTrue(out.indexOf("Hello Santa Claus how are you?") > -1); } public void testTransform3() throws Exception { result.setParse(false); result.setLocation("XSLTResultTest3.xsl"); result.execute(mai); String out = response.getContentAsString(); assertTrue(out.startsWith("<?xml version=\"1.0\" encoding=\"UTF-8\"?>")); assertTrue(out.indexOf("<html xmlns=\"http://www.w3.org/TR/xhtml1/strict\"") > -1); assertTrue(out.indexOf("Hello Santa Claus how are you?") > -1); assertTrue(out.indexOf("WebWork in Action by Patrick and Jason") > -1); assertTrue(out.indexOf("XWork not in Action by Superman") > -1); } public void testTransformWithBoolean() throws Exception { result.setParse(false); result.setLocation("XSLTResultTest5.xsl"); result.execute(mai); String out = response.getContentAsString(); assertTrue(out.startsWith("<?xml version=\"1.0\" encoding=\"UTF-8\"?>")); assertTrue(out.indexOf("<html xmlns=\"http://www.w3.org/TR/xhtml1/strict\"") > -1); assertTrue(out.indexOf("Hello Santa Claus how are you?") > -1); assertTrue(out.indexOf("You are active: true") > -1); } public void testTransform4WithDocumentInclude() throws Exception { result = new XSLTResult(){ protected URIResolver getURIResolver() { return new URIResolver() { public Source resolve(String href, String base) throws TransformerException { return new StreamSource(ClassLoaderUtil.getResourceAsStream(href, this.getClass())); } }; } }; result.setParse(false); result.setLocation("XSLTResultTest4.xsl"); result.execute(mai); String out = response.getContentAsString(); assertTrue(out.startsWith("<?xml version=\"1.0\" encoding=\"UTF-8\"?>")); assertTrue(out.indexOf("<validators>") > -1); } public void testTransform4WithBadDocumentInclude() throws Exception { result = new XSLTResult(){ protected URIResolver getURIResolver() { return new URIResolver() { public Source resolve(String href, String base) throws TransformerException { return new StreamSource(ClassLoaderUtil.getResourceAsStream(href, this.getClass())); } }; } }; result.setParse(false); result.setLocation("XSLTResultTest4.badinclude.xsl"); try { result.execute(mai); fail("Should have thrown an exception"); } catch (Exception ex) { assertEquals("Error transforming result", ex.getMessage()); } } public void testTransformWithError() throws Exception { result = new XSLTResult(){ protected URIResolver getURIResolver() { return new URIResolver() { public Source resolve(String href, String base) throws TransformerException { throw new TransformerException("Some random error"); } }; } }; result.setLocation("XSLTResultTest4.xsl"); try { result.execute(mai); fail("Should have thrown an exception"); } catch (Exception ex) { assertEquals("Error transforming result", ex.getMessage()); } } public void testStatusCode() throws Exception { result.setParse(false); result.setLocation("XSLTResultTest.xsl"); result.setStatus("302"); result.execute(mai); String out = response.getContentAsString(); assertEquals(302, response.getStatus()); assertTrue(out.startsWith("<?xml version=\"1.0\" encoding=\"UTF-8\"?>")); assertTrue(out.indexOf("<result xmlns=\"http://www.w3.org/TR/xhtml1/strict\"") > -1); } protected void setUp() throws Exception { super.setUp(); request = new MockHttpServletRequest(); response = new MockHttpServletResponse(); servletContext = new MockServletContext(); result = new XSLTResult(); stack = ActionContext.getContext().getValueStack(); MyAction action = new MyAction(); mai = new com.opensymphony.xwork2.mock.MockActionInvocation(); mai.setAction(action); mai.setStack(stack); mai.setInvocationContext(ActionContext.getContext()); stack.push(action); ActionContext.getContext().put(ServletActionContext.HTTP_REQUEST, request); ActionContext.getContext().put(ServletActionContext.HTTP_RESPONSE, response); ActionContext.getContext().put(ServletActionContext.SERVLET_CONTEXT, servletContext); } protected void tearDown() throws Exception { super.tearDown(); request = null; response = null; servletContext = null; result = null; stack = null; mai = null; } private class MyAction implements Action { public String execute() throws Exception { return SUCCESS; } public String getMyLocation() { return ("XSLTResultTest.xsl"); } public String getUsername() { return "Santa Claus"; } public boolean isActive() { return true; } public List getBooks() { List list = new ArrayList(); list.add(new Book("WebWork in Action", "Patrick and Jason")); list.add(null); list.add(new Book("XWork not in Action", "Superman")); return list; } } public class Book { private String title; private String author; public Book(String title, String author) { this.title = title; this.author = author; } public String getTitle() { return title; } public String getAuthor() { return author; } } }