summaryrefslogtreecommitdiffstats
path: root/luni/src/test/java/libcore/xml/DomSerializationTest.java
blob: 9015f97c4d629b42eeaae1604612a1cbb05d918d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
/*
 * Copyright (C) 2011 The Android Open Source Project
 *
 * 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 libcore.xml;

import java.io.StringWriter;
import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerFactory;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamResult;
import junit.framework.TestCase;
import org.w3c.dom.Attr;
import org.w3c.dom.Document;
import org.w3c.dom.Element;

public final class DomSerializationTest extends TestCase {
    private DocumentBuilder documentBuilder;
    private Transformer transformer;

    @Override protected void setUp() throws Exception {
        DocumentBuilderFactory builderFactory = DocumentBuilderFactory.newInstance();
        documentBuilder = builderFactory.newDocumentBuilder();
        transformer = TransformerFactory.newInstance().newTransformer();
    }

    public void testWriteDocument() throws Exception {
        Document document = documentBuilder.newDocument();
        Element foo = document.createElement("foo");
        Attr quux = document.createAttribute("quux");
        quux.setValue("abc");
        foo.setAttributeNode(quux);
        foo.appendChild(document.createElement("bar"));
        foo.appendChild(document.createElement("baz"));
        document.appendChild(foo);
        assertXmlEquals("<foo quux=\"abc\"><bar/><baz/></foo>", document);
    }

    public void testWriteSpecialCharactersInText() throws Exception {
        Document document = documentBuilder.newDocument();
        Element foo = document.createElement("foo");
        foo.appendChild(document.createTextNode("5'8\", 5 < 6 & 7 > 3!"));
        document.appendChild(foo);
        assertXmlEquals("<foo>5'8\", 5 &lt; 6 &amp; 7 &gt; 3!</foo>", document);
    }

    private String toXml(Document document) throws Exception {
        StringWriter stringWriter = new StringWriter();
        transformer.transform(new DOMSource(document), new StreamResult(stringWriter));
        return stringWriter.toString();
    }

    private void assertXmlEquals(String expectedXml, Document document) throws Exception {
        String actual = toXml(document);
        String declarationA = "<?xml version=\"1.0\" encoding=\"UTF-8\"?>";
        String declarationB = "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\"?>";
        assertTrue(actual, actual.equals(declarationA + expectedXml)
                || actual.equals(declarationB + expectedXml));
    }
}