| /* |
| * 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.Arrays; |
| 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.setStylesheetLocation(null); |
| result.execute(mai); |
| fail("Should have thrown an IllegalArgumentException"); |
| } catch (IllegalArgumentException e) { |
| // success |
| } |
| } |
| |
| public void testNoFileFound() throws Exception { |
| try { |
| result.setParse(false); |
| result.setStylesheetLocation("nofile.xsl"); |
| result.execute(mai); |
| fail("Should have thrown a TransformerException"); |
| } catch (TransformerException e) { |
| // success |
| } |
| } |
| |
| public void testSimpleTransform() throws Exception { |
| result.setParse(false); |
| result.setStylesheetLocation("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 testSimpleTransform5() throws Exception { |
| result.setParse(false); |
| result.setStylesheetLocation("XSLTResultTest6.xsl"); |
| result.execute(mai); |
| |
| String out = response.getContentAsString(); |
| assertTrue(out.startsWith("<?xml version=\"1.0\" encoding=\"UTF-8\"?>")); |
| assertTrue(out.contains("<title>WebWork in Action</title>")); |
| assertTrue(out.contains("<author>Patrick and Jason</author>")); |
| assertTrue(out.contains("<editions><edition value=\"I\">I</edition><edition value=\"IV\">IV</edition></editions>")); |
| assertTrue(out.contains("<book><title/><author/><editions/></book>")); |
| assertTrue(out.contains("<title>XWork not in Action</title>")); |
| assertTrue(out.contains("<author>Superman</author>")); |
| assertTrue(out.contains("<editions><edition value=\"1234\">1234</edition><edition value=\"345\">345</edition><edition value=\"6667\">6667</edition></editions>")); |
| } |
| |
| public void testSimpleTransformParse() throws Exception { |
| result.setParse(true); |
| result.setStylesheetLocation("${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.setStylesheetLocation("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.setStylesheetLocation("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.setStylesheetLocation("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.setStylesheetLocation("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.setStylesheetLocation("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.setStylesheetLocation("XSLTResultTest4.xsl"); |
| try { |
| result.execute(mai); |
| fail("Should have thrown an exception"); |
| } catch (Exception ex) { |
| assertEquals("Error transforming result", ex.getMessage()); |
| } |
| } |
| |
| public void testTransformWithBadCharacter() throws Exception { |
| result = new XSLTResult(); |
| result.setStylesheetLocation("XSLTResultTest.bad.character.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.setStylesheetLocation("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); |
| } |
| |
| public void testEncoding() throws Exception { |
| result.setParse(false); |
| result.setStylesheetLocation("XSLTResultTest.xsl"); |
| result.setEncoding("ISO-8859-1"); |
| result.execute(mai); |
| |
| String actual = response.getCharacterEncoding(); |
| |
| assertEquals(actual, "ISO-8859-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", Arrays.asList("I", "IV"))); |
| list.add(null); |
| list.add(new Book("XWork not in Action", "Superman", Arrays.asList("1234", "345", "6667"))); |
| return list; |
| } |
| |
| } |
| |
| public class Book { |
| |
| private String title; |
| private String author; |
| private List<String> editions; |
| |
| public Book(String title, String author, List<String> editions) { |
| this.title = title; |
| this.author = author; |
| this.editions = editions; |
| } |
| |
| public String getTitle() { |
| return title; |
| } |
| |
| public String getAuthor() { |
| return author; |
| } |
| |
| public List<String> getEditions() { |
| return editions; |
| } |
| } |
| } |