yanglib/mdsal-yanglib-rfc7895/src/test/java/org/opendaylight/mdsal/yanglib/rfc7895/YangModuleLibrarySupportTest.java

   1 /*
   2  * Copyright (c) 2020 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.mdsal.yanglib.rfc7895;
   9
  10 import static org.junit.Assert.assertEquals;
  11
  12 import java.util.Collections;
  13 import java.util.Map;
  14 import org.junit.Before;
  15 import org.junit.Test;
  16 import org.opendaylight.mdsal.binding.dom.codec.api.BindingCodecTree;
  17 import org.opendaylight.mdsal.binding.dom.codec.api.BindingDataObjectCodecTreeNode;
  18 import org.opendaylight.mdsal.binding.dom.codec.impl.DefaultBindingCodecTreeFactory;
  19 import org.opendaylight.mdsal.binding.generator.impl.DefaultBindingRuntimeGenerator;
  20 import org.opendaylight.mdsal.binding.runtime.api.BindingRuntimeContext;
  21 import org.opendaylight.mdsal.binding.runtime.api.BindingRuntimeGenerator;
  22 import org.opendaylight.mdsal.binding.runtime.spi.BindingRuntimeHelpers;
  23 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Uri;
  24 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.library.rev160621.ModulesState;
  25 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.library.rev160621.RevisionIdentifier;
  26 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.library.rev160621.module.list.CommonLeafs.Revision;
  27 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.library.rev160621.module.list.Module;
  28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.library.rev160621.module.list.Module.ConformanceType;
  29 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.library.rev160621.module.list.ModuleBuilder;
  30 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.library.rev160621.module.list.ModuleKey;
  31 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.YangIdentifier;
  32 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  33 import org.opendaylight.yangtools.yang.parser.api.YangParserException;
  34 import org.opendaylight.yangtools.yang.parser.api.YangParserFactory;
  35 import org.opendaylight.yangtools.yang.parser.impl.DefaultYangParserFactory;
  36
  37 public class YangModuleLibrarySupportTest {
  38
  39     private static final BindingRuntimeGenerator BINDING_RUNTIME_GENERATOR = new DefaultBindingRuntimeGenerator();
  40
  41     private static final YangParserFactory YANG_PARSER_FACTORY = new DefaultYangParserFactory();
  42
  43     private BindingRuntimeContext runtimeContext;
  44     private BindingCodecTree codecTree;
  45     private YangModuleLibrarySupport yangLib;
  46
  47     @Before
  48     public void setUp() throws YangParserException {
  49         runtimeContext = BindingRuntimeHelpers.createRuntimeContext();
  50         final DefaultBindingCodecTreeFactory codecFactory = new DefaultBindingCodecTreeFactory();
  51         codecTree = codecFactory.create(runtimeContext);
  52
  53         yangLib = new YangModuleLibrarySupport(YANG_PARSER_FACTORY, BINDING_RUNTIME_GENERATOR, codecFactory);
  54     }
  55
  56     @Test
  57     public void testModulesState() {
  58         final BindingDataObjectCodecTreeNode<ModulesState> legacyCodec =
  59                 codecTree.getSubtreeCodec(InstanceIdentifier.create(ModulesState.class));
  60         final ModulesState modulesState = legacyCodec.deserialize(
  61                 yangLib.newContentBuilder().defaultContext(
  62                         runtimeContext.getEffectiveModelContext()).formatYangLibraryContent());
  63
  64         assertEquals(3, modulesState.nonnullModule().size());
  65         assertEquals(createControlModules(), modulesState.getModule());
  66     }
  67
  68     private static Map<ModuleKey, Module> createControlModules() {
  69         return Map.of(
  70             new ModuleKey(new YangIdentifier("ietf-yang-library"), new Revision(new RevisionIdentifier("2016-06-21"))),
  71             createModule("ietf-yang-library", "urn:ietf:params:xml:ns:yang:ietf-yang-library", "2016-06-21"),
  72             new ModuleKey(new YangIdentifier("ietf-inet-types"), new Revision(new RevisionIdentifier("2013-07-15"))),
  73             createModule("ietf-inet-types", "urn:ietf:params:xml:ns:yang:ietf-inet-types", "2013-07-15"),
  74             new ModuleKey(new YangIdentifier("ietf-yang-types"), new Revision(new RevisionIdentifier("2013-07-15"))),
  75             createModule("ietf-yang-types", "urn:ietf:params:xml:ns:yang:ietf-yang-types", "2013-07-15"));
  76     }
  77
  78     private static Module createModule(final String name, final String namespace, final String revision) {
  79         return new ModuleBuilder()
  80                 .setName(new YangIdentifier(name))
  81                 .setNamespace(new Uri(namespace))
  82                 .setRevision(new Revision(new RevisionIdentifier(revision)))
  83                 .setConformanceType(ConformanceType.Implement)
  84                 .setFeature(Collections.emptyList())
  85                 .build();
  86     }
  87 }