Projects moved under correct parent.

[controller.git] / opendaylight / sal / yang-prototype / yang / yang-model-parser-impl / src / test / java / org / opendaylight / controller / yang / parser / impl / TestUtils.java

diff --git a/opendaylight/sal/yang-prototype/yang/yang-model-parser-impl/src/test/java/org/opendaylight/controller/yang/parser/impl/TestUtils.java b/opendaylight/sal/yang-prototype/yang/yang-model-parser-impl/src/test/java/org/opendaylight/controller/yang/parser/impl/TestUtils.java
new file mode 100644 (file)
index 0000000..fed8cde
--- /dev/null
+++ b/opendaylight/sal/yang-prototype/yang/yang-model-parser-impl/src/test/java/org/opendaylight/controller/yang/parser/impl/TestUtils.java
@@ -0,0 +1,147 @@
+/*
+ * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+package org.opendaylight.controller.yang.parser.impl;
+
+import java.io.File;
+import java.io.FileNotFoundException;
+import java.io.IOException;
+import java.io.InputStream;
+import java.net.URI;
+import java.text.DateFormat;
+import java.text.ParseException;
+import java.text.SimpleDateFormat;
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.Date;
+import java.util.HashSet;
+import java.util.List;
+import java.util.Set;
+
+import org.opendaylight.controller.yang.common.QName;
+import org.opendaylight.controller.yang.model.api.Module;
+import org.opendaylight.controller.yang.model.api.ModuleImport;
+import org.opendaylight.controller.yang.model.api.SchemaContext;
+import org.opendaylight.controller.yang.model.api.SchemaPath;
+import org.opendaylight.controller.yang.model.api.TypeDefinition;
+import org.opendaylight.controller.yang.model.parser.api.YangModelParser;
+
+final class TestUtils {
+
+    private TestUtils() {
+    }
+
+    public static Set<Module> loadModules(String resourceDirectory) throws FileNotFoundException {
+        YangModelParser parser = new YangParserImpl();
+        final File testDir = new File(resourceDirectory);
+        final String[] fileList = testDir.list();
+        final List<File> testFiles = new ArrayList<File>();
+        if(fileList == null) {
+            throw new FileNotFoundException(resourceDirectory);
+        }
+        for (int i = 0; i < fileList.length; i++) {
+            String fileName = fileList[i];
+            testFiles.add(new File(testDir, fileName));
+        }
+        return parser.parseYangModels(testFiles);
+    }
+
+    public static Set<Module> loadModules(List<InputStream> input) throws IOException {
+        final YangModelParser parser = new YangParserImpl();
+        final Set<Module> modules = new HashSet<Module>(
+                parser.parseYangModelsFromStreams(input));
+        for(InputStream stream : input) {
+            stream.close();
+        }
+        return modules;
+    }
+
+    public static Module loadModule(final InputStream stream) throws
+            IOException {
+        final YangModelParser parser = new YangParserImpl();
+        final List<InputStream> input = Collections.singletonList(stream);
+        final Set<Module> modules = new HashSet<Module>(
+                parser.parseYangModelsFromStreams(input));
+        stream.close();
+        return modules.iterator().next();
+    }
+
+    public static Module loadModuleWithContext(final InputStream stream, final SchemaContext context) throws IOException {
+        final YangModelParser parser = new YangParserImpl();
+        final List<InputStream> input = Collections.singletonList(stream);
+        final Set<Module> modules = new HashSet<Module>(parser.parseYangModelsFromStreams(input, context));
+        stream.close();
+        return modules.iterator().next();
+    }
+
+    public static Set<Module> loadModulesWithContext(final List<InputStream> input, final SchemaContext context) throws IOException {
+        final YangModelParser parser = new YangParserImpl();
+        final Set<Module> modules = new HashSet<Module>(parser.parseYangModelsFromStreams(input, context));
+        for(InputStream is : input) {
+            if(is != null) {
+                is.close();
+            }
+        }
+        return modules;
+    }
+
+    public static Module findModule(Set<Module> modules, String moduleName) {
+        Module result = null;
+        for (Module module : modules) {
+            if (module.getName().equals(moduleName)) {
+                result = module;
+                break;
+            }
+        }
+        return result;
+    }
+
+    public static ModuleImport findImport(Set<ModuleImport> imports,
+            String prefix) {
+        ModuleImport result = null;
+        for (ModuleImport moduleImport : imports) {
+            if (moduleImport.getPrefix().equals(prefix)) {
+                result = moduleImport;
+                break;
+            }
+        }
+        return result;
+    }
+
+    public static TypeDefinition<?> findTypedef(
+            Set<TypeDefinition<?>> typedefs, String name) {
+        TypeDefinition<?> result = null;
+        for (TypeDefinition<?> td : typedefs) {
+            if (td.getQName().getLocalName().equals(name)) {
+                result = td;
+                break;
+            }
+        }
+        return result;
+    }
+
+    public static SchemaPath createPath(boolean absolute, URI namespace,
+            Date revision, String prefix, String... names) {
+        List<QName> path = new ArrayList<QName>();
+        for (String name : names) {
+            path.add(new QName(namespace, revision, prefix, name));
+        }
+        return new SchemaPath(path, absolute);
+    }
+
+    public static Date createDate(String date) {
+        Date result;
+        final DateFormat simpleDateFormat = new SimpleDateFormat("yyyy-MM-dd");
+        try {
+            result = simpleDateFormat.parse(date);
+        } catch (ParseException e) {
+            result = null;
+        }
+        return result;
+    }
+
+}