binding2/mdsal-binding2-generator-impl/src/test/java/org/opendaylight/mdsal/binding/javav2/generator/impl/YangTemplateTest.java

   1 /*
   2  * Copyright (c) 2017 Cisco Systems, Inc. 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
   9 package org.opendaylight.mdsal.binding.javav2.generator.impl;
  10
  11 import static org.junit.Assert.assertTrue;
  12 import com.google.common.annotations.Beta;
  13 import java.io.FileNotFoundException;
  14 import java.net.URISyntaxException;
  15 import java.util.Iterator;
  16 import java.util.Set;
  17 import org.junit.Assert;
  18 import org.junit.Before;
  19 import org.junit.Test;
  20 import org.opendaylight.mdsal.binding.javav2.generator.impl.txt.yangTemplateForModule;
  21 import org.opendaylight.mdsal.binding.javav2.generator.util.YangSnippetCleaner;
  22 import org.opendaylight.yangtools.yang.model.api.Module;
  23 import org.opendaylight.yangtools.yang.parser.spi.meta.ReactorException;
  24 import org.opendaylight.yangtools.yang.test.util.YangParserTestUtils;
  25
  26 @Beta
  27 public class YangTemplateTest {
  28
  29     private Set<Module> modules;
  30
  31     @Before
  32     public void setup() throws URISyntaxException, ReactorException, FileNotFoundException {
  33         this.modules = YangParserTestUtils.parseYangSources("/yang-template").getModules();
  34     }
  35
  36     @Test
  37     public void printYangSnippetForModule() throws Exception{
  38         Iterator<Module> iterator = this.modules.iterator();
  39         String moduleBody = yangTemplateForModule.render(iterator.next()).body().trim();
  40         String cleanedModuleBody = YangSnippetCleaner.clean(moduleBody);
  41         assertTrue(cleanedModuleBody.contains("yang-template-import"));
  42         assertTrue(cleanedModuleBody.contains("    deviation /yti:yti-leaf {"));
  43         assertTrue(cleanedModuleBody.contains("            default \"def\";"));
  44         assertTrue(cleanedModuleBody.contains("    extension ext;"));
  45         assertTrue(cleanedModuleBody.contains("    yti:ext;"));
  46         assertTrue(cleanedModuleBody.contains("    yti:ext-arg \"arg\""));
  47
  48         moduleBody = yangTemplateForModule.render(iterator.next()).body().trim();
  49         cleanedModuleBody = YangSnippetCleaner.clean(moduleBody);
  50         assertTrue(cleanedModuleBody.contains("module yang-template-test {"));
  51         assertTrue(cleanedModuleBody
  52                 .contains("    import yang-template-import { prefix \"yti\"; revision-date 2016-06-23; }"));
  53         assertTrue(cleanedModuleBody.contains("    anydata simple-anydata;"));
  54         assertTrue(cleanedModuleBody.contains("    container simple-container-with-action {"));
  55         assertTrue(cleanedModuleBody.contains("    leaf-list simple-leaf-list {"));
  56         assertTrue(cleanedModuleBody.contains("    leaf-list simple-leaf-list-userordered {"));
  57         assertTrue(cleanedModuleBody.contains("    list simple-list {"));
  58         assertTrue(cleanedModuleBody.contains("        key \"simple-list-leaf-1\";"));
  59         assertTrue(cleanedModuleBody.contains("        unique \"simple-list-leaf-2\";"));
  60         assertTrue(cleanedModuleBody.contains("        action act {"));
  61         assertTrue(cleanedModuleBody.contains("    list simple-list-more-arg-in-unique {"));
  62         assertTrue(cleanedModuleBody.contains("        ordered-by user;"));
  63         assertTrue(cleanedModuleBody.contains("    choice simple-choice {"));
  64         assertTrue(cleanedModuleBody.contains("    anyxml simple-anyxml;"));
  65         assertTrue(cleanedModuleBody.contains("        uses simple-grouping {"));
  66         assertTrue(cleanedModuleBody.contains("            refine simple-grouping-leaf {"));
  67         assertTrue(cleanedModuleBody.contains("    grouping simple-grouping {"));
  68         assertTrue(cleanedModuleBody.contains("    augment \"/simple-container-uses\" {"));
  69         assertTrue(cleanedModuleBody.contains("    extension simple-extension {"));
  70         assertTrue(cleanedModuleBody.contains("    feature simple-feature {"));
  71         assertTrue(cleanedModuleBody.contains("    identity simple-identity {"));
  72         assertTrue(cleanedModuleBody.contains("    notification simple-notification {"));
  73         assertTrue(cleanedModuleBody.contains("    rpc simple-rpc {"));
  74     }
  75 }