Merge branch 'milanperera-master'

components/policy-mgt/org.wso2.carbon.policy.mgt.core/src/main/java/org/wso2/carbon/policy/mgt/core/dao/FeatureDAO.java

@@ -53,4 +53,6 @@ public interface FeatureDAO {
 
     boolean deleteFeaturesOfProfile(Profile profile) throws FeatureManagerDAOException;
 
+    boolean deleteFeaturesOfProfile(int profileId) throws FeatureManagerDAOException;
+
 }

components/policy-mgt/org.wso2.carbon.policy.mgt.core/src/main/java/org/wso2/carbon/policy/mgt/core/dao/ProfileDAO.java

@@ -32,6 +32,8 @@ public interface ProfileDAO {
 
     boolean deleteProfile(Profile profile) throws ProfileManagerDAOException;
 
+    boolean deleteProfile(int policyId) throws ProfileManagerDAOException;
+
     Profile getProfiles(int profileId) throws ProfileManagerDAOException;
 
     List<Profile> getAllProfiles() throws ProfileManagerDAOException;

components/policy-mgt/org.wso2.carbon.policy.mgt.core/src/main/java/org/wso2/carbon/policy/mgt/core/dao/impl/FeatureDAOImpl.java

@@ -261,6 +261,28 @@ public class FeatureDAOImpl implements FeatureDAO {
         }
     }
 
+    @Override
+    public boolean deleteFeaturesOfProfile(int profileId) throws FeatureManagerDAOException {
+        Connection conn;
+        PreparedStatement stmt = null;
+
+        try {
+            conn = this.getConnection();
+            String query = "DELETE FROM DM_PROFILE_FEATURES WHERE PROFILE_ID = ?";
+            stmt = conn.prepareStatement(query);
+            stmt.setInt(1, profileId);
+            stmt.executeUpdate();
+            return true;
+
+        } catch (SQLException e) {
+            String msg = "Error occurred while deleting the feature related to a profile.";
+            log.error(msg);
+            throw new FeatureManagerDAOException(msg, e);
+        } finally {
+            PolicyManagementDAOUtil.cleanupResources(stmt, null);
+        }
+    }
+
 
     @Override
     public List<ProfileFeature> getAllProfileFeatures() throws FeatureManagerDAOException {

components/policy-mgt/org.wso2.carbon.policy.mgt.core/src/main/java/org/wso2/carbon/policy/mgt/core/dao/impl/ProfileDAOImpl.java

@@ -154,6 +154,29 @@ public class ProfileDAOImpl implements ProfileDAO {
         }
     }
 
+    @Override
+    public boolean deleteProfile(int profileId) throws ProfileManagerDAOException {
+        Connection conn;
+        PreparedStatement stmt = null;
+
+        try {
+            conn = this.getConnection();
+            String query = "DELETE FROM DM_PROFILE WHERE ID = ?";
+            stmt = conn.prepareStatement(query);
+            stmt.setInt(1, profileId);
+            stmt.executeUpdate();
+            return true;
+
+        } catch (SQLException e) {
+            String msg = "Error occurred while deleting the profile from the data base.";
+            log.error(msg);
+            throw new ProfileManagerDAOException(msg, e);
+        } finally {
+            PolicyManagementDAOUtil.cleanupResources(stmt, null);
+        }
+    }
+
+
     @Override
     public Profile getProfiles(int profileId) throws ProfileManagerDAOException {
 

components/policy-mgt/org.wso2.carbon.policy.mgt.core/src/main/java/org/wso2/carbon/policy/mgt/core/mgt/impl/PolicyManagerImpl.java

@@ -126,7 +126,7 @@ public class PolicyManagerImpl implements PolicyManager {
                 log.warn("Error occurred while roll backing the transaction.");
             }
             String msg = "Error occurred while adding the policy (" +
-                    policy.getId() + " - " + policy.getPolicyName() + ")";
+                         policy.getId() + " - " + policy.getPolicyName() + ")";
             log.error(msg, e);
             throw new PolicyManagementException(msg, e);
 
@@ -137,7 +137,7 @@ public class PolicyManagerImpl implements PolicyManager {
                 log.warn("Error occurred while roll backing the transaction.");
             }
             String msg = "Error occurred while adding the profile related to policy (" +
-                    policy.getId() + " - " + policy.getPolicyName() + ")";
+                         policy.getId() + " - " + policy.getPolicyName() + ")";
             log.error(msg, e);
             throw new PolicyManagementException(msg, e);
         } catch (FeatureManagerDAOException e) {
@@ -147,7 +147,7 @@ public class PolicyManagerImpl implements PolicyManager {
                 log.warn("Error occurred while roll backing the transaction.");
             }
             String msg = "Error occurred while adding the features of profile related to policy (" +
-                    policy.getId() + " - " + policy.getPolicyName() + ")";
+                         policy.getId() + " - " + policy.getPolicyName() + ")";
             log.error(msg, e);
             throw new PolicyManagementException(msg, e);
         }
