Allow other statuses to update the device status

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

@@ -2744,11 +2744,9 @@ public class DeviceManagementProviderServiceImpl implements DeviceManagementProv
             return false; //New status is similar to current
         }
         int tenantId = this.getTenantId();
-        switch (newStatus) {
+        if (EnrolmentInfo.Status.REMOVED == newStatus) {
-            case ACTIVE:
+            isDeviceUpdated = disenrollDevice(deviceIdentifier);
-                isDeviceUpdated = updateEnrollment(deviceId, enrolmentInfo, tenantId);
+        } else {
-                break;
-            case INACTIVE:
             enrolmentInfo = device.getEnrolmentInfo();
             if (enrolmentInfo.getStatus() != newStatus) {
                 enrolmentInfo.setStatus(newStatus);
@@ -2756,12 +2754,6 @@ public class DeviceManagementProviderServiceImpl implements DeviceManagementProv
             } else {
                 isDeviceUpdated = false;
             }
-                break;
-            case REMOVED:
-                isDeviceUpdated = disenrollDevice(deviceIdentifier);
-                break;
-            default:
-                throw new DeviceManagementException("Invalid status retrieved. Status : " + newStatus);
         }
         return isDeviceUpdated;
     }