parser/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.hamcrest.MatcherAssert.assertThat;
  13 import static org.junit.Assert.assertEquals;
  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.repo.api.YangTextSchemaSource;
  32 import org.opendaylight.yangtools.yang.parser.api.YangParserConfiguration;
  33 import org.opendaylight.yangtools.yang.parser.api.YangSyntaxErrorException;
  34 import org.opendaylight.yangtools.yang.parser.rfc7950.reactor.RFC7950Reactors;
  35 import org.opendaylight.yangtools.yang.parser.rfc7950.repo.YangStatementStreamSource;
  36 import org.opendaylight.yangtools.yang.parser.spi.meta.ModelProcessingPhase;
  37 import org.opendaylight.yangtools.yang.parser.spi.meta.ReactorException;
  38 import org.opendaylight.yangtools.yang.parser.stmt.reactor.CrossSourceStatementReactor;
  39
  40 public class NetconfTest {
  41     private static final QName FILTER = QName.create(NetconfConstants.RFC6241_MODULE, "filter");
  42
  43     private static CrossSourceStatementReactor reactor;
  44
  45     @BeforeClass
  46     public static void createReactor() {
  47         reactor = RFC7950Reactors.defaultReactorBuilder()
  48                 .addStatementSupport(ModelProcessingPhase.FULL_DECLARATION,
  49                     new GetFilterElementAttributesStatementSupport(YangParserConfiguration.DEFAULT))
  50                 .build();
  51     }
  52
  53     @AfterClass
  54     public static void freeReactor() {
  55         reactor = null;
  56     }
  57
  58     @Test
  59     public void testResolution() throws ReactorException, IOException, YangSyntaxErrorException {
  60         final SchemaContext context = reactor.newBuild()
  61                 .addLibSources(YangStatementStreamSource.create(
  62                     YangTextSchemaSource.forResource("/ietf-inet-types@2013-07-15.yang")))
  63                 .addSource(YangStatementStreamSource.create(
  64                     YangTextSchemaSource.forResource("/ietf-netconf@2011-06-01.yang")))
  65                 .buildEffective();
  66
  67         final Module module = context.findModule(NetconfConstants.RFC6241_MODULE).get();
  68         final Collection<? extends RpcDefinition> rpcs = module.getRpcs();
  69         assertEquals(13, rpcs.size());
  70         final Iterator<? extends RpcDefinition> it = module.getRpcs().iterator();
  71         // get-config
  72         assertExtension(true, it.next());
  73         assertExtension(false, it.next());
  74         assertExtension(false, it.next());
  75         assertExtension(false, it.next());
  76         assertExtension(false, it.next());
  77         assertExtension(false, it.next());
  78         // get
  79         assertExtension(true, it.next());
  80         it.forEachRemaining(def -> assertExtension(false, def));
  81     }
  82
  83     private static void assertExtension(final boolean expected, final RpcDefinition def) {
  84         final Optional<DataSchemaNode> optFilter = def.getInput().findDataTreeChild(FILTER);
  85         assertEquals(expected, optFilter.isPresent());
  86         optFilter.ifPresent(filter -> {
  87             assertThat(filter, is(instanceOf(AnyxmlSchemaNode.class)));
  88             assertTrue(GetFilterElementAttributesSchemaNode.findIn((AnyxmlSchemaNode) filter).isPresent());
  89         });
  90     }
  91 }