22# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).
33from datetime import date
44from unittest .mock import patch
5- from odoo . tests import Form
5+
66from freezegun import freeze_time
7+
8+ from odoo .tests import Form
9+
710from .test_forecast_line import BaseForecastRoleTest
811
912
@@ -74,9 +77,8 @@ def test_write_trigger_field_updates_forecast_lines(self):
7477 # 'forecast_date_end' is in _update_forecast_lines_trigger_fields().
7578 line ._write ({"forecast_date_end" : date (2022 , 3 , 31 )})
7679
77- mock_update .assert_called_once_with (
78- line ,
79- ), "_update_forecast_lines must be called once when a trigger field is written"
80+ # _update_forecast_lines must be called once when a trigger field is written
81+ mock_update .assert_called_once_with (line )
8082
8183 @freeze_time ("2022-01-01" )
8284 def test_write_non_trigger_field_does_not_update_forecast_lines (self ):
@@ -91,8 +93,8 @@ def test_write_non_trigger_field_does_not_update_forecast_lines(self):
9193 # 'customer_lead' is not a trigger field.
9294 line ._write ({"customer_lead" : 5.0 })
9395
94- mock_update . assert_not_called (
95- ), "_update_forecast_lines must NOT be called for non-trigger fields"
96+ # _update_forecast_lines must NOT be called for non-trigger fields
97+ mock_update . assert_not_called ()
9698
9799 @freeze_time ("2022-01-01" )
98100 def test_draft_sale_order_without_dates_no_forecast (self ):
@@ -190,9 +192,7 @@ def test_confirm_order_sale_order_create_project_task_with_forecast_line(self):
190192 def test_timesheet_create_project_with_template_and_role_writes_tasks (self ):
191193 # 1. Template project with a task
192194 template = self .ProjectProject .create ({"name" : "Dev Template" })
193- self .ProjectTask .create (
194- {"name" : "Template Task" , "project_id" : template .id }
195- )
195+ self .ProjectTask .create ({"name" : "Template Task" , "project_id" : template .id })
196196
197197 # 2. Product: project_only + template + forecast role
198198 product = self .ProductProduct .create (
@@ -306,4 +306,4 @@ def test_timesheet_create_project_without_role_skips_task_write(self):
306306 self .assertFalse (
307307 task .date_end ,
308308 "date_end must remain unset when the template block is skipped" ,
309- )
309+ )
0 commit comments