PolicyNotificationController extends RestrictedBaseController { @RequestMapping(value = {"/watchPolicy"}, method = {org.springframework.web.bind.annotation.RequestMethod.POST}) public ModelAndView watchPolicy(HttpServletRequest request, HttpServletResponse response) throws IOException { StringBuilder path = new StringBuilder(); String responseValue = ""; try { response.setCharacterEncoding(PolicyUtils.CHARACTER_ENCODING); request.setCharacterEncoding(PolicyUtils.CHARACTER_ENCODING); String userId = UserUtils.getUserSession(request).getOrgUserId(); logger.info("userid info: " + userId); ObjectMapper mapper = new ObjectMapper(); mapper.configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false); JsonNode root = mapper.readTree(request.getReader()); String name = root.get("watchData").get("name").toString(); JsonNode pathList = root.get("watchData").get("path"); String finalName; if (pathList.isArray()) { ArrayNode arrayNode = (ArrayNode) pathList; for (int i = 0; i < arrayNode.size(); i++) { JsonNode individualElement = arrayNode.get(i); if (i == 0) { path.append(individualElement.toString().replace("\"", "").trim()); } else { path.append(File.separator + individualElement.toString().replace("\"", "").trim()); } } } if (pathList.size() > 0) { finalName = path + File.separator + name.replace("\"", "").trim(); } else { finalName = name.replace("\"", "").trim(); } if (finalName.contains("\\")) { finalName = finalName.replace("\\", "\\\\"); } String query = "from WatchPolicyNotificationTable where POLICYNAME = :finalName and LOGINIDS = :userId"; SimpleBindings params = new SimpleBindings(); params.put("finalName", finalName); params.put("userId", userId); List<Object> watchList = commonClassDao.getDataByQuery(query, params); if (watchList.isEmpty()) { if (finalName.contains("\\\\")) { finalName = finalName.replace("\\\\", File.separator); } WatchPolicyNotificationTable watch = new WatchPolicyNotificationTable(); watch.setPolicyName(finalName); watch.setLoginIds(userId); commonClassDao.save(watch); responseValue = "You have Subscribed Successfully"; } else { commonClassDao.delete(watchList.get(0)); responseValue = "You have UnSubscribed Successfully"; } response.setContentType(PolicyUtils.APPLICATION_JSON); response.getWriter().write(new JSONObject("{watchData: " + mapper.writeValueAsString(responseValue) + "}").toString()); } catch (Exception e) { logger.error("Error druing watchPolicy function " + e); response.getWriter().write(PolicyUtils.CATCH_EXCEPTION); } return null; }  @RequestMapping(value = {"/watchPolicy"}, method = {org.springframework.web.bind.annotation.RequestMethod.POST}) ModelAndView watchPolicy(HttpServletRequest request, HttpServletResponse response);  }
@Test public void testWatch() throws IOException { PowerMockito.mockStatic(UserUtils.class); User user = new User(); user.setOrgUserId("testID"); when(UserUtils.getUserSession(any())).thenReturn(user); CommonClassDao dao = Mockito.mock(CommonClassDao.class); Mockito.when(dao.getDataByQuery(any(), any())).thenReturn(Collections.emptyList()); PolicyNotificationController controller = new PolicyNotificationController(); controller.commonClassDao = dao; MockHttpServletRequest request = new MockHttpServletRequest(); request.setBodyContent("{\n\"watchData\": {\"name\": \"testVal\",\"path\": \"testPath\"\n}}\n"); MockHttpServletResponse response = new MockHttpServletResponse(); ModelAndView model = controller.watchPolicy(request, response); assertNull(model); assertEquals(response.getStatusCode(), HttpServletResponse.SC_OK); request.setBodyContent("{\n\"watchData\": {\"name\": \"testVal\",\"nopath\": \"testPath\"\n}}\n"); response = new MockHttpServletResponse(); model = controller.watchPolicy(request, response); assertNull(model); assertEquals(response.getStatusCode(), HttpServletResponse.SC_OK); }