[IMP] l10n_br_mdfe: add make pdf tests

l10n_br_mdfe/tests/__init__.py

@@ -4,3 +4,4 @@
 from . import test_mdfe_import
 from . import test_mdfe_structure
 from . import test_mdfe_res_partner
+from . import test_mdfe_damdfe

l10n_br_mdfe/tests/test_mdfe_damdfe.py

@@ -0,0 +1,39 @@
+# Copyright 2024 - TODAY, Marcel Savegnago <[email protected]>
+# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).
+
+from odoo.exceptions import UserError
+from odoo.tests.common import TransactionCase
+
+
+class TestDamdfeGeneration(TransactionCase):
+    def setUp(self):
+        super().setUp()
+
+    def test_generate_damdfe_brazil_fiscal_report(self):
+        mdfe = self.env.ref("l10n_br_mdfe.demo_mdfe_lc_modal_rodoviario")
+        mdfe.action_document_confirm()
+        mdfe.view_pdf()
+
+        self.assertTrue(mdfe.file_report_id)
+
+    def test_generate_damdfe_document_type_error(self):
+        damdfe_report = self.env["ir.actions.report"].search(
+            [("report_name", "=", "main_template_damdfe")]
+        )
+        mdfe = self.env.ref("l10n_br_mdfe.demo_mdfe_lc_modal_rodoviario")
+        mdfe.document_type_id = self.env.ref("l10n_br_fiscal.document_01")
+        mdfe.action_document_confirm()
+        with self.assertRaises(UserError) as captured_exception:
+            damdfe_report._render_qweb_pdf([mdfe.id])
+        self.assertEqual(
+            captured_exception.exception.args[0],
+            "You can only print a DAMDFE of a MDFe(58).",
+        )
+
+    def test_generate_damdfe_brazil_fiscal_report_partner(self):
+        mdfe = self.env.ref("l10n_br_mdfe.demo_mdfe_lc_modal_rodoviario")
+        mdfe.action_document_confirm()
+        mdfe.issuer = "partner"
+        mdfe.view_pdf()
+
+        self.assertTrue(mdfe.file_report_id)