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