yang/odlext-parser-support/src/test/java/org/opendaylight/yangtools/odlext/parser/Bug3874ExtensionTest.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 package org.opendaylight.yangtools.odlext.parser;
   9
  10 import static org.junit.Assert.assertEquals;
  11 import static org.junit.Assert.assertTrue;
  12
  13 import java.net.URI;
  14 import java.util.List;
  15 import org.junit.AfterClass;
  16 import org.junit.BeforeClass;
  17 import org.junit.Test;
  18 import org.opendaylight.yangtools.odlext.model.api.OpenDaylightExtensionsStatements;
  19 import org.opendaylight.yangtools.odlext.model.api.YangModeledAnyxmlSchemaNode;
  20 import org.opendaylight.yangtools.yang.common.QName;
  21 import org.opendaylight.yangtools.yang.common.QNameModule;
  22 import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
  23 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
  24 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
  25 import org.opendaylight.yangtools.yang.model.api.SchemaNode;
  26 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
  27 import org.opendaylight.yangtools.yang.model.api.UnknownSchemaNode;
  28 import org.opendaylight.yangtools.yang.model.repo.api.YangTextSchemaSource;
  29 import org.opendaylight.yangtools.yang.model.util.SchemaContextUtil;
  30 import org.opendaylight.yangtools.yang.parser.rfc7950.reactor.RFC7950Reactors;
  31 import org.opendaylight.yangtools.yang.parser.rfc7950.repo.YangStatementStreamSource;
  32 import org.opendaylight.yangtools.yang.parser.spi.meta.ModelProcessingPhase;
  33 import org.opendaylight.yangtools.yang.parser.stmt.reactor.CrossSourceStatementReactor;
  34
  35 public class Bug3874ExtensionTest {
  36     private static CrossSourceStatementReactor reactor;
  37
  38     @BeforeClass
  39     public static void createReactor() {
  40         reactor = RFC7950Reactors.vanillaReactorBuilder()
  41                 .addStatementSupport(ModelProcessingPhase.FULL_DECLARATION,
  42                     AnyxmlSchemaLocationStatementSupport.getInstance())
  43                 .addNamespaceSupport(ModelProcessingPhase.FULL_DECLARATION, AnyxmlSchemaLocationNamespace.BEHAVIOUR)
  44                 .overrideStatementSupport(ModelProcessingPhase.FULL_DECLARATION,
  45                     AnyxmlStatementSupportOverride.getInstance())
  46                 .build();
  47     }
  48
  49     @AfterClass
  50     public static void freeReactor() {
  51         reactor = null;
  52     }
  53
  54     @Test
  55     public void test() throws Exception {
  56         SchemaContext context = reactor.newBuild()
  57                 .addSource(YangStatementStreamSource.create(YangTextSchemaSource.forResource("/bugs/bug3874/foo.yang")))
  58                 .addSource(YangStatementStreamSource.create(
  59                     YangTextSchemaSource.forResource("/bugs/bug3874/yang-ext.yang")))
  60                 .buildEffective();
  61
  62         QNameModule foo = QNameModule.create(URI.create("foo"));
  63         QName myContainer2QName = QName.create(foo, "my-container-2");
  64         QName myAnyXmlDataQName = QName.create(foo, "my-anyxml-data");
  65
  66         DataSchemaNode dataChildByName = context.findDataChildByName(myAnyXmlDataQName).get();
  67         assertTrue(dataChildByName instanceof YangModeledAnyxmlSchemaNode);
  68         YangModeledAnyxmlSchemaNode yangModeledAnyXml = (YangModeledAnyxmlSchemaNode) dataChildByName;
  69
  70         SchemaNode myContainer2 = SchemaContextUtil.findDataSchemaNode(context,
  71             SchemaPath.create(true, myContainer2QName));
  72         assertTrue(myContainer2 instanceof ContainerSchemaNode);
  73         assertEquals(myContainer2, yangModeledAnyXml.getSchemaOfAnyXmlData());
  74
  75         List<UnknownSchemaNode> unknownSchemaNodes = yangModeledAnyXml.getUnknownSchemaNodes();
  76         assertEquals(1, unknownSchemaNodes.size());
  77
  78         UnknownSchemaNode next = unknownSchemaNodes.iterator().next();
  79         assertTrue(next instanceof AnyxmlSchemaLocationEffectiveStatementImpl);
  80         AnyxmlSchemaLocationEffectiveStatementImpl anyxmlSchemaLocationUnknownNode =
  81                 (AnyxmlSchemaLocationEffectiveStatementImpl) next;
  82         assertEquals(OpenDaylightExtensionsStatements.ANYXML_SCHEMA_LOCATION.getStatementName(),
  83             anyxmlSchemaLocationUnknownNode.getNodeType());
  84         assertEquals(OpenDaylightExtensionsStatements.ANYXML_SCHEMA_LOCATION.getStatementName(),
  85             anyxmlSchemaLocationUnknownNode.getQName());
  86     }
  87 }