Fix config fetching and device type filtering for tenants

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

@@ -1410,29 +1410,26 @@ public class DeviceManagementProviderServiceImpl implements DeviceManagementProv
             publicSharedDeviceTypesInDB = deviceTypeDAO.getSharedDeviceTypes();
             Map<DeviceTypeServiceIdentifier, DeviceManagementService> registeredTypes =
                     pluginRepository.getAllDeviceManagementServices(tenantId);
-            Set<String> deviceTypeSetForTenant = new HashSet<>();
+            // Get the device from the public space, however if there is another device with same name then give
-
+            // priority to that
-            if (registeredTypes != null) {
+            if (deviceTypesProvidedByTenant != null) {
-                if (deviceTypesProvidedByTenant != null) {
+                for (DeviceType deviceType : deviceTypesProvidedByTenant) {
-                    for (DeviceType deviceType : deviceTypesProvidedByTenant) {
+                    deviceTypesResponse.add(deviceType.getName());
-                        DeviceTypeServiceIdentifier providerKey = new DeviceTypeServiceIdentifier(deviceType.getName(), tenantId);
+                }
-                        if (registeredTypes.get(providerKey) != null || deviceType.getDeviceTypeMetaDefinition() != null) {
+            }
-                            deviceTypesResponse.add(deviceType.getName());
+            if (publicSharedDeviceTypesInDB != null) {
-                            deviceTypeSetForTenant.add(deviceType.getName());
+                for (String deviceType: publicSharedDeviceTypesInDB) {
-                        }
+                    if (!deviceTypesResponse.contains(deviceType)) {
+                        deviceTypesResponse.add(deviceType);
                     }
                 }
-                // Get the device from the public space, however if there is another device with same name then give
+            }
-                // priority to that
+            if (registeredTypes != null) {
-                if (publicSharedDeviceTypesInDB != null) {
+                for (DeviceTypeServiceIdentifier deviceType: registeredTypes.keySet()) {
-                    for (String deviceType : publicSharedDeviceTypesInDB) {
+                    if (!deviceTypesResponse.contains(deviceType.getDeviceType())) {
-                        DeviceTypeServiceIdentifier providerKey = new DeviceTypeServiceIdentifier(deviceType);
+                        deviceTypesResponse.add(deviceType.getDeviceType());
-                        if (registeredTypes.get(providerKey) != null && !deviceTypeSetForTenant.contains(deviceType)) {
-                            deviceTypesResponse.add(deviceType);
-                        }
                     }
                 }
-
             }
         } catch (DeviceManagementDAOException e) {
             String msg = "Error occurred while obtaining the device types.";
@@ -3489,12 +3486,12 @@ public class DeviceManagementProviderServiceImpl implements DeviceManagementProv
         }
 
         try {
-            Device device = this.getDevice(new DeviceIdentifier(deviceProperties.getDeviceIdentifier(),
-                                                                deviceProperties.getDeviceTypeName()), false);
-            String owner = device.getEnrolmentInfo().getOwner();
             PrivilegedCarbonContext.startTenantFlow();
             PrivilegedCarbonContext ctx = PrivilegedCarbonContext.getThreadLocalCarbonContext();
             ctx.setTenantId(Integer.parseInt(deviceProperties.getTenantId()), true);
+            Device device = this.getDevice(new DeviceIdentifier(deviceProperties.getDeviceIdentifier(),
+                                                                deviceProperties.getDeviceTypeName()), false);
+            String owner = device.getEnrolmentInfo().getOwner();
             PlatformConfiguration configuration = this.getConfiguration(device.getType());
             List<ConfigurationEntry> configurationEntries = new ArrayList<>();
             if (configuration != null) {