mirror of
https://github.com/OCA/knowledge.git
synced 2025-07-13 07:24:48 -06:00
77 lines
3.2 KiB
Python
77 lines
3.2 KiB
Python
# 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 .test_attachment_zipped_download import TestAttachmentZippedDownloadBase
|
|
|
|
|
|
class TestMixin(TestAttachmentZippedDownloadBase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
super().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["res.partner"].create({"name": "Test partner 1"})
|
|
cls.partner_2 = cls.env["res.partner"].create({"name": "Test partner 2"})
|
|
cls.partner_3 = cls.env["res.partner"].create({"name": "Test partner 3"})
|
|
|
|
cls.partner_1_f1 = cls._create_attachment(
|
|
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.uid,
|
|
"partner_1-f2.txt",
|
|
model="res.partner",
|
|
res_id=cls.partner_1.id,
|
|
)
|
|
cls.partner_2_f1 = cls._create_attachment(
|
|
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"], f"/web/content/{self.partner_2_f1.id}?download=1"
|
|
)
|
|
|
|
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
|
|
)
|