Converting DOM Document to File

Issue: BAEL-2167
2018-10-12 05:59:44 +01:00
parent 90867eb6b4
commit f824da90ba
2 changed files with 93 additions and 0 deletions
@@ -0,0 +1,52 @@
+package com.baeldung.xml;
+
+import javax.xml.parsers.DocumentBuilder;
+import javax.xml.parsers.DocumentBuilderFactory;
+import javax.xml.parsers.ParserConfigurationException;
+
+import org.apache.commons.io.FileUtils;
+import org.junit.After;
+import org.junit.Test;
+import org.w3c.dom.Document;
+import org.w3c.dom.Element;
+
+import java.io.File;
+
+public class XMLDocumentWriterUnitTest {
+
+    @Test
+    public void givenXMLDocumentWhenWriteIsCalledThenXMLIsWrittenToFile() throws Exception {
+        Document document = createSampleDocument();
+        new XMLDocumentWriter().write(document, "company_simple.xml", false, false);
+    }
+
+    @Test
+    public void givenXMLDocumentWhenWriteIsCalledWithPrettyPrintThenFormattedXMLIsWrittenToFile() throws Exception {
+        Document document = createSampleDocument();
+        new XMLDocumentWriter().write(document, "company_prettyprinted.xml", false, true);
+    }
+
+    private Document createSampleDocument() throws ParserConfigurationException {
+        DocumentBuilderFactory documentBuilderFactory = DocumentBuilderFactory.newInstance();
+        DocumentBuilder documentBuilder = documentBuilderFactory.newDocumentBuilder();
+        Document document = documentBuilder.newDocument();
+        Element companyElement = document.createElement("Company");
+        document.appendChild(companyElement);
+        Element departmentElement = document.createElement("Department");
+        departmentElement.setAttribute("name", "Sales");
+        companyElement.appendChild(departmentElement);
+        Element employee1 = document.createElement("Employee");
+        employee1.setAttribute("name", "John Smith");
+        departmentElement.appendChild(employee1);
+        Element employee2 = document.createElement("Employee");
+        employee2.setAttribute("name", "Tim Dellor");
+        departmentElement.appendChild(employee2);
+        return document;
+    }
+
+    @After
+    public void cleanUp() throws Exception {
+        FileUtils.deleteQuietly(new File("company_simple.xml"));
+        FileUtils.deleteQuietly(new File("company_prettyprinted.xml"));
+    }
+}