1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17 package org.springframework.xml.transform;
18
19 import java.io.StringReader;
20 import javax.xml.stream.XMLEventReader;
21 import javax.xml.stream.XMLInputFactory;
22 import javax.xml.stream.XMLStreamReader;
23 import javax.xml.transform.Result;
24 import javax.xml.transform.Transformer;
25 import javax.xml.transform.TransformerFactory;
26
27 import org.custommonkey.xmlunit.XMLTestCase;
28
29 public class StaxSourceTest extends XMLTestCase {
30
31 private static final String XML = "<root xmlns='namespace'><child/></root>";
32
33 private Transformer transformer;
34
35 private XMLInputFactory inputFactory;
36
37 protected void setUp() throws Exception {
38 TransformerFactory transformerFactory = TransformerFactory.newInstance();
39 transformer = transformerFactory.newTransformer();
40 inputFactory = XMLInputFactory.newInstance();
41 }
42
43 public void testStreamReaderSource() throws Exception {
44 XMLStreamReader streamReader = inputFactory.createXMLStreamReader(new StringReader(XML));
45 StaxSource source = new StaxSource(streamReader);
46 assertEquals("Invalid streamReader returned", streamReader, source.getXMLStreamReader());
47 assertNull("EventReader returned", source.getXMLEventReader());
48 Result result = new StringResult();
49 transformer.transform(source, result);
50 assertXMLEqual("Invalid result", XML, result.toString());
51 }
52
53 public void testEventReaderSource() throws Exception {
54 XMLEventReader eventReader = inputFactory.createXMLEventReader(new StringReader(XML));
55 StaxSource source = new StaxSource(eventReader);
56 assertEquals("Invalid eventReader returned", eventReader, source.getXMLEventReader());
57 assertNull("StreamReader returned", source.getXMLStreamReader());
58 Result result = new StringResult();
59 transformer.transform(source, result);
60 assertXMLEqual("Invalid result", XML, result.toString());
61 }
62 }