Simple Xslt payload transformer

org.springframework.integration.xml/src/test/java/org/springframework/integration/xml/transformer/XsltPayloadTransformerTest.java

@@ -0,0 +1,104 @@
+/*
+ * Copyright 2002-2008 the original author or authors.
+ *
+ * Licensed 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.springframework.integration.xml.transformer;
+
+import static org.junit.Assert.assertEquals;
+
+import java.io.StringReader;
+
+import javax.xml.parsers.DocumentBuilderFactory;
+import javax.xml.transform.Source;
+import javax.xml.transform.TransformerFactory;
+import javax.xml.transform.dom.DOMResult;
+
+import org.junit.Before;
+import org.junit.Test;
+import org.springframework.core.io.ByteArrayResource;
+import org.springframework.core.io.Resource;
+import org.springframework.integration.message.GenericMessage;
+import org.springframework.integration.message.MessagingException;
+import org.springframework.integration.message.StringMessage;
+import org.springframework.xml.transform.StringSource;
+import org.w3c.dom.Document;
+import org.xml.sax.InputSource;
+
+/**
+ * 
+ * @author Jonas Partner
+ *
+ */
+public class XsltPayloadTransformerTest {
+
+	XsltPayloadTransformer transformer;
+	
+	@Before
+	public void setUp() throws Exception{
+		transformer = new XsltPayloadTransformer(getXslResource());
+	}
+	
+	@Test
+	public void testDocumentAsPayload() throws Exception {
+		Document input = DocumentBuilderFactory.newInstance().newDocumentBuilder().parse(
+				new InputSource(new StringReader(getInputString())));
+		GenericMessage<Document> documentMessage = new GenericMessage<Document>(input);
+		transformer.transform(documentMessage);
+		String rootNodeName = ((Document)documentMessage.getPayload()).getDocumentElement().getNodeName();
+		assertEquals("Wrong name for root element after transform", "bob", rootNodeName);
+	}
+	
+	@Test
+	public void testXmlAsStringPayload() throws Exception {
+		StringMessage message = new StringMessage(getInputString());
+		transformer.transform(message);
+		String rootNodeName = DocumentBuilderFactory.newInstance().newDocumentBuilder().parse(new InputSource(new StringReader(message.getPayload()))).getDocumentElement().getNodeName();
+		assertEquals("Wrong name for root element after transform", "bob", rootNodeName);
+	}
+	
+	@Test
+	public void testSourceAsPayload() throws Exception {
+		GenericMessage<Source> message = new GenericMessage<Source>(new StringSource(getInputString()));
+		transformer.transform(message);
+
+		DOMResult result = new DOMResult();
+		TransformerFactory.newInstance().newTransformer().transform(message.getPayload(), result);
+		String rootNodeName = ((Document)result.getNode()).getDocumentElement().getNodeName();
+		assertEquals("Wrong name for root element after transform", "bob", rootNodeName);
+	}
+	
+	@Test(expected = MessagingException.class)
+	public void testNonXmlString() throws Exception {
+		transformer.transform(new StringMessage("test"));
+	}
+	
+	@Test(expected = MessagingException.class)
+	public void testUnsupportedPayloadType() throws Exception {
+		transformer.transform(new GenericMessage<Long>(new Long(12)));
+	}
+	
+	
+	
+	
+
+	public String getInputString() {
+		return "<?xml version=\"1.0\" encoding=\"ISO-8859-1\"?><order><orderItem>test</orderItem></order>";
+	}
+
+	public Resource getXslResource() throws Exception{
+		String xsl = "<?xml version=\"1.0\" encoding=\"ISO-8859-1\"?><xsl:stylesheet version=\"1.0\" xmlns:xsl=\"http://www.w3.org/1999/XSL/Transform\"><xsl:template match=\"order\"><bob>test</bob></xsl:template></xsl:stylesheet>";
+		return new ByteArrayResource(xsl.getBytes("UTF-8"));
+	}
+
+}