opendaylight/md-sal/sal-rest-docgen/src/test/java/org/opendaylight/controller/sal/rest/doc/impl/ModelGeneratorTest.java

   1 package org.opendaylight.controller.sal.rest.doc.impl;
   2
   3 import com.google.common.base.Preconditions;
   4 import org.json.JSONObject;
   5 import org.junit.Assert;
   6 import org.junit.Before;
   7 import org.junit.Test;
   8 import org.opendaylight.yangtools.yang.model.api.Module;
   9 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
  10 import org.opendaylight.yangtools.yang.parser.impl.YangParserImpl;
  11
  12 import java.io.File;
  13 import java.util.HashSet;
  14 import java.util.Map;
  15
  16
  17 public class ModelGeneratorTest {
  18
  19     private DocGenTestHelper helper;
  20     private SchemaContext schemaContext;
  21
  22     @Before
  23     public void setUp() throws Exception {
  24         helper = new DocGenTestHelper();
  25         helper.setUp();
  26         schemaContext = new YangParserImpl().resolveSchemaContext(new HashSet<Module>(helper.getModules().values()));
  27     }
  28
  29     @Test
  30     public void testConvertToJsonSchema() throws Exception {
  31
  32         Preconditions.checkArgument(helper.getModules() != null, "No modules found");
  33
  34         ModelGenerator generator = new ModelGenerator();
  35
  36         for (Map.Entry<File, Module> m : helper.getModules().entrySet()) {
  37             if (m.getKey().getAbsolutePath().endsWith("opflex.yang")) {
  38
  39                 JSONObject jsonObject = generator.convertToJsonSchema(m.getValue(), schemaContext);
  40                 Assert.assertNotNull(jsonObject);
  41             }
  42         }
  43
  44     }
  45 }