commit
fbb9ab4cb3
|
@ -4,6 +4,7 @@
|
|||
import json
|
||||
import logging
|
||||
|
||||
from werkzeug.exceptions import InternalServerError
|
||||
from werkzeug.urls import url_decode
|
||||
|
||||
from odoo.http import (
|
||||
|
@ -54,6 +55,7 @@ class ReportController(report.ReportController):
|
|||
def report_download(self, data, context=None):
|
||||
requestcontent = json.loads(data)
|
||||
url, report_type = requestcontent[0], requestcontent[1]
|
||||
reportname = ""
|
||||
try:
|
||||
if report_type == "csv":
|
||||
reportname = url.split("/report/csv/")[1].split("?")[0]
|
||||
|
@ -102,4 +104,5 @@ class ReportController(report.ReportController):
|
|||
_logger.exception("Error while generating report %s", reportname)
|
||||
se = _serialize_exception(e)
|
||||
error = {"code": 200, "message": "Odoo Server Error", "data": se}
|
||||
return request.make_response(html_escape(json.dumps(error)))
|
||||
res = request.make_response(html_escape(json.dumps(error)))
|
||||
raise InternalServerError(response=res) from e
|
||||
|
|
|
@ -1,9 +1,15 @@
|
|||
# Copyright 2019 Creu Blanca
|
||||
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html).
|
||||
import json
|
||||
import logging
|
||||
from io import StringIO
|
||||
from unittest import mock
|
||||
|
||||
from odoo import http
|
||||
from odoo.tests import common
|
||||
from odoo.tools import mute_logger
|
||||
|
||||
from odoo.addons.web.controllers.report import ReportController
|
||||
|
||||
_logger = logging.getLogger(__name__)
|
||||
try:
|
||||
|
@ -12,6 +18,14 @@ except ImportError:
|
|||
_logger.debug("Can not import csv.")
|
||||
|
||||
|
||||
class TestCsvException(Exception):
|
||||
def __init__(self, message):
|
||||
"""
|
||||
:param message: exception message and frontend modal content
|
||||
"""
|
||||
super().__init__(message)
|
||||
|
||||
|
||||
class TestReport(common.TransactionCase):
|
||||
def setUp(self):
|
||||
super().setUp()
|
||||
|
@ -55,3 +69,49 @@ class TestReport(common.TransactionCase):
|
|||
# Typical call from render
|
||||
objs = self.csv_report._get_objs_for_report(self.docs.ids, {})
|
||||
self.assertEqual(objs, self.docs)
|
||||
|
||||
|
||||
class TestCsvReport(common.HttpCase):
|
||||
"""
|
||||
Some tests calling controller
|
||||
"""
|
||||
|
||||
def setUp(self):
|
||||
super().setUp()
|
||||
self.report_object = self.env["ir.actions.report"]
|
||||
self.csv_report = self.env["report.report_csv.abstract"].with_context(
|
||||
active_model="res.partner"
|
||||
)
|
||||
self.report_name = "report_csv.partner_csv"
|
||||
self.report = self.report_object._get_report_from_name(self.report_name)
|
||||
self.docs = self.env["res.company"].search([], limit=1).partner_id
|
||||
self.session = self.authenticate("admin", "admin")
|
||||
|
||||
def test_csv(self):
|
||||
filename = self.get_report_headers().headers.get("Content-Disposition")
|
||||
self.assertTrue(".csv" in filename)
|
||||
|
||||
@mute_logger("odoo.addons.web.controllers.report")
|
||||
def test_pdf_error(self):
|
||||
with mock.patch.object(
|
||||
ReportController, "report_routes"
|
||||
) as route_patch, self.assertLogs(
|
||||
"odoo.addons.report_csv.controllers.main", level=logging.ERROR
|
||||
) as cm:
|
||||
route_patch.side_effect = TestCsvException("Test")
|
||||
self.get_report_headers(
|
||||
suffix="/report/pdf/test/10", f_type="qweb-pdf"
|
||||
).headers.get("Content-Disposition")
|
||||
[msg] = cm.output
|
||||
self.assertIn("Error while generating report", msg)
|
||||
|
||||
def get_report_headers(
|
||||
self, suffix="/report/csv/report_csv.partner_csv/1", f_type="csv"
|
||||
):
|
||||
return self.url_open(
|
||||
url="/report/download",
|
||||
data={
|
||||
"data": json.dumps([suffix, f_type]),
|
||||
"csrf_token": http.Request.csrf_token(self),
|
||||
},
|
||||
)
|
||||
|
|
Loading…
Reference in New Issue