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