yang/yang-data-impl/src/test/java/org/opendaylight/yangtools/yang/data/impl/schema/transform/dom/serializer/DomSerializerTestUtils.java

   1 /*
   2  * Copyright (c) 2016 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8
   9 package org.opendaylight.yangtools.yang.data.impl.schema.transform.dom.serializer;
  10
  11 import static org.junit.Assert.assertNotNull;
  12 import static org.junit.Assert.assertTrue;
  13
  14 import java.io.IOException;
  15 import java.io.StringWriter;
  16 import java.text.ParseException;
  17 import javax.xml.transform.OutputKeys;
  18 import javax.xml.transform.Transformer;
  19 import javax.xml.transform.TransformerException;
  20 import javax.xml.transform.TransformerFactory;
  21 import javax.xml.transform.TransformerFactoryConfigurationError;
  22 import javax.xml.transform.dom.DOMSource;
  23 import javax.xml.transform.stream.StreamResult;
  24 import org.mockito.Mockito;
  25 import org.opendaylight.yangtools.yang.common.QName;
  26 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  27 import org.opendaylight.yangtools.yang.data.impl.TestUtils;
  28 import org.opendaylight.yangtools.yang.data.impl.codec.TypeDefinitionAwareCodec;
  29 import org.opendaylight.yangtools.yang.data.impl.codec.xml.XmlCodecProvider;
  30 import org.opendaylight.yangtools.yang.data.impl.codec.xml.XmlDocumentUtils;
  31 import org.opendaylight.yangtools.yang.data.impl.schema.transform.base.serializer.NodeSerializerDispatcher;
  32 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
  33 import org.opendaylight.yangtools.yang.model.parser.api.YangSyntaxErrorException;
  34 import org.opendaylight.yangtools.yang.parser.spi.meta.ReactorException;
  35 import org.opendaylight.yangtools.yang.parser.spi.source.StatementStreamSource;
  36 import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.YangStatementSourceImpl;
  37 import org.w3c.dom.Document;
  38 import org.w3c.dom.Element;
  39 import org.w3c.dom.Node;
  40
  41 public class DomSerializerTestUtils {
  42     static final Document DOC = XmlDocumentUtils.getDocument();
  43     static final Element DATA = DOC.createElement("data");
  44     static final NodeSerializerDispatcher MOCK_DISPATCHER = Mockito.mock(NodeSerializerDispatcher.class);
  45     static final XmlCodecProvider CODEC_PROVIDER = TypeDefinitionAwareCodec::from;
  46
  47     private DomSerializerTestUtils() {
  48         throw new UnsupportedOperationException("Utility class");
  49     }
  50
  51     public static SchemaContext getSchemaContext() throws ReactorException, IOException, YangSyntaxErrorException {
  52         final StatementStreamSource source = new YangStatementSourceImpl("/dom-serializer-test/serializer-test.yang", false);
  53         final SchemaContext schemaContext = TestUtils.parseYangSources(source);
  54
  55         assertNotNull("Schema context must not be null.", schemaContext);
  56         return schemaContext;
  57     }
  58
  59     public static YangInstanceIdentifier.NodeIdentifier getNodeIdentifier(final String localName, final String
  60             namespace, final String revision) throws ParseException {
  61         return new YangInstanceIdentifier.NodeIdentifier(QName.create(namespace, revision, localName));
  62     }
  63
  64     public static String toString(final Node xml) {
  65         try {
  66             final Transformer transformer = TransformerFactory.newInstance().newTransformer();
  67             transformer.setOutputProperty(OutputKeys.INDENT, "yes");
  68             transformer.setOutputProperty("{http://xml.apache.org/xslt}indent-amount", "4");
  69
  70             final StreamResult result = new StreamResult(new StringWriter());
  71             final DOMSource source = new DOMSource(xml);
  72             transformer.transform(source, result);
  73
  74             return result.getWriter().toString();
  75         } catch (IllegalArgumentException | TransformerFactoryConfigurationError | TransformerException e) {
  76             throw new RuntimeException("Unable to serialize xml element " + xml, e);
  77         }
  78     }
  79
  80     public static QName generateQname(String name) {
  81         return QName.create("dom-serializer-test", "2016-01-01", name);
  82     }
  83
  84     public static void testResults(String node, Element element) {
  85         DATA.appendChild(element);
  86         final String tempXml = toString(DATA);
  87
  88         assertNotNull(tempXml);
  89         assertTrue(tempXml.contains(node));
  90     }
  91 }