diff --git a/storage_media/models/storage_media.py b/storage_media/models/storage_media.py index b7337537cb..b37907d842 100644 --- a/storage_media/models/storage_media.py +++ b/storage_media/models/storage_media.py @@ -38,3 +38,7 @@ def _get_default_backend_id(self): return self.env["storage.backend"]._get_backend_id_from_param( self.env, "storage.media.backend_id" ) + + def unlink(self): + files = self.mapped("file_id") + return super().unlink() and files.unlink() diff --git a/storage_media/tests/test_storage_media.py b/storage_media/tests/test_storage_media.py index 64f5a83d15..5d1dc4a123 100644 --- a/storage_media/tests/test_storage_media.py +++ b/storage_media/tests/test_storage_media.py @@ -27,3 +27,10 @@ def test_create_media(self): media = self.env["storage.media"].create({"name": self.filename}) self.assertEqual(media.file_type, "media") self.assertIsNotNone(media.backend_id) + + def test_unlink(self): + media = self.env["storage.media"].create({"name": self.filename}) + stfile = media.file_id + media.unlink() + self.assertEqual(stfile.to_delete, True) + self.assertEqual(stfile.active, False)