opendaylight/sal/yang-prototype/code-generator/yang-model-parser-impl/src/test/java/org/opendaylight/controller/yang/model/parser/impl/TestUtils.java

   1 /*
   2  * Copyright (c) 2013 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 package org.opendaylight.controller.yang.model.parser.impl;
   9
  10 import java.io.File;
  11 import java.net.URI;
  12 import java.text.DateFormat;
  13 import java.text.ParseException;
  14 import java.text.SimpleDateFormat;
  15 import java.util.ArrayList;
  16 import java.util.Date;
  17 import java.util.List;
  18 import java.util.Set;
  19
  20 import org.opendaylight.controller.yang.common.QName;
  21 import org.opendaylight.controller.yang.model.api.Module;
  22 import org.opendaylight.controller.yang.model.api.ModuleImport;
  23 import org.opendaylight.controller.yang.model.api.SchemaPath;
  24 import org.opendaylight.controller.yang.model.api.TypeDefinition;
  25 import org.opendaylight.controller.yang.model.parser.api.YangModelParser;
  26
  27 final class TestUtils {
  28
  29     private TestUtils() {
  30     }
  31
  32     public static Set<Module> loadModules(String resourceDirectory) {
  33         YangModelParser parser = new YangModelParserImpl();
  34         File testDir = new File(resourceDirectory);
  35         String[] fileList = testDir.list();
  36         String[] testFiles = new String[fileList.length];
  37         for (int i = 0; i < fileList.length; i++) {
  38             String fileName = fileList[i];
  39             File file = new File(testDir, fileName);
  40             testFiles[i] = file.getAbsolutePath();
  41         }
  42         return parser.parseYangModels(testFiles);
  43     }
  44
  45     public static Module findModule(Set<Module> modules, String moduleName) {
  46         Module result = null;
  47         for (Module module : modules) {
  48             if (module.getName().equals(moduleName)) {
  49                 result = module;
  50                 break;
  51             }
  52         }
  53         return result;
  54     }
  55
  56     public static ModuleImport findImport(Set<ModuleImport> imports,
  57             String prefix) {
  58         ModuleImport result = null;
  59         for (ModuleImport moduleImport : imports) {
  60             if (moduleImport.getPrefix().equals(prefix)) {
  61                 result = moduleImport;
  62                 break;
  63             }
  64         }
  65         return result;
  66     }
  67
  68     public static TypeDefinition<?> findTypedef(
  69             Set<TypeDefinition<?>> typedefs, String name) {
  70         TypeDefinition<?> result = null;
  71         for (TypeDefinition<?> td : typedefs) {
  72             if (td.getQName().getLocalName().equals(name)) {
  73                 result = td;
  74                 break;
  75             }
  76         }
  77         return result;
  78     }
  79
  80     public static SchemaPath createPath(boolean absolute, URI namespace,
  81             Date revision, String prefix, String... names) {
  82         List<QName> path = new ArrayList<QName>();
  83         for (String name : names) {
  84             path.add(new QName(namespace, revision, prefix, name));
  85         }
  86         return new SchemaPath(path, absolute);
  87     }
  88
  89     public static Date createDate(String date) {
  90         Date result;
  91         final DateFormat simpleDateFormat = new SimpleDateFormat("yyyy-MM-dd");
  92         try {
  93             result = simpleDateFormat.parse(date);
  94         } catch (ParseException e) {
  95             result = null;
  96         }
  97         return result;
  98     }
  99
 100 }