Re-implement model type value for Resource Mapping

[aai/babel.git] / src / main / java / org / onap / aai / babel / xml / generator / data / WidgetMapping.java

diff --git a/src/main/java/org/onap/aai/babel/xml/generator/data/WidgetMapping.java b/src/main/java/org/onap/aai/babel/xml/generator/data/WidgetMapping.java
index 2157728..2b298da 100644 (file)
--- a/src/main/java/org/onap/aai/babel/xml/generator/data/WidgetMapping.java
+++ b/src/main/java/org/onap/aai/babel/xml/generator/data/WidgetMapping.java
@@ -21,13 +21,24 @@
 
 package org.onap.aai.babel.xml.generator.data;
 
-import java.util.Map;
-
 public class WidgetMapping {
 
     String prefix;
     String type = "resource"; // Default type is Resource (not Widget)
     String widget;
-    boolean deleteFlag;
-    Map<String, Object> properties;
+    boolean deleteFlag = true;
+
+    public void setType(String type) {
+        this.type = type;
+    }
+
+    public void setWidget(String widgetName) {
+        this.widget = widgetName;
+    }
+
+    @Override
+    public String toString() {
+        return "WidgetMapping [prefix=" + prefix + ", type=" + type + ", widget=" + widget + ", deleteFlag="
+                + deleteFlag + "]";
+    }
 }