2 * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.
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
8 package org.opendaylight.yangtools.yang.stmt;
11 import java.io.IOException;
13 import java.nio.file.Path;
14 import java.util.ArrayList;
15 import java.util.Collection;
16 import java.util.List;
18 import org.eclipse.jdt.annotation.NonNull;
19 import org.eclipse.jdt.annotation.Nullable;
20 import org.opendaylight.yangtools.yang.common.QName;
21 import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
22 import org.opendaylight.yangtools.yang.model.api.Module;
23 import org.opendaylight.yangtools.yang.model.api.ModuleImport;
24 import org.opendaylight.yangtools.yang.model.api.TypeDefinition;
25 import org.opendaylight.yangtools.yang.model.repo.api.YangTextSchemaSource;
26 import org.opendaylight.yangtools.yang.model.repo.api.YinTextSchemaSource;
27 import org.opendaylight.yangtools.yang.parser.rfc7950.reactor.RFC7950Reactors;
28 import org.opendaylight.yangtools.yang.parser.rfc7950.repo.YangStatementStreamSource;
29 import org.opendaylight.yangtools.yang.parser.rfc7950.repo.YinStatementStreamSource;
30 import org.opendaylight.yangtools.yang.parser.rfc7950.repo.YinTextToDomTransformer;
31 import org.opendaylight.yangtools.yang.parser.spi.meta.ReactorException;
32 import org.opendaylight.yangtools.yang.parser.spi.source.StatementStreamSource;
33 import org.opendaylight.yangtools.yang.parser.stmt.reactor.CrossSourceStatementReactor.BuildAction;
34 import org.xml.sax.SAXException;
36 public final class TestUtils {
41 public static @NonNull List<StatementStreamSource> loadSources(final String resourceDirectory)
43 return loadSources(TestUtils.class, resourceDirectory);
46 public static @NonNull List<StatementStreamSource> loadSources(final Class<?> cls, final String resourceDirectory)
48 final var files = new File(cls.getResource(resourceDirectory).toURI())
49 .listFiles(StmtTestUtils.YANG_FILE_FILTER);
50 final var sources = new ArrayList<StatementStreamSource>(files.length);
51 for (var file : files) {
52 sources.add(YangStatementStreamSource.create(YangTextSchemaSource.forPath(file.toPath())));
57 public static EffectiveModelContext loadModules(final String resourceDirectory) throws Exception {
58 return loadModules(TestUtils.class, resourceDirectory);
61 public static EffectiveModelContext loadModules(final String resourceDirectory,
62 final @Nullable Set<QName> supportedFeatures) throws Exception {
63 return loadModules(TestUtils.class, resourceDirectory, supportedFeatures);
66 public static EffectiveModelContext loadModules(final Class<?> cls, final String resourceDirectory)
68 return loadModules(cls, resourceDirectory, null);
71 public static EffectiveModelContext loadModules(final Class<?> cls, final String resourceDirectory,
72 final @Nullable Set<QName> supportedFeatures) throws Exception {
73 final var action = RFC7950Reactors.defaultReactor().newBuild()
74 .addSources(loadSources(cls, resourceDirectory));
75 if (supportedFeatures != null) {
76 action.setSupportedFeatures(supportedFeatures);
78 return action.buildEffective();
81 public static EffectiveModelContext parseYangSource(final String... yangSourceFilePath) throws Exception {
82 final var reactor = RFC7950Reactors.defaultReactor().newBuild();
83 for (var resourcePath : yangSourceFilePath) {
84 reactor.addSource(YangStatementStreamSource.create(YangTextSchemaSource.forPath(Path.of(
85 TestUtils.class.getResource(resourcePath).toURI()))));
87 return reactor.buildEffective();
90 // FIXME: these remain unaudited
92 public static EffectiveModelContext loadYinModules(final URI resourceDirectory)
93 throws ReactorException, SAXException, IOException {
94 final BuildAction reactor = RFC7950Reactors.defaultReactor().newBuild();
96 for (File file : new File(resourceDirectory).listFiles()) {
97 reactor.addSource(YinStatementStreamSource.create(YinTextToDomTransformer.transformSource(
98 YinTextSchemaSource.forPath(file.toPath()))));
101 return reactor.buildEffective();
104 public static Module loadYinModule(final YinTextSchemaSource source) throws ReactorException, SAXException,
106 return RFC7950Reactors.defaultReactor().newBuild()
107 .addSource(YinStatementStreamSource.create(YinTextToDomTransformer.transformSource(source)))
109 .getModules().iterator().next();
112 public static ModuleImport findImport(final Collection<? extends ModuleImport> imports, final String prefix) {
113 for (ModuleImport moduleImport : imports) {
114 if (moduleImport.getPrefix().equals(prefix)) {
121 public static TypeDefinition<?> findTypedef(final Collection<? extends TypeDefinition<?>> typedefs,
123 for (TypeDefinition<?> td : typedefs) {
124 if (td.getQName().getLocalName().equals(name)) {