01: /*
02: * Copyright 2006 the original author or authors.
03: *
04: * Licensed under the Apache License, Version 2.0 (the "License");
05: * you may not use this file except in compliance with the License.
06: * You may obtain a copy of the License at
07: *
08: * http://www.apache.org/licenses/LICENSE-2.0
09: *
10: * Unless required by applicable law or agreed to in writing, software
11: * distributed under the License is distributed on an "AS IS" BASIS,
12: * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13: * See the License for the specific language governing permissions and
14: * limitations under the License.
15: */
16:
17: package org.springframework.oxm.xstream;
18:
19: import java.io.ByteArrayInputStream;
20: import java.io.StringReader;
21: import java.util.Properties;
22:
23: import javax.xml.parsers.DocumentBuilder;
24: import javax.xml.parsers.DocumentBuilderFactory;
25: import javax.xml.stream.XMLInputFactory;
26: import javax.xml.stream.XMLStreamReader;
27: import javax.xml.transform.dom.DOMSource;
28: import javax.xml.transform.stream.StreamSource;
29:
30: import junit.framework.TestCase;
31: import org.w3c.dom.Document;
32: import org.xml.sax.InputSource;
33:
34: import org.springframework.xml.transform.StaxSource;
35:
36: public class XStreamUnmarshallerTest extends TestCase {
37:
38: protected static final String INPUT_STRING = "<flight><flightNumber>42</flightNumber></flight>";
39:
40: private XStreamMarshaller unmarshaller;
41:
42: protected void setUp() throws Exception {
43: unmarshaller = new XStreamMarshaller();
44: Properties aliases = new Properties();
45: aliases.setProperty("flight", Flight.class.getName());
46: unmarshaller.setAliases(aliases);
47: }
48:
49: private void testFlight(Object o) {
50: assertTrue("Unmarshalled object is not Flights",
51: o instanceof Flight);
52: Flight flight = (Flight) o;
53: assertNotNull("Flight is null", flight);
54: assertEquals("Number is invalid", 42L, flight.getFlightNumber());
55: }
56:
57: public void testUnmarshalDomSource() throws Exception {
58: DocumentBuilder builder = DocumentBuilderFactory.newInstance()
59: .newDocumentBuilder();
60: Document document = builder.parse(new InputSource(
61: new StringReader(INPUT_STRING)));
62: DOMSource source = new DOMSource(document);
63: Object flight = unmarshaller.unmarshal(source);
64: testFlight(flight);
65: }
66:
67: public void testUnmarshalStaxSourceXmlStreamReader()
68: throws Exception {
69: XMLInputFactory inputFactory = XMLInputFactory.newInstance();
70: XMLStreamReader streamReader = inputFactory
71: .createXMLStreamReader(new StringReader(INPUT_STRING));
72: StaxSource source = new StaxSource(streamReader);
73: Object flights = unmarshaller.unmarshal(source);
74: testFlight(flights);
75: }
76:
77: public void testUnmarshalStreamSourceInputStream() throws Exception {
78: StreamSource source = new StreamSource(
79: new ByteArrayInputStream(INPUT_STRING.getBytes("UTF-8")));
80: Object flights = unmarshaller.unmarshal(source);
81: testFlight(flights);
82: }
83:
84: public void testUnmarshalStreamSourceReader() throws Exception {
85: StreamSource source = new StreamSource(new StringReader(
86: INPUT_STRING));
87: Object flights = unmarshaller.unmarshal(source);
88: testFlight(flights);
89: }
90: }
|