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

tracking_manager: refactor test

pull/2789/head
Sébastien BEAU 2023-01-10 12:27:09 +01:00 committed by Christopher Rogos
parent 20c4babf64
commit a40c2d958e
2 changed files with 87 additions and 123 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tracking_manager/models/models.py
View File

@ -132,9 +132,10 @@ class Base(models.AbstractModel):
def create(self, list_vals): def create(self, list_vals):
self = self._tm_init_tracker() self = self._tm_init_tracker()
records = super().create(list_vals) records = super().create(list_vals)
records._tm_notify_owner("create") for record in records:
record._tm_notify_owner("create")
self._tm_post_message() self._tm_post_message()
return records return records.with_context(tm_tracker=None)
def unlink(self): def unlink(self):
self = self._tm_init_tracker() self = self._tm_init_tracker()

201
tracking_manager/tests/test_tracking_manager.py
View File

@ -9,67 +9,59 @@ class TestTrackingManager(SavepointCase):
@classmethod @classmethod
def setUpClass(cls): def setUpClass(cls):
super().setUpClass() super().setUpClass()
# Using the model res.partner for testing : cls.partner_categ_1, cls.partner_categ_2, cls.partner_categ_3 = cls.env[
# - street field as regular field "res.partner.category"
# - parent_name as related field ].create(
# - category_id as many2many field
# - bank_ids as one2many field with acc_number field to test
# changes.
cls.partner_1 = cls.env.ref("base.res_partner_1")
cls.partner_2 = cls.env.ref("base.res_partner_2")
cls.partner_1.bank_ids = [(6, 0, cls.env.ref("base.bank_partner_demo").ids)]
cls.partner_model = cls.env["ir.model"].search(
[("model", "=", cls.partner_1._name)], limit=1
)
cls.bank_partner_2 = cls.env["res.partner.bank"].create(
{
"acc_number": "1234567890",
"partner_id": cls.partner_1.id,
}
)
cls.partner_categ_1 = cls.env["res.partner.category"].create({"name": "FOO"})
cls.partner_categ_2 = cls.env["res.partner.category"].create({"name": "BAR"})
cls.partner_categ_3 = cls.env["res.partner.category"].create({"name": "TOOH"})
cls.partner_1.category_id = cls.partner_categ_1
def _active_tracking(self, fields_list, model="res.partner"):
self.partner_model.active_custom_tracking = True
for field in self._get_fields(fields_list, model=model):
field.custom_tracking = True
def _get_fields(self, fields_list, model="res.partner"):
return self.env["ir.model.fields"].search(
[ [
("model_id.model", "=", model), {"name": "FOO"},
("name", "in", fields_list), {"name": "BAR"},
{"name": "TOOH"},
] ]
) )
cls.partner = cls.env["res.partner"].create(
{
"name": "Foo",
"bank_ids": [(0, 0, {"acc_number": "007"})],
"category_id": [(6, 0, [cls.partner_categ_1.id])],
}
)
cls.partner.message_ids.unlink()
cls.partner_model = cls.env.ref("base.model_res_partner")
cls._active_tracking(["bank_ids", "category_id"])
@classmethod
def _active_tracking(cls, fields_list):
cls.partner_model.active_custom_tracking = True
for field in cls._get_fields(fields_list):
field.custom_tracking = True
@classmethod
def _get_fields(cls, fields_list):
return cls.partner_model.field_id.filtered(lambda s: s.name in fields_list)
def test_not_tracked(self): def test_not_tracked(self):
self.partner_model.active_custom_tracking = True field = self._get_fields(["name"])[0]
field = self._get_fields(["category_id"])[0]
self.assertFalse(field.native_tracking) self.assertFalse(field.native_tracking)
self.assertFalse(field.custom_tracking) self.assertFalse(field.custom_tracking)
def test_native_tracked(self): def test_native_tracked(self):
self.partner_model.active_custom_tracking = True
field = self._get_fields(["email"])[0] field = self._get_fields(["email"])[0]
self.assertTrue(field.native_tracking) self.assertTrue(field.native_tracking)
self.assertTrue(field.custom_tracking) self.assertTrue(field.custom_tracking)
def test_update_tracked(self): def test_update_tracked(self):
self.partner_model.active_custom_tracking = True field = self._get_fields(["name"])[0]
field = self._get_fields(["category_id"])[0]
self.assertFalse(field.native_tracking) self.assertFalse(field.native_tracking)
self.partner_model.automatic_custom_tracking = True self.partner_model.automatic_custom_tracking = True
self.partner_model.update_custom_tracking() self.partner_model.update_custom_tracking()
self.assertTrue(field.custom_tracking) self.assertTrue(field.custom_tracking)
@property
def messages(self):
return self.partner.message_ids
def test_m2m_add_line(self): def test_m2m_add_line(self):
initial_msg = self.partner_1.message_ids self.partner.write(
self._active_tracking(["category_id"])
self.partner_1.write(
{ {
"category_id": [ "category_id": [
( (
@ -83,22 +75,16 @@ class TestTrackingManager(SavepointCase):
] ]
} }
) )
after_msg = self.partner_1.message_ids self.assertEqual(len(self.messages), 1)
self.assertEqual(len(initial_msg) + 1, len(after_msg)) self.assertIn("New", self.messages.body)
self.assertTrue("New" in after_msg[0].body)
def test_m2m_delete_line(self): def test_m2m_delete_line(self):
initial_msg = self.partner_1.message_ids self.partner.write({"category_id": [(6, 0, [])]})
self._active_tracking(["category_id"]) self.assertEqual(len(self.messages), 1)
self.partner_1.write({"category_id": [(6, 0, [])]}) self.assertIn("Delete", self.messages.body)
after_msg = self.partner_1.message_ids
self.assertEqual(len(initial_msg) + 1, len(after_msg))
self.assertTrue("Delete" in after_msg[0].body)
def test_m2m_multi_line(self): def test_m2m_multi_line(self):
initial_msg = self.partner_1.message_ids self.partner.write(
self._active_tracking(["category_id"])
self.partner_1.write(
{ {
"category_id": [ "category_id": [
( (
@ -112,94 +98,71 @@ class TestTrackingManager(SavepointCase):
] ]
} }
) )
after_msg = self.partner_1.message_ids self.assertEqual(len(self.messages), 1)
self.assertEqual(len(initial_msg) + 1, len(after_msg)) self.assertEqual(self.messages.body.count("New"), 2)
self.assertEqual(after_msg[0].body.count("New"), 2) self.assertEqual(self.messages.body.count("Delete"), 1)
self.assertEqual(after_msg[0].body.count("Delete"), 1)
def test_o2m_add(self): def test_o2m_create_indirectly(self):
initial_msg = self.partner_1.message_ids self.partner.write({"bank_ids": [(0, 0, {"acc_number": "1234567890"})]})
self._active_tracking(["bank_ids"]) self.assertEqual(len(self.messages), 1)
self.partner_1.write({"bank_ids": [(0, 0, {"acc_number": "1234567890"})]}) self.assertEqual(self.messages.body.count("New"), 1)
after_msg = self.partner_1.message_ids
self.assertEqual(len(initial_msg) + 1, len(after_msg))
self.assertTrue("New" in after_msg[0].body)
def test_o2m_delete(self): def test_o2m_unlink_indirectly(self):
self._active_tracking(["bank_ids"]) self.partner.write({"bank_ids": [(3, self.partner.bank_ids[0].id)]})
initial_msg = self.partner_1.message_ids self.assertEqual(len(self.messages), 1)
self.partner_1.write({"bank_ids": [(3, self.partner_1.bank_ids[0].id)]}) self.assertIn("Delete", self.messages.body)
after_msg = self.partner_1.message_ids
self.assertEqual(len(initial_msg) + 1, len(after_msg))
self.assertTrue("Delete" in after_msg[0].body)
def test_o2m_change_in_line(self): def test_o2m_write_indirectly(self):
initial_msg = self.partner_1.message_ids self.partner.write(
self._active_tracking(["bank_ids"])
self.partner_1.write(
{ {
"bank_ids": [(1, self.partner_1.bank_ids[0].id, {"acc_number": "123"})], "bank_ids": [(1, self.partner.bank_ids[0].id, {"acc_number": "123"})],
} }
) )
after_msg = self.partner_1.message_ids self.assertEqual(len(self.messages), 1)
self.assertEqual(len(initial_msg) + 1, len(after_msg)) self.assertIn("Change", self.messages.body)
self.assertTrue("Change" in after_msg[0].body)
def test_o2m_change_in_line_not_tracked(self): def test_o2m_write_indirectly_on_not_tracked_fields(self):
initial_msg = self.partner_1.message_ids
self._active_tracking(["bank_ids"])
# Active custom tracking on res.partner.bank and remove tracking on acc_number # Active custom tracking on res.partner.bank and remove tracking on acc_number
bank_model = self.env["ir.model"].search( bank_model = self.env["ir.model"].search([("model", "=", "res.partner.bank")])
[("model", "=", self.bank_partner_2._name)], limit=1
)
bank_model.active_custom_tracking = True bank_model.active_custom_tracking = True
acc_number = bank_model.field_id.filtered(lambda x: x.name == "acc_number") acc_number = bank_model.field_id.filtered(lambda x: x.name == "acc_number")
acc_number.custom_tracking = False acc_number.custom_tracking = False
self.partner_1.write( self.partner.write(
{ {
"bank_ids": [(1, self.partner_1.bank_ids[0].id, {"acc_number": "456"})], "bank_ids": [(1, self.partner.bank_ids[0].id, {"acc_number": "123"})],
} }
) )
after_msg = self.partner_1.message_ids self.assertEqual(len(self.messages), 0)
self.assertEqual(len(initial_msg), len(after_msg))
def test_o2m_multi_line(self): def test_o2m_create_and_unlink_indirectly(self):
initial_msg = self.partner_1.message_ids self.partner.write(
self._active_tracking(["bank_ids"]) {
self.partner_1.bank_ids = [ "bank_ids": [
(3, self.partner_1.bank_ids[0].id), (3, self.partner.bank_ids[0].id),
(0, 0, {"acc_number": "1234567890"}), (0, 0, {"acc_number": "1234567890"}),
] ]
after_msg = self.partner_1.message_ids }
self.assertEqual(len(initial_msg) + 1, len(after_msg)) )
self.assertEqual(after_msg[0].body.count("New"), 1) self.assertEqual(len(self.messages), 1)
self.assertEqual(after_msg[0].body.count("Delete"), 1) self.assertEqual(self.messages.body.count("New"), 1)
self.assertEqual(self.messages.body.count("Delete"), 1)
def test_create_o2m_directly(self): def test_o2m_create_directly(self):
initial_msg = self.partner_1.message_ids
self._active_tracking(["bank_ids"])
self.env["res.partner.bank"].create( self.env["res.partner.bank"].create(
{ {
"acc_number": "1234567890", "acc_number": "1234567890",
"partner_id": self.partner_1.id, "partner_id": self.partner.id,
} }
) )
after_msg = self.partner_1.message_ids self.assertEqual(len(self.messages), 1)
self.assertEqual(len(initial_msg) + 1, len(after_msg)) self.assertEqual(self.messages.body.count("New"), 1)
self.assertEqual(after_msg[0].body.count("New"), 1)
def test_unlink_o2m_directly(self): def test_o2m_unlink_directly(self):
initial_msg = self.partner_1.message_ids self.partner.bank_ids.unlink()
self._active_tracking(["bank_ids"]) self.assertEqual(len(self.messages), 1)
self.bank_partner_2.unlink() self.assertEqual(self.messages.body.count("Delete"), 1)
after_msg = self.partner_1.message_ids
self.assertEqual(len(initial_msg) + 1, len(after_msg))
self.assertEqual(after_msg[0].body.count("Delete"), 1)
def test_update_o2m_directly(self): def test_o2m_update_directly(self):
initial_msg = self.partner_1.message_ids self.partner.bank_ids.write({"acc_number": "0987654321"})
self._active_tracking(["bank_ids"]) self.assertEqual(len(self.messages), 1)
self.bank_partner_2.write({"acc_number": "0987654321"}) self.assertEqual(self.messages.body.count("Change :"), 1)
after_msg = self.partner_1.message_ids
self.assertEqual(len(initial_msg) + 1, len(after_msg))
self.assertEqual(after_msg[0].body.count("Change :"), 1)