oca-technical/odoo-bringout-oca-dms-dms/dms/tests/test_storage_database.py
2025-08-29 15:43:03 +02:00

84 lines
3.9 KiB
Python

# Copyright 2017-2019 MuK IT GmbH.
# Copyright 2020 Creu Blanca
# Copyright 2022 Víctor Martínez
# License LGPL-3.0 or later (http://www.gnu.org/licenses/lgpl).
from odoo.tests.common import users
from odoo.tools import mute_logger
from .common import StorageDatabaseBaseCase
class StorageDatabaseTestCase(StorageDatabaseBaseCase):
@users("dms-manager")
def test_action_storage_migrate(self):
self.storage.action_storage_migrate()
@users("dms-manager", "dms-user")
def test_count_storage_directories(self):
self.assertTrue(self.storage.count_storage_directories)
@users("dms-manager", "dms-user")
def test_count_storage_files(self):
self.assertTrue(self.storage.count_storage_files)
@users("dms-manager")
@mute_logger("odoo.models.unlink")
def test_file_migrate(self):
root_directory = self.storage.root_directory_ids.filtered(
lambda x: x.is_root_directory
)
file_01 = self.create_file(directory=root_directory)
self.assertEqual(file_01.storage_id, self.storage)
self.assertEqual(file_01.storage_id.save_type, "database")
self.assertEqual(file_01.save_type, "database")
self.storage.write({"save_type": "file"})
directory = self.create_directory(storage=self.storage)
file_02 = self.create_file(directory=directory)
self.assertEqual(file_01.storage_id, self.storage)
self.assertEqual(file_01.storage_id.save_type, "file")
self.assertEqual(file_01.save_type, "database")
self.assertEqual(file_02.storage_id, self.storage)
self.assertEqual(file_02.storage_id.save_type, "file")
self.assertEqual(file_02.save_type, "file")
self.storage.action_storage_migrate()
self.assertEqual(file_01.storage_id, self.storage)
self.assertEqual(file_01.storage_id.save_type, "file")
self.assertEqual(file_01.save_type, "file")
self.assertEqual(file_02.storage_id, self.storage)
self.assertEqual(file_02.storage_id.save_type, "file")
self.assertEqual(file_02.save_type, "file")
self.storage.write({"save_type": "database"})
self.storage.action_storage_migrate()
self.assertEqual(file_01.storage_id, self.storage)
self.assertEqual(file_01.storage_id.save_type, "database")
self.assertEqual(file_01.save_type, "database")
self.assertEqual(file_02.storage_id, self.storage)
self.assertEqual(file_02.storage_id.save_type, "database")
self.assertEqual(file_02.save_type, "database")
self.storage.write({"save_type": "attachment"})
model_partner = self.env.ref("base.model_res_partner")
partner = self.partner_model.create({"name": "Test partner"})
directory = self.directory_model.create(
{
"name": self.partner_model._name,
"model_id": model_partner.id,
"res_id": partner.id,
"parent_id": self.storage.root_directory_ids[0].id,
"storage_id": self.storage.id,
}
)
file_03 = self.create_file(directory=directory)
self.assertEqual(file_02.storage_id, self.storage)
self.assertEqual(file_02.storage_id.save_type, "attachment")
self.assertEqual(file_02.save_type, "database")
self.assertEqual(file_03.storage_id, self.storage)
self.assertEqual(file_03.storage_id.save_type, "attachment")
self.assertEqual(file_03.save_type, "database")
self.storage.action_storage_migrate()
self.assertEqual(file_02.storage_id, self.storage)
self.assertEqual(file_02.storage_id.save_type, "attachment")
self.assertEqual(file_02.save_type, "database")
self.assertEqual(file_03.storage_id, self.storage)
self.assertEqual(file_03.storage_id.save_type, "attachment")
self.assertEqual(file_03.save_type, "database")