Fixed the streams and the imports issues

[dcaegen2/platform/cli.git] / blueprint-generator / src / main / java / org / onap / blueprintgenerator / models / dmaapbp / DmaapBlueprint.java

diff --git a/blueprint-generator/src/main/java/org/onap/blueprintgenerator/models/dmaapbp/DmaapBlueprint.java b/blueprint-generator/src/main/java/org/onap/blueprintgenerator/models/dmaapbp/DmaapBlueprint.java
index ff93307..d3cff67 100644 (file)
--- a/blueprint-generator/src/main/java/org/onap/blueprintgenerator/models/dmaapbp/DmaapBlueprint.java
+++ b/blueprint-generator/src/main/java/org/onap/blueprintgenerator/models/dmaapbp/DmaapBlueprint.java
@@ -49,13 +49,13 @@ public class DmaapBlueprint extends Blueprint{
                //set the imports
                Imports imps = new Imports();
                if(importPath.equals("")) {
-                       imps.createDmaapImports();
+                       bp.setImports(imps.createDmaapImports());
                }
                else {
-                       imps.createImportsFromFile(importPath);
+                       bp.setImports(imps.createImportsFromFile(importPath));
                }
 
-               bp.setImports(imps.getImports());
+               //bp.setImports(imps.getImports());
 
                //set and create the node templates
                TreeMap<String, Node> nodeTemplate = new TreeMap();