yang/yang-data-codec-xml/src/test/java/org/opendaylight/yangtools/yang/data/codec/xml/Yangtools821Test.java

   1 /*
   2  * Copyright (c) 2018 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 package org.opendaylight.yangtools.yang.data.codec.xml;
   9
  10 import java.io.InputStream;
  11 import javax.xml.stream.XMLStreamReader;
  12 import org.junit.Before;
  13 import org.junit.Test;
  14 import org.opendaylight.yangtools.util.xml.UntrustedXML;
  15 import org.opendaylight.yangtools.yang.common.QName;
  16 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  17 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  18 import org.opendaylight.yangtools.yang.data.api.schema.stream.NormalizedNodeStreamWriter;
  19 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTree;
  20 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeCandidate;
  21 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeConfiguration;
  22 import org.opendaylight.yangtools.yang.data.api.schema.tree.DataTreeModification;
  23 import org.opendaylight.yangtools.yang.data.impl.leafref.LeafRefContext;
  24 import org.opendaylight.yangtools.yang.data.impl.leafref.LeafRefDataValidationFailedException;
  25 import org.opendaylight.yangtools.yang.data.impl.leafref.LeafRefValidation;
  26 import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNormalizedNodeStreamWriter;
  27 import org.opendaylight.yangtools.yang.data.impl.schema.NormalizedNodeResult;
  28 import org.opendaylight.yangtools.yang.data.impl.schema.tree.InMemoryDataTreeFactory;
  29 import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
  30 import org.opendaylight.yangtools.yang.model.api.Module;
  31 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
  32 import org.opendaylight.yangtools.yang.test.util.YangParserTestUtils;
  33
  34 public class Yangtools821Test {
  35     private static final String NS = "urn:opendaylight:params:xml:ns:yang:foo";
  36     private static final String REV = "2018-07-18";
  37     private static final QName ROOT = QName.create(NS, REV, "root");
  38     private static final YangInstanceIdentifier ROOT_ID = YangInstanceIdentifier.of(ROOT);
  39
  40     private SchemaContext schemaContext;
  41     private LeafRefContext leafRefContext;
  42     private DataTree dataTree;
  43
  44     @Before
  45     public void setup() {
  46         schemaContext = YangParserTestUtils.parseYangResource("/yangtools821/foo.yang");
  47         leafRefContext = LeafRefContext.create(schemaContext);
  48         dataTree = new InMemoryDataTreeFactory().create(DataTreeConfiguration.DEFAULT_CONFIGURATION, schemaContext);
  49     }
  50
  51     @Test
  52     public void testValidRefFromAugmentation() throws Exception {
  53         final NormalizedNode<?, ?> root = readNode(schemaContext, "/yangtools821/foo-valid.xml");
  54         final DataTreeModification writeModification = dataTree.takeSnapshot().newModification();
  55         writeModification.write(ROOT_ID, root);
  56         writeModification.ready();
  57         final DataTreeCandidate writeContributorsCandidate = dataTree.prepare(writeModification);
  58
  59         LeafRefValidation.validate(writeContributorsCandidate, leafRefContext);
  60         dataTree.commit(writeContributorsCandidate);
  61     }
  62
  63     @Test(expected = LeafRefDataValidationFailedException.class)
  64     public void testInvalidRefFromAugmentation() throws Exception {
  65         final NormalizedNode<?, ?> root = readNode(schemaContext, "/yangtools821/foo-invalid.xml");
  66         final DataTreeModification writeModification = dataTree.takeSnapshot().newModification();
  67         writeModification.write(ROOT_ID, root);
  68         writeModification.ready();
  69         final DataTreeCandidate writeContributorsCandidate = dataTree.prepare(writeModification);
  70
  71         LeafRefValidation.validate(writeContributorsCandidate, leafRefContext);
  72     }
  73
  74     @Test
  75     public void testValidRefInContainerFromAugmentation() throws Exception {
  76         final NormalizedNode<?, ?> root = readNode(schemaContext, "/yangtools821/foo-ref-in-container-valid.xml");
  77         final DataTreeModification writeModification = dataTree.takeSnapshot().newModification();
  78         writeModification.write(ROOT_ID, root);
  79         writeModification.ready();
  80         final DataTreeCandidate writeContributorsCandidate = dataTree.prepare(writeModification);
  81
  82         LeafRefValidation.validate(writeContributorsCandidate, leafRefContext);
  83         dataTree.commit(writeContributorsCandidate);
  84     }
  85
  86     @Test(expected = LeafRefDataValidationFailedException.class)
  87     public void testInvalidRefInContainerFromAugmentation() throws Exception {
  88         final NormalizedNode<?, ?> root = readNode(schemaContext, "/yangtools821/foo-ref-in-container-invalid.xml");
  89         final DataTreeModification writeModification = dataTree.takeSnapshot().newModification();
  90         writeModification.write(ROOT_ID, root);
  91         writeModification.ready();
  92         final DataTreeCandidate writeContributorsCandidate = dataTree.prepare(writeModification);
  93
  94         LeafRefValidation.validate(writeContributorsCandidate, leafRefContext);
  95     }
  96
  97     private static NormalizedNode<?, ?> readNode(final SchemaContext context, final String filename) throws Exception {
  98         final InputStream resourceAsStream = Yangtools821Test.class.getResourceAsStream(filename);
  99         final XMLStreamReader reader = UntrustedXML.createXMLStreamReader(resourceAsStream);
 100         final NormalizedNodeResult result = new NormalizedNodeResult();
 101         final NormalizedNodeStreamWriter streamWriter = ImmutableNormalizedNodeStreamWriter.from(result);
 102         final Module fooModule = context.getModules().iterator().next();
 103         final ContainerSchemaNode rootContainer = (ContainerSchemaNode) fooModule.findDataChildByName(ROOT).get();
 104         final XmlParserStream xmlParser = XmlParserStream.create(streamWriter, context, rootContainer);
 105         xmlParser.parse(reader);
 106         return result.getResult();
 107     }
 108 }