@@ -210,7 +210,7 @@ public class PolicyManagerImpl implements PolicyManager {
                 log.warn("Error occurred while roll backing the transaction.");
             }
             String msg = "Error occurred while updating the policy ("
-                    + policy.getId() + " - " + policy.getPolicyName() + ")";
+                         + policy.getId() + " - " + policy.getPolicyName() + ")";
             log.error(msg, e);
             throw new PolicyManagementException(msg, e);
         }
@@ -240,12 +240,14 @@ public class PolicyManagerImpl implements PolicyManager {
     @Override
     public boolean deletePolicy(Policy policy) throws PolicyManagementException {
 
-        boolean bool;
         try {
             PolicyManagementDAOFactory.beginTransaction();
-            bool = policyDAO.deletePolicy(policy);
+            policyDAO.deleteAllPolicyRelatedConfigs(policy.getId());
+            policyDAO.deletePolicy(policy.getId());
+            featureDAO.deleteFeaturesOfProfile(policy.getProfileId());
+            profileDAO.deleteProfile(policy.getProfileId());
             PolicyManagementDAOFactory.commitTransaction();
-
+            return true;
         } catch (PolicyManagerDAOException e) {
             try {
                 PolicyManagementDAOFactory.rollbackTransaction();
@@ -253,21 +255,51 @@ public class PolicyManagerImpl implements PolicyManager {
                 log.warn("Error occurred while roll backing the transaction.");
             }
             String msg = "Error occurred while deleting the policy ("
-                    + policy.getId() + " - " + policy.getPolicyName() + ")";
+                         + policy.getId() + " - " + policy.getPolicyName() + ")";
+            log.error(msg, e);
+            throw new PolicyManagementException(msg, e);
+        } catch (ProfileManagerDAOException e) {
+            try {
+                PolicyManagementDAOFactory.rollbackTransaction();
+            } catch (PolicyManagerDAOException e1) {
+                log.warn("Error occurred while roll backing the transaction.");
+            }
+            String msg = "Error occurred while deleting the profile for policy ("
+                         + policy.getId() + ")";
+            log.error(msg, e);
+            throw new PolicyManagementException(msg, e);
+        } catch (FeatureManagerDAOException e) {
+            try {
+                PolicyManagementDAOFactory.rollbackTransaction();
+            } catch (PolicyManagerDAOException e1) {
+                log.warn("Error occurred while roll backing the transaction.");
+            }
+            String msg = "Error occurred while deleting the profile features for policy ("
+                         + policy.getId() + ")";
             log.error(msg, e);
             throw new PolicyManagementException(msg, e);
         }
-        return bool;
+
     }
 
     @Override
     public boolean deletePolicy(int policyId) throws PolicyManagementException {
-        boolean bool;
+
         try {
             PolicyManagementDAOFactory.beginTransaction();
+            Policy policy = policyDAO.getPolicy(policyId);
             policyDAO.deleteAllPolicyRelatedConfigs(policyId);
-            bool = policyDAO.deletePolicy(policyId);
+            policyDAO.deletePolicy(policyId);
+
+            if (log.isDebugEnabled()) {
+                log.debug("Profile ID: " + policy.getProfileId());
+            }
+
+            featureDAO.deleteFeaturesOfProfile(policy.getProfileId());
+
+            profileDAO.deleteProfile(policy.getProfileId());
             PolicyManagementDAOFactory.commitTransaction();
+            return true;
 
         } catch (PolicyManagerDAOException e) {
             try {
@@ -276,16 +308,35 @@ public class PolicyManagerImpl implements PolicyManager {
                 log.warn("Error occurred while roll backing the transaction.");
             }
             String msg = "Error occurred while deleting the policy ("
-                    + policyId + ")";
+                         + policyId + ")";
+            log.error(msg, e);
+            throw new PolicyManagementException(msg, e);
+        } catch (ProfileManagerDAOException e) {
+            try {
+                PolicyManagementDAOFactory.rollbackTransaction();
+            } catch (PolicyManagerDAOException e1) {
+                log.warn("Error occurred while roll backing the transaction.");
+            }
+            String msg = "Error occurred while deleting the profile for policy ("
+                         + policyId + ")";
+            log.error(msg, e);
+            throw new PolicyManagementException(msg, e);
+        } catch (FeatureManagerDAOException e) {
+            try {
+                PolicyManagementDAOFactory.rollbackTransaction();
+            } catch (PolicyManagerDAOException e1) {
+                log.warn("Error occurred while roll backing the transaction.");
+            }
+            String msg = "Error occurred while deleting the profile features for policy ("
+                         + policyId + ")";
             log.error(msg, e);
             throw new PolicyManagementException(msg, e);
         }
-        return bool;
     }
 
     @Override
-    public Policy addPolicyToDevice(List<DeviceIdentifier> deviceIdentifierList, Policy policy) throws
+    public Policy addPolicyToDevice(List<DeviceIdentifier> deviceIdentifierList, Policy policy)
-            PolicyManagementException {
+            throws PolicyManagementException {
 
         try {
             PolicyManagementDAOFactory.beginTransaction();
@@ -319,7 +370,7 @@ public class PolicyManagerImpl implements PolicyManager {
                 log.warn("Error occurred while roll backing the transaction.");
             }
             String msg = "Error occurred while adding the policy ("
-                    + policy.getId() + " - " + policy.getPolicyName() + ")";
+                         + policy.getId() + " - " + policy.getPolicyName() + ")";
             log.error(msg, e);
             throw new PolicyManagementException(msg, e);
         } catch (DeviceManagementException e) {
@@ -366,7 +417,7 @@ public class PolicyManagerImpl implements PolicyManager {
                 log.warn("Error occurred while roll backing the transaction.");
             }
             String msg = "Error occurred while adding the policy ("
-                    + policy.getId() + " - " + policy.getPolicyName() + ")";
+                         + policy.getId() + " - " + policy.getPolicyName() + ")";
             log.error(msg, e);
             throw new PolicyManagementException(msg, e);
         }
@@ -404,7 +455,7 @@ public class PolicyManagerImpl implements PolicyManager {
                 log.warn("Error occurred while roll backing the transaction.");
             }
             String msg = "Error occurred while adding the policy ("
-                    + policy.getId() + " - " + policy.getPolicyName() + ") to user list.";
+                         + policy.getId() + " - " + policy.getPolicyName() + ") to user list.";
             log.error(msg, e);
             throw new PolicyManagementException(msg, e);
         }
@@ -538,12 +589,12 @@ public class PolicyManagerImpl implements PolicyManager {
             Collections.sort(policies);
         } catch (PolicyManagerDAOException e) {
             String msg = "Error occurred while getting the policies for device identifier (" +
-                    deviceIdentifier.getId() + " - " + deviceIdentifier.getType() + ")";
+                         deviceIdentifier.getId() + " - " + deviceIdentifier.getType() + ")";
             log.error(msg, e);
             throw new PolicyManagementException(msg, e);
         } catch (DeviceManagementException e) {
             String msg = "Error occurred while getting device related to device identifier (" +
-                    deviceIdentifier.getId() + " - " + deviceIdentifier.getType() + ")";
+                         deviceIdentifier.getId() + " - " + deviceIdentifier.getType() + ")";
             log.error(msg, e);
             throw new PolicyManagementException(msg, e);
         }
@@ -658,8 +709,7 @@ public class PolicyManagerImpl implements PolicyManager {
 
     @Override
     public void addAppliedPolicyFeaturesToDevice(DeviceIdentifier deviceIdentifier, int policyId,
-                                                 List<ProfileFeature> profileFeatures) throws
+                                                 List<ProfileFeature> profileFeatures) throws PolicyManagementException {
-            PolicyManagementException {
 
         int deviceId = -1;
         try {
@@ -681,7 +731,7 @@ public class PolicyManagerImpl implements PolicyManager {
                 log.warn("Error occurred while roll backing the transaction.");
             }
             String msg = "Error occurred while adding the evaluated policy to device (" +
-                    deviceId + " - " + policyId + ")";
+                         deviceId + " - " + policyId + ")";
             log.error(msg, e);
             throw new PolicyManagementException(msg, e);
         } catch (DeviceManagementException e) {
@@ -693,8 +743,8 @@ public class PolicyManagerImpl implements PolicyManager {
     }
 
     @Override
-    public void addAppliedPolicyToDevice(DeviceIdentifier deviceIdentifier, Policy policy) throws
+    public void addAppliedPolicyToDevice(DeviceIdentifier deviceIdentifier, Policy policy)
-            PolicyManagementException {
+            throws PolicyManagementException {
 
         int deviceId = -1;
         try {
@@ -721,7 +771,7 @@ public class PolicyManagerImpl implements PolicyManager {
                 log.warn("Error occurred while roll backing the transaction.");
             }
             String msg = "Error occurred while adding the evaluated policy to device (" +
-                    deviceId + " - " + policy.getId() + ")";
+                         deviceId + " - " + policy.getId() + ")";
             log.error(msg, e);
             throw new PolicyManagementException(msg, e);
         } catch (DeviceManagementException e) {
@@ -766,7 +816,7 @@ public class PolicyManagerImpl implements PolicyManager {
             return true;
         } catch (PolicyManagerDAOException e) {
             String msg = "Error occurred while setting the policy has applied to device (" +
-                    deviceIdentifier.getId() + ")";
+                         deviceIdentifier.getId() + ")";
             log.error(msg, e);
             throw new PolicyManagementException(msg, e);
         } catch (DeviceManagementException e) {

components/policy-mgt/org.wso2.carbon.policy.mgt.core/src/test/java/org/wso2/carbon/policy/mgt/core/PolicyDAOTestCase.java

@@ -137,8 +137,10 @@ public class PolicyDAOTestCase extends BasePolicyManagementDAOTest {
     }
 
     @Test(dependsOnMethods = ("addProfileFeatures"))
-    public void addPolicy() throws PolicyManagementException {
+    public void addPolicy() throws PolicyManagementException, ProfileManagementException {
-
+        ProfileManager profileManager = new ProfileManagerImpl();
+        profile = ProfileCreator.getProfile(featureList);
+        profileManager.addProfile(profile);
         PolicyManager policyManager = new PolicyManagerImpl();
         policy = PolicyCreator.createPolicy(profile);
         policyManager.addPolicy(policy);
@@ -173,8 +175,10 @@ public class PolicyDAOTestCase extends BasePolicyManagementDAOTest {
     }
 
     @Test(dependsOnMethods = ("addPolicyToDevice"))
-    public void addNewPolicy() throws PolicyManagementException {
+    public void addNewPolicy() throws PolicyManagementException, ProfileManagementException {
-
+        ProfileManager profileManager = new ProfileManagerImpl();
+        profile = ProfileCreator.getProfile(featureList);
+        profileManager.addProfile(profile);
         PolicyManager policyManager = new PolicyManagerImpl();
         policy = PolicyCreator.createPolicy2(profile);
         policyManager.addPolicy(policy);
@@ -182,8 +186,10 @@ public class PolicyDAOTestCase extends BasePolicyManagementDAOTest {
 
 
     @Test(dependsOnMethods = ("addPolicyToDevice"))
-    public void addThirdPolicy() throws PolicyManagementException {
+    public void addThirdPolicy() throws PolicyManagementException, ProfileManagementException {
-
+        ProfileManager profileManager = new ProfileManagerImpl();
+        profile = ProfileCreator.getProfile(featureList);
+        profileManager.addProfile(profile);
         PolicyManager policyManager = new PolicyManagerImpl();
         policy = PolicyCreator.createPolicy4(profile);
         policyManager.addPolicy(policy);
@@ -277,7 +283,10 @@ public class PolicyDAOTestCase extends BasePolicyManagementDAOTest {
     }
 
     @Test(dependsOnMethods = ("getRoleRelatedPolicy"))
-    public void addSecondPolicy() throws PolicyManagementException {
+    public void addSecondPolicy() throws PolicyManagementException, ProfileManagementException {
+        ProfileManager profileManager = new ProfileManagerImpl();
+        profile = ProfileCreator.getProfile(featureList);
+        profileManager.addProfile(profile);
         PolicyManager policyManager = new PolicyManagerImpl();
         policy = PolicyCreator.createPolicy3(profile);
         policyManager.addPolicy(policy);