Odoo18-Base/addons/sale_project/tests/test_sale_project.py
2025-03-10 11:12:23 +07:00

335 lines
16 KiB
Python

# -*- coding: utf-8 -*-
# Part of Odoo. See LICENSE file for full copyright and licensing details.
from odoo import Command
from odoo.tests.common import users
from odoo.addons.base.tests.common import TransactionCaseWithUserDemo
class TestSaleProject(TransactionCaseWithUserDemo):
@classmethod
def setUpClass(cls):
super().setUpClass()
cls.analytic_plan = cls.env['account.analytic.plan'].create({
'name': 'Plan Test',
'company_id': False,
})
cls.analytic_account_sale = cls.env['account.analytic.account'].create({
'name': 'Project for selling timesheet - AA',
'plan_id': cls.analytic_plan.id,
'code': 'AA-2030'
})
# Create projects
cls.project_global = cls.env['project.project'].create({
'name': 'Global Project',
'analytic_account_id': cls.analytic_account_sale.id,
'allow_billable': True,
})
cls.project_template = cls.env['project.project'].create({
'name': 'Project TEMPLATE for services',
})
cls.project_template_state = cls.env['project.task.type'].create({
'name': 'Only stage in project template',
'sequence': 1,
'project_ids': [(4, cls.project_template.id)]
})
# Create service products
uom_hour = cls.env.ref('uom.product_uom_hour')
cls.product_order_service1 = cls.env['product.product'].create({
'name': "Service Ordered, create no task",
'standard_price': 11,
'list_price': 13,
'type': 'service',
'invoice_policy': 'order',
'uom_id': uom_hour.id,
'uom_po_id': uom_hour.id,
'default_code': 'SERV-ORDERED1',
'service_tracking': 'no',
'project_id': False,
})
cls.product_order_service2 = cls.env['product.product'].create({
'name': "Service Ordered, create task in global project",
'standard_price': 30,
'list_price': 90,
'type': 'service',
'invoice_policy': 'order',
'uom_id': uom_hour.id,
'uom_po_id': uom_hour.id,
'default_code': 'SERV-ORDERED2',
'service_tracking': 'task_global_project',
'project_id': cls.project_global.id,
})
cls.product_order_service3 = cls.env['product.product'].create({
'name': "Service Ordered, create task in new project",
'standard_price': 10,
'list_price': 20,
'type': 'service',
'invoice_policy': 'order',
'uom_id': uom_hour.id,
'uom_po_id': uom_hour.id,
'default_code': 'SERV-ORDERED3',
'service_tracking': 'task_in_project',
'project_id': False, # will create a project
})
cls.product_order_service4 = cls.env['product.product'].create({
'name': "Service Ordered, create project only",
'standard_price': 15,
'list_price': 30,
'type': 'service',
'invoice_policy': 'order',
'uom_id': uom_hour.id,
'uom_po_id': uom_hour.id,
'default_code': 'SERV-ORDERED4',
'service_tracking': 'project_only',
'project_id': False,
})
# Create partner
cls.partner = cls.env['res.partner'].create({'name': "Mur en béton"})
def test_sale_order_with_project_task(self):
SaleOrder = self.env['sale.order'].with_context(tracking_disable=True)
SaleOrderLine = self.env['sale.order.line'].with_context(tracking_disable=True)
sale_order = SaleOrder.create({
'partner_id': self.partner.id,
'partner_invoice_id': self.partner.id,
'partner_shipping_id': self.partner.id,
})
so_line_order_no_task = SaleOrderLine.create({
'product_id': self.product_order_service1.id,
'product_uom_qty': 10,
'order_id': sale_order.id,
})
so_line_order_task_in_global = SaleOrderLine.create({
'product_id': self.product_order_service2.id,
'product_uom_qty': 10,
'order_id': sale_order.id,
})
so_line_order_new_task_new_project = SaleOrderLine.create({
'product_id': self.product_order_service3.id,
'product_uom_qty': 10,
'order_id': sale_order.id,
})
so_line_order_only_project = SaleOrderLine.create({
'product_id': self.product_order_service4.id,
'product_uom_qty': 10,
'order_id': sale_order.id,
})
sale_order.action_confirm()
# service_tracking 'no'
self.assertFalse(so_line_order_no_task.project_id, "The project should not be linked to no task product")
self.assertFalse(so_line_order_no_task.task_id, "The task should not be linked to no task product")
# service_tracking 'task_global_project'
self.assertFalse(so_line_order_task_in_global.project_id, "Only task should be created, project should not be linked")
self.assertEqual(self.project_global.tasks.sale_line_id, so_line_order_task_in_global, "Global project's task should be linked to so line")
# service_tracking 'task_in_project'
self.assertTrue(so_line_order_new_task_new_project.project_id, "Sales order line should be linked to newly created project")
self.assertTrue(so_line_order_new_task_new_project.task_id, "Sales order line should be linked to newly created task")
# service_tracking 'project_only'
self.assertFalse(so_line_order_only_project.task_id, "Task should not be created")
self.assertTrue(so_line_order_only_project.project_id, "Sales order line should be linked to newly created project")
self.assertEqual(self.project_global._get_sale_order_items(), self.project_global.sale_line_id | self.project_global.tasks.sale_line_id, 'The _get_sale_order_items should returns all the SOLs linked to the project and its active tasks.')
sale_order_2 = SaleOrder.create({
'partner_id': self.partner.id,
'partner_invoice_id': self.partner.id,
'partner_shipping_id': self.partner.id,
})
sale_line_1_order_2 = SaleOrderLine.create({
'product_id': self.product_order_service1.id,
'product_uom_qty': 10,
'product_uom': self.product_order_service1.uom_id.id,
'price_unit': self.product_order_service1.list_price,
'order_id': sale_order_2.id,
})
section_sale_line_order_2 = SaleOrderLine.create({
'display_type': 'line_section',
'name': 'Test Section',
'order_id': sale_order_2.id,
})
note_sale_line_order_2 = SaleOrderLine.create({
'display_type': 'line_note',
'name': 'Test Note',
'order_id': sale_order_2.id,
})
sale_order_2.action_confirm()
task = self.env['project.task'].create({
'name': 'Task',
'sale_line_id': sale_line_1_order_2.id,
'project_id': self.project_global.id,
})
self.assertEqual(task.sale_line_id, sale_line_1_order_2)
self.assertIn(task.sale_line_id, self.project_global._get_sale_order_items())
self.assertEqual(self.project_global._get_sale_orders(), sale_order | sale_order_2)
sale_order_lines = sale_order.order_line + sale_line_1_order_2 # exclude the Section and Note Sales Order Items
sale_items_data = self.project_global._get_sale_items(with_action=False)
self.assertEqual(sale_items_data['total'], len(sale_order_lines - so_line_order_new_task_new_project - so_line_order_only_project),
"Should be all the sale items linked to the global project.")
expected_sale_line_dict = {
sol_read['id']: sol_read
for sol_read in sale_order_lines.read(['display_name', 'product_uom_qty', 'qty_delivered', 'qty_invoiced', 'product_uom'])
}
actual_sol_ids = []
for line in sale_items_data['data']:
sol_id = line['id']
actual_sol_ids.append(sol_id)
self.assertIn(sol_id, expected_sale_line_dict)
self.assertDictEqual(line, expected_sale_line_dict[sol_id])
self.assertNotIn(section_sale_line_order_2.id, actual_sol_ids, 'The section Sales Order Item should not be takken into account in the Sales section of project.')
self.assertNotIn(note_sale_line_order_2.id, actual_sol_ids, 'The note Sales Order Item should not be takken into account in the Sales section of project.')
def test_sol_product_type_update(self):
sale_order = self.env['sale.order'].with_context(tracking_disable=True).create({
'partner_id': self.partner.id,
'partner_invoice_id': self.partner.id,
'partner_shipping_id': self.partner.id,
})
self.product_order_service3.type = 'consu'
sale_order_line = self.env['sale.order.line'].create({
'order_id': sale_order.id,
'name': self.product_order_service3.name,
'product_id': self.product_order_service3.id,
'product_uom_qty': 5,
'product_uom': self.product_order_service3.uom_id.id,
'price_unit': self.product_order_service3.list_price
})
self.assertFalse(sale_order_line.is_service, "As the product is consumable, the SOL should not be a service")
self.product_order_service3.type = 'service'
self.assertTrue(sale_order_line.is_service, "As the product is a service, the SOL should be a service")
@users('demo')
def test_cancel_so_linked_to_project(self):
""" Test that cancelling a SO linked to a project will not raise an error """
# Ensure user don't have edit right access to the project
group_sale_manager = self.env.ref('sales_team.group_sale_manager')
group_project_user = self.env.ref('project.group_project_user')
self.env.user.write({'groups_id': [(6, 0, [group_sale_manager.id, group_project_user.id])]})
sale_order = self.env['sale.order'].with_context(tracking_disable=True).create({
'partner_id': self.partner.id,
'partner_invoice_id': self.partner.id,
'partner_shipping_id': self.partner.id,
'project_id': self.project_global.id,
})
sale_order_line = self.env['sale.order.line'].create({
'name': self.product_order_service2.name,
'product_id': self.product_order_service2.id,
'order_id': sale_order.id,
})
self.assertFalse(self.project_global.tasks.sale_line_id, "The project tasks should not be linked to the SOL")
sale_order.action_confirm()
self.assertEqual(self.project_global.tasks.sale_line_id.id, sale_order_line.id, "The project tasks should be linked to the SOL from the SO")
self.project_global.sale_line_id = sale_order_line
sale_order.with_context({'disable_cancel_warning': True}).action_cancel()
self.assertFalse(self.project_global.sale_line_id, "The project should not be linked to the SOL anymore")
def test_create_task_from_template_line(self):
"""
When we add an SOL from a template that is a service that has a service_policy that will generate a task,
even if default_task_id is present in the context, a new task should be created when confirming the SO.
"""
default_task = self.env['project.task'].with_context(tracking_disable=True).create({
'name': 'Task',
'project_id': self.project_global.id
})
sale_order = self.env['sale.order'].with_context(tracking_disable=True, default_task_id=default_task.id).create({
'partner_id': self.partner.id,
})
quotation_template = self.env['sale.order.template'].create({
'name': 'Test quotation',
})
quotation_template.write({
'sale_order_template_line_ids': [
Command.set(
self.env['sale.order.template.line'].create([{
'name': self.product_order_service2.display_name,
'sale_order_template_id': quotation_template.id,
'product_id': self.product_order_service2.id,
'product_uom_id': self.product_order_service2.uom_id.id,
}, {
'name': self.product_order_service3.display_name,
'sale_order_template_id': quotation_template.id,
'product_id': self.product_order_service3.id,
'product_uom_id': self.product_order_service3.uom_id.id,
}]).ids
)
]
})
sale_order.with_context(default_task_id=default_task.id).write({
'sale_order_template_id': quotation_template.id,
})
sale_order.with_context(default_task_id=default_task.id)._onchange_sale_order_template_id()
self.assertFalse(sale_order.order_line.mapped('task_id'),
"SOL should have no related tasks, because they are from services that generates a task")
sale_order.action_confirm()
self.assertEqual(sale_order.tasks_count, 2, "SO should have 2 related tasks")
self.assertNotIn(default_task, sale_order.tasks_ids, "SO should link to the default task from the context")
def test_include_archived_projects_in_stat_btn_related_view(self):
"""Checks if the project stat-button action includes both archived and active projects."""
# Setup
project_A = self.env['project.project'].create({'name': 'Project_A'})
project_B = self.env['project.project'].create({'name': 'Project_B'})
product_A = self.env['product.product'].create({
'name': 'product A',
'list_price': 1.0,
'type': 'service',
'service_tracking': 'task_global_project',
'project_id':project_A.id,
})
product_B = self.env['product.product'].create({
'name': 'product B',
'list_price': 2.0,
'type': 'service',
'service_tracking': 'task_global_project',
'project_id':project_B.id,
})
sale_order = self.env['sale.order'].with_context(tracking_disable=True).create({
'partner_id': self.partner.id,
'partner_invoice_id': self.partner.id,
'partner_shipping_id': self.partner.id,
})
SaleOrderLine = self.env['sale.order.line'].with_context(tracking_disable=True)
SaleOrderLine.create({
'name': product_A.name,
'product_id': product_A.id,
'product_uom_qty': 10,
'price_unit': product_A.list_price,
'order_id': sale_order.id,
})
SaleOrderLine.create({
'name': product_B.name,
'product_id': product_B.id,
'product_uom_qty': 10,
'price_unit': product_B.list_price,
'order_id': sale_order.id,
})
# Check if button action includes both projects BEFORE archivization
action = sale_order.action_view_project_ids()
self.assertEqual(len(action['domain'][0][2]), 2, "Domain should contain 2 projects.")
# Check if button action includes both projects AFTER archivization
project_B.write({'active': False})
action = sale_order.action_view_project_ids()
self.assertEqual(len(action['domain'][0][2]), 2, "Domain should contain 2 projects. (one archived, one not)")