Techsystech
/
server-tools
mirror of https://github.com/OCA/server-tools.git
3
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[IMP] database_cleanup: improve tests

pull/2547/head
Maksym Yankin 2023-02-09 14:39:14 +02:00
parent e937f03017
commit 00cdfb2c71
9 changed files with 201 additions and 196 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
database_cleanup/tests/common.py
View File

@ -1,23 +1,27 @@
# Copyright 2021 Camptocamp SA # Copyright 2021 Camptocamp SA
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html). # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
from odoo.modules.registry import Registry from contextlib import contextmanager
from odoo.tests.common import TransactionCase
import odoo
from odoo.tests import common
from odoo.tests.common import BaseCase, tagged
ADMIN_USER_ID = common.ADMIN_USER_ID
class Common(TransactionCase): @contextmanager
@classmethod def environment():
def setUpClass(cls): """Return an environment with a new cursor for the current database; the
super().setUpClass() cursor is committed and closed after the context block.
# this reloads our registry, and we don't want to run tests twice """
# we also need the original registry for further tests, so save a registry = odoo.registry(common.get_db_name())
# reference to it with registry.cursor() as cr:
cls.original_registry = Registry.registries[cls.env.cr.dbname] yield odoo.api.Environment(cr, ADMIN_USER_ID, {})
@classmethod
def tearDownClass(cls): # Use post_install to get all models loaded more info: odoo/odoo#13458
super().tearDownClass() @tagged("post_install", "-at_install")
# Force rollback to avoid unstable test database class Common(BaseCase):
cls.env.cr.rollback() def setUp(self):
# reset afterwards super().setUp()
Registry.registries[cls.env.cr.dbname] = cls.original_registry

27
database_cleanup/tests/test_create_indexes.py
View File

@ -3,23 +3,24 @@
from odoo.tests.common import tagged from odoo.tests.common import tagged
from .common import Common from .common import Common, environment
# Use post_install to get all models loaded more info: odoo/odoo#13458 # Use post_install to get all models loaded more info: odoo/odoo#13458
@tagged("post_install", "-at_install") @tagged("post_install", "-at_install")
class TestCreateIndexesLine(Common): class TestCreateIndexesLine(Common):
@classmethod def setUp(self):
def setUpClass(cls): super().setUp()
super().setUpClass() with environment() as env:
# delete some index and check if our module recreated it # delete some index and check if our module recreated it
cls.env.cr.execute("drop index res_partner_name_index") env.cr.execute("drop index res_partner_name_index")
def test_deleted_index(self): def test_deleted_index(self):
wizard = self.env["cleanup.create_indexes.wizard"].create({}) with environment() as env:
wizard.purge_all() wizard = env["cleanup.create_indexes.wizard"].create({})
self.env.cr.execute( wizard.purge_all()
"select indexname from pg_indexes where " env.cr.execute(
"indexname='res_partner_name_index' and tablename='res_partner' " "select indexname from pg_indexes where "
) "indexname='res_partner_name_index' and tablename='res_partner' "
self.assertEqual(self.env.cr.rowcount, 1) )
self.assertEqual(env.cr.rowcount, 1)

62
database_cleanup/tests/test_purge_columns.py
View File

