Add YinXMLEventReaderFactory
[yangtools.git] / yang / yang-model-export / src / test / java / org / opendaylight / yangtools / yang / model / export / SchemaContextEmitterTest.java
index 791dd0c49cddb2839a11129ae5d8859243b0cc8d..f1c151d3abd54847d1f1a58860a7c90b9eab5de6 100644 (file)
@@ -8,55 +8,15 @@
 
 package org.opendaylight.yangtools.yang.model.export;
 
-import static org.junit.Assert.assertEquals;
-import static org.junit.Assert.assertNotEquals;
-import static org.junit.Assert.assertNotNull;
-
-import java.io.BufferedOutputStream;
-import java.io.ByteArrayOutputStream;
 import java.io.IOException;
-import java.io.OutputStream;
 import javax.xml.stream.XMLStreamException;
-import org.custommonkey.xmlunit.Diff;
-import org.custommonkey.xmlunit.ElementNameAndAttributeQualifier;
-import org.custommonkey.xmlunit.XMLAssert;
-import org.custommonkey.xmlunit.XMLUnit;
 import org.junit.Test;
-import org.opendaylight.yangtools.yang.model.api.Module;
-import org.opendaylight.yangtools.yang.model.api.SchemaContext;
-import org.opendaylight.yangtools.yang.model.export.YinExportUtils;
-import org.opendaylight.yangtools.yang.test.util.YangParserTestUtils;
-import org.w3c.dom.Document;
 import org.xml.sax.SAXException;
 
-public class SchemaContextEmitterTest {
+public class SchemaContextEmitterTest extends AbstractYinExportTest {
 
     @Test
     public void testSchemaContextEmitter() throws IOException, XMLStreamException, SAXException {
-        final SchemaContext schemaContext = YangParserTestUtils.parseYangResourceDirectory(
-            "/schema-context-emitter-test");
-        assertNotNull(schemaContext);
-        assertEquals(1, schemaContext.getModules().size());
-
-        final OutputStream byteArrayOutputStream = new ByteArrayOutputStream();
-        final BufferedOutputStream bufferedOutputStream = new BufferedOutputStream(byteArrayOutputStream);
-
-        for (final Module module : schemaContext.getModules()) {
-            YinExportUtils.writeModuleToOutputStream(schemaContext, module, bufferedOutputStream);
-        }
-
-        final String output = byteArrayOutputStream.toString();
-        assertNotNull(output);
-        assertNotEquals(0, output.length());
-
-        final Document doc = YinExportTestUtils.loadDocument("/schema-context-emitter-test/foo.yin");
-        final String expected = YinExportTestUtils.toString(doc.getDocumentElement());
-
-        XMLUnit.setIgnoreWhitespace(true);
-        XMLUnit.setNormalize(true);
-
-        final Diff diff = new Diff(expected, output);
-        diff.overrideElementQualifier(new ElementNameAndAttributeQualifier());
-        XMLAssert.assertXMLEqual(diff, true);
+        exportYinModules("/schema-context-emitter-test", "/schema-context-emitter-test");
     }
 }