yanglib/mdsal-yanglib-rfc8525/src/test/java/org/opendaylight/mdsal/yanglib/rfc8525/YangLibrarySupportTest.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.rfc8525;
   9
  10 import static org.junit.Assert.assertEquals;
  11
  12 import java.util.Collections;
  13 import java.util.HashMap;
  14 import java.util.Map;
  15 import org.junit.Before;
  16 import org.junit.Test;
  17 import org.opendaylight.mdsal.binding.dom.codec.api.BindingCodecTree;
  18 import org.opendaylight.mdsal.binding.dom.codec.api.BindingDataObjectCodecTreeNode;
  19 import org.opendaylight.mdsal.binding.dom.codec.impl.DefaultBindingCodecTreeFactory;
  20 import org.opendaylight.mdsal.binding.generator.impl.DefaultBindingRuntimeGenerator;
  21 import org.opendaylight.mdsal.binding.runtime.api.BindingRuntimeContext;
  22 import org.opendaylight.mdsal.binding.runtime.api.BindingRuntimeGenerator;
  23 import org.opendaylight.mdsal.binding.runtime.spi.BindingRuntimeHelpers;
  24 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Uri;
  25 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.library.rev190104.RevisionIdentifier;
  26 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.library.rev190104.YangLibrary;
  27 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.library.rev190104.module.set.parameters.Module;
  28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.library.rev190104.module.set.parameters.ModuleBuilder;
  29 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.library.rev190104.module.set.parameters.ModuleKey;
  30 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.library.rev190104.yang.library.parameters.ModuleSet;
  31 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.library.rev190104.yang.library.parameters.ModuleSetKey;
  32 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.YangIdentifier;
  33 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  34 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
  35 import org.opendaylight.yangtools.yang.parser.api.YangParserFactory;
  36 import org.opendaylight.yangtools.yang.parser.impl.DefaultYangParserFactory;
  37
  38 public class YangLibrarySupportTest {
  39
  40     private static final BindingRuntimeGenerator BINDING_RUNTIME_GENERATOR = new DefaultBindingRuntimeGenerator();
  41
  42     private static final YangParserFactory YANG_PARSER_FACTORY = new DefaultYangParserFactory();
  43
  44     private YangLibrarySupport yangLib;
  45     private BindingRuntimeContext runtimeContext;
  46     private BindingCodecTree codecTree;
  47
  48     @Before
  49     public void setUp() throws Exception {
  50         runtimeContext = BindingRuntimeHelpers.createRuntimeContext();
  51         final DefaultBindingCodecTreeFactory codecFactory = new DefaultBindingCodecTreeFactory();
  52         yangLib = new YangLibrarySupport(YANG_PARSER_FACTORY, BINDING_RUNTIME_GENERATOR, codecFactory);
  53         codecTree = codecFactory.create(runtimeContext);
  54     }
  55
  56     @Test
  57     public void testFormatSchema() {
  58         final BindingDataObjectCodecTreeNode<YangLibrary> codec =
  59                 codecTree.getSubtreeCodec(InstanceIdentifier.create(YangLibrary.class));
  60
  61         final ContainerNode nonLegacyContent = yangLib.newContentBuilder()
  62                 .defaultContext(runtimeContext.getEffectiveModelContext()).formatYangLibraryContent();
  63         final YangLibrary yangLibrary = codec.deserialize(nonLegacyContent);
  64
  65         assertEquals(1, yangLibrary.nonnullModuleSet().size());
  66         final ModuleSet moduleSet = yangLibrary.nonnullModuleSet().get(new ModuleSetKey("ODL_modules"));
  67         assertEquals(4, moduleSet.nonnullModule().size());
  68         assertEquals(moduleSet.getModule(), createControlModules());
  69     }
  70
  71     private Map<ModuleKey, Module> createControlModules() {
  72         final Map<ModuleKey, Module> modules = new HashMap<>();
  73         modules.put(new ModuleKey(new YangIdentifier("ietf-yang-library")),
  74                 createModule("ietf-yang-library", "urn:ietf:params:xml:ns:yang:ietf-yang-library", "2019-01-04"));
  75         modules.put(new ModuleKey(new YangIdentifier("ietf-inet-types")),
  76                 createModule("ietf-inet-types", "urn:ietf:params:xml:ns:yang:ietf-inet-types", "2013-07-15"));
  77         modules.put(new ModuleKey(new YangIdentifier("ietf-datastores")),
  78                 createModule("ietf-datastores", "urn:ietf:params:xml:ns:yang:ietf-datastores", "2018-02-14"));
  79         modules.put(new ModuleKey(new YangIdentifier("ietf-yang-types")),
  80                 createModule("ietf-yang-types", "urn:ietf:params:xml:ns:yang:ietf-yang-types", "2013-07-15"));
  81         return modules;
  82     }
  83
  84     private Module createModule(final String name, final String namespace, final String revision) {
  85         return new ModuleBuilder().setName(new YangIdentifier(name))
  86                 .setNamespace(new Uri(namespace))
  87                 .setRevision(new RevisionIdentifier(revision))
  88                 .setFeature(Collections.emptyList())
  89                 .setSubmodule(Collections.emptyMap())
  90                 .build();
  91     }
  92 }