1 files changed, 22 insertions, 29 deletions
diff --git a/openecomp-be/lib/openecomp-heat-lib/src/main/java/org/openecomp/sdc/heat/datatypes/model/PolicyTypes.java b/openecomp-be/lib/openecomp-heat-lib/src/main/java/org/openecomp/sdc/heat/datatypes/model/PolicyTypes.java
index 3195e61fe5..77a7cd4552 100644
--- a/openecomp-be/lib/openecomp-heat-lib/src/main/java/org/openecomp/sdc/heat/datatypes/model/PolicyTypes.java
+++ b/openecomp-be/lib/openecomp-heat-lib/src/main/java/org/openecomp/sdc/heat/datatypes/model/PolicyTypes.java
@@ -7,9 +7,9 @@
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at
- * 
+ *
  *      http://www.apache.org/licenses/LICENSE-2.0
- * 
+ *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
@@ -24,38 +24,31 @@ import java.util.HashMap;
 import java.util.Map;
 import java.util.Objects;
 
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+
+@Getter
+@AllArgsConstructor
 public enum PolicyTypes {
-  AFFINITY("affinity"),
-  ANTI_AFFINITY("anti-affinity");
+    AFFINITY("affinity"),
+    ANTI_AFFINITY("anti-affinity");
 
-  private static Map<String, PolicyTypes> stringToPolicyTypesMap;
+    private static Map<String, PolicyTypes> stringToPolicyTypesMap;
 
-  static {
-    stringToPolicyTypesMap = new HashMap<>();
-    for (PolicyTypes type : PolicyTypes.values()) {
-      stringToPolicyTypesMap.put(type.policy, type);
+    static {
+        stringToPolicyTypesMap = new HashMap<>();
+        for (PolicyTypes type : PolicyTypes.values()) {
+            stringToPolicyTypesMap.put(type.policy, type);
+        }
     }
-  }
-
-  private String policy;
-
-  PolicyTypes(String policy) {
-    this.policy = policy;
-  }
 
-  public static PolicyTypes findByPolicy(String policy) {
-    return stringToPolicyTypesMap.get(policy);
-  }
+    private String policy;
 
-  public static boolean isGivenPolicyValid(String policyToCheck) {
-    return Objects.nonNull(findByPolicy(policyToCheck));
-  }
-
-  public String getPolicy() {
-    return policy;
-  }
+    public static PolicyTypes findByPolicy(String policy) {
+        return stringToPolicyTypesMap.get(policy);
+    }
 
-  public void setPolicy(String policy) {
-    this.policy = policy;
-  }
+    public static boolean isGivenPolicyValid(String policyToCheck) {
+        return Objects.nonNull(findByPolicy(policyToCheck));
+    }
 }