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.binding.runtime.api.BindingRuntimeContext;
  18 import org.opendaylight.binding.runtime.api.BindingRuntimeGenerator;
  19 import org.opendaylight.binding.runtime.spi.BindingRuntimeHelpers;
  20 import org.opendaylight.mdsal.binding.dom.codec.api.BindingCodecTree;
  21 import org.opendaylight.mdsal.binding.dom.codec.api.BindingDataObjectCodecTreeNode;
  22 import org.opendaylight.mdsal.binding.dom.codec.impl.DefaultBindingCodecTreeFactory;
  23 import org.opendaylight.mdsal.binding.generator.impl.DefaultBindingRuntimeGenerator;
  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.model.parser.api.YangParserFactory;
  36 import org.opendaylight.yangtools.yang.parser.impl.YangParserFactoryImpl;
  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 YangParserFactoryImpl();
  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,
  53                 codecFactory);
  54         codecTree = codecFactory.create(runtimeContext);
  55     }
  56
  57     @Test
  58     public void testFormatSchema() {
  59         final BindingDataObjectCodecTreeNode<YangLibrary> codec =
  60                 codecTree.getSubtreeCodec(InstanceIdentifier.create(YangLibrary.class));
  61
  62         final ContainerNode nonLegacyContent = yangLib.newContentBuilder()
  63                 .defaultContext(runtimeContext.getEffectiveModelContext()).formatYangLibraryContent();
  64         final YangLibrary yangLibrary = codec.deserialize(nonLegacyContent);
  65
  66         assertEquals(1, yangLibrary.nonnullModuleSet().size());
  67         final ModuleSet moduleSet = yangLibrary.nonnullModuleSet().get(new ModuleSetKey("ODL_modules"));
  68         assertEquals(4, moduleSet.nonnullModule().size());
  69         assertEquals(moduleSet.getModule(), createControlModules());
  70     }
  71
  72     private Map<ModuleKey, Module> createControlModules() {
  73         final Map<ModuleKey, Module> modules = new HashMap<>();
  74         modules.put(new ModuleKey(new YangIdentifier("ietf-yang-library")),
  75                 createModule("ietf-yang-library", "urn:ietf:params:xml:ns:yang:ietf-yang-library", "2019-01-04"));
  76         modules.put(new ModuleKey(new YangIdentifier("ietf-inet-types")),
  77                 createModule("ietf-inet-types", "urn:ietf:params:xml:ns:yang:ietf-inet-types", "2013-07-15"));
  78         modules.put(new ModuleKey(new YangIdentifier("ietf-datastores")),
  79                 createModule("ietf-datastores", "urn:ietf:params:xml:ns:yang:ietf-datastores", "2018-02-14"));
  80         modules.put(new ModuleKey(new YangIdentifier("ietf-yang-types")),
  81                 createModule("ietf-yang-types", "urn:ietf:params:xml:ns:yang:ietf-yang-types", "2013-07-15"));
  82         return modules;
  83     }
  84
  85     private Module createModule(final String name, final String namespace, final String revision) {
  86         return new ModuleBuilder().setName(new YangIdentifier(name))
  87                 .setNamespace(new Uri(namespace))
  88                 .setRevision(new RevisionIdentifier(revision))
  89                 .setFeature(Collections.emptyList())
  90                 .setSubmodule(Collections.emptyMap())
  91                 .build();
  92     }
  93 }