Browse Source

Improve tests

pull/347/head
Laurent Mignon 8 years ago
committed by Laurent Mignon (ACSONE)
parent
commit
ef81ba21cf
  1. 2
      report_py3o/py3o_parser.py
  2. 16
      report_py3o/tests/test_report_py3o.py

2
report_py3o/py3o_parser.py

@ -180,7 +180,7 @@ class Py3oParser(report_sxw):
# we do nice chunked reading from the network...
chunk_size = 1024
with NamedTemporaryFile(
suffix=filetype.human_ext,
suffix=filetype,
prefix='py3o-template-'
) as fd:
for chunk in r.iter_content(chunk_size):

16
report_py3o/tests/test_report_py3o.py

@ -19,9 +19,21 @@ class TestReportPy3o(TransactionCase):
'Py3oParser.create_single_pdf') as patched_pdf:
# test the call the the create method inside our custom parser
report.render_report(self.env.user.ids,
report.report_name,
{})
report.report_name,
{})
self.assertEqual(1, patched_pdf.call_count)
res = report.render_report(
self.env.user.ids, report.report_name, {})
self.assertTrue(res)
py3o_server = self.env['py3o.server'].create({"url": "http://dummy"})
# check the call to the fusion server
report.write({"py3o_filetype": "pdf",
"py3o_server_id": py3o_server.id})
with mock.patch('requests.post') as patched_post:
magick_response = mock.MagicMock()
magick_response.status_code = 200
patched_post.return_value = magick_response
magick_response.iter_content.return_value = "test result"
res = report.render_report(
self.env.user.ids, report.report_name, {})
self.assertEqual(('test result', '.pdf'), res)
Loading…
Cancel
Save