From 017f36c7a74c52ca9abe05607920dcb3c524f0ad Mon Sep 17 00:00:00 2001 From: "Wang,Frank(gw1218)" Date: Thu, 13 Sep 2018 21:09:47 -0500 Subject: [PATCH] Added validation on matching required fields Import policy should report to end users if there are missing required matching fields in policy template. Issue-ID: POLICY-1117 Change-Id: Iac4485259b73784e641f42a22d5a12c524f4b8e2 Signed-off-by: Wang,Frank(gw1218) Signed-off-by: Wang, Frank(gw1218) --- .../PolicyExportAndImportController.java | 1092 +++++++++++++------- .../PolicyExportAndImportControllerTest.java | 121 ++- 2 files changed, 777 insertions(+), 436 deletions(-) diff --git a/POLICY-SDK-APP/src/main/java/org/onap/policy/controller/PolicyExportAndImportController.java b/POLICY-SDK-APP/src/main/java/org/onap/policy/controller/PolicyExportAndImportController.java index 40056d995..6c7a0aff3 100644 --- a/POLICY-SDK-APP/src/main/java/org/onap/policy/controller/PolicyExportAndImportController.java +++ b/POLICY-SDK-APP/src/main/java/org/onap/policy/controller/PolicyExportAndImportController.java @@ -20,7 +20,9 @@ package org.onap.policy.controller; - +import com.fasterxml.jackson.databind.DeserializationFeature; +import com.fasterxml.jackson.databind.JsonNode; +import com.fasterxml.jackson.databind.ObjectMapper; import java.io.BufferedWriter; import java.io.File; import java.io.FileInputStream; @@ -33,11 +35,9 @@ import java.util.Iterator; import java.util.LinkedHashMap; import java.util.List; import java.util.Set; - import javax.script.SimpleBindings; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; - import org.apache.poi.hssf.usermodel.HSSFRow; import org.apache.poi.hssf.usermodel.HSSFSheet; import org.apache.poi.hssf.usermodel.HSSFWorkbook; @@ -52,6 +52,11 @@ import org.onap.policy.rest.adapter.PolicyExportAdapter; import org.onap.policy.rest.dao.CommonClassDao; import org.onap.policy.rest.jpa.ActionBodyEntity; import org.onap.policy.rest.jpa.ConfigurationDataEntity; +import org.onap.policy.rest.jpa.DCAEuuid; +import org.onap.policy.rest.jpa.GroupPolicyScopeList; +import org.onap.policy.rest.jpa.MicroServiceConfigName; +import org.onap.policy.rest.jpa.MicroServiceLocation; +import org.onap.policy.rest.jpa.MicroServiceModels; import org.onap.policy.rest.jpa.PolicyEditorScopes; import org.onap.policy.rest.jpa.PolicyEntity; import org.onap.policy.rest.jpa.PolicyVersion; @@ -60,386 +65,713 @@ import org.onap.policy.utils.UserUtils.Pair; import org.onap.policy.xacml.api.XACMLErrorConstants; import org.onap.portalsdk.core.controller.RestrictedBaseController; import org.onap.portalsdk.core.web.support.UserUtils; -import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Controller; import org.springframework.web.bind.annotation.RequestMapping; -import com.fasterxml.jackson.databind.DeserializationFeature; -import com.fasterxml.jackson.databind.JsonNode; -import com.fasterxml.jackson.databind.ObjectMapper; - - @Controller @RequestMapping("/") public class PolicyExportAndImportController extends RestrictedBaseController { - private static Logger logger = FlexLogger.getLogger(PolicyExportAndImportController.class); - - private static String superAdmin = "super-admin"; - private static String superEditor = "super-editor"; - private static String admin = "admin"; - private static String editor = "editor"; - private static String policyName = "policyName"; - private static String configurationName = "configurationName"; - private static String configurationbody = "configurationbody"; - private static String config = "Config_"; - - private static CommonClassDao commonClassDao; - - - private PolicyController policyController; - public PolicyController getPolicyController() { - return policyController; - } - - public void setPolicyController(PolicyController policyController) { - this.policyController = policyController; - } - - public static CommonClassDao getCommonClassDao() { - return commonClassDao; - } - - public static void setCommonClassDao(CommonClassDao commonClassDao) { - PolicyExportAndImportController.commonClassDao = commonClassDao; - } - - @Autowired - private PolicyExportAndImportController(CommonClassDao commonClassDao){ - PolicyExportAndImportController.commonClassDao = commonClassDao; - } - - public PolicyExportAndImportController(){ - // Empty constructor - } - - @RequestMapping(value={"/policy_download/exportPolicy.htm"}, method={org.springframework.web.bind.annotation.RequestMethod.POST}) - public void exportPolicy(HttpServletRequest request, HttpServletResponse response) throws IOException{ - try(HSSFWorkbook workBook2 = new HSSFWorkbook()){ - String file; - ArrayList selectedPolicy = new ArrayList<>(); - ObjectMapper mapper = new ObjectMapper(); - mapper.configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false); - JsonNode root = mapper.readTree(request.getReader()); - PolicyExportAdapter adapter = mapper.readValue(root.get("exportData").toString(), PolicyExportAdapter.class); - for (Object policyId : adapter.getPolicyDatas()) { - LinkedHashMap selected = (LinkedHashMap)policyId; - String policyWithScope = selected.get(policyName).toString() + "." + selected.get("activeVersion").toString() + ".xml"; - String scope = policyWithScope.substring(0 , policyWithScope.lastIndexOf(File.separator)).replace(File.separator, "."); - String policyName = policyWithScope.substring(policyWithScope.lastIndexOf(File.separator)+1); - selectedPolicy.add(policyName+":"+scope); - } - List entityData = commonClassDao.getMultipleDataOnAddingConjunction(PolicyEntity.class, "policyName:scope", selectedPolicy); - - HSSFSheet sheet = workBook2.createSheet("PolicyEntity"); - - HSSFRow headingRow = sheet.createRow(0); - headingRow.createCell(0).setCellValue(policyName); - headingRow.createCell(1).setCellValue("scope"); - headingRow.createCell(2).setCellValue("version"); - headingRow.createCell(3).setCellValue("policyData"); - headingRow.createCell(4).setCellValue("description"); - headingRow.createCell(5).setCellValue(configurationName); - headingRow.createCell(6).setCellValue("bodySize"); - headingRow.createCell(7).setCellValue(configurationbody); - - short rowNo = 1; - for (Object object : entityData) { - PolicyEntity policyEntity = (PolicyEntity) object; - HSSFRow row = sheet.createRow(rowNo); - row.createCell(0).setCellValue(policyEntity.getPolicyName()); - row.createCell(1).setCellValue(policyEntity.getScope()); - row.createCell(2).setCellValue(policyEntity.getVersion()); - row.createCell(3).setCellValue(policyEntity.getPolicyData()); - row.createCell(4).setCellValue(policyEntity.getDescription()); - if(!policyEntity.getPolicyName().contains("Decision_")){ - if(policyEntity.getConfigurationData() != null){ - row.createCell(5).setCellValue(policyEntity.getConfigurationData().getConfigurationName()); - String body = policyEntity.getConfigurationData().getConfigBody(); - if(policyEntity.getPolicyName().contains("Config_BRMS_Param_")){ - int index = 0; - int arraySize = 0; - while (index < body.length()) { - if(arraySize == 0){ - row.createCell(7).setCellValue(body.substring(index, Math.min(index + 30000, body.length()))); - }else{ - headingRow.createCell(7 + arraySize).setCellValue(configurationbody+arraySize); - row.createCell(7 + arraySize).setCellValue(body.substring(index, Math.min(index + 30000, body.length()))); - } - index += 30000; - arraySize += 1; - } - row.createCell(6).setCellValue(arraySize); - }else{ - row.createCell(6).setCellValue(0); - row.createCell(7).setCellValue(body); - } - } - if(policyEntity.getActionBodyEntity() != null){ - row.createCell(5).setCellValue(policyEntity.getActionBodyEntity().getActionBodyName()); - row.createCell(6).setCellValue(0); - row.createCell(7).setCellValue(policyEntity.getActionBodyEntity().getActionBody()); - } - }else{ - row.createCell(5).setCellValue(""); - row.createCell(6).setCellValue(0); - row.createCell(7).setCellValue(""); - } - rowNo++; - } - - String tmp = System.getProperty("catalina.base") + File.separator + "webapps" + File.separator + "temp"; - String deleteCheckPath = tmp + File.separator + "PolicyExport.xls"; - File deleteCheck = new File(deleteCheckPath); - if(deleteCheck.exists() && deleteCheck.delete()){ - logger.info("Deleted the file from system before exporting a new file."); - } - File temPath = new File(tmp); - if(!temPath.exists()){ - temPath.mkdir(); - } - - file = temPath + File.separator + "PolicyExport.xls"; - File filepath = new File(file); - FileOutputStream fos = new FileOutputStream(filepath); - workBook2.write(fos); - fos.flush(); - - response.setCharacterEncoding("UTF-8"); - response.setContentType("application / json"); - request.setCharacterEncoding("UTF-8"); - - PrintWriter out = response.getWriter(); - String successMap = file.substring(file.lastIndexOf("webapps")+8); - String responseString = mapper.writeValueAsString(successMap); - JSONObject j = new JSONObject("{data: " + responseString + "}"); - out.write(j.toString()); - }catch(Exception e){ - logger.error(XACMLErrorConstants.ERROR_SYSTEM_ERROR+"Exception Occured while Exporting Policies"+e); - } - } - - //Policy Import - public JSONObject importRepositoryFile(String file, HttpServletRequest request) throws IOException{ - boolean configExists = false; - boolean actionExists = false; - String configName = null; - String scope; - boolean finalColumn; - PolicyController controller = policyController != null ? getPolicyController() : new PolicyController(); - String userId = UserUtils.getUserSession(request).getOrgUserId(); - UserInfo userInfo = (UserInfo) commonClassDao.getEntityItem(UserInfo.class, "userLoginId", userId); - - //Check if the Role and Scope Size are Null get the values from db. - List userRoles = controller.getRoles(userId); - Pair, List> pair = org.onap.policy.utils.UserUtils.checkRoleAndScope(userRoles); - List roles = pair.u; - Set scopes = pair.t; - - try(FileInputStream excelFile = new FileInputStream(new File(file)); HSSFWorkbook workbook = new HSSFWorkbook(excelFile)){ - Sheet datatypeSheet = workbook.getSheetAt(0); - Iterator rowIterator = datatypeSheet.iterator(); - - while (rowIterator.hasNext()) { - finalColumn = false; - PolicyEntity policyEntity = new PolicyEntity(); - ConfigurationDataEntity configurationDataEntity = new ConfigurationDataEntity(); - ActionBodyEntity actionBodyEntity = new ActionBodyEntity(); - PolicyVersion policyVersion = new PolicyVersion(); - Row currentRow = rowIterator.next(); - if (currentRow.getRowNum() == 0) { - continue; - } - Iterator cellIterator = currentRow.cellIterator(); - StringBuilder body = new StringBuilder(); - int bodySize = 0; - int setBodySize = 0; - boolean configurationBodySet = false; - while (cellIterator.hasNext()) { - Cell cell = cellIterator.next(); - if (policyName.equalsIgnoreCase(getCellHeaderName(cell))) { - policyEntity.setPolicyName(cell.getStringCellValue()); - finalColumn = false; - configurationBodySet = false; - configExists = false; - actionExists = false; - } - if ("scope".equalsIgnoreCase(getCellHeaderName(cell))) { - policyEntity.setScope(cell.getStringCellValue()); - } - if ("policyData".equalsIgnoreCase(getCellHeaderName(cell))) { - policyEntity.setPolicyData(cell.getStringCellValue()); - } - if ("description".equalsIgnoreCase(getCellHeaderName(cell))) { - policyEntity.setDescription(cell.getStringCellValue()); - } - if (configurationbody.equalsIgnoreCase(getCellHeaderName(cell))) { - if(policyEntity.getPolicyName().contains(config)){ - if(policyEntity.getPolicyName().contains("Config_BRMS_Param_")){ - setBodySize += 1; - } - if(setBodySize == bodySize){ - finalColumn = true; - configurationBodySet = true; - }else if(setBodySize == 0){ - configurationBodySet = true; - } - configExists = true; - body.append(cell.getStringCellValue()); - }else if(policyEntity.getPolicyName().contains("Action_")){ - actionExists = true; - actionBodyEntity.setActionBody(cell.getStringCellValue()); - } - } - if ("bodySize".equalsIgnoreCase(getCellHeaderName(cell))) { - if(cell.getNumericCellValue() < 1){ - finalColumn = true; - }else{ - bodySize = (int) cell.getNumericCellValue(); - } - } - if (configurationName.equalsIgnoreCase(getCellHeaderName(cell))) { - configName = cell.getStringCellValue(); - if(policyEntity.getPolicyName().contains(config)){ - configurationDataEntity.setConfigurationName(cell.getStringCellValue()); - }else if(policyEntity.getPolicyName().contains("Action_")){ - actionBodyEntity.setActionBodyName(cell.getStringCellValue()); - } - } - - if(finalColumn && configurationBodySet){ - configurationDataEntity.setConfigBody(body.toString()); - scope = policyEntity.getScope().replace(".", File.separator); - String query = "FROM PolicyEntity where policyName = :policyName and scope = :policyScope"; - SimpleBindings params = new SimpleBindings(); - params.put(policyName, policyEntity.getPolicyName()); - params.put("policyScope", policyEntity.getScope()); - List queryData = controller.getDataByQuery(query, params); - if(!queryData.isEmpty()){ - continue; - } - if (roles.contains(superAdmin) || roles.contains(superEditor)) { - //1. if Role contains super admin create scope. - //2. if Role contains super editor don't create new scope and add to list to show to user. - - PolicyEditorScopes policyEditorScope = (PolicyEditorScopes) commonClassDao.getEntityItem(PolicyEditorScopes.class, "scopeName", scope); - if(policyEditorScope == null){ - if(roles.contains(superAdmin)){ - PolicyEditorScopes policyEditorScopeEntity = new PolicyEditorScopes(); - policyEditorScopeEntity.setScopeName(scope); - policyEditorScopeEntity.setUserCreatedBy(userInfo); - policyEditorScopeEntity.setUserModifiedBy(userInfo); - commonClassDao.save(policyEditorScopeEntity); - }else{ - //Add Error Message a new Scope Exists, contact super-admin to create a new scope - continue; - } - } - } - if (roles.contains(admin) || roles.contains(editor)) { - if(scopes.isEmpty()){ - logger.error("No Scopes has been Assigned to the User. Please, Contact Super-Admin"); - }else{ - //1. if Role contains admin, then check if parent scope has role admin, if not don't create a scope and add to list. - if(roles.contains(admin)){ - String scopeCheck = scope.substring(0, scope.lastIndexOf('.')); - if(scopes.contains(scopeCheck)){ - PolicyEditorScopes policyEditorScopeEntity = new PolicyEditorScopes(); - policyEditorScopeEntity.setScopeName(scope); - policyEditorScopeEntity.setUserCreatedBy(userInfo); - policyEditorScopeEntity.setUserModifiedBy(userInfo); - commonClassDao.save(policyEditorScopeEntity); - }else{ - continue; - } - }else{ - continue; - } - } - } - - if(configExists){ - if(configName.endsWith("json")){ - configurationDataEntity.setConfigType("JSON"); - }else if(configName.endsWith("txt")){ - configurationDataEntity.setConfigType("OTHER"); - }else if(configName.endsWith("xml")){ - configurationDataEntity.setConfigType("XML"); - }else if(configName.endsWith("properties")){ - configurationDataEntity.setConfigType("PROPERTIES"); - } - configurationDataEntity.setDeleted(false); - configurationDataEntity.setCreatedBy(userId); - configurationDataEntity.setModifiedBy(userId); - commonClassDao.save(configurationDataEntity); - writeConfigurationFile(configurationDataEntity); - } - if(actionExists){ - actionBodyEntity.setDeleted(false); - actionBodyEntity.setCreatedBy(userId); - actionBodyEntity.setModifiedBy(userId); - commonClassDao.save(actionBodyEntity); - writeActionBodyFile(actionBodyEntity); - } - if(configName != null){ - if(configName.contains(config)){ - ConfigurationDataEntity configuration = (ConfigurationDataEntity) commonClassDao.getEntityItem(ConfigurationDataEntity.class, configurationName, configName); - policyEntity.setConfigurationData(configuration); - }else{ - ActionBodyEntity actionBody = (ActionBodyEntity) commonClassDao.getEntityItem(ActionBodyEntity.class, "actionBodyName", configName); - policyEntity.setActionBodyEntity(actionBody); - } - } - policyEntity.setCreatedBy(userId); - policyEntity.setModifiedBy(userId); - policyEntity.setDeleted(false); - commonClassDao.save(policyEntity); - - policyVersion = new PolicyVersion(); - String policyName = policyEntity.getPolicyName().replace(".xml", ""); - int version = Integer.parseInt(policyName.substring(policyName.lastIndexOf('.')+1)); - policyName = policyName.substring(0, policyName.lastIndexOf('.')); - - policyVersion.setPolicyName(scope.replace(".", File.separator) + File.separator + policyName); - policyVersion.setActiveVersion(version); - policyVersion.setHigherVersion(version); - policyVersion.setCreatedBy(userId); - policyVersion.setModifiedBy(userId); - commonClassDao.save(policyVersion); - - //Notify Other paps regarding Export Policy. - PolicyRestController restController = new PolicyRestController(); - restController.notifyOtherPAPSToUpdateConfigurations("exportPolicy", configName, null); - } - } - } - }catch(IOException e){ - logger.error("Exception Occured While importing the Policy"+e); - } - return null; - } - - private void writeConfigurationFile(ConfigurationDataEntity configurationDataEntity){ - try(FileWriter fw = new FileWriter(PolicyController.getConfigHome() + File.separator + configurationDataEntity.getConfigurationName())){ - BufferedWriter bw = new BufferedWriter(fw); - bw.write(configurationDataEntity.getConfigBody()); - bw.close(); - } catch (IOException e) { - logger.error("Exception Occured While cloning the configuration file",e); - } - } - - private void writeActionBodyFile(ActionBodyEntity actionBodyEntity){ - try(FileWriter fw = new FileWriter(PolicyController.getActionHome() + File.separator + actionBodyEntity.getActionBodyName())) { - BufferedWriter bw = new BufferedWriter(fw); - bw.write(actionBodyEntity.getActionBody()); - bw.close(); - } catch (IOException e) { - logger.error("Exception Occured While cloning the configuration file",e); - } - } - - //return the column header name value - private String getCellHeaderName(Cell cell){ - return cell.getSheet().getRow(0).getCell(cell.getColumnIndex()).getRichStringCellValue().toString(); - } -} \ No newline at end of file + + private static Logger logger = FlexLogger.getLogger(PolicyExportAndImportController.class); + + private static String superAdmin = "super-admin"; + private static String superEditor = "super-editor"; + private static String admin = "admin"; + private static String editor = "editor"; + private static String policyName = "policyName"; + private static String configurationName = "configurationName"; + private static String configurationbody = "configurationbody"; + private static String config = "Config_"; + private static final String CONFIG_MS = "Config_MS_"; + private static final String NOTVALID = " is not valid."; + private static final String POLICY = "Policy:"; + private static final String BODYSIZE = "bodySize"; + private static final String DECISION_MS = "Decision_MS_"; + private static final String ACTION = "Action_"; + private static CommonClassDao commonClassDao; + + + private PolicyController policyController; + + public PolicyController getPolicyController() { + return policyController; + } + + public void setPolicyController(PolicyController policyController) { + this.policyController = policyController; + } + + public static CommonClassDao getCommonClassDao() { + return commonClassDao; + } + + public static void setCommonClassDao(CommonClassDao commonClassDao) { + PolicyExportAndImportController.commonClassDao = commonClassDao; + } + + public PolicyExportAndImportController() { + // Empty constructor + } + + /** + * This is for downloading existing policy. + * + * @param request HttpServletRequest + * @param response HttpServletResponse + * @throws IOException error out + */ + @RequestMapping(value = {"/policy_download/exportPolicy.htm"}, + method = {org.springframework.web.bind.annotation.RequestMethod.POST}) + public void exportPolicy(HttpServletRequest request, HttpServletResponse response) throws IOException { + try (HSSFWorkbook workBook2 = new HSSFWorkbook()) { + ArrayList selectedPolicy = new ArrayList<>(); + ObjectMapper mapper = new ObjectMapper(); + mapper.configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false); + JsonNode root = mapper.readTree(request.getReader()); + PolicyExportAdapter adapter = + mapper.readValue(root.get("exportData").toString(), PolicyExportAdapter.class); + for (Object policyId : adapter.getPolicyDatas()) { + LinkedHashMap selected = (LinkedHashMap) policyId; + String policyWithScope = + selected.get(policyName).toString() + "." + selected.get("activeVersion").toString() + ".xml"; + String scope = + policyWithScope.substring(0, policyWithScope.lastIndexOf(File.separator)).replace( + File.separator, "."); + String policyNamel = policyWithScope.substring(policyWithScope.lastIndexOf(File.separator) + 1); + selectedPolicy.add(policyNamel + ":" + scope); + } + + HSSFSheet sheet = workBook2.createSheet("PolicyEntity"); + HSSFRow headingRow = sheet.createRow(0); + headingRow.createCell(0).setCellValue(policyName); + headingRow.createCell(1).setCellValue("scope"); + headingRow.createCell(2).setCellValue("version"); + headingRow.createCell(3).setCellValue("policyData"); + headingRow.createCell(4).setCellValue("description"); + headingRow.createCell(5).setCellValue(configurationName); + headingRow.createCell(6).setCellValue(BODYSIZE); + headingRow.createCell(7).setCellValue(configurationbody); + + List entityData = + commonClassDao.getMultipleDataOnAddingConjunction(PolicyEntity.class, "policyName:scope", + selectedPolicy); + processEntityData(entityData, sheet, headingRow); // + String tmp = System.getProperty("catalina.base") + File.separator + "webapps" + File.separator + "temp"; + String deleteCheckPath = tmp + File.separator + "PolicyExport.xls"; + File deleteCheck = new File(deleteCheckPath); + if (deleteCheck.exists() && deleteCheck.delete()) { + logger.info("Deleted the file from system before exporting a new file."); + } + File temPath = new File(tmp); + if (!temPath.exists()) { + temPath.mkdir(); + } + + String file = temPath + File.separator + "PolicyExport.xls"; + File filepath = new File(file); + FileOutputStream fos = new FileOutputStream(filepath); + workBook2.write(fos); + fos.flush(); + + response.setContentType("application / json"); + request.setCharacterEncoding("UTF-8"); + + PrintWriter out = response.getWriter(); + String successMap = file.substring(file.lastIndexOf("webapps") + 8); + String responseString = mapper.writeValueAsString(successMap); + JSONObject j = new JSONObject("{data: " + responseString + "}"); + out.write(j.toString()); + } catch (Exception e) { + logger.error(XACMLErrorConstants.ERROR_SYSTEM_ERROR + "Exception Occured while Exporting Policies" + e); + } + } + + private void processEntityData(List entityData, HSSFSheet sheet, HSSFRow headingRow) { + + short rowNo = 1; + for (Object object : entityData) { + PolicyEntity policyEntity = (PolicyEntity) object; + HSSFRow row = sheet.createRow(rowNo); + row.createCell(0).setCellValue(policyEntity.getPolicyName()); + row.createCell(1).setCellValue(policyEntity.getScope()); + row.createCell(2).setCellValue(policyEntity.getVersion()); + row.createCell(3).setCellValue(policyEntity.getPolicyData()); + row.createCell(4).setCellValue(policyEntity.getDescription()); + if (policyEntity.getPolicyName().contains(DECISION_MS) + || !policyEntity.getPolicyName().contains("Decision_")) { + if (policyEntity.getConfigurationData() != null) { + row.createCell(5).setCellValue(policyEntity.getConfigurationData().getConfigurationName()); + String body = policyEntity.getConfigurationData().getConfigBody(); + row = populateConfigParam(policyEntity, row, headingRow, body); + } + populateActionBodyEntity(policyEntity, row); + } else { + row.createCell(5).setCellValue(""); + row.createCell(6).setCellValue(0); + row.createCell(7).setCellValue(""); + } + rowNo++; + } + } + + private HSSFRow populateActionBodyEntity(PolicyEntity policyEntity, HSSFRow row) { + if (policyEntity.getActionBodyEntity() != null) { + row.createCell(5).setCellValue(policyEntity.getActionBodyEntity().getActionBodyName()); + row.createCell(6).setCellValue(0); + row.createCell(7).setCellValue(policyEntity.getActionBodyEntity().getActionBody()); + } + return row; + } + + private HSSFRow populateConfigParam(PolicyEntity policyEntity, HSSFRow row, HSSFRow headingRow, String body) { + + if (policyEntity.getPolicyName().contains("Config_BRMS_Param_")) { + int index = 0; + int arraySize = 0; + while (index < body.length()) { + if (arraySize == 0) { + row.createCell(7).setCellValue(body.substring(index, Math.min(index + 30000, body.length()))); + } else { + headingRow.createCell(7 + arraySize).setCellValue(configurationbody + arraySize); + row.createCell(7 + arraySize).setCellValue( + body.substring(index, Math.min(index + 30000, body.length()))); + } + index += 30000; + arraySize += 1; + } + row.createCell(6).setCellValue(arraySize); + } else { + row.createCell(6).setCellValue(0); + row.createCell(7).setCellValue(body); + } + return row; + } + + + + /** + * This is to upload a policy and save it to database. + * + * @param file String + * @param request HttpServletRequest + * @return JSONObject + * @throws IOException error out + */ + public JSONObject importRepositoryFile(String file, HttpServletRequest request) throws IOException { + boolean configExists = false; + boolean actionExists = false; + String configName = null; + boolean finalColumn; + PolicyController controller = policyController != null ? getPolicyController() : new PolicyController(); + String userId = UserUtils.getUserSession(request).getOrgUserId(); + UserInfo userInfo = (UserInfo) commonClassDao.getEntityItem(UserInfo.class, "userLoginId", userId); + + // Check if the Role and Scope Size are Null get the values from db. + List userRoles = controller.getRoles(userId); + Pair, List> pair = org.onap.policy.utils.UserUtils.checkRoleAndScope(userRoles); + List roles = pair.u; + Set scopes = pair.t; + + try (FileInputStream excelFile = new FileInputStream(new File(file)); + HSSFWorkbook workbook = new HSSFWorkbook(excelFile)) { + Sheet datatypeSheet = workbook.getSheetAt(0); + Iterator rowIterator = datatypeSheet.iterator(); + + while (rowIterator.hasNext()) { + finalColumn = false; + PolicyEntity policyEntity = new PolicyEntity(); + ConfigurationDataEntity configurationDataEntity = new ConfigurationDataEntity(); + ActionBodyEntity actionBodyEntity = new ActionBodyEntity(); + Row currentRow = rowIterator.next(); + if (currentRow.getRowNum() == 0) { + continue; + } + Iterator cellIterator = currentRow.cellIterator(); + StringBuilder body = new StringBuilder(); + int bodySize = 0; + int setBodySize = 0; + boolean configurationBodySet = false; + while (cellIterator.hasNext()) { + Cell cell = cellIterator.next(); + if (policyName.equalsIgnoreCase(getCellHeaderName(cell))) { + policyEntity.setPolicyName(cell.getStringCellValue()); + finalColumn = false; + configurationBodySet = false; + configExists = false; + actionExists = false; + } + policyEntity = populatePolicyEntity(cell, policyEntity); + setBodySize = getSetBodySize(policyEntity, cell, setBodySize); + finalColumn = isFinalColumn(policyEntity, cell, setBodySize, setBodySize, finalColumn); + configurationBodySet = + isConfigurationBodySet(policyEntity, cell, setBodySize, setBodySize, configurationBodySet); + configExists = isConfigExists(policyEntity, cell, configExists); + body = addCellValue(policyEntity, cell, body); + actionExists = isActionExists(policyEntity, cell, actionExists); + actionBodyEntity = setActionBodyObject(policyEntity, actionBodyEntity, cell); + bodySize = getBobySize(bodySize, cell); // + configName = getConfigName(cell, configName); + configurationDataEntity = + populateConfigurationDataEntity(policyEntity, configurationDataEntity, cell); + actionBodyEntity = populateActionBodyObject(policyEntity, actionBodyEntity, cell); + JSONObject response = validatRequiredValue(policyEntity, body, finalColumn, configurationBodySet); + if (response != null) { + return response; + } + savePolicyEntiies(finalColumn, configurationBodySet, configurationDataEntity, policyEntity, + controller, roles, userInfo, scopes, configName, userId, configExists, actionExists, + actionBodyEntity, body); + + } + } + } catch (IOException e) { + logger.error("Exception Occured While importing the Policy" + e); + } + return null; + } + + private void writeConfigurationFile(ConfigurationDataEntity configurationDataEntity) { + try (FileWriter fw = + new FileWriter(PolicyController.getConfigHome() + File.separator + + configurationDataEntity.getConfigurationName())) { + BufferedWriter bw = new BufferedWriter(fw); + bw.write(configurationDataEntity.getConfigBody()); + bw.close(); + } catch (IOException e) { + logger.error("Exception Occured While cloning the configuration file", e); + } + } + + private void writeActionBodyFile(ActionBodyEntity actionBodyEntity) { + try (FileWriter fw = + new FileWriter(PolicyController.getActionHome() + File.separator + actionBodyEntity.getActionBodyName())) { + BufferedWriter bw = new BufferedWriter(fw); + bw.write(actionBodyEntity.getActionBody()); + bw.close(); + } catch (IOException e) { + logger.error("Exception Occured While cloning the configuration file", e); + } + } + + // return the column header name value + private String getCellHeaderName(Cell cell) { + return cell.getSheet().getRow(0).getCell(cell.getColumnIndex()).getRichStringCellValue().toString(); + } + + /** + * This is to validate all matching required fields. + * + * @param policyName String + * @param jsonString String + * @return String + */ + public String validatMatchRequiredFields(String policyName, String jsonString) { + + try { + JSONObject jsonObject = new JSONObject(jsonString); + String configName = jsonObject.getString("configName"); + String uuid = jsonObject.getString("uuid"); + String erorMsg = validConfigName(configName); + if (erorMsg != null) { + return erorMsg; + } + erorMsg = validUuid(uuid); + if (erorMsg != null) { + return erorMsg; + } + String location = jsonObject.getString("location"); + erorMsg = validLocation(location); + if (erorMsg != null) { + return erorMsg; + } + String policyScope = jsonObject.getString("policyScope"); + erorMsg = validPolicyScope(policyScope); + if (erorMsg != null) { + return erorMsg; + } + String msVersion = jsonObject.getString("version"); + String msService = jsonObject.getString("service"); + if (!isAttributeObjectFound(msService, msVersion)) { + return POLICY + policyName + " MS Service: " + msService + " and MS Version: " + msVersion + NOTVALID; + } + + } catch (Exception e) { + logger.error("Exception Occured While validating required fields", e); + } + + return null; + } + + private JSONObject validatRequiredValue(PolicyEntity policyEntity, StringBuilder body, boolean finalColumn, + boolean configurationBodySet) { + if (finalColumn && configurationBodySet && (policyEntity.getPolicyName().contains(CONFIG_MS))) { + String errorMsg = validatMatchRequiredFields(policyEntity.getPolicyName(), body.toString()); + if (errorMsg != null) { + logger.error("errorMsg => " + errorMsg); + JSONObject response = new JSONObject(); + response.append("error", errorMsg); + return response; + } + } + return null; + } + + private String validConfigName(String configName) { + String message = null; + if (configName != null) { + List configNames = commonClassDao.getDataByColumn(MicroServiceConfigName.class, "name"); + if (configNames != null + && (!(configNames.stream().filter(o -> o.equals(configName)).findFirst().isPresent()))) { + message = POLICY + policyName + " configName: " + configName + NOTVALID; + } + } else { + message = POLICY + policyName + "configName is null"; + } + return message; + } + + private String validUuid(String uuid) { + String message = null; + if (uuid != null) { + List uuids = commonClassDao.getDataByColumn(DCAEuuid.class, "name"); + if (uuids != null && !(uuids.stream().filter(o -> o.equals(uuid)).findFirst().isPresent())) { + message = POLICY + policyName + " uuid: " + uuid + NOTVALID; + } + } else { + message = POLICY + policyName + "uuid is null"; + } + return message; + } + + private String validLocation(String location) { + String message = null; + if (location != null) { + List locations = commonClassDao.getDataByColumn(MicroServiceLocation.class, "name"); + if ((locations != null && !(locations.stream().filter(o -> o.equals(location)).findFirst().isPresent()))) { + message = POLICY + policyName + " location: " + location + NOTVALID; + } + } else { + message = POLICY + policyName + "location is null"; + } + return message; + } + + private String validPolicyScope(String policyScope) { + String message = null; + if (policyScope != null) { + List foundData = + commonClassDao.checkDuplicateEntry(policyScope, "groupList", GroupPolicyScopeList.class); + if (foundData == null || foundData.isEmpty()) { + message = POLICY + policyName + " policyScope: " + policyScope + NOTVALID; + } + } else { + message = POLICY + policyName + "policyScope is null"; + } + return message; + } + + private boolean isAttributeObjectFound(String msService, String msVersion) { + if (msService == null) { + return false; + } + + if (msVersion == null) { + return false; + } + MicroServiceModels workingModel = null; + List microServiceModelsData = + commonClassDao.getDataById(MicroServiceModels.class, "modelName", msService); + if (microServiceModelsData != null) { + for (int i = 0; i < microServiceModelsData.size(); i++) { + workingModel = (MicroServiceModels) microServiceModelsData.get(i); + if (workingModel != null && workingModel.getVersion() != null + && workingModel.getVersion().equals(msVersion)) { + return true; + } + } + } + return false; + } + + private PolicyEntity populatePolicyEntity(Cell cell, PolicyEntity policyEntityObject) { + + if (policyEntityObject == null) { + return null; + } + + PolicyEntity policyEntity = policyEntityObject; + if ("scope".equalsIgnoreCase(getCellHeaderName(cell))) { + policyEntity.setScope(cell.getStringCellValue()); + } + if ("policyData".equalsIgnoreCase(getCellHeaderName(cell))) { + policyEntity.setPolicyData(cell.getStringCellValue()); + } + if ("description".equalsIgnoreCase(getCellHeaderName(cell))) { + policyEntity.setDescription(cell.getStringCellValue()); + } + + return policyEntity; + } + + private void saveConfigurePolicy(String configName, ConfigurationDataEntity configurationDataEntity, String userId, + boolean configExists) { + if (configExists) { + if (configName.endsWith("json")) { + configurationDataEntity.setConfigType("JSON"); + } else if (configName.endsWith("txt")) { + configurationDataEntity.setConfigType("OTHER"); + } else if (configName.endsWith("xml")) { + configurationDataEntity.setConfigType("XML"); + } else if (configName.endsWith("properties")) { + configurationDataEntity.setConfigType("PROPERTIES"); + } + configurationDataEntity.setDeleted(false); + configurationDataEntity.setCreatedBy(userId); + configurationDataEntity.setModifiedBy(userId); + commonClassDao.save(configurationDataEntity); + writeConfigurationFile(configurationDataEntity); + } + } + + private void saveActionPolicy(ActionBodyEntity actionBodyEntity, String userId, boolean actionExists) { + if (actionExists) { + actionBodyEntity.setDeleted(false); + actionBodyEntity.setCreatedBy(userId); + actionBodyEntity.setModifiedBy(userId); + commonClassDao.save(actionBodyEntity); + writeActionBodyFile(actionBodyEntity); + } + } + + private boolean isContinue(List roles, String scope, UserInfo userInfo) { + if (roles.contains(superAdmin) || roles.contains(superEditor)) { + // 1. if Role contains super admin create scope. + // 2. if Role contains super editor don't create new scope and add to list to show to user. + PolicyEditorScopes policyEditorScope = + (PolicyEditorScopes) commonClassDao.getEntityItem(PolicyEditorScopes.class, "scopeName", scope); + if (policyEditorScope == null) { + if (roles.contains(superAdmin)) { + PolicyEditorScopes policyEditorScopeEntity = new PolicyEditorScopes(); + policyEditorScopeEntity.setScopeName(scope); + policyEditorScopeEntity.setUserCreatedBy(userInfo); + policyEditorScopeEntity.setUserModifiedBy(userInfo); + commonClassDao.save(policyEditorScopeEntity); + } else { + // Add Error Message a new Scope Exists, contact super-admin to create a new scope + return true; + } + } + } + return false; + } + + private boolean isContinue(List roles, String scope, UserInfo userInfo, Set scopes) { + if (roles.contains(admin) || roles.contains(editor)) { + if (scopes.isEmpty()) { + logger.error("No Scopes has been Assigned to the User. Please, Contact Super-Admin"); + } else { + // 1. if Role contains admin, then check if parent scope has role admin, if not don't + // create a scope and add to list. + if (roles.contains(admin)) { + String scopeCheck = scope.substring(0, scope.lastIndexOf('.')); + if (scopes.contains(scopeCheck)) { + PolicyEditorScopes policyEditorScopeEntity = new PolicyEditorScopes(); + policyEditorScopeEntity.setScopeName(scope); + policyEditorScopeEntity.setUserCreatedBy(userInfo); + policyEditorScopeEntity.setUserModifiedBy(userInfo); + commonClassDao.save(policyEditorScopeEntity); + } else { + return true; + } + } else { + return true; + } + } + } + return false; + } + + private boolean isContinue(List queryData, List roles, String scope, UserInfo userInfo, + Set scopes) { + if (!queryData.isEmpty()) { + return true; + } + if (isContinue(roles, scope, userInfo)) { + return true; + } + return isContinue(roles, scope, userInfo, scopes); + } + + private void savePolicyEntity(PolicyEntity policyEntity, String configName, String userId) { + if (configName != null) { + if (configName.contains(config) || configName.contains(DECISION_MS)) { + ConfigurationDataEntity configuration = + (ConfigurationDataEntity) commonClassDao.getEntityItem(ConfigurationDataEntity.class, + configurationName, configName); + policyEntity.setConfigurationData(configuration); + } else { + ActionBodyEntity actionBody = + (ActionBodyEntity) commonClassDao.getEntityItem(ActionBodyEntity.class, "actionBodyName", + configName); + policyEntity.setActionBodyEntity(actionBody); + } + } + policyEntity.setCreatedBy(userId); + policyEntity.setModifiedBy(userId); + policyEntity.setDeleted(false); + commonClassDao.save(policyEntity); + } + + private void saveVersion(PolicyEntity policyEntity, String scope, String userId) { + PolicyVersion policyVersion = new PolicyVersion(); + String policyNamel = policyEntity.getPolicyName().replace(".xml", ""); + int version = Integer.parseInt(policyNamel.substring(policyNamel.lastIndexOf('.') + 1)); + policyNamel = policyNamel.substring(0, policyNamel.lastIndexOf('.')); + + policyVersion.setPolicyName(scope.replace(".", File.separator) + File.separator + policyNamel); + policyVersion.setActiveVersion(version); + policyVersion.setHigherVersion(version); + policyVersion.setCreatedBy(userId); + policyVersion.setModifiedBy(userId); + commonClassDao.save(policyVersion); + } + + private int getSetBodySize(PolicyEntity policyEntity, Cell cell, int setBodySize) { + int setBodySizel = setBodySize; + if (configurationbody.equalsIgnoreCase(getCellHeaderName(cell)) + && ((policyEntity.getPolicyName().contains(config) || policyEntity.getPolicyName() + .contains(DECISION_MS)) && (policyEntity.getPolicyName().contains("Config_BRMS_Param_")))) { + setBodySizel += 1; + } + return setBodySizel; + } + + private boolean isFinalColumn(PolicyEntity policyEntity, Cell cell, int setBodySize, int bodySize, + boolean finalColumn) { + boolean finalColumnl = finalColumn; + if (configurationbody.equalsIgnoreCase(getCellHeaderName(cell)) + && (policyEntity.getPolicyName().contains(config) || policyEntity.getPolicyName().contains(DECISION_MS)) + && setBodySize == bodySize) { + finalColumnl = true; + } + if (BODYSIZE.equalsIgnoreCase(getCellHeaderName(cell)) && cell.getNumericCellValue() < 1) { + finalColumnl = true; + } + return finalColumnl; + } + + private boolean isConfigurationBodySet(PolicyEntity policyEntity, Cell cell, int setBodySize, int bodySize, + boolean configurationBodySet) { + boolean configurationBodySetl = configurationBodySet; + if (configurationbody.equalsIgnoreCase(getCellHeaderName(cell)) + && (policyEntity.getPolicyName().contains(config) || policyEntity.getPolicyName().contains(DECISION_MS)) + && (setBodySize == bodySize)) { + configurationBodySetl = true; + } + if (configurationbody.equalsIgnoreCase(getCellHeaderName(cell)) + && (policyEntity.getPolicyName().contains(config) || policyEntity.getPolicyName().contains(DECISION_MS)) + && (setBodySize == 0)) { + configurationBodySetl = true; + } + return configurationBodySetl; + } + + + private boolean isConfigExists(PolicyEntity policyEntity, Cell cell, boolean configExists) { + boolean configExistsl = configExists; + if (configurationbody.equalsIgnoreCase(getCellHeaderName(cell)) + && (policyEntity.getPolicyName().contains(config) || policyEntity.getPolicyName().contains(DECISION_MS))) { + configExistsl = true; + } + return configExistsl; + } + + + private boolean isActionExists(PolicyEntity policyEntity, Cell cell, boolean actionExists) { + boolean actionExistsl = actionExists; + if (configurationbody.equalsIgnoreCase(getCellHeaderName(cell)) + && policyEntity.getPolicyName().contains(ACTION)) { + actionExistsl = true; + } + return actionExistsl; + } + + private int getBobySize(int bodySize, Cell cell) { + int bodySizel = bodySize; + if (BODYSIZE.equalsIgnoreCase(getCellHeaderName(cell)) && cell.getNumericCellValue() >= 1) { + bodySizel = (int) cell.getNumericCellValue(); + } + return bodySizel; + } + + private StringBuilder addCellValue(PolicyEntity policyEntity, Cell cell, StringBuilder body) { + if (configurationbody.equalsIgnoreCase(getCellHeaderName(cell)) + && (policyEntity.getPolicyName().contains(config) || policyEntity.getPolicyName().contains(DECISION_MS))) { + body.append(cell.getStringCellValue()); + } + + return body; + } + + private ActionBodyEntity setActionBodyObject(PolicyEntity policyEntity, ActionBodyEntity actionBodyEntity, + Cell cell) { + if (configurationbody.equalsIgnoreCase(getCellHeaderName(cell)) + && (policyEntity.getPolicyName().contains(ACTION))) { + actionBodyEntity.setActionBody(cell.getStringCellValue()); + } + return actionBodyEntity; + } + + private ActionBodyEntity populateActionBodyObject(PolicyEntity policyEntity, ActionBodyEntity actionBodyEntity, + Cell cell) { + if (configurationName.equalsIgnoreCase(getCellHeaderName(cell)) + && policyEntity.getPolicyName().contains(ACTION)) { + actionBodyEntity.setActionBodyName(cell.getStringCellValue()); + } + return actionBodyEntity; + } + + private ConfigurationDataEntity populateConfigurationDataEntity(PolicyEntity policyEntity, + ConfigurationDataEntity configurationDataEntity, Cell cell) { + if (configurationName.equalsIgnoreCase(getCellHeaderName(cell)) + && (policyEntity.getPolicyName().contains(config) || policyEntity.getPolicyName().contains(DECISION_MS))) { + configurationDataEntity.setConfigurationName(cell.getStringCellValue()); + } + return configurationDataEntity; + } + + private String getConfigName(Cell cell, String configName) { + String configNameL = configName; + if (configurationName.equalsIgnoreCase(getCellHeaderName(cell))) { + configNameL = cell.getStringCellValue(); + } + return configNameL; + } + + private void savePolicyEntiies(boolean finalColumn, boolean configurationBodySet, + ConfigurationDataEntity configurationDataEntity, PolicyEntity policyEntity, PolicyController controller, + List roles, UserInfo userInfo, Set scopes, String configName, String userId, + boolean configExists, boolean actionExists, ActionBodyEntity actionBodyEntity, StringBuilder body) { + + if (finalColumn && configurationBodySet) { + configurationDataEntity.setConfigBody(body.toString()); + String scope = policyEntity.getScope().replace(".", File.separator); + String query = "FROM PolicyEntity where policyName = :policyName and scope = :policyScope"; + SimpleBindings params = new SimpleBindings(); + params.put(policyName, policyEntity.getPolicyName()); + params.put("policyScope", policyEntity.getScope()); + List queryData = controller.getDataByQuery(query, params); + + if (isContinue(queryData, roles, scope, userInfo, scopes)) { + return; + } + saveConfigurePolicy(configName, configurationDataEntity, userId, configExists); // + saveActionPolicy(actionBodyEntity, userId, actionExists); // + savePolicyEntity(policyEntity, configName, userId);// + saveVersion(policyEntity, scope, userId); // + // Notify Other paps regarding Export Policy. + PolicyRestController restController = new PolicyRestController(); + restController.notifyOtherPAPSToUpdateConfigurations("exportPolicy", configName, null); + } + } +} diff --git a/POLICY-SDK-APP/src/test/java/org/onap/policy/controller/PolicyExportAndImportControllerTest.java b/POLICY-SDK-APP/src/test/java/org/onap/policy/controller/PolicyExportAndImportControllerTest.java index 358108fa1..1e4e62da1 100644 --- a/POLICY-SDK-APP/src/test/java/org/onap/policy/controller/PolicyExportAndImportControllerTest.java +++ b/POLICY-SDK-APP/src/test/java/org/onap/policy/controller/PolicyExportAndImportControllerTest.java @@ -21,6 +21,7 @@ package org.onap.policy.controller; import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertNull; +import static org.junit.Assert.assertTrue; import static org.mockito.Mockito.when; import static org.mockito.Matchers.any; import static org.mockito.Matchers.eq; @@ -47,62 +48,70 @@ import com.mockrunner.mock.web.MockHttpServletResponse; @RunWith(PowerMockRunner.class) public class PolicyExportAndImportControllerTest { - @Test - public void testSetAndGet(){ - PolicyExportAndImportController controller = new PolicyExportAndImportController(); - PolicyController policyController = new PolicyController(); - controller.setPolicyController(policyController); - assertEquals(controller.getPolicyController(), policyController); - CommonClassDao commonClassDao = new CommonClassDaoImpl(); - PolicyExportAndImportController.setCommonClassDao(commonClassDao); - assertEquals(PolicyExportAndImportController.getCommonClassDao(), commonClassDao); - } - - @Test - public void testExport() throws IOException { - PolicyExportAndImportController controller = new PolicyExportAndImportController(); - MockHttpServletRequest request = new MockHttpServletRequest(); - request.setBodyContent("{\n\"exportData\": {}\n}\n"); - MockHttpServletResponse response = new MockHttpServletResponse(); - - // Test negative case - controller.exportPolicy(request, response); - assertEquals(response.getStatusCode(), HttpServletResponse.SC_OK); - } - - @PrepareForTest({UserUtils.class}) - @Test - public void testImport() throws IOException { - // Mock user utilities - PowerMockito.mockStatic(UserUtils.class); - User user = new User(); - when(UserUtils.getUserSession(any())).thenReturn(user); + @Test + public void testSetAndGet() { + PolicyExportAndImportController controller = new PolicyExportAndImportController(); + PolicyController policyController = new PolicyController(); + controller.setPolicyController(policyController); + assertEquals(controller.getPolicyController(), policyController); + CommonClassDao commonClassDao = new CommonClassDaoImpl(); + PolicyExportAndImportController.setCommonClassDao(commonClassDao); + assertEquals(PolicyExportAndImportController.getCommonClassDao(), commonClassDao); + } - // Mock dao - UserInfo info = new UserInfo(); - ConfigurationDataEntity configEntity = new ConfigurationDataEntity(); - CommonClassDao commonClassDao = Mockito.mock(CommonClassDaoImpl.class); - when(commonClassDao.getEntityItem(eq(UserInfo.class), any(), any())).thenReturn(info); - when(commonClassDao.getEntityItem(eq(ConfigurationDataEntity.class), any(), any())).thenReturn(configEntity); - when(commonClassDao.getDataById(any(), any(), any())).thenReturn(Collections.emptyList()); + @Test + public void testExport() throws IOException { + PolicyExportAndImportController controller = new PolicyExportAndImportController(); + MockHttpServletRequest request = new MockHttpServletRequest(); + request.setBodyContent("{\n\"exportData\": {}\n}\n"); + MockHttpServletResponse response = new MockHttpServletResponse(); - // Test import - ClassLoader classLoader = getClass().getClassLoader(); - PolicyController policyController = new PolicyController(); - PolicyController.setCommonClassDao(commonClassDao); - PolicyExportAndImportController controller = new PolicyExportAndImportController(); - PolicyExportAndImportController.setCommonClassDao(commonClassDao); - controller.setPolicyController(policyController); - HttpServletRequest request = new MockHttpServletRequest(); - - // Test negative case - String file = new File(classLoader.getResource("Config_BRMS_Raw_TestBRMSRawPolicy.1.xml").getFile()).getAbsolutePath(); - JSONObject json = controller.importRepositoryFile(file, request); - assertNull(json); - - // Another negative case - file = new File(classLoader.getResource("PolicyExport.xls").getFile()).getAbsolutePath(); - json = controller.importRepositoryFile(file, request); - assertNull(json); - } + // Test negative case + controller.exportPolicy(request, response); + assertEquals(response.getStatusCode(), HttpServletResponse.SC_OK); + } + + @PrepareForTest({UserUtils.class}) + @Test + public void testImport() throws IOException { + // Mock user utilities + PowerMockito.mockStatic(UserUtils.class); + User user = new User(); + when(UserUtils.getUserSession(any())).thenReturn(user); + + // Mock dao + UserInfo info = new UserInfo(); + ConfigurationDataEntity configEntity = new ConfigurationDataEntity(); + CommonClassDao commonClassDao = Mockito.mock(CommonClassDaoImpl.class); + when(commonClassDao.getEntityItem(eq(UserInfo.class), any(), any())).thenReturn(info); + when(commonClassDao.getEntityItem(eq(ConfigurationDataEntity.class), any(), any())).thenReturn(configEntity); + when(commonClassDao.getDataById(any(), any(), any())).thenReturn(Collections.emptyList()); + + // Test import + PolicyController policyController = new PolicyController(); + PolicyController.setCommonClassDao(commonClassDao); + PolicyExportAndImportController controller = new PolicyExportAndImportController(); + PolicyExportAndImportController.setCommonClassDao(commonClassDao); + controller.setPolicyController(policyController); + HttpServletRequest request = new MockHttpServletRequest(); + ClassLoader classLoader = getClass().getClassLoader(); + + // Test negative case + String file = + new File(classLoader.getResource("Config_BRMS_Raw_TestBRMSRawPolicy.1.xml").getFile()) + .getAbsolutePath(); + JSONObject json = controller.importRepositoryFile(file, request); + assertNull(json); + + // Another negative case + file = new File(classLoader.getResource("PolicyExport.xls").getFile()).getAbsolutePath(); + json = controller.importRepositoryFile(file, request); + assertNull(json); + + // test validation + String jsonString = "{ configName:\"abc\", uuid:\"someone\", location:\"somewhere\", policyScope:\"test\", " + + "service:\"sdnc\", version:\"1810\"}"; + String errorMsg = controller.validatMatchRequiredFields("TestPolicy", jsonString); + assertTrue(errorMsg != null); + } } -- 2.16.6