mirror of
https://gitlab.com/flectra-community/partner-contact.git
synced 2024-12-23 12:51:46 +00:00
126 lines
4.7 KiB
Python
126 lines
4.7 KiB
Python
# Copyright 2020 Ecosoft Co., Ltd (http://ecosoft.co.th/)
|
|
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
|
|
|
|
from flectra.tests.common import SavepointCase
|
|
|
|
|
|
class TestPartnerCompanyGroup(SavepointCase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
super().setUpClass()
|
|
cls.partner_model = cls.env["res.partner"]
|
|
cls.company = cls.partner_model.create(
|
|
{"name": "Test Company", "company_type": "company"}
|
|
)
|
|
cls.contact = cls.partner_model.create(
|
|
{"name": "Test Contact", "type": "contact", "parent_id": cls.company.id}
|
|
)
|
|
currency = cls.env.ref("base.USD")
|
|
cls.pricelist1 = cls.env["product.pricelist"].create(
|
|
{"name": "Pricelist 01", "currency_id": currency.id}
|
|
)
|
|
cls.pricelist2 = cls.env["product.pricelist"].create(
|
|
{"name": "Pricelist 02", "currency_id": currency.id}
|
|
)
|
|
cls.company_group1 = cls.env["res.partner"].create(
|
|
{
|
|
"name": "Company Group 01",
|
|
"is_company": True,
|
|
"property_product_pricelist": cls.pricelist1.id,
|
|
}
|
|
)
|
|
cls.company_group2 = cls.env["res.partner"].create(
|
|
{
|
|
"name": "Company Group 02",
|
|
"is_company": True,
|
|
"property_product_pricelist": cls.pricelist2.id,
|
|
}
|
|
)
|
|
cls.partner1 = cls.env["res.partner"].create(
|
|
{
|
|
"name": "Partner 01",
|
|
"is_company": True,
|
|
"property_product_pricelist": cls.pricelist1.id,
|
|
"company_group_id": cls.company_group1.id,
|
|
}
|
|
)
|
|
cls.partner2 = cls.env["res.partner"].create(
|
|
{
|
|
"name": "Partner 02",
|
|
"is_company": True,
|
|
"property_product_pricelist": cls.pricelist1.id,
|
|
"company_group_id": cls.company_group1.id,
|
|
}
|
|
)
|
|
|
|
def test_partner_company_group(self):
|
|
self.company.write({"company_group_id": self.company.id})
|
|
self.assertEqual(self.company.company_group_id, self.contact.company_group_id)
|
|
|
|
def test_01_change_pricelist_partner(self):
|
|
self.partner1.property_product_pricelist = self.pricelist2
|
|
res = self.partner1._onchange_property_product_pricelist()
|
|
self.assertEqual(
|
|
{
|
|
"warning": {
|
|
"title": "Warning",
|
|
"message": "The company group Company Group 01 has the pricelist "
|
|
"Pricelist 01 (USD), that is different than the "
|
|
"pricelist set on this contact",
|
|
}
|
|
},
|
|
res,
|
|
)
|
|
self.partner1.property_product_pricelist = self.pricelist1
|
|
res = self.partner1._onchange_property_product_pricelist()
|
|
self.assertEqual({}, res)
|
|
|
|
def test_02_change_company_group_partner(self):
|
|
self.partner1.company_group_id = self.company_group2
|
|
res = self.partner1._onchange_company_group_id()
|
|
self.assertEqual(
|
|
{
|
|
"warning": {
|
|
"title": "Warning",
|
|
"message": "The company group Company Group 02 has the pricelist "
|
|
"Pricelist 02 (USD), that is different than the "
|
|
"pricelist set on this contact",
|
|
}
|
|
},
|
|
res,
|
|
)
|
|
self.partner1.company_group_id = self.company_group1
|
|
res = self.partner1._onchange_company_group_id()
|
|
self.assertEqual({}, res)
|
|
|
|
def test_03_change_pricelist_company_group(self):
|
|
self.company_group1.property_product_pricelist = self.pricelist2
|
|
res = self.company_group1._onchange_property_product_pricelist()
|
|
self.assertEqual(
|
|
{
|
|
"warning": {
|
|
"title": "Warning",
|
|
"message": "This contact has members of a company group with "
|
|
"different pricelists, the members are:\n"
|
|
"\t- Partner 01\n\t- Partner 02\n",
|
|
}
|
|
},
|
|
res,
|
|
)
|
|
self.partner1.property_product_pricelist = self.pricelist2
|
|
res = self.company_group1._onchange_property_product_pricelist()
|
|
self.assertEqual(
|
|
{
|
|
"warning": {
|
|
"title": "Warning",
|
|
"message": "This contact has members of a company group with "
|
|
"different pricelists, the members are:\n"
|
|
"\t- Partner 02\n",
|
|
}
|
|
},
|
|
res,
|
|
)
|
|
self.partner2.property_product_pricelist = self.pricelist2
|
|
res = self.company_group1._onchange_property_product_pricelist()
|
|
self.assertEqual({}, res)
|