mirror of
https://gitlab.com/flectra-community/reporting-engine.git
synced 2024-11-15 02:32:05 +00:00
58 lines
1.9 KiB
Python
58 lines
1.9 KiB
Python
|
# Copyright 2019 Creu Blanca
|
||
|
|
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html).
|
||
|
|
import logging
|
||
|
|
from io import StringIO
|
||
|
|
|
||
|
|
from flectra.tests import common
|
||
|
|
|
||
|
|
_logger = logging.getLogger(__name__)
|
||
|
|
try:
|
||
|
|
import csv
|
||
|
|
except ImportError:
|
||
|
|
_logger.debug("Can not import csv.")
|
||
|
|
|
||
|
|
|
||
|
|
class TestReport(common.TransactionCase):
|
||
|
|
def setUp(self):
|
||
|
|
super().setUp()
|
||
|
|
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 = report_object._get_report_from_name(self.report_name)
|
||
|
|
self.docs = self.env["res.company"].search([], limit=1).partner_id
|
||
|
|
|
||
|
|
def test_report(self):
|
||
|
|
# Test if not res:
|
||
|
|
report = self.report
|
||
|
|
self.assertEqual(report.report_type, "csv")
|
||
|
|
rep = report._render(self.docs.ids, {})
|
||
|
|
str_io = StringIO(rep[0])
|
||
|
|
dict_report = list(csv.DictReader(str_io, delimiter=";", quoting=csv.QUOTE_ALL))
|
||
|
|
self.assertEqual(self.docs.name, dict(dict_report[0])["name"])
|
||
|
|
|
||
|
|
def test_id_retrieval(self):
|
||
|
|
|
||
|
|
# Typical call from WebUI with wizard
|
||
|
|
objs = self.csv_report._get_objs_for_report(
|
||
|
|
False, {"context": {"active_ids": self.docs.ids}}
|
||
|
|
)
|
||
|
|
self.assertEqual(objs, self.docs)
|
||
|
|
|
||
|
|
# Typical call from within code not to report_action
|
||
|
|
objs = self.csv_report.with_context(
|
||
|
|
active_ids=self.docs.ids
|
||
|
|
)._get_objs_for_report(False, False)
|
||
|
|
self.assertEqual(objs, self.docs)
|
||
|
|
|
||
|
|
# Typical call from WebUI
|
||
|
|
objs = self.csv_report._get_objs_for_report(
|
||
|
|
self.docs.ids, {"data": [self.report_name, self.report.report_type]}
|
||
|
|
)
|
||
|
|
self.assertEqual(objs, self.docs)
|
||
|
|
|
||
|
|
# Typical call from render
|
||
|
|
objs = self.csv_report._get_objs_for_report(self.docs.ids, {})
|
||
|
|
self.assertEqual(objs, self.docs)
|