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