1/*
2 * Copyright (c) 2014, 2015, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation.
8 *
9 * This code is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
12 * version 2 for more details (a copy is included in the LICENSE file that
13 * accompanied this code).
14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20 * or visit www.oracle.com if you need additional information or have any
21 * questions.
22 */
23
24package transform.util;
25
26import java.io.InputStream;
27
28import javax.xml.parsers.DocumentBuilder;
29import javax.xml.transform.Result;
30import javax.xml.transform.Source;
31import javax.xml.transform.dom.DOMResult;
32import javax.xml.transform.dom.DOMSource;
33
34import org.testng.Assert;
35import org.w3c.dom.Document;
36
37public class DOMUtil extends TransformerUtil {
38
39    DocumentBuilder docBuilder = null;
40
41    private static DOMUtil instance = null;
42
43    /** Creates a new instance of DOMUtil */
44    private DOMUtil() throws Exception {
45        if (docBuilder == null)
46            docBuilder = getDomParser();
47    }
48
49    public static synchronized DOMUtil getInstance() throws Exception {
50        if (instance == null)
51            instance = new DOMUtil();
52        return instance;
53    }
54
55    public Source prepareSource(InputStream is) throws Exception {
56        docBuilder.reset();
57        Document doc1 = docBuilder.parse(is);
58        return new DOMSource(doc1);
59    }
60
61    public Result prepareResult() {
62        Document target = docBuilder.newDocument();
63        return new DOMResult(target);
64    }
65
66    public void checkResult(Result result, String version) {
67        String resultVersion = ((Document) ((DOMResult) result).getNode()).getXmlVersion();
68        Assert.assertTrue(version.equals(resultVersion), "Expected XML Version is 1.1, but actual version " + resultVersion);
69    }
70}
71