@ -5,39 +5,43 @@ from psycopg2 import ProgrammingError
from odoo.tests.common import tagged from odoo.tests.common import tagged
from odoo.tools import mute_logger from odoo.tools import mute_logger
from .common import Common from .common import Common, environment
# Use post_install to get all models loaded more info: odoo/odoo#13458 # Use post_install to get all models loaded more info: odoo/odoo#13458
@tagged("post_install", "-at_install") @tagged("post_install", "-at_install")
class TestCleanupPurgeLineColumn(Common): class TestCleanupPurgeLineColumn(Common):
@classmethod def setUp(self):
def setUpClass(cls): super().setUp()
super().setUpClass() with environment() as env:
# create an orphaned column # create an orphaned column
cls.env.cr.execute( env.cr.execute(
"alter table res_partner add column database_cleanup_test int" "alter table res_partner add column database_cleanup_test int"
) )
def test_empty_column(self): def test_empty_column(self):
# We need use a model that is not blocked (Avoid use res.users) with environment() as env:
partner_model = self.env["ir.model"].search( # We need use a model that is not blocked (Avoid use res.users)
[("model", "=", "res.partner")], limit=1 partner_model = env["ir.model"].search(
) [("model", "=", "res.partner")], limit=1
wizard = self.env["cleanup.purge.wizard.column"].create( )
{ wizard = env["cleanup.purge.wizard.column"].create(
"purge_line_ids": [ {
( "purge_line_ids": [
0, (
0, 0,
{"model_id": partner_model.id, "name": "database_cleanup_test"}, 0,
) {
] "model_id": partner_model.id,
} "name": "database_cleanup_test",
) },
wizard.purge_all() )
# must be removed by the wizard ]
with self.assertRaises(ProgrammingError): }
with self.env.registry.cursor() as cr: )
with mute_logger("odoo.sql_db"): wizard.purge_all()
cr.execute("select database_cleanup_test from res_partner") # must be removed by the wizard
with self.assertRaises(ProgrammingError):
with env.registry.cursor() as cr:
with mute_logger("odoo.sql_db"):
cr.execute("select database_cleanup_test from res_partner")

39
database_cleanup/tests/test_purge_data.py
View File

@ -3,29 +3,30 @@
from odoo.tests.common import tagged from odoo.tests.common import tagged
from .common import Common from .common import Common, environment
# Use post_install to get all models loaded more info: odoo/odoo#13458 # Use post_install to get all models loaded more info: odoo/odoo#13458
@tagged("post_install", "-at_install") @tagged("post_install", "-at_install")
class TestCleanupPurgeLineData(Common): class TestCleanupPurgeLineData(Common):
@classmethod def setUp(self):
def setUpClass(cls): super().setUp()
super().setUpClass() with environment() as env:
# create a data entry pointing nowhere # create a data entry pointing nowhere
cls.env.cr.execute("select max(id) + 1 from res_users") env.cr.execute("select max(id) + 1 from res_users")
cls.env["ir.model.data"].create( env["ir.model.data"].create(
{ {
"module": "database_cleanup", "module": "database_cleanup",
"name": "test_no_data_entry", "name": "test_no_data_entry",
"model": "res.users", "model": "res.users",
"res_id": cls.env.cr.fetchone()[0], "res_id": env.cr.fetchone()[0],
} }
) )
def test_pointing_nowhere(self): def test_pointing_nowhere(self):
wizard = self.env["cleanup.purge.wizard.data"].create({}) with environment() as env:
wizard.purge_all() wizard = env["cleanup.purge.wizard.data"].create({})
# must be removed by the wizard wizard.purge_all()
with self.assertRaises(ValueError): # must be removed by the wizard
self.env.ref("database_cleanup.test_no_data_entry") with self.assertRaises(ValueError):
env.ref("database_cleanup.test_no_data_entry")

53
database_cleanup/tests/test_purge_menus.py
View File

@ -3,37 +3,38 @@
from odoo.tests.common import tagged from odoo.tests.common import tagged
from .common import Common from .common import Common, environment
# Use post_install to get all models loaded more info: odoo/odoo#13458 # Use post_install to get all models loaded more info: odoo/odoo#13458
@tagged("post_install", "-at_install") @tagged("post_install", "-at_install")
class TestCleanupPurgeLineMenu(Common): class TestCleanupPurgeLineMenu(Common):
@classmethod def setUp(self):
def setUpClass(cls): super().setUp()
super().setUpClass() with environment() as env:
# create a new empty menu # create a new empty menu
cls.menu = cls.env["ir.ui.menu"].create({"name": "database_cleanup_test"}) self.menu = env["ir.ui.menu"].create({"name": "database_cleanup_test"})
def test_empty_menu(self): def test_empty_menu(self):
wizard = self.env["cleanup.purge.wizard.menu"].create( with environment() as env:
{ wizard = env["cleanup.purge.wizard.menu"].create(
"purge_line_ids": [ {
( "purge_line_ids": [
0, (
0, 0,
{ 0,
"menu_id": self.menu.id, {
}, "menu_id": self.menu.id,
) },
] )
} ]
) }
wizard.purge_all() )
self.assertFalse( wizard.purge_all()
self.env["ir.ui.menu"].search( self.assertFalse(
[ env["ir.ui.menu"].search(
("name", "=", "database_cleanup_test"), [
] ("name", "=", "database_cleanup_test"),
]
)
) )
)

60
database_cleanup/tests/test_purge_models.py
View File

@ -3,44 +3,38 @@
from odoo.tests.common import tagged from odoo.tests.common import tagged
from .common import Common from .common import Common, environment
# Use post_install to get all models loaded more info: odoo/odoo#13458 # Use post_install to get all models loaded more info: odoo/odoo#13458
@tagged("post_install", "-at_install") @tagged("post_install", "-at_install")
class TestCleanupPurgeLineColumn(Common): class TestCleanupPurgeLineColumn(Common):
@classmethod def setUp(self):
def setUpClass(cls): super().setUp()
super().setUpClass() with environment() as env:
# create a nonexistent model # create a nonexistent model
cls.model_name = "x_database.cleanup.test.model" self.model_name = "x_database.cleanup.test.model"
cls.model_values = { self.model_values = {
"name": "Database cleanup test model", "name": "Database cleanup test model",
"model": cls.model_name, "model": self.model_name,
} }
cls.model = cls.env["ir.model"].create(cls.model_values) self.model = env["ir.model"].create(self.model_values)
cls.env.cr.execute( env.cr.execute(
"insert into ir_attachment (name, res_model, res_id, type) values " "insert into ir_attachment (name, res_model, res_id, type) values "
"('test attachment', %s, 42, 'binary')", "('test attachment', %s, 42, 'binary')",
[cls.model_name], [self.model_name],
) )
cls.env.registry.models.pop(cls.model_name) env.registry.models.pop(self.model_name)
@classmethod
def tearDownClass(cls):
"""We recreate the model to avoid registry Exception at loading"""
super().tearDownClass()
# FIXME: issue origin is not clear but it must be addressed.
cls.model = cls.env["ir.model"].create(cls.model_values)
def test_empty_model(self): def test_empty_model(self):
wizard = self.env["cleanup.purge.wizard.model"].create({}) with environment() as env:
wizard.purge_all() wizard = env["cleanup.purge.wizard.model"].create({})
# must be removed by the wizard wizard.purge_all()
self.assertFalse( # must be removed by the wizard
self.env["ir.model"].search( self.assertFalse(
[ env["ir.model"].search(
("model", "=", self.model_name), [
] ("model", "=", self.model_name),
]
)
) )
)

33
database_cleanup/tests/test_purge_modules.py
View File

@ -3,26 +3,27 @@
from odoo.tests.common import tagged from odoo.tests.common import tagged
from .common import Common from .common import Common, environment
# Use post_install to get all models loaded more info: odoo/odoo#13458 # Use post_install to get all models loaded more info: odoo/odoo#13458
@tagged("post_install", "-at_install") @tagged("post_install", "-at_install")
class TestCleanupPurgeLineModule(Common): class TestCleanupPurgeLineModule(Common):
@classmethod def setUp(self):
def setUpClass(cls): super().setUp()
super().setUpClass() with environment() as env:
# create a nonexistent module # create a nonexistent module
cls.module = cls.env["ir.module.module"].create( self.module = env["ir.module.module"].create(
{ {
"name": "database_cleanup_test", "name": "database_cleanup_test",
"state": "to upgrade", "state": "to upgrade",
} }
) )
def test_remove_to_upgrade_module(self): def test_remove_to_upgrade_module(self):
wizard = self.env["cleanup.purge.wizard.module"].create({}) with environment() as env:
module_names = wizard.purge_line_ids.filtered(lambda x: not x.purged).mapped( wizard = env["cleanup.purge.wizard.module"].create({})
"name" module_names = wizard.purge_line_ids.filtered(
) lambda x: not x.purged
self.assertTrue("database_cleanup_test" in module_names) ).mapped("name")
self.assertTrue("database_cleanup_test" in module_names)

62
database_cleanup/tests/test_purge_properties.py
View File

@ -3,42 +3,44 @@
from odoo.tests.common import tagged from odoo.tests.common import tagged
from .common import Common from .common import Common, environment
# Use post_install to get all models loaded more info: odoo/odoo#13458 # Use post_install to get all models loaded more info: odoo/odoo#13458
@tagged("post_install", "-at_install") @tagged("post_install", "-at_install")
class TestCleanupPurgeLineProperty(Common): class TestCleanupPurgeLineProperty(Common):
@classmethod def setUp(self):
def setUpClass(cls): super().setUp()
super().setUpClass() with environment() as env:
# Create one property for tests # Create one property for tests
cls.partner_name_field_id = cls.env["ir.model.fields"].search( self.partner_name_field_id = env["ir.model.fields"].search(
[("name", "=", "name"), ("model_id.model", "=", "res.partner")], limit=1 [("name", "=", "name"), ("model_id.model", "=", "res.partner")], limit=1
) )
def test_property_to_not_removed(self): def test_property_to_not_removed(self):
self.property = self.env["ir.property"].create( with environment() as env:
{ self.property = env["ir.property"].create(
"fields_id": self.partner_name_field_id.id, {
"type": "char", "fields_id": self.partner_name_field_id.id,
"value_text": "My default partner name", "type": "char",
"res_id": False, "value_text": "My default partner name",
} "res_id": False,
) }
wizard = self.env["cleanup.purge.wizard.property"].create({}) )
wizard.purge_all() wizard = env["cleanup.purge.wizard.property"].create({})
self.assertTrue(self.property.exists()) wizard.purge_all()
self.assertTrue(self.property.exists())
def test_property_no_value(self): def test_property_no_value(self):
self.property = self.env["ir.property"].create( with environment() as env:
{ self.property = env["ir.property"].create(
"fields_id": self.partner_name_field_id.id, {
"type": "char", "fields_id": self.partner_name_field_id.id,
"value_text": False, "type": "char",
"res_id": False, "value_text": False,
} "res_id": False,
) }
wizard = self.env["cleanup.purge.wizard.property"].create({}) )
wizard.purge_all() wizard = env["cleanup.purge.wizard.property"].create({})
self.assertFalse(self.property.exists()) wizard.purge_all()
self.assertFalse(self.property.exists())

23
database_cleanup/tests/test_purge_tables.py
View File

@ -5,22 +5,19 @@ from psycopg2 import ProgrammingError
from odoo.tests.common import tagged from odoo.tests.common import tagged
from odoo.tools import mute_logger from odoo.tools import mute_logger
from .common import Common from .common import Common, environment
# Use post_install to get all models loaded more info: odoo/odoo#13458 # Use post_install to get all models loaded more info: odoo/odoo#13458
@tagged("post_install", "-at_install") @tagged("post_install", "-at_install")
class TestCleanupPurgeLineTable(Common): class TestCleanupPurgeLineTable(Common):
@classmethod
def setUpClass(cls):
super().setUpClass()
# create an orphaned table
cls.env.cr.execute("create table database_cleanup_test (test int)")
def test_empty_table(self): def test_empty_table(self):
wizard = self.env["cleanup.purge.wizard.table"].create({}) with environment() as env:
wizard.purge_all() # create an orphaned table
with self.assertRaises(ProgrammingError): env.cr.execute("create table database_cleanup_test (test int)")
with self.env.registry.cursor() as cr: wizard = env["cleanup.purge.wizard.table"].create({})
with mute_logger("odoo.sql_db"): wizard.purge_all()
cr.execute("select * from database_cleanup_test") with self.assertRaises(ProgrammingError):
with env.registry.cursor() as cr:
with mute_logger("odoo.sql_db"):
cr.execute("select * from database_cleanup_test")