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.api.TypeDefinition;
  34 import org.opendaylight.yangtools.yang.model.parser.api.YangSyntaxErrorException;
  35 import org.opendaylight.yangtools.yang.parser.spi.meta.ReactorException;
  36 import org.opendaylight.yangtools.yang.parser.spi.source.StatementStreamSource;
  37 import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.YangStatementSourceImpl;
  38 import org.w3c.dom.Document;
  39 import org.w3c.dom.Element;
  40 import org.w3c.dom.Node;
  41
  42 public class DomSerializerTestUtils {
  43     static final Document DOC = XmlDocumentUtils.getDocument();
  44     static final Element DATA = DOC.createElement("data");
  45     static final NodeSerializerDispatcher MOCK_DISPATCHER = Mockito.mock(NodeSerializerDispatcher.class);
  46     static final XmlCodecProvider CODEC_PROVIDER = new XmlCodecProvider() {
  47         @Override
  48         public TypeDefinitionAwareCodec<Object, ? extends TypeDefinition<?>> codecFor(TypeDefinition<?> baseType) {
  49             return TypeDefinitionAwareCodec.from(baseType);
  50         }
  51     };
  52
  53     private DomSerializerTestUtils() {
  54         throw new UnsupportedOperationException("Utility class");
  55     }
  56
  57     public static SchemaContext getSchemaContext() throws ReactorException, IOException, YangSyntaxErrorException {
  58         final StatementStreamSource source = new YangStatementSourceImpl("/dom-serializer-test/serializer-test.yang", false);
  59         final SchemaContext schemaContext = TestUtils.parseYangSources(source);
  60
  61         assertNotNull("Schema context must not be null.", schemaContext);
  62         return schemaContext;
  63     }
  64
  65     public static YangInstanceIdentifier.NodeIdentifier getNodeIdentifier(final String localName, final String
  66             namespace, final String revision) throws ParseException {
  67         return new YangInstanceIdentifier.NodeIdentifier(QName.create(namespace, revision, localName));
  68     }
  69
  70     public static String toString(final Node xml) {
  71         try {
  72             final Transformer transformer = TransformerFactory.newInstance().newTransformer();
  73             transformer.setOutputProperty(OutputKeys.INDENT, "yes");
  74             transformer.setOutputProperty("{http://xml.apache.org/xslt}indent-amount", "4");
  75
  76             final StreamResult result = new StreamResult(new StringWriter());
  77             final DOMSource source = new DOMSource(xml);
  78             transformer.transform(source, result);
  79
  80             return result.getWriter().toString();
  81         } catch (IllegalArgumentException | TransformerFactoryConfigurationError | TransformerException e) {
  82             throw new RuntimeException("Unable to serialize xml element " + xml, e);
  83         }
  84     }
  85
  86     public static QName generateQname(String name) {
  87         return QName.create("dom-serializer-test", "2016-01-01", name);
  88     }
  89
  90     public static void testResults(String node, Element element) {
  91         DATA.appendChild(element);
  92         final String tempXml = toString(DATA);
  93
  94         assertNotNull(tempXml);
  95         assertTrue(tempXml.contains(node));
  96     }
  97 }