Reduce technical debt and add JUnit

[policy/engine.git] / PolicyEngineAPI / src / main / java / org / onap / policy / api / PolicyParameters.java

diff --git a/PolicyEngineAPI/src/main/java/org/onap/policy/api/PolicyParameters.java b/PolicyEngineAPI/src/main/java/org/onap/policy/api/PolicyParameters.java
index 2062638..cb51e48 100644 (file)
--- a/PolicyEngineAPI/src/main/java/org/onap/policy/api/PolicyParameters.java
+++ b/PolicyEngineAPI/src/main/java/org/onap/policy/api/PolicyParameters.java
@@ -40,6 +40,7 @@ public class PolicyParameters {
        private String onapName;
        private String configName;
        private Map<AttributeType, Map<String,String>> attributes;
+       private Map<String, String> treatments;
        private String configBody;
        private PolicyType configBodyType;
        private String actionPerformer;
@@ -52,7 +53,7 @@ public class PolicyParameters {
        private String priority;
        private RuleProvider ruleProvider;
        private String controllerName;
-       private ArrayList<String> dependencyNames;
+       private List<String> dependencyNames;
        private Date ttlDate;
        private boolean guard = false;
        private String riskLevel = "5";
@@ -547,7 +548,7 @@ public class PolicyParameters {
      * 
      * @return ArrayList of String(s) format of dependency names.
      */
-    public ArrayList<String> getDependencyNames() {
+    public List<String> getDependencyNames() {
         return dependencyNames;
     }
 
@@ -556,7 +557,7 @@ public class PolicyParameters {
      * 
      * @param dependencyNames ArrayList of String(s). 
      */
-    public void setDependencyNames(ArrayList<String> dependencyNames) {
+    public void setDependencyNames(List<String> dependencyNames) {
         this.dependencyNames = dependencyNames;
     }
        
@@ -568,6 +569,24 @@ public class PolicyParameters {
                this.extendedOption = extendedOption;
        }
        
+       /**
+     * Gets Allowed Treatments Map for Rainy Day Decision Policy
+     * 
+     * @return Map of String format for treatments per errorcode
+     */
+       public Map<String, String> getTreatments() {
+               return treatments;
+       }
+
+       /**
+        * Sets Allowed Treatments Map for Rainy Day Decision Policy 
+        * 
+        * @param treatments Map that contains the treatment per errorcode 
+        */
+       public void setTreatments(Map<String, String> treatments) {
+               this.treatments = treatments;
+       }
+       
        @Override
        public String toString() {
                return "PolicyParameters [ policyName=" + policyName + ", policyDescription=" + policyDescription + ", onapName="+ onapName 
@@ -576,6 +595,6 @@ public class PolicyParameters {
                                + ",dynamicRuleAlgorithmField1=" + dynamicRuleAlgorithmField1 + ",dynamicRuleAlgorithmField2=" + dynamicRuleAlgorithmField2 
                                + ", actionPerformer=" + actionPerformer + ", actionAttribute=" + actionAttribute + ", priority=" + priority  
                                + ", ruleProvider= " + ruleProvider + ", riskLevel= " + riskLevel + ", riskType= " + riskType + ", extendedOption= " + extendedOption
-                               + "]";
+                               + ", treatments= " + treatments + "]";
        }
 }