91 lines
3 KiB
Python
91 lines
3 KiB
Python
from odoo.tests import TransactionCase, tagged, Form
|
|
|
|
|
|
@tagged('-at_install', 'post_install')
|
|
class FSMVisitTest(TransactionCase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
super().setUpClass()
|
|
|
|
def test_create_visit_sets_name_on_section(self):
|
|
so = self.generate_sale_order()
|
|
self.add_service_so_line(so)
|
|
|
|
visit = self.generate_visit(so)
|
|
|
|
self.assertTrue(visit.so_section_id)
|
|
self.assertEqual(visit.so_section_id.name, visit.label)
|
|
|
|
def test_change_visit_section_name(self):
|
|
so = self.generate_sale_order()
|
|
visit = self.generate_visit(so, label="First Label")
|
|
line = visit.so_section_id
|
|
|
|
line.name = "Second Label"
|
|
|
|
self.assertEqual(visit.label, "Second Label")
|
|
|
|
def test_change_visit_label_changes_section_name(self):
|
|
so = self.generate_sale_order()
|
|
visit = self.generate_visit(so, label="First Label")
|
|
line = visit.so_section_id
|
|
|
|
visit.label = "Second Label"
|
|
|
|
self.assertEqual(line.name, "Second Label")
|
|
|
|
def test_visit_completes_when_task_completes(self):
|
|
so = self.generate_sale_order()
|
|
visit = self.generate_visit(so)
|
|
sol = self.add_service_so_line(so, task=True)
|
|
so.action_confirm()
|
|
task = so.order_line.filtered(lambda l: l.task_id).task_id
|
|
|
|
task.action_fsm_validate()
|
|
|
|
self.assertTrue(visit.is_completed)
|
|
|
|
def generate_visit(self, sale_order, label="Test Label"):
|
|
return self.env['bemade_fsm.visit'].create([{
|
|
'sale_order_id': sale_order.id,
|
|
'label': label,
|
|
}])
|
|
|
|
def generate_sale_order(self, partner_id=None):
|
|
so = self.env['sale.order'].create({
|
|
'partner_id': partner_id or self.generate_partner().id,
|
|
'client_order_ref': 'Test',
|
|
})
|
|
return so
|
|
|
|
def add_service_so_line(self, sale_order, task: bool = False):
|
|
""" Generates a sales order line for a service product.
|
|
|
|
:param sale_order: The sales order to which the new line is to be added
|
|
:param task: If true, the created line will be for a product with service_tracking=task_global_project
|
|
"""
|
|
if not task:
|
|
service_tracking = 'no'
|
|
project = False
|
|
else:
|
|
project = self.env.ref("industry_fsm.fsm_project")
|
|
service_tracking = 'task_global_project'
|
|
product = self.env['product.product'].create({
|
|
'name': 'Test Product',
|
|
'type': 'service',
|
|
'service_tracking': service_tracking,
|
|
'project_id': project and project.id,
|
|
'price': 100.0,
|
|
'sale_ok': True,
|
|
})
|
|
return self.env['sale.order.line'].create({
|
|
'product_id': product.id,
|
|
'order_id': sale_order.id,
|
|
})
|
|
|
|
def generate_partner(self):
|
|
partner = self.env['res.partner'].create({
|
|
'name': 'Test Partner',
|
|
'company_type': 'company',
|
|
})
|
|
return partner
|