[FIX]:hr_pms_property: fix commit in test

This commit is contained in:
irluidev
2024-10-15 17:25:34 +02:00
parent ab79537fc5
commit aed9da9885

View File

@@ -4,6 +4,7 @@
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl). # License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).
from odoo.tests.common import TransactionCase from odoo.tests.common import TransactionCase
class TestPmsHrProperty(TransactionCase): class TestPmsHrProperty(TransactionCase):
def setUp(self): def setUp(self):
super(TestPmsHrProperty, self).setUp() super(TestPmsHrProperty, self).setUp()
@@ -11,96 +12,73 @@ class TestPmsHrProperty(TransactionCase):
self.HrEmployee = self.env["hr.employee"] self.HrEmployee = self.env["hr.employee"]
self.Company = self.env["res.company"] self.Company = self.env["res.company"]
# Company creation self.company1 = self.Company.create({"name": "Company 1"})
self.company1 = self.Company.create(
{"name": "Company 1"}
)
# Property creation
self.property1 = self.PmsProperty.create( self.property1 = self.PmsProperty.create(
{ {"name": "Property 1", "company_id": self.company1.id}
"name": "Property 1",
"company_id": self.company1.id
}
) )
# User creation
user_dict = { user_dict = {
"name": "User 1", "name": "User 1",
"login": "tua@example.com", "login": "tua@example.com",
"password": "base-test-passwd" "password": "base-test-passwd",
} }
self.user_test = self.env["res.users"].create(user_dict) self.user_test = self.env["res.users"].create(user_dict)
user_dict2 = { user_dict2 = {
"name": "User 2", "name": "User 2",
"login": "user2@example.com", "login": "user2@example.com",
"password": "base-test-passwd" "password": "base-test-passwd",
} }
self.user_test2 = self.env["res.users"].create(user_dict2) self.user_test2 = self.env["res.users"].create(user_dict2)
# Employee creation
employee_dict = { employee_dict = {
"name": "Employee 1", "name": "Employee 1",
"user_id": self.user_test.id, "user_id": self.user_test.id,
"address_id": self.user_test.partner_id.id "address_id": self.user_test.partner_id.id,
} }
self.employee = self.env["hr.employee"].create(employee_dict) self.employee = self.env["hr.employee"].create(employee_dict)
employee_dict2 = { employee_dict2 = {
"name": "Employee 2", "name": "Employee 2",
"user_id": self.user_test2.id, "user_id": self.user_test2.id,
"address_id": self.user_test2.partner_id.id "address_id": self.user_test2.partner_id.id,
} }
self.employee2 = self.env["hr.employee"].create(employee_dict2) self.employee2 = self.env["hr.employee"].create(employee_dict2)
def test_employee_assignment(self): def test_employee_assignment(self):
"""Check if employees are correctly assigned to properties""" """Check if employees are correctly assigned to properties"""
self.employee.write( self.employee.write({"property_ids": [(4, self.property1.id)]})
{"property_ids": [(4, self.property1.id)]}
)
assigned_employees = self.property1.employee_ids assigned_employees = self.property1.employee_ids
self.assertIn( self.assertIn(
self.employee, self.employee,
assigned_employees, assigned_employees,
"The employee is not correctly assigned to the property." "The employee is not correctly assigned to the property.",
) )
def test_employee_removal(self): def test_employee_removal(self):
"""Check if employees can be unassigned from properties correctly""" """Check if employees can be unassigned from properties correctly"""
self.employee.write( self.employee.write({"property_ids": [(4, self.property1.id)]})
{"property_ids": [(4, self.property1.id)]}
)
self.employee.write( self.employee.write({"property_ids": [(3, self.property1.id)]})
{"property_ids": [(3, self.property1.id)]}
)
assigned_employees = self.property1.employee_ids assigned_employees = self.property1.employee_ids
self.assertNotIn( self.assertNotIn(
self.employee, self.employee,
assigned_employees, assigned_employees,
"The employee is still assigned to the property after removal." "The employee is still assigned to the property after removal.",
) )
def test_multiple_employees_assignment(self): def test_multiple_employees_assignment(self):
"""Check if multiple employees can be assigned to a single property""" """Check if multiple employees can be assigned to a single property"""
self.employee.write( self.employee.write({"property_ids": [(4, self.property1.id)]})
{"property_ids": [(4, self.property1.id)]} self.employee2.write({"property_ids": [(4, self.property1.id)]})
)
self.employee2.write(
{"property_ids": [(4, self.property1.id)]}
)
assigned_employees = self.property1.employee_ids assigned_employees = self.property1.employee_ids
self.assertIn( self.assertIn(
self.employee, self.employee, assigned_employees, "Employee 1 is not correctly assigned."
assigned_employees,
"Employee 1 is not correctly assigned."
) )
self.assertIn( self.assertIn(
self.employee2, self.employee2, assigned_employees, "Employee 2 is not correctly assigned."
assigned_employees,
"Employee 2 is not correctly assigned."
) )