[IMP] attachment_zipped_download: provide ir.attachment.action_download mixin

This helps to download multiple attachments from any models.
2025-12-21 04:42:18 -06:00 · 2023-04-28 16:13:20 +02:00
parent 6984c9caf5
commit 66dd5ab063
14 changed files with 433 additions and 14 deletions
--- a/attachment_zipped_download/tests/test_ir_attachment_action_download.py
+++ b/attachment_zipped_download/tests/test_ir_attachment_action_download.py
@@ -0,0 +1,81 @@
+# Copyright 2023 Foodles (https://www.foodles.com/)
+# @author Pierre Verkest <pierreverkest84@gmail.com>
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
+from odoo_test_helper import FakeModelLoader
+
+from odoo.tests import SavepointCase
+
+from .test_attachment_zipped_download import TestAttachmentZippedDownloadBase
+
+
+class TestMixin(SavepointCase, TestAttachmentZippedDownloadBase):
+    @classmethod
+    def setUpClass(cls):
+        super(TestMixin, cls).setUpClass()
+        cls.loader = FakeModelLoader(cls.env, cls.__module__)
+        cls.addClassCleanup(cls.loader.restore_registry)
+        cls.loader.backup_registry()
+
+        # Imported Test model must be done after the backup_registry
+        from .models.res_partner import ResPartner
+
+        cls.loader.update_registry((ResPartner,))
+
+        cls.partner_1 = cls.env.ref("base.res_partner_1")
+        cls.partner_2 = cls.env.ref("base.res_partner_2")
+        cls.partner_3 = cls.env.ref("base.res_partner_3")
+
+        cls.partner_1_f1 = cls._create_attachment(
+            cls.env,
+            cls.env.uid,
+            "partner_1-f1.txt",
+            model="res.partner",
+            res_id=cls.partner_1.id,
+        )
+        cls.partner_1_f2 = cls._create_attachment(
+            cls.env,
+            cls.env.uid,
+            "partner_1-f2.txt",
+            model="res.partner",
+            res_id=cls.partner_1.id,
+        )
+        cls.partner_2_f1 = cls._create_attachment(
+            cls.env,
+            cls.env.uid,
+            "partner_2-f1.txt",
+            model="res.partner",
+            res_id=cls.partner_2.id,
+        )
+
+    def test_action_download_attachments_no_attachment(self):
+        action = self.partner_3.action_download_attachments()
+        self.assertEqual(action["type"], "ir.actions.client")
+        self.assertEqual(action["tag"], "display_notification")
+
+    def test_action_download_attachments_one_attachment(self):
+        action = (self.partner_2 | self.partner_3).action_download_attachments()
+        self.assertEqual(action["type"], "ir.actions.act_url")
+        self.assertEqual(action["target"], "self")
+        self.assertEqual(
+            action["url"], "/web/content/%s?download=1" % self.partner_2_f1.id
+        )
+
+    def test_action_download_attachments_two_attachment_one_record(self):
+        action = (self.partner_1).action_download_attachments()
+        self.assertEqual(action["type"], "ir.actions.act_url")
+        self.assertEqual(action["target"], "self")
+        self.assertTrue(action["url"].startswith("/web/attachment/download_zip?ids="))
+        ids = sorted(map(int, action["url"].split("=")[1].split(",")))
+        self.assertEqual(ids, (self.partner_1_f1 | self.partner_1_f2).ids)
+
+    def test_action_download_attachments_three_attachment_n_records(self):
+        action = (
+            self.partner_1 | self.partner_2 | self.partner_3
+        ).action_download_attachments()
+        self.assertEqual(action["type"], "ir.actions.act_url")
+        self.assertEqual(action["target"], "self")
+        self.assertTrue(action["url"].startswith("/web/attachment/download_zip?ids="))
+        ids = sorted(map(int, action["url"].split("=")[1].split(",")))
+        self.assertEqual(
+            ids, (self.partner_1_f1 + self.partner_1_f2 + self.partner_2_f1).ids
+        )