# Copyright 2019 Ecosoft Co., Ltd (http://ecosoft.co.th/)
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html)
from flectra.exceptions import UserError
from flectra.tests import common
from flectra.tests.common import Form


class TestJobChannel(common.TransactionCase):
    def setUp(self):
        super().setUp()
        self.print_doc = self.env.ref("report_async.report_async_print_document")
        self.test_rec = self.env.ref("base.module_mail")
        self.test_rpt = self.env.ref("base.ir_module_reference_print")

    def _print_wizard(self, res):
        obj = self.env[res["res_model"]]
        with Form(
            obj.with_context(
                active_model=self.print_doc._name,
                active_id=self.print_doc.id,
                async_process=res["context"].get("async_process"),
            )
        ) as form:
            form.reference = f"{self.test_rec._name},{self.test_rec.id}"
            form.action_report_id = self.test_rpt
        print_wizard = form.save()
        return print_wizard

    def test_1_run_now(self):
        """Run now will return report action as normal"""
        res = self.print_doc.run_now()
        report_action = self._print_wizard(res).print_report()
        self.assertEqual(report_action["type"], "ir.actions.report")

    def test_2_run_async(self):
        """Run background will return nothing, job started"""
        self.print_doc.write({"allow_async": False})
        with self.assertRaises(UserError):
            self.print_doc.run_async()
        self.print_doc.write({"allow_async": True, "email_notify": True})
        res = self.print_doc.run_async()
        print_wizard = self._print_wizard(res)
        report_action = print_wizard.print_report()
        self.assertEqual(report_action, {})  # Do not run report yet
        self.assertEqual(self.print_doc.job_status, "pending")  # Job started
        # Test produce file (as queue will not run in test mode)
        docids = [print_wizard.reference.id]
        data = None
        report_id = self.test_rpt.id
        user_id = self.env.user.id
        self.print_doc.run_report(docids, data, report_id, user_id)
        # Check name of the newly producted file
        # Note: on env with test-enable, always fall back to render_qweb_html
        self.assertIn(self.test_rpt.name, self.print_doc.file_ids[0].name)
        # View fileds/jobs
        self.print_doc.view_files()
        self.print_doc.view_jobs()