Fixes for device enrollment status change issues

components/device-mgt/org.wso2.carbon.device.mgt.core/src/main/java/org/wso2/carbon/device/mgt/core/operation/mgt/dao/impl/GenericOperationDAOImpl.java

@@ -674,6 +674,7 @@ public class GenericOperationDAOImpl implements OperationDAO {
             while (rs.next()) {
                 if (enrolmentId == 0) {
                     activity = new Activity();
+                    activity.setActivityId(DeviceManagementConstants.OperationAttributes.ACTIVITY + operationId);
                     activity.setType(Activity.Type.valueOf(rs.getString("OPERATION_TYPE")));
                     activity.setCreatedTimeStamp(new java.util.Date(rs.getLong(("CREATED_TIMESTAMP")) * 1000).toString());
                     activity.setCode(rs.getString("OPERATION_CODE"));

components/device-mgt/org.wso2.carbon.device.mgt.core/src/main/java/org/wso2/carbon/device/mgt/core/service/DeviceManagementProviderServiceImpl.java

@@ -2750,14 +2750,10 @@ public class DeviceManagementProviderServiceImpl implements DeviceManagementProv
         if (EnrolmentInfo.Status.REMOVED == newStatus) {
             isDeviceUpdated = disenrollDevice(deviceIdentifier);
         } else {
-            enrolmentInfo = device.getEnrolmentInfo();
+            enrolmentInfo.setStatus(newStatus);
-            if (enrolmentInfo.getStatus() != newStatus) {
+            isDeviceUpdated = updateEnrollment(deviceId, enrolmentInfo, tenantId);
-                enrolmentInfo.setStatus(newStatus);
-                isDeviceUpdated = updateEnrollment(deviceId, enrolmentInfo, tenantId);
-            } else {
-                isDeviceUpdated = false;
-            }
         }
+        this.removeDeviceFromCache(deviceIdentifier);
         return isDeviceUpdated;
     }
 
@@ -2793,15 +2789,12 @@ public class DeviceManagementProviderServiceImpl implements DeviceManagementProv
         }
         boolean isUpdatedEnrollment = false;
         try {
-            DeviceManagementDAOFactory.openConnection();
+            DeviceManagementDAOFactory.beginTransaction();
             int updatedRows = enrollmentDAO.updateEnrollment(enrolmentInfo, tenantId);
+            DeviceManagementDAOFactory.commitTransaction();
             if (updatedRows > 0) {
                 isUpdatedEnrollment = true;
             }
-        } catch (SQLException e) {
-            String msg = "Error occurred while opening a connection to the data source";
-            log.error(msg, e);
-            throw new DeviceManagementException(msg, e);
         } catch (DeviceManagementDAOException e) {
             DeviceManagementDAOFactory.rollbackTransaction();
             String msg = "Error occurred while updating the enrollment information device for" +
@@ -3319,6 +3312,11 @@ public class DeviceManagementProviderServiceImpl implements DeviceManagementProv
             DeviceManagementDAOFactory.beginTransaction();
             success = deviceDAO.setEnrolmentStatusInBulk(deviceType, status, getTenantId(), deviceList);
             DeviceManagementDAOFactory.commitTransaction();
+            if (success) {
+                for (String id : deviceList) {
+                    this.removeDeviceFromCache(new DeviceIdentifier(id, deviceType));
+                }
+            }
         } catch (DeviceManagementDAOException e) {
             DeviceManagementDAOFactory.rollbackTransaction();
             String msg = "Error occurred in while updating status of devices :" + deviceType + " status : " + status;