[DEV] test excption handler with response enttity

atms-api/src/main/java/pwc/taxtech/atms/controller/ReportController.java

@@ -33,10 +33,11 @@ public class ReportController {
         return reportService.getReportTemplate(projectId, EnumServiceType.getEnumByCode(serviceType), period);
     }
 
-    @RequestMapping(value = "generateByTotal/{projectId}/{mergeManual}/{period}", method = RequestMethod.POST, produces = MediaType.APPLICATION_JSON_UTF8_VALUE)
-    public OperationResultDto generateAllData(@PathVariable String projectId, @PathVariable Integer period, @RequestParam Optional<String> generator,
-                                              @PathVariable Boolean mergeManual) {
-        return reportService.generateData(projectId, EnumServiceType.VAT, mergeManual, period, null, generator);
+    @RequestMapping(value = "generateByTotal/{projectId}/{mergeManual}/{period}", method = RequestMethod.POST,
+            produces = MediaType.APPLICATION_JSON_UTF8_VALUE)
+    public ResponseEntity generateAllData(@PathVariable String projectId, @PathVariable Integer period,
+                                          @RequestParam Optional<String> generator, @PathVariable Boolean mergeManual) {
+        return ResponseEntity.ok(reportService.generateData(projectId, EnumServiceType.VAT, mergeManual, period, null, generator));
     }
 
     @RequestMapping(value = "getRunningJob/{projectId}/{period}", method = RequestMethod.GET, produces = MediaType.APPLICATION_JSON_UTF8_VALUE)
@@ -105,12 +106,12 @@ public class ReportController {
     }
 
     @RequestMapping(value = "addCellManualData", method = RequestMethod.POST, produces = MediaType.APPLICATION_JSON_UTF8_VALUE)
-    public OperationResultDto addCellManualDataSource(@RequestBody ManualDataSourceDto data, @RequestHeader String from) {
+    public ResponseEntity addCellManualDataSource(@RequestBody ManualDataSourceDto data, @RequestHeader String from) {
         String projectId = StringUtils.EMPTY;
         if (StringUtils.isNotBlank(from) && from.split("@").length > 0) {
             projectId = from.split("@")[0];
         }
-        return reportService.addCellManualDataSource(data, from);
+        return ResponseEntity.ok(reportService.addCellManualDataSource(data, from));
     }
 
     @RequestMapping(value = "addDataSource", method = RequestMethod.POST, produces = MediaType.APPLICATION_JSON_UTF8_VALUE)