Fix passing favicon image bean when retrieving existing logo

components/device-mgt/org.wso2.carbon.device.mgt.core/src/main/java/org/wso2/carbon/device/mgt/core/metadata/mgt/WhiteLabelManagementServiceImpl.java

@@ -248,24 +248,24 @@ public class WhiteLabelManagementServiceImpl implements WhiteLabelManagementServ
                 existingFaviconImage = WhiteLabelStorageUtil.getWhiteLabelImageFile(theme.getFaviconImage(), WhiteLabelImage.ImageName.FAVICON);
             }
             if (theme.getLogoImage().getImageLocationType() == WhiteLabelImage.ImageLocationType.CUSTOM_FILE) {
-                existingLogoImage = WhiteLabelStorageUtil.getWhiteLabelImageFile(theme.getFaviconImage(), WhiteLabelImage.ImageName.LOGO);
+                existingLogoImage = WhiteLabelStorageUtil.getWhiteLabelImageFile(theme.getLogoImage(), WhiteLabelImage.ImageName.LOGO);
             }
             storeWhiteLabelImageIfRequired(createWhiteLabelTheme.getFavicon(), WhiteLabelImage.ImageName.FAVICON, tenantId);
             storeWhiteLabelImageIfRequired(createWhiteLabelTheme.getLogo(), WhiteLabelImage.ImageName.LOGO, tenantId);
             WhiteLabelTheme whiteLabelTheme = constructWhiteLabelTheme(createWhiteLabelTheme);
             Metadata metadataWhiteLabelTheme = constructWhiteLabelThemeMetadata(whiteLabelTheme);
-
+            try {
                 MetadataManagementDAOFactory.beginTransaction();
                 metadataDAO.updateMetadata(tenantId, metadataWhiteLabelTheme);
                 MetadataManagementDAOFactory.commitTransaction();
                 if (log.isDebugEnabled()) {
-                log.debug("Metadata entry created successfully. " + createWhiteLabelTheme.toString());
+                    log.debug("Metadata entry created successfully. " + createWhiteLabelTheme);
                 }
                 return whiteLabelTheme;
             } catch (MetadataManagementDAOException e) {
                 MetadataManagementDAOFactory.rollbackTransaction();
                 restoreWhiteLabelImages(existingFaviconImage, existingLogoImage, tenantId);
-            String msg = "Error occurred while creating the metadata entry. " + createWhiteLabelTheme.toString();
+                String msg = "Error occurred while creating the metadata entry. " + createWhiteLabelTheme;
                 log.error(msg, e);
                 throw new MetadataManagementException(msg, e);
             } catch (TransactionManagementException e) {
@@ -273,12 +273,13 @@ public class WhiteLabelManagementServiceImpl implements WhiteLabelManagementServ
                 String msg = "Error occurred while opening a connection to the data source";
                 log.error(msg, e);
                 throw new MetadataManagementException("Error occurred while creating metadata record", e);
+            } finally {
+                MetadataManagementDAOFactory.closeConnection();
+            }
         } catch (NotFoundException e) {
             String msg = "Error occurred while retrieving existing white label theme";
             log.error(msg, e);
             throw new MetadataManagementException(msg, e);
-        } finally {
-            MetadataManagementDAOFactory.closeConnection();
         }
     }
 

components/device-mgt/org.wso2.carbon.device.mgt.core/src/main/java/org/wso2/carbon/device/mgt/core/metadata/mgt/util/WhiteLabelStorageUtil.java

@@ -116,7 +116,7 @@ public class WhiteLabelStorageUtil {
      * @return white label image file {@link File}
      */
     public static File getWhiteLabelImageFile(WhiteLabelImage image, WhiteLabelImage.ImageName imageName)
-            throws MetadataManagementException, NotFoundException {
+            throws MetadataManagementException {
         String fullPathToImage = getPathToImage(image, imageName);
         return new File(fullPathToImage);
     }