Yang-maven-plugin refactored + fixed bugs.

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

diff --git a/opendaylight/sal/yang-prototype/code-generator/yang-model-parser-impl/src/test/java/org/opendaylight/controller/yang/parser/impl/TestUtils.java b/opendaylight/sal/yang-prototype/code-generator/yang-model-parser-impl/src/test/java/org/opendaylight/controller/yang/parser/impl/TestUtils.java
index 3d9df349e4ec5a390518b67fdaadb5640b51dd86..d14c54219e0fd5e1c1d9f3cdae3b0512f8dd654c 100644 (file)
--- a/opendaylight/sal/yang-prototype/code-generator/yang-model-parser-impl/src/test/java/org/opendaylight/controller/yang/parser/impl/TestUtils.java
+++ b/opendaylight/sal/yang-prototype/code-generator/yang-model-parser-impl/src/test/java/org/opendaylight/controller/yang/parser/impl/TestUtils.java
@@ -18,6 +18,7 @@ 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;
 
@@ -54,7 +55,8 @@ final class TestUtils {
         for(String path : pathToYangFile) {
             input.add(TestUtils.class.getResourceAsStream(path));
         }
-        Set<Module> modules = parser.parseYangModelsFromStreams(input);
+        Set<Module> modules = new HashSet<Module>(
+                parser.parseYangModelsFromStreams(input));
         for(InputStream stream : input) {
             stream.close();
         }
@@ -65,7 +67,8 @@ final class TestUtils {
         YangModelParser parser = new YangParserImpl();
         InputStream stream = TestUtils.class.getResourceAsStream(pathToYangFile);
         List<InputStream> input = Collections.singletonList(stream);
-        Set<Module> modules = parser.parseYangModelsFromStreams(input);
+        Set<Module> modules = new HashSet<Module>(
+                parser.parseYangModelsFromStreams(input));
         stream.close();
         return modules.iterator().next();
     }