yang/rfc6241-parser-support/src/test/java/org/opendaylight/yangtools/rfc6241/parser/NetconfTest.java

   1 /*
   2  * Copyright (c) 2019 PANTHEON.tech, s.r.o. 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.rfc6241.parser;
   9
  10 import static org.hamcrest.CoreMatchers.instanceOf;
  11 import static org.hamcrest.CoreMatchers.is;
  12 import static org.junit.Assert.assertEquals;
  13 import static org.junit.Assert.assertThat;
  14 import static org.junit.Assert.assertTrue;
  15
  16 import java.io.IOException;
  17 import java.util.Collection;
  18 import java.util.Iterator;
  19 import java.util.Optional;
  20 import org.junit.AfterClass;
  21 import org.junit.BeforeClass;
  22 import org.junit.Test;
  23 import org.opendaylight.yangtools.rfc6241.model.api.GetFilterElementAttributesSchemaNode;
  24 import org.opendaylight.yangtools.rfc6241.model.api.NetconfConstants;
  25 import org.opendaylight.yangtools.yang.common.QName;
  26 import org.opendaylight.yangtools.yang.model.api.AnyXmlSchemaNode;
  27 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
  28 import org.opendaylight.yangtools.yang.model.api.Module;
  29 import org.opendaylight.yangtools.yang.model.api.RpcDefinition;
  30 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
  31 import org.opendaylight.yangtools.yang.model.parser.api.YangSyntaxErrorException;
  32 import org.opendaylight.yangtools.yang.model.repo.api.YangTextSchemaSource;
  33 import org.opendaylight.yangtools.yang.parser.rfc7950.reactor.RFC7950Reactors;
  34 import org.opendaylight.yangtools.yang.parser.rfc7950.repo.YangStatementStreamSource;
  35 import org.opendaylight.yangtools.yang.parser.spi.meta.ModelProcessingPhase;
  36 import org.opendaylight.yangtools.yang.parser.spi.meta.ReactorException;
  37 import org.opendaylight.yangtools.yang.parser.stmt.reactor.CrossSourceStatementReactor;
  38
  39 public class NetconfTest {
  40     private static final QName FILTER = QName.create(NetconfConstants.RFC6241_MODULE, "filter");
  41
  42     private static CrossSourceStatementReactor reactor;
  43
  44     @BeforeClass
  45     public static void createReactor() {
  46         reactor = RFC7950Reactors.defaultReactorBuilder()
  47                 .addStatementSupport(ModelProcessingPhase.FULL_DECLARATION,
  48                     GetFilterElementAttributesStatementSupport.getInstance())
  49                 .build();
  50     }
  51
  52     @AfterClass
  53     public static void freeReactor() {
  54         reactor = null;
  55     }
  56
  57     @Test
  58     public void testResolution() throws ReactorException, IOException, YangSyntaxErrorException {
  59         final SchemaContext context = reactor.newBuild()
  60                 .addLibSources(YangStatementStreamSource.create(
  61                     YangTextSchemaSource.forResource("/ietf-inet-types@2013-07-15.yang")))
  62                 .addSource(YangStatementStreamSource.create(
  63                     YangTextSchemaSource.forResource("/ietf-netconf@2011-06-01.yang")))
  64                 .buildEffective();
  65
  66         final Module module = context.findModule(NetconfConstants.RFC6241_MODULE).get();
  67         final Collection<? extends RpcDefinition> rpcs = module.getRpcs();
  68         assertEquals(13, rpcs.size());
  69         final Iterator<? extends RpcDefinition> it = module.getRpcs().iterator();
  70         // get-config
  71         assertExtension(true, it.next());
  72         assertExtension(false, it.next());
  73         assertExtension(false, it.next());
  74         assertExtension(false, it.next());
  75         assertExtension(false, it.next());
  76         assertExtension(false, it.next());
  77         // get
  78         assertExtension(true, it.next());
  79         it.forEachRemaining(def -> assertExtension(false, def));
  80     }
  81
  82     private static void assertExtension(final boolean expected, final RpcDefinition def) {
  83         final Optional<DataSchemaNode> optFilter = def.getInput().findDataTreeChild(FILTER);
  84         assertEquals(expected, optFilter.isPresent());
  85         optFilter.ifPresent(filter -> {
  86             assertThat(filter, is(instanceOf(AnyXmlSchemaNode.class)));
  87             assertTrue(GetFilterElementAttributesSchemaNode.findIn((AnyXmlSchemaNode) filter).isPresent());
  88         });
  89     }
  90 }