[MIG] dms_attachment_link: Finish migration

Author: victoralmau

dms_attachment_link/models/__init__.py

@@ -1,2 +1,3 @@
 from . import dms_file
 from . import ir_attachment
+from . import ir_binary

dms_attachment_link/models/ir_attachment.py

@@ -9,7 +9,7 @@ class IrAttachment(models.Model):
 
     dms_file_id = fields.Many2one(comodel_name="dms.file")
 
-    @api.depends("dms_file_id.content")
+    @api.depends("dms_file_id", "dms_file_id.content")
     def _compute_datas(self):
         """Get the contents of the attachment directly from the DMS file."""
         _self = self.filtered("dms_file_id")

dms_attachment_link/models/ir_binary.py

@@ -0,0 +1,15 @@
+# Copyright 2024 Tecnativa - Víctor Martínez
+# License LGPL-3.0 or later (http://www.gnu.org/licenses/lgpl).
+
+from odoo import models
+
+
+class IrBinary(models.AbstractModel):
+    _inherit = "ir.binary"
+
+    def _record_to_stream(self, record, field_name):
+        """We need to overwrite for the download and preview to be correct."""
+        if record._name == "ir.attachment" and record.dms_file_id:
+            record = record.dms_file_id
+            field_name = "content"
+        return super()._record_to_stream(record=record, field_name=field_name)

dms_attachment_link/tests/test_dms_attachment_link.py

@@ -1,14 +1,15 @@
 # Copyright 2023 Tecnativa - Víctor Martínez
 # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
 
-from odoo.tests import common
+from odoo.addons.base.tests.common import BaseCommon
 
 
-class TestDmsAttachmentLink(common.TransactionCase):
-    def setUp(self):
-        super().setUp()
-        self.partner = self.env["res.partner"].create({"name": "Test partner"})
-        self.dms_file = self.env.ref("dms.file_01_demo")
+class TestDmsAttachmentLink(BaseCommon):
+    @classmethod
+    def setUpClass(cls):
+        super().setUpClass()
+        cls.partner = cls.env["res.partner"].create({"name": "Test partner"})
+        cls.dms_file = cls.env.ref("dms.file_01_demo")
 
     def test_add_url_attachment(self):
         attachment = self.dms_file.with_